{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.999705680368881,
  "eval_steps": 500,
  "global_step": 7644,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.003924261748258609,
      "grad_norm": 3.25,
      "learning_rate": 8.695652173913044e-06,
      "loss": 1.8783,
      "step": 10
    },
    {
      "epoch": 0.007848523496517218,
      "grad_norm": 2.5625,
      "learning_rate": 1.739130434782609e-05,
      "loss": 2.1068,
      "step": 20
    },
    {
      "epoch": 0.011772785244775827,
      "grad_norm": 1.3203125,
      "learning_rate": 2.608695652173913e-05,
      "loss": 1.7524,
      "step": 30
    },
    {
      "epoch": 0.015697046993034435,
      "grad_norm": 1.1640625,
      "learning_rate": 3.478260869565218e-05,
      "loss": 1.7851,
      "step": 40
    },
    {
      "epoch": 0.019621308741293045,
      "grad_norm": 3.9375,
      "learning_rate": 4.347826086956522e-05,
      "loss": 1.7849,
      "step": 50
    },
    {
      "epoch": 0.023545570489551654,
      "grad_norm": 0.953125,
      "learning_rate": 5.217391304347826e-05,
      "loss": 1.54,
      "step": 60
    },
    {
      "epoch": 0.02746983223781026,
      "grad_norm": 1.5078125,
      "learning_rate": 6.086956521739131e-05,
      "loss": 1.7694,
      "step": 70
    },
    {
      "epoch": 0.03139409398606887,
      "grad_norm": 0.828125,
      "learning_rate": 6.956521739130436e-05,
      "loss": 1.4367,
      "step": 80
    },
    {
      "epoch": 0.035318355734327476,
      "grad_norm": 1.125,
      "learning_rate": 7.82608695652174e-05,
      "loss": 1.6799,
      "step": 90
    },
    {
      "epoch": 0.03924261748258609,
      "grad_norm": 3.09375,
      "learning_rate": 8.695652173913044e-05,
      "loss": 1.563,
      "step": 100
    },
    {
      "epoch": 0.043166879230844696,
      "grad_norm": 0.8671875,
      "learning_rate": 9.565217391304348e-05,
      "loss": 1.5597,
      "step": 110
    },
    {
      "epoch": 0.04709114097910331,
      "grad_norm": 1.3515625,
      "learning_rate": 0.00010434782608695653,
      "loss": 1.6684,
      "step": 120
    },
    {
      "epoch": 0.051015402727361915,
      "grad_norm": 0.73828125,
      "learning_rate": 0.00011304347826086956,
      "loss": 1.4209,
      "step": 130
    },
    {
      "epoch": 0.05493966447562052,
      "grad_norm": 1.0859375,
      "learning_rate": 0.00012173913043478263,
      "loss": 1.6828,
      "step": 140
    },
    {
      "epoch": 0.058863926223879134,
      "grad_norm": 2.5,
      "learning_rate": 0.00013043478260869567,
      "loss": 1.5066,
      "step": 150
    },
    {
      "epoch": 0.06278818797213774,
      "grad_norm": 0.83203125,
      "learning_rate": 0.0001391304347826087,
      "loss": 1.6399,
      "step": 160
    },
    {
      "epoch": 0.06671244972039635,
      "grad_norm": 1.359375,
      "learning_rate": 0.00014782608695652173,
      "loss": 1.6942,
      "step": 170
    },
    {
      "epoch": 0.07063671146865495,
      "grad_norm": 0.734375,
      "learning_rate": 0.0001565217391304348,
      "loss": 1.4899,
      "step": 180
    },
    {
      "epoch": 0.07456097321691357,
      "grad_norm": 0.9296875,
      "learning_rate": 0.00016521739130434784,
      "loss": 1.7132,
      "step": 190
    },
    {
      "epoch": 0.07848523496517218,
      "grad_norm": 2.359375,
      "learning_rate": 0.00017391304347826088,
      "loss": 1.5505,
      "step": 200
    },
    {
      "epoch": 0.08240949671343079,
      "grad_norm": 0.76171875,
      "learning_rate": 0.00018260869565217392,
      "loss": 1.7034,
      "step": 210
    },
    {
      "epoch": 0.08633375846168939,
      "grad_norm": 1.3203125,
      "learning_rate": 0.00019130434782608697,
      "loss": 1.7085,
      "step": 220
    },
    {
      "epoch": 0.090258020209948,
      "grad_norm": 0.66015625,
      "learning_rate": 0.0002,
      "loss": 1.5324,
      "step": 230
    },
    {
      "epoch": 0.09418228195820662,
      "grad_norm": 0.9375,
      "learning_rate": 0.00019999910223238217,
      "loss": 1.8071,
      "step": 240
    },
    {
      "epoch": 0.09810654370646522,
      "grad_norm": 2.125,
      "learning_rate": 0.00019999640894564831,
      "loss": 1.5573,
      "step": 250
    },
    {
      "epoch": 0.10203080545472383,
      "grad_norm": 0.69921875,
      "learning_rate": 0.00019999192018815736,
      "loss": 1.7348,
      "step": 260
    },
    {
      "epoch": 0.10595506720298244,
      "grad_norm": 1.234375,
      "learning_rate": 0.0001999856360405066,
      "loss": 1.7666,
      "step": 270
    },
    {
      "epoch": 0.10987932895124104,
      "grad_norm": 0.64453125,
      "learning_rate": 0.00019997755661553006,
      "loss": 1.5174,
      "step": 280
    },
    {
      "epoch": 0.11380359069949966,
      "grad_norm": 0.83203125,
      "learning_rate": 0.00019996768205829665,
      "loss": 1.772,
      "step": 290
    },
    {
      "epoch": 0.11772785244775827,
      "grad_norm": 1.859375,
      "learning_rate": 0.00019995601254610757,
      "loss": 1.5657,
      "step": 300
    },
    {
      "epoch": 0.12165211419601688,
      "grad_norm": 0.703125,
      "learning_rate": 0.000199942548288493,
      "loss": 1.7429,
      "step": 310
    },
    {
      "epoch": 0.12557637594427548,
      "grad_norm": 1.046875,
      "learning_rate": 0.0001999272895272084,
      "loss": 1.7322,
      "step": 320
    },
    {
      "epoch": 0.1295006376925341,
      "grad_norm": 0.64453125,
      "learning_rate": 0.0001999102365362303,
      "loss": 1.5078,
      "step": 330
    },
    {
      "epoch": 0.1334248994407927,
      "grad_norm": 0.7890625,
      "learning_rate": 0.00019989138962175104,
      "loss": 1.7882,
      "step": 340
    },
    {
      "epoch": 0.1373491611890513,
      "grad_norm": 1.6796875,
      "learning_rate": 0.00019987074912217369,
      "loss": 1.554,
      "step": 350
    },
    {
      "epoch": 0.1412734229373099,
      "grad_norm": 0.6640625,
      "learning_rate": 0.00019984831540810567,
      "loss": 1.7348,
      "step": 360
    },
    {
      "epoch": 0.14519768468556854,
      "grad_norm": 1.0625,
      "learning_rate": 0.00019982408888235225,
      "loss": 1.735,
      "step": 370
    },
    {
      "epoch": 0.14912194643382715,
      "grad_norm": 0.57421875,
      "learning_rate": 0.00019979806997990918,
      "loss": 1.4742,
      "step": 380
    },
    {
      "epoch": 0.15304620818208575,
      "grad_norm": 0.7890625,
      "learning_rate": 0.00019977025916795502,
      "loss": 1.7792,
      "step": 390
    },
    {
      "epoch": 0.15697046993034436,
      "grad_norm": 1.609375,
      "learning_rate": 0.00019974065694584279,
      "loss": 1.5286,
      "step": 400
    },
    {
      "epoch": 0.16089473167860296,
      "grad_norm": 0.69140625,
      "learning_rate": 0.00019970926384509072,
      "loss": 1.7143,
      "step": 410
    },
    {
      "epoch": 0.16481899342686157,
      "grad_norm": 1.0546875,
      "learning_rate": 0.00019967608042937302,
      "loss": 1.717,
      "step": 420
    },
    {
      "epoch": 0.16874325517512018,
      "grad_norm": 0.52734375,
      "learning_rate": 0.00019964110729450964,
      "loss": 1.465,
      "step": 430
    },
    {
      "epoch": 0.17266751692337878,
      "grad_norm": 0.72265625,
      "learning_rate": 0.00019960434506845553,
      "loss": 1.7703,
      "step": 440
    },
    {
      "epoch": 0.1765917786716374,
      "grad_norm": 1.484375,
      "learning_rate": 0.00019956579441128943,
      "loss": 1.4954,
      "step": 450
    },
    {
      "epoch": 0.180516040419896,
      "grad_norm": 0.6875,
      "learning_rate": 0.0001995254560152019,
      "loss": 1.7071,
      "step": 460
    },
    {
      "epoch": 0.18444030216815463,
      "grad_norm": 0.99609375,
      "learning_rate": 0.0001994833306044831,
      "loss": 1.6897,
      "step": 470
    },
    {
      "epoch": 0.18836456391641324,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00019943941893550967,
      "loss": 1.444,
      "step": 480
    },
    {
      "epoch": 0.19228882566467184,
      "grad_norm": 0.72265625,
      "learning_rate": 0.00019939372179673105,
      "loss": 1.7892,
      "step": 490
    },
    {
      "epoch": 0.19621308741293045,
      "grad_norm": 1.5625,
      "learning_rate": 0.00019934624000865543,
      "loss": 1.4801,
      "step": 500
    },
    {
      "epoch": 0.20013734916118905,
      "grad_norm": 0.6328125,
      "learning_rate": 0.00019929697442383512,
      "loss": 1.6799,
      "step": 510
    },
    {
      "epoch": 0.20406161090944766,
      "grad_norm": 0.953125,
      "learning_rate": 0.00019924592592685103,
      "loss": 1.7007,
      "step": 520
    },
    {
      "epoch": 0.20798587265770627,
      "grad_norm": 0.55859375,
      "learning_rate": 0.0001991930954342969,
      "loss": 1.4269,
      "step": 530
    },
    {
      "epoch": 0.21191013440596487,
      "grad_norm": 0.7265625,
      "learning_rate": 0.00019913848389476285,
      "loss": 1.7524,
      "step": 540
    },
    {
      "epoch": 0.21583439615422348,
      "grad_norm": 1.3203125,
      "learning_rate": 0.00019908209228881827,
      "loss": 1.4737,
      "step": 550
    },
    {
      "epoch": 0.21975865790248209,
      "grad_norm": 0.609375,
      "learning_rate": 0.0001990239216289944,
      "loss": 1.6668,
      "step": 560
    },
    {
      "epoch": 0.2236829196507407,
      "grad_norm": 0.8984375,
      "learning_rate": 0.00019896397295976584,
      "loss": 1.6983,
      "step": 570
    },
    {
      "epoch": 0.22760718139899933,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00019890224735753215,
      "loss": 1.4106,
      "step": 580
    },
    {
      "epoch": 0.23153144314725793,
      "grad_norm": 0.69140625,
      "learning_rate": 0.00019883874593059825,
      "loss": 1.7042,
      "step": 590
    },
    {
      "epoch": 0.23545570489551654,
      "grad_norm": 1.4609375,
      "learning_rate": 0.0001987734698191546,
      "loss": 1.4864,
      "step": 600
    },
    {
      "epoch": 0.23937996664377514,
      "grad_norm": 0.6171875,
      "learning_rate": 0.00019870642019525678,
      "loss": 1.6671,
      "step": 610
    },
    {
      "epoch": 0.24330422839203375,
      "grad_norm": 0.921875,
      "learning_rate": 0.00019863759826280447,
      "loss": 1.6773,
      "step": 620
    },
    {
      "epoch": 0.24722849014029236,
      "grad_norm": 0.515625,
      "learning_rate": 0.00019856700525751965,
      "loss": 1.4164,
      "step": 630
    },
    {
      "epoch": 0.25115275188855096,
      "grad_norm": 0.66796875,
      "learning_rate": 0.00019849464244692467,
      "loss": 1.7153,
      "step": 640
    },
    {
      "epoch": 0.25507701363680957,
      "grad_norm": 1.4921875,
      "learning_rate": 0.00019842051113031924,
      "loss": 1.4368,
      "step": 650
    },
    {
      "epoch": 0.2590012753850682,
      "grad_norm": 0.66796875,
      "learning_rate": 0.00019834461263875728,
      "loss": 1.6844,
      "step": 660
    },
    {
      "epoch": 0.2629255371333268,
      "grad_norm": 0.93359375,
      "learning_rate": 0.00019826694833502295,
      "loss": 1.6616,
      "step": 670
    },
    {
      "epoch": 0.2668497988815854,
      "grad_norm": 0.49609375,
      "learning_rate": 0.00019818751961360622,
      "loss": 1.4094,
      "step": 680
    },
    {
      "epoch": 0.270774060629844,
      "grad_norm": 0.6640625,
      "learning_rate": 0.00019810632790067772,
      "loss": 1.7346,
      "step": 690
    },
    {
      "epoch": 0.2746983223781026,
      "grad_norm": 1.4453125,
      "learning_rate": 0.0001980233746540633,
      "loss": 1.4831,
      "step": 700
    },
    {
      "epoch": 0.2786225841263612,
      "grad_norm": 0.60546875,
      "learning_rate": 0.00019793866136321773,
      "loss": 1.6569,
      "step": 710
    },
    {
      "epoch": 0.2825468458746198,
      "grad_norm": 0.90234375,
      "learning_rate": 0.00019785218954919797,
      "loss": 1.6338,
      "step": 720
    },
    {
      "epoch": 0.2864711076228785,
      "grad_norm": 0.451171875,
      "learning_rate": 0.00019776396076463595,
      "loss": 1.3856,
      "step": 730
    },
    {
      "epoch": 0.2903953693711371,
      "grad_norm": 0.6875,
      "learning_rate": 0.00019767397659371057,
      "loss": 1.7281,
      "step": 740
    },
    {
      "epoch": 0.2943196311193957,
      "grad_norm": 1.3671875,
      "learning_rate": 0.0001975822386521193,
      "loss": 1.4777,
      "step": 750
    },
    {
      "epoch": 0.2982438928676543,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00019748874858704923,
      "loss": 1.6089,
      "step": 760
    },
    {
      "epoch": 0.3021681546159129,
      "grad_norm": 0.91015625,
      "learning_rate": 0.0001973935080771474,
      "loss": 1.6463,
      "step": 770
    },
    {
      "epoch": 0.3060924163641715,
      "grad_norm": 0.5390625,
      "learning_rate": 0.00019729651883249074,
      "loss": 1.4142,
      "step": 780
    },
    {
      "epoch": 0.3100166781124301,
      "grad_norm": 0.69140625,
      "learning_rate": 0.00019719778259455532,
      "loss": 1.6992,
      "step": 790
    },
    {
      "epoch": 0.3139409398606887,
      "grad_norm": 1.3515625,
      "learning_rate": 0.00019709730113618508,
      "loss": 1.4576,
      "step": 800
    },
    {
      "epoch": 0.3178652016089473,
      "grad_norm": 0.59375,
      "learning_rate": 0.00019699507626156,
      "loss": 1.6192,
      "step": 810
    },
    {
      "epoch": 0.32178946335720593,
      "grad_norm": 0.87890625,
      "learning_rate": 0.0001968911098061637,
      "loss": 1.6478,
      "step": 820
    },
    {
      "epoch": 0.32571372510546454,
      "grad_norm": 0.482421875,
      "learning_rate": 0.0001967854036367506,
      "loss": 1.3934,
      "step": 830
    },
    {
      "epoch": 0.32963798685372314,
      "grad_norm": 0.671875,
      "learning_rate": 0.00019667795965131214,
      "loss": 1.7188,
      "step": 840
    },
    {
      "epoch": 0.33356224860198175,
      "grad_norm": 1.4921875,
      "learning_rate": 0.00019656877977904298,
      "loss": 1.462,
      "step": 850
    },
    {
      "epoch": 0.33748651035024035,
      "grad_norm": 0.6015625,
      "learning_rate": 0.00019645786598030616,
      "loss": 1.6335,
      "step": 860
    },
    {
      "epoch": 0.34141077209849896,
      "grad_norm": 0.859375,
      "learning_rate": 0.00019634522024659804,
      "loss": 1.6204,
      "step": 870
    },
    {
      "epoch": 0.34533503384675757,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00019623084460051248,
      "loss": 1.3418,
      "step": 880
    },
    {
      "epoch": 0.3492592955950162,
      "grad_norm": 0.66796875,
      "learning_rate": 0.00019611474109570444,
      "loss": 1.679,
      "step": 890
    },
    {
      "epoch": 0.3531835573432748,
      "grad_norm": 1.4453125,
      "learning_rate": 0.00019599691181685334,
      "loss": 1.4345,
      "step": 900
    },
    {
      "epoch": 0.3571078190915334,
      "grad_norm": 0.578125,
      "learning_rate": 0.00019587735887962533,
      "loss": 1.6147,
      "step": 910
    },
    {
      "epoch": 0.361032080839792,
      "grad_norm": 0.890625,
      "learning_rate": 0.00019575608443063558,
      "loss": 1.6158,
      "step": 920
    },
    {
      "epoch": 0.3649563425880506,
      "grad_norm": 0.44921875,
      "learning_rate": 0.00019563309064740954,
      "loss": 1.3699,
      "step": 930
    },
    {
      "epoch": 0.36888060433630926,
      "grad_norm": 0.640625,
      "learning_rate": 0.0001955083797383439,
      "loss": 1.6874,
      "step": 940
    },
    {
      "epoch": 0.37280486608456787,
      "grad_norm": 1.3671875,
      "learning_rate": 0.000195381953942667,
      "loss": 1.4276,
      "step": 950
    },
    {
      "epoch": 0.37672912783282647,
      "grad_norm": 0.58203125,
      "learning_rate": 0.00019525381553039852,
      "loss": 1.6062,
      "step": 960
    },
    {
      "epoch": 0.3806533895810851,
      "grad_norm": 0.91796875,
      "learning_rate": 0.00019512396680230882,
      "loss": 1.6286,
      "step": 970
    },
    {
      "epoch": 0.3845776513293437,
      "grad_norm": 0.53515625,
      "learning_rate": 0.0001949924100898776,
      "loss": 1.3761,
      "step": 980
    },
    {
      "epoch": 0.3885019130776023,
      "grad_norm": 0.66015625,
      "learning_rate": 0.00019485914775525193,
      "loss": 1.6746,
      "step": 990
    },
    {
      "epoch": 0.3924261748258609,
      "grad_norm": 1.3203125,
      "learning_rate": 0.000194724182191204,
      "loss": 1.4136,
      "step": 1000
    },
    {
      "epoch": 0.3963504365741195,
      "grad_norm": 0.640625,
      "learning_rate": 0.0001945875158210881,
      "loss": 1.5886,
      "step": 1010
    },
    {
      "epoch": 0.4002746983223781,
      "grad_norm": 0.91796875,
      "learning_rate": 0.00019444915109879704,
      "loss": 1.6172,
      "step": 1020
    },
    {
      "epoch": 0.4041989600706367,
      "grad_norm": 0.45703125,
      "learning_rate": 0.00019430909050871815,
      "loss": 1.3672,
      "step": 1030
    },
    {
      "epoch": 0.4081232218188953,
      "grad_norm": 0.62890625,
      "learning_rate": 0.00019416733656568868,
      "loss": 1.6882,
      "step": 1040
    },
    {
      "epoch": 0.4120474835671539,
      "grad_norm": 1.2421875,
      "learning_rate": 0.00019402389181495062,
      "loss": 1.3849,
      "step": 1050
    },
    {
      "epoch": 0.41597174531541253,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00019387875883210507,
      "loss": 1.5925,
      "step": 1060
    },
    {
      "epoch": 0.41989600706367114,
      "grad_norm": 0.8671875,
      "learning_rate": 0.00019373194022306576,
      "loss": 1.5587,
      "step": 1070
    },
    {
      "epoch": 0.42382026881192975,
      "grad_norm": 0.427734375,
      "learning_rate": 0.0001935834386240127,
      "loss": 1.3806,
      "step": 1080
    },
    {
      "epoch": 0.42774453056018835,
      "grad_norm": 0.6484375,
      "learning_rate": 0.00019343325670134428,
      "loss": 1.6765,
      "step": 1090
    },
    {
      "epoch": 0.43166879230844696,
      "grad_norm": 1.2578125,
      "learning_rate": 0.00019328139715162995,
      "loss": 1.4104,
      "step": 1100
    },
    {
      "epoch": 0.43559305405670556,
      "grad_norm": 0.56640625,
      "learning_rate": 0.00019312786270156136,
      "loss": 1.5651,
      "step": 1110
    },
    {
      "epoch": 0.43951731580496417,
      "grad_norm": 0.85546875,
      "learning_rate": 0.0001929726561079037,
      "loss": 1.5937,
      "step": 1120
    },
    {
      "epoch": 0.4434415775532228,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00019281578015744604,
      "loss": 1.3398,
      "step": 1130
    },
    {
      "epoch": 0.4473658393014814,
      "grad_norm": 0.640625,
      "learning_rate": 0.00019265723766695133,
      "loss": 1.646,
      "step": 1140
    },
    {
      "epoch": 0.45129010104974004,
      "grad_norm": 1.2109375,
      "learning_rate": 0.00019249703148310586,
      "loss": 1.401,
      "step": 1150
    },
    {
      "epoch": 0.45521436279799865,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00019233516448246814,
      "loss": 1.6092,
      "step": 1160
    },
    {
      "epoch": 0.45913862454625726,
      "grad_norm": 0.82421875,
      "learning_rate": 0.00019217163957141716,
      "loss": 1.5747,
      "step": 1170
    },
    {
      "epoch": 0.46306288629451586,
      "grad_norm": 0.41796875,
      "learning_rate": 0.00019200645968610035,
      "loss": 1.3217,
      "step": 1180
    },
    {
      "epoch": 0.46698714804277447,
      "grad_norm": 0.6171875,
      "learning_rate": 0.0001918396277923807,
      "loss": 1.6309,
      "step": 1190
    },
    {
      "epoch": 0.4709114097910331,
      "grad_norm": 1.1796875,
      "learning_rate": 0.0001916711468857837,
      "loss": 1.367,
      "step": 1200
    },
    {
      "epoch": 0.4748356715392917,
      "grad_norm": 0.58203125,
      "learning_rate": 0.00019150101999144338,
      "loss": 1.5816,
      "step": 1210
    },
    {
      "epoch": 0.4787599332875503,
      "grad_norm": 0.8515625,
      "learning_rate": 0.00019132925016404804,
      "loss": 1.5709,
      "step": 1220
    },
    {
      "epoch": 0.4826841950358089,
      "grad_norm": 0.408203125,
      "learning_rate": 0.0001911558404877855,
      "loss": 1.3288,
      "step": 1230
    },
    {
      "epoch": 0.4866084567840675,
      "grad_norm": 0.64453125,
      "learning_rate": 0.0001909807940762876,
      "loss": 1.6266,
      "step": 1240
    },
    {
      "epoch": 0.4905327185323261,
      "grad_norm": 1.3046875,
      "learning_rate": 0.00019080411407257431,
      "loss": 1.3431,
      "step": 1250
    },
    {
      "epoch": 0.4944569802805847,
      "grad_norm": 0.58203125,
      "learning_rate": 0.00019062580364899732,
      "loss": 1.5936,
      "step": 1260
    },
    {
      "epoch": 0.4983812420288433,
      "grad_norm": 0.84375,
      "learning_rate": 0.00019044586600718322,
      "loss": 1.6004,
      "step": 1270
    },
    {
      "epoch": 0.5023055037771019,
      "grad_norm": 0.44921875,
      "learning_rate": 0.00019026430437797567,
      "loss": 1.3021,
      "step": 1280
    },
    {
      "epoch": 0.5062297655253606,
      "grad_norm": 0.625,
      "learning_rate": 0.00019008112202137777,
      "loss": 1.641,
      "step": 1290
    },
    {
      "epoch": 0.5101540272736191,
      "grad_norm": 1.234375,
      "learning_rate": 0.0001898963222264932,
      "loss": 1.3817,
      "step": 1300
    },
    {
      "epoch": 0.5140782890218778,
      "grad_norm": 0.52734375,
      "learning_rate": 0.00018970990831146744,
      "loss": 1.5528,
      "step": 1310
    },
    {
      "epoch": 0.5180025507701363,
      "grad_norm": 0.8515625,
      "learning_rate": 0.00018952188362342804,
      "loss": 1.5717,
      "step": 1320
    },
    {
      "epoch": 0.521926812518395,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00018933225153842446,
      "loss": 1.3188,
      "step": 1330
    },
    {
      "epoch": 0.5258510742666536,
      "grad_norm": 0.61328125,
      "learning_rate": 0.00018914101546136766,
      "loss": 1.6275,
      "step": 1340
    },
    {
      "epoch": 0.5297753360149122,
      "grad_norm": 1.296875,
      "learning_rate": 0.00018894817882596877,
      "loss": 1.3886,
      "step": 1350
    },
    {
      "epoch": 0.5336995977631708,
      "grad_norm": 0.5625,
      "learning_rate": 0.00018875374509467755,
      "loss": 1.5805,
      "step": 1360
    },
    {
      "epoch": 0.5376238595114294,
      "grad_norm": 0.84765625,
      "learning_rate": 0.00018855771775862012,
      "loss": 1.572,
      "step": 1370
    },
    {
      "epoch": 0.541548121259688,
      "grad_norm": 0.4296875,
      "learning_rate": 0.00018836010033753637,
      "loss": 1.3052,
      "step": 1380
    },
    {
      "epoch": 0.5454723830079466,
      "grad_norm": 0.625,
      "learning_rate": 0.00018816089637971675,
      "loss": 1.642,
      "step": 1390
    },
    {
      "epoch": 0.5493966447562052,
      "grad_norm": 1.234375,
      "learning_rate": 0.0001879601094619385,
      "loss": 1.3427,
      "step": 1400
    },
    {
      "epoch": 0.5533209065044639,
      "grad_norm": 0.5390625,
      "learning_rate": 0.0001877577431894015,
      "loss": 1.5367,
      "step": 1410
    },
    {
      "epoch": 0.5572451682527224,
      "grad_norm": 0.8046875,
      "learning_rate": 0.00018755380119566341,
      "loss": 1.562,
      "step": 1420
    },
    {
      "epoch": 0.5611694300009811,
      "grad_norm": 0.45703125,
      "learning_rate": 0.0001873482871425747,
      "loss": 1.3385,
      "step": 1430
    },
    {
      "epoch": 0.5650936917492396,
      "grad_norm": 0.609375,
      "learning_rate": 0.0001871412047202125,
      "loss": 1.6053,
      "step": 1440
    },
    {
      "epoch": 0.5690179534974983,
      "grad_norm": 1.2578125,
      "learning_rate": 0.00018693255764681474,
      "loss": 1.3783,
      "step": 1450
    },
    {
      "epoch": 0.572942215245757,
      "grad_norm": 0.5546875,
      "learning_rate": 0.0001867223496687131,
      "loss": 1.5609,
      "step": 1460
    },
    {
      "epoch": 0.5768664769940155,
      "grad_norm": 0.84375,
      "learning_rate": 0.00018651058456026592,
      "loss": 1.5268,
      "step": 1470
    },
    {
      "epoch": 0.5807907387422742,
      "grad_norm": 0.404296875,
      "learning_rate": 0.00018629726612379033,
      "loss": 1.3114,
      "step": 1480
    },
    {
      "epoch": 0.5847150004905327,
      "grad_norm": 0.5859375,
      "learning_rate": 0.000186082398189494,
      "loss": 1.6552,
      "step": 1490
    },
    {
      "epoch": 0.5886392622387914,
      "grad_norm": 1.1328125,
      "learning_rate": 0.00018586598461540644,
      "loss": 1.3315,
      "step": 1500
    },
    {
      "epoch": 0.5925635239870499,
      "grad_norm": 0.546875,
      "learning_rate": 0.00018564802928730965,
      "loss": 1.5536,
      "step": 1510
    },
    {
      "epoch": 0.5964877857353086,
      "grad_norm": 0.859375,
      "learning_rate": 0.00018542853611866825,
      "loss": 1.5711,
      "step": 1520
    },
    {
      "epoch": 0.6004120474835671,
      "grad_norm": 0.421875,
      "learning_rate": 0.00018520750905055947,
      "loss": 1.2923,
      "step": 1530
    },
    {
      "epoch": 0.6043363092318258,
      "grad_norm": 0.62109375,
      "learning_rate": 0.00018498495205160226,
      "loss": 1.5839,
      "step": 1540
    },
    {
      "epoch": 0.6082605709800843,
      "grad_norm": 1.2265625,
      "learning_rate": 0.00018476086911788585,
      "loss": 1.3458,
      "step": 1550
    },
    {
      "epoch": 0.612184832728343,
      "grad_norm": 0.5390625,
      "learning_rate": 0.00018453526427289835,
      "loss": 1.5177,
      "step": 1560
    },
    {
      "epoch": 0.6161090944766016,
      "grad_norm": 0.8125,
      "learning_rate": 0.0001843081415674542,
      "loss": 1.533,
      "step": 1570
    },
    {
      "epoch": 0.6200333562248602,
      "grad_norm": 0.435546875,
      "learning_rate": 0.00018407950507962165,
      "loss": 1.282,
      "step": 1580
    },
    {
      "epoch": 0.6239576179731188,
      "grad_norm": 0.609375,
      "learning_rate": 0.0001838493589146494,
      "loss": 1.6295,
      "step": 1590
    },
    {
      "epoch": 0.6278818797213774,
      "grad_norm": 1.203125,
      "learning_rate": 0.00018361770720489286,
      "loss": 1.3054,
      "step": 1600
    },
    {
      "epoch": 0.631806141469636,
      "grad_norm": 0.55859375,
      "learning_rate": 0.0001833845541097402,
      "loss": 1.5162,
      "step": 1610
    },
    {
      "epoch": 0.6357304032178946,
      "grad_norm": 0.84375,
      "learning_rate": 0.00018314990381553727,
      "loss": 1.5543,
      "step": 1620
    },
    {
      "epoch": 0.6396546649661532,
      "grad_norm": 0.4296875,
      "learning_rate": 0.00018291376053551292,
      "loss": 1.3095,
      "step": 1630
    },
    {
      "epoch": 0.6435789267144119,
      "grad_norm": 0.60546875,
      "learning_rate": 0.0001826761285097029,
      "loss": 1.6443,
      "step": 1640
    },
    {
      "epoch": 0.6475031884626704,
      "grad_norm": 1.28125,
      "learning_rate": 0.0001824370120048739,
      "loss": 1.3545,
      "step": 1650
    },
    {
      "epoch": 0.6514274502109291,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00018219641531444714,
      "loss": 1.5222,
      "step": 1660
    },
    {
      "epoch": 0.6553517119591877,
      "grad_norm": 0.80078125,
      "learning_rate": 0.0001819543427584209,
      "loss": 1.5274,
      "step": 1670
    },
    {
      "epoch": 0.6592759737074463,
      "grad_norm": 0.46875,
      "learning_rate": 0.0001817107986832932,
      "loss": 1.3046,
      "step": 1680
    },
    {
      "epoch": 0.663200235455705,
      "grad_norm": 0.59765625,
      "learning_rate": 0.00018146578746198376,
      "loss": 1.6112,
      "step": 1690
    },
    {
      "epoch": 0.6671244972039635,
      "grad_norm": 1.2109375,
      "learning_rate": 0.00018121931349375538,
      "loss": 1.3335,
      "step": 1700
    },
    {
      "epoch": 0.6710487589522222,
      "grad_norm": 0.546875,
      "learning_rate": 0.000180971381204135,
      "loss": 1.5279,
      "step": 1710
    },
    {
      "epoch": 0.6749730207004807,
      "grad_norm": 0.79296875,
      "learning_rate": 0.00018072199504483427,
      "loss": 1.4884,
      "step": 1720
    },
    {
      "epoch": 0.6788972824487394,
      "grad_norm": 0.44140625,
      "learning_rate": 0.00018047115949366952,
      "loss": 1.2858,
      "step": 1730
    },
    {
      "epoch": 0.6828215441969979,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00018021887905448147,
      "loss": 1.6059,
      "step": 1740
    },
    {
      "epoch": 0.6867458059452566,
      "grad_norm": 1.1328125,
      "learning_rate": 0.0001799651582570543,
      "loss": 1.3141,
      "step": 1750
    },
    {
      "epoch": 0.6906700676935151,
      "grad_norm": 0.546875,
      "learning_rate": 0.00017971000165703434,
      "loss": 1.5152,
      "step": 1760
    },
    {
      "epoch": 0.6945943294417738,
      "grad_norm": 0.80078125,
      "learning_rate": 0.00017945341383584818,
      "loss": 1.5056,
      "step": 1770
    },
    {
      "epoch": 0.6985185911900323,
      "grad_norm": 0.435546875,
      "learning_rate": 0.00017919539940062067,
      "loss": 1.2746,
      "step": 1780
    },
    {
      "epoch": 0.702442852938291,
      "grad_norm": 0.609375,
      "learning_rate": 0.00017893596298409182,
      "loss": 1.6109,
      "step": 1790
    },
    {
      "epoch": 0.7063671146865496,
      "grad_norm": 1.1953125,
      "learning_rate": 0.00017867510924453393,
      "loss": 1.3086,
      "step": 1800
    },
    {
      "epoch": 0.7102913764348082,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00017841284286566783,
      "loss": 1.5316,
      "step": 1810
    },
    {
      "epoch": 0.7142156381830668,
      "grad_norm": 0.79296875,
      "learning_rate": 0.00017814916855657872,
      "loss": 1.5172,
      "step": 1820
    },
    {
      "epoch": 0.7181398999313254,
      "grad_norm": 0.40625,
      "learning_rate": 0.00017788409105163176,
      "loss": 1.2837,
      "step": 1830
    },
    {
      "epoch": 0.722064161679584,
      "grad_norm": 0.58984375,
      "learning_rate": 0.00017761761511038695,
      "loss": 1.5799,
      "step": 1840
    },
    {
      "epoch": 0.7259884234278426,
      "grad_norm": 1.203125,
      "learning_rate": 0.0001773497455175137,
      "loss": 1.3279,
      "step": 1850
    },
    {
      "epoch": 0.7299126851761012,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00017708048708270496,
      "loss": 1.5233,
      "step": 1860
    },
    {
      "epoch": 0.7338369469243599,
      "grad_norm": 0.78515625,
      "learning_rate": 0.00017680984464059078,
      "loss": 1.5245,
      "step": 1870
    },
    {
      "epoch": 0.7377612086726185,
      "grad_norm": 0.39453125,
      "learning_rate": 0.00017653782305065158,
      "loss": 1.2797,
      "step": 1880
    },
    {
      "epoch": 0.7416854704208771,
      "grad_norm": 0.58984375,
      "learning_rate": 0.00017626442719713084,
      "loss": 1.6374,
      "step": 1890
    },
    {
      "epoch": 0.7456097321691357,
      "grad_norm": 1.171875,
      "learning_rate": 0.00017598966198894745,
      "loss": 1.3099,
      "step": 1900
    },
    {
      "epoch": 0.7495339939173943,
      "grad_norm": 0.515625,
      "learning_rate": 0.00017571353235960752,
      "loss": 1.5077,
      "step": 1910
    },
    {
      "epoch": 0.7534582556656529,
      "grad_norm": 0.80078125,
      "learning_rate": 0.00017543604326711592,
      "loss": 1.5213,
      "step": 1920
    },
    {
      "epoch": 0.7573825174139115,
      "grad_norm": 0.376953125,
      "learning_rate": 0.00017515719969388697,
      "loss": 1.3016,
      "step": 1930
    },
    {
      "epoch": 0.7613067791621702,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00017487700664665535,
      "loss": 1.5944,
      "step": 1940
    },
    {
      "epoch": 0.7652310409104287,
      "grad_norm": 1.1640625,
      "learning_rate": 0.00017459546915638593,
      "loss": 1.3176,
      "step": 1950
    },
    {
      "epoch": 0.7691553026586874,
      "grad_norm": 0.55078125,
      "learning_rate": 0.0001743125922781836,
      "loss": 1.5129,
      "step": 1960
    },
    {
      "epoch": 0.7730795644069459,
      "grad_norm": 0.796875,
      "learning_rate": 0.00017402838109120228,
      "loss": 1.4998,
      "step": 1970
    },
    {
      "epoch": 0.7770038261552046,
      "grad_norm": 0.392578125,
      "learning_rate": 0.0001737428406985541,
      "loss": 1.2621,
      "step": 1980
    },
    {
      "epoch": 0.7809280879034631,
      "grad_norm": 0.60546875,
      "learning_rate": 0.00017345597622721727,
      "loss": 1.5645,
      "step": 1990
    },
    {
      "epoch": 0.7848523496517218,
      "grad_norm": 1.1796875,
      "learning_rate": 0.00017316779282794458,
      "loss": 1.304,
      "step": 2000
    },
    {
      "epoch": 0.7887766113999803,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00017287829567517047,
      "loss": 1.4972,
      "step": 2010
    },
    {
      "epoch": 0.792700873148239,
      "grad_norm": 0.8125,
      "learning_rate": 0.00017258748996691828,
      "loss": 1.4643,
      "step": 2020
    },
    {
      "epoch": 0.7966251348964976,
      "grad_norm": 0.38671875,
      "learning_rate": 0.00017229538092470708,
      "loss": 1.2648,
      "step": 2030
    },
    {
      "epoch": 0.8005493966447562,
      "grad_norm": 0.59375,
      "learning_rate": 0.00017200197379345751,
      "loss": 1.5823,
      "step": 2040
    },
    {
      "epoch": 0.8044736583930148,
      "grad_norm": 1.1796875,
      "learning_rate": 0.00017170727384139807,
      "loss": 1.3042,
      "step": 2050
    },
    {
      "epoch": 0.8083979201412734,
      "grad_norm": 0.5234375,
      "learning_rate": 0.00017141128635997027,
      "loss": 1.4864,
      "step": 2060
    },
    {
      "epoch": 0.812322181889532,
      "grad_norm": 0.7890625,
      "learning_rate": 0.00017111401666373356,
      "loss": 1.5021,
      "step": 2070
    },
    {
      "epoch": 0.8162464436377906,
      "grad_norm": 0.41796875,
      "learning_rate": 0.00017081547009027012,
      "loss": 1.271,
      "step": 2080
    },
    {
      "epoch": 0.8201707053860493,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00017051565200008886,
      "loss": 1.5518,
      "step": 2090
    },
    {
      "epoch": 0.8240949671343079,
      "grad_norm": 1.1171875,
      "learning_rate": 0.00017021456777652926,
      "loss": 1.292,
      "step": 2100
    },
    {
      "epoch": 0.8280192288825665,
      "grad_norm": 0.5234375,
      "learning_rate": 0.00016991222282566464,
      "loss": 1.4496,
      "step": 2110
    },
    {
      "epoch": 0.8319434906308251,
      "grad_norm": 0.77734375,
      "learning_rate": 0.00016960862257620507,
      "loss": 1.4803,
      "step": 2120
    },
    {
      "epoch": 0.8358677523790837,
      "grad_norm": 0.39453125,
      "learning_rate": 0.00016930377247940004,
      "loss": 1.2479,
      "step": 2130
    },
    {
      "epoch": 0.8397920141273423,
      "grad_norm": 0.58203125,
      "learning_rate": 0.0001689976780089405,
      "loss": 1.5399,
      "step": 2140
    },
    {
      "epoch": 0.8437162758756009,
      "grad_norm": 1.28125,
      "learning_rate": 0.00016869034466086044,
      "loss": 1.2961,
      "step": 2150
    },
    {
      "epoch": 0.8476405376238595,
      "grad_norm": 0.51953125,
      "learning_rate": 0.00016838177795343846,
      "loss": 1.4839,
      "step": 2160
    },
    {
      "epoch": 0.8515647993721182,
      "grad_norm": 0.8046875,
      "learning_rate": 0.00016807198342709857,
      "loss": 1.4934,
      "step": 2170
    },
    {
      "epoch": 0.8554890611203767,
      "grad_norm": 0.392578125,
      "learning_rate": 0.0001677609666443105,
      "loss": 1.2432,
      "step": 2180
    },
    {
      "epoch": 0.8594133228686354,
      "grad_norm": 0.609375,
      "learning_rate": 0.00016744873318949032,
      "loss": 1.6118,
      "step": 2190
    },
    {
      "epoch": 0.8633375846168939,
      "grad_norm": 1.265625,
      "learning_rate": 0.00016713528866889966,
      "loss": 1.3155,
      "step": 2200
    },
    {
      "epoch": 0.8672618463651526,
      "grad_norm": 0.53515625,
      "learning_rate": 0.00016682063871054534,
      "loss": 1.4913,
      "step": 2210
    },
    {
      "epoch": 0.8711861081134111,
      "grad_norm": 0.78515625,
      "learning_rate": 0.00016650478896407825,
      "loss": 1.4827,
      "step": 2220
    },
    {
      "epoch": 0.8751103698616698,
      "grad_norm": 0.380859375,
      "learning_rate": 0.00016618774510069185,
      "loss": 1.2543,
      "step": 2230
    },
    {
      "epoch": 0.8790346316099283,
      "grad_norm": 0.609375,
      "learning_rate": 0.00016586951281302045,
      "loss": 1.5636,
      "step": 2240
    },
    {
      "epoch": 0.882958893358187,
      "grad_norm": 1.1875,
      "learning_rate": 0.0001655500978150369,
      "loss": 1.2843,
      "step": 2250
    },
    {
      "epoch": 0.8868831551064456,
      "grad_norm": 0.53515625,
      "learning_rate": 0.00016522950584195,
      "loss": 1.4772,
      "step": 2260
    },
    {
      "epoch": 0.8908074168547042,
      "grad_norm": 0.8046875,
      "learning_rate": 0.00016490774265010168,
      "loss": 1.4683,
      "step": 2270
    },
    {
      "epoch": 0.8947316786029628,
      "grad_norm": 0.416015625,
      "learning_rate": 0.0001645848140168633,
      "loss": 1.2446,
      "step": 2280
    },
    {
      "epoch": 0.8986559403512214,
      "grad_norm": 0.625,
      "learning_rate": 0.00016426072574053237,
      "loss": 1.5365,
      "step": 2290
    },
    {
      "epoch": 0.9025802020994801,
      "grad_norm": 1.1875,
      "learning_rate": 0.00016393548364022803,
      "loss": 1.2834,
      "step": 2300
    },
    {
      "epoch": 0.9065044638477386,
      "grad_norm": 0.53515625,
      "learning_rate": 0.0001636090935557868,
      "loss": 1.489,
      "step": 2310
    },
    {
      "epoch": 0.9104287255959973,
      "grad_norm": 0.765625,
      "learning_rate": 0.00016328156134765763,
      "loss": 1.4761,
      "step": 2320
    },
    {
      "epoch": 0.9143529873442559,
      "grad_norm": 0.4140625,
      "learning_rate": 0.00016295289289679674,
      "loss": 1.2639,
      "step": 2330
    },
    {
      "epoch": 0.9182772490925145,
      "grad_norm": 0.609375,
      "learning_rate": 0.000162623094104562,
      "loss": 1.5386,
      "step": 2340
    },
    {
      "epoch": 0.9222015108407731,
      "grad_norm": 1.2109375,
      "learning_rate": 0.00016229217089260692,
      "loss": 1.2766,
      "step": 2350
    },
    {
      "epoch": 0.9261257725890317,
      "grad_norm": 0.55078125,
      "learning_rate": 0.00016196012920277438,
      "loss": 1.4832,
      "step": 2360
    },
    {
      "epoch": 0.9300500343372903,
      "grad_norm": 0.78515625,
      "learning_rate": 0.00016162697499698986,
      "loss": 1.4795,
      "step": 2370
    },
    {
      "epoch": 0.9339742960855489,
      "grad_norm": 0.400390625,
      "learning_rate": 0.0001612927142571546,
      "loss": 1.2476,
      "step": 2380
    },
    {
      "epoch": 0.9378985578338075,
      "grad_norm": 0.59765625,
      "learning_rate": 0.00016095735298503788,
      "loss": 1.5386,
      "step": 2390
    },
    {
      "epoch": 0.9418228195820662,
      "grad_norm": 1.1640625,
      "learning_rate": 0.00016062089720216956,
      "loss": 1.2613,
      "step": 2400
    },
    {
      "epoch": 0.9457470813303247,
      "grad_norm": 0.5390625,
      "learning_rate": 0.0001602833529497318,
      "loss": 1.4806,
      "step": 2410
    },
    {
      "epoch": 0.9496713430785834,
      "grad_norm": 0.76953125,
      "learning_rate": 0.00015994472628845055,
      "loss": 1.473,
      "step": 2420
    },
    {
      "epoch": 0.9535956048268419,
      "grad_norm": 0.40625,
      "learning_rate": 0.0001596050232984868,
      "loss": 1.2207,
      "step": 2430
    },
    {
      "epoch": 0.9575198665751006,
      "grad_norm": 0.609375,
      "learning_rate": 0.00015926425007932747,
      "loss": 1.5097,
      "step": 2440
    },
    {
      "epoch": 0.9614441283233591,
      "grad_norm": 1.1640625,
      "learning_rate": 0.00015892241274967578,
      "loss": 1.2636,
      "step": 2450
    },
    {
      "epoch": 0.9653683900716178,
      "grad_norm": 0.53125,
      "learning_rate": 0.00015857951744734146,
      "loss": 1.5006,
      "step": 2460
    },
    {
      "epoch": 0.9692926518198763,
      "grad_norm": 0.78515625,
      "learning_rate": 0.00015823557032913044,
      "loss": 1.4665,
      "step": 2470
    },
    {
      "epoch": 0.973216913568135,
      "grad_norm": 0.375,
      "learning_rate": 0.00015789057757073445,
      "loss": 1.23,
      "step": 2480
    },
    {
      "epoch": 0.9771411753163936,
      "grad_norm": 0.58203125,
      "learning_rate": 0.00015754454536662,
      "loss": 1.5229,
      "step": 2490
    },
    {
      "epoch": 0.9810654370646522,
      "grad_norm": 1.140625,
      "learning_rate": 0.0001571974799299172,
      "loss": 1.272,
      "step": 2500
    },
    {
      "epoch": 0.9849896988129109,
      "grad_norm": 0.515625,
      "learning_rate": 0.0001568493874923084,
      "loss": 1.4653,
      "step": 2510
    },
    {
      "epoch": 0.9889139605611694,
      "grad_norm": 0.77734375,
      "learning_rate": 0.00015650027430391585,
      "loss": 1.4253,
      "step": 2520
    },
    {
      "epoch": 0.9928382223094281,
      "grad_norm": 0.423828125,
      "learning_rate": 0.0001561501466331899,
      "loss": 1.2399,
      "step": 2530
    },
    {
      "epoch": 0.9967624840576866,
      "grad_norm": 0.6484375,
      "learning_rate": 0.00015579901076679623,
      "loss": 1.5164,
      "step": 2540
    },
    {
      "epoch": 0.9999018934562935,
      "eval_loss": 1.2656188011169434,
      "eval_runtime": 528.4959,
      "eval_samples_per_second": 47.304,
      "eval_steps_per_second": 11.826,
      "step": 2548
    },
    {
      "epoch": 1.0006867458059452,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00015544687300950307,
      "loss": 1.2517,
      "step": 2550
    },
    {
      "epoch": 1.0046110075542038,
      "grad_norm": 0.55859375,
      "learning_rate": 0.0001550937396840679,
      "loss": 1.3853,
      "step": 2560
    },
    {
      "epoch": 1.0085352693024625,
      "grad_norm": 0.80078125,
      "learning_rate": 0.00015473961713112405,
      "loss": 1.0944,
      "step": 2570
    },
    {
      "epoch": 1.0124595310507212,
      "grad_norm": 0.60546875,
      "learning_rate": 0.00015438451170906673,
      "loss": 1.1831,
      "step": 2580
    },
    {
      "epoch": 1.0163837927989796,
      "grad_norm": 0.6796875,
      "learning_rate": 0.00015402842979393884,
      "loss": 1.2688,
      "step": 2590
    },
    {
      "epoch": 1.0203080545472383,
      "grad_norm": 0.7265625,
      "learning_rate": 0.00015367137777931673,
      "loss": 0.9534,
      "step": 2600
    },
    {
      "epoch": 1.024232316295497,
      "grad_norm": 0.55078125,
      "learning_rate": 0.00015331336207619506,
      "loss": 1.3796,
      "step": 2610
    },
    {
      "epoch": 1.0281565780437556,
      "grad_norm": 0.7890625,
      "learning_rate": 0.000152954389112872,
      "loss": 1.0844,
      "step": 2620
    },
    {
      "epoch": 1.032080839792014,
      "grad_norm": 0.5234375,
      "learning_rate": 0.00015259446533483356,
      "loss": 1.1626,
      "step": 2630
    },
    {
      "epoch": 1.0360051015402727,
      "grad_norm": 0.68359375,
      "learning_rate": 0.00015223359720463795,
      "loss": 1.2734,
      "step": 2640
    },
    {
      "epoch": 1.0399293632885314,
      "grad_norm": 0.6953125,
      "learning_rate": 0.0001518717912017997,
      "loss": 0.9391,
      "step": 2650
    },
    {
      "epoch": 1.04385362503679,
      "grad_norm": 0.53515625,
      "learning_rate": 0.00015150905382267296,
      "loss": 1.3889,
      "step": 2660
    },
    {
      "epoch": 1.0477778867850485,
      "grad_norm": 0.84375,
      "learning_rate": 0.00015114539158033529,
      "loss": 1.1101,
      "step": 2670
    },
    {
      "epoch": 1.0517021485333071,
      "grad_norm": 0.5234375,
      "learning_rate": 0.00015078081100447036,
      "loss": 1.171,
      "step": 2680
    },
    {
      "epoch": 1.0556264102815658,
      "grad_norm": 0.67578125,
      "learning_rate": 0.00015041531864125082,
      "loss": 1.2797,
      "step": 2690
    },
    {
      "epoch": 1.0595506720298244,
      "grad_norm": 0.65625,
      "learning_rate": 0.0001500489210532209,
      "loss": 0.9225,
      "step": 2700
    },
    {
      "epoch": 1.063474933778083,
      "grad_norm": 0.578125,
      "learning_rate": 0.00014968162481917836,
      "loss": 1.3852,
      "step": 2710
    },
    {
      "epoch": 1.0673991955263415,
      "grad_norm": 0.7734375,
      "learning_rate": 0.00014931343653405653,
      "loss": 1.1023,
      "step": 2720
    },
    {
      "epoch": 1.0713234572746002,
      "grad_norm": 0.498046875,
      "learning_rate": 0.00014894436280880576,
      "loss": 1.1635,
      "step": 2730
    },
    {
      "epoch": 1.0752477190228589,
      "grad_norm": 0.70703125,
      "learning_rate": 0.00014857441027027484,
      "loss": 1.2726,
      "step": 2740
    },
    {
      "epoch": 1.0791719807711175,
      "grad_norm": 0.6328125,
      "learning_rate": 0.000148203585561092,
      "loss": 0.9296,
      "step": 2750
    },
    {
      "epoch": 1.083096242519376,
      "grad_norm": 0.55078125,
      "learning_rate": 0.00014783189533954555,
      "loss": 1.3673,
      "step": 2760
    },
    {
      "epoch": 1.0870205042676346,
      "grad_norm": 0.828125,
      "learning_rate": 0.00014745934627946432,
      "loss": 1.0808,
      "step": 2770
    },
    {
      "epoch": 1.0909447660158933,
      "grad_norm": 0.4921875,
      "learning_rate": 0.00014708594507009806,
      "loss": 1.1457,
      "step": 2780
    },
    {
      "epoch": 1.094869027764152,
      "grad_norm": 0.6875,
      "learning_rate": 0.00014671169841599694,
      "loss": 1.2882,
      "step": 2790
    },
    {
      "epoch": 1.0987932895124104,
      "grad_norm": 0.59765625,
      "learning_rate": 0.00014633661303689157,
      "loss": 0.9345,
      "step": 2800
    },
    {
      "epoch": 1.102717551260669,
      "grad_norm": 0.60546875,
      "learning_rate": 0.00014596069566757207,
      "loss": 1.3616,
      "step": 2810
    },
    {
      "epoch": 1.1066418130089277,
      "grad_norm": 0.80859375,
      "learning_rate": 0.0001455839530577673,
      "loss": 1.0942,
      "step": 2820
    },
    {
      "epoch": 1.1105660747571864,
      "grad_norm": 0.51171875,
      "learning_rate": 0.00014520639197202354,
      "loss": 1.1486,
      "step": 2830
    },
    {
      "epoch": 1.1144903365054448,
      "grad_norm": 0.6875,
      "learning_rate": 0.0001448280191895831,
      "loss": 1.2617,
      "step": 2840
    },
    {
      "epoch": 1.1184145982537035,
      "grad_norm": 0.59375,
      "learning_rate": 0.00014444884150426269,
      "loss": 0.9443,
      "step": 2850
    },
    {
      "epoch": 1.1223388600019621,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00014406886572433113,
      "loss": 1.3774,
      "step": 2860
    },
    {
      "epoch": 1.1262631217502208,
      "grad_norm": 0.82421875,
      "learning_rate": 0.00014368809867238753,
      "loss": 1.0782,
      "step": 2870
    },
    {
      "epoch": 1.1301873834984795,
      "grad_norm": 0.494140625,
      "learning_rate": 0.00014330654718523847,
      "loss": 1.1409,
      "step": 2880
    },
    {
      "epoch": 1.134111645246738,
      "grad_norm": 0.70703125,
      "learning_rate": 0.00014292421811377533,
      "loss": 1.2942,
      "step": 2890
    },
    {
      "epoch": 1.1380359069949966,
      "grad_norm": 0.61328125,
      "learning_rate": 0.00014254111832285126,
      "loss": 0.9422,
      "step": 2900
    },
    {
      "epoch": 1.1419601687432552,
      "grad_norm": 0.578125,
      "learning_rate": 0.00014215725469115807,
      "loss": 1.3708,
      "step": 2910
    },
    {
      "epoch": 1.1458844304915137,
      "grad_norm": 0.8046875,
      "learning_rate": 0.0001417726341111025,
      "loss": 1.0782,
      "step": 2920
    },
    {
      "epoch": 1.1498086922397723,
      "grad_norm": 0.52734375,
      "learning_rate": 0.00014138726348868249,
      "loss": 1.1471,
      "step": 2930
    },
    {
      "epoch": 1.153732953988031,
      "grad_norm": 0.7109375,
      "learning_rate": 0.00014100114974336352,
      "loss": 1.2699,
      "step": 2940
    },
    {
      "epoch": 1.1576572157362897,
      "grad_norm": 0.56640625,
      "learning_rate": 0.00014061429980795383,
      "loss": 0.9462,
      "step": 2950
    },
    {
      "epoch": 1.1615814774845483,
      "grad_norm": 0.546875,
      "learning_rate": 0.00014022672062848036,
      "loss": 1.3579,
      "step": 2960
    },
    {
      "epoch": 1.1655057392328068,
      "grad_norm": 0.81640625,
      "learning_rate": 0.00013983841916406382,
      "loss": 1.0755,
      "step": 2970
    },
    {
      "epoch": 1.1694300009810654,
      "grad_norm": 0.51953125,
      "learning_rate": 0.00013944940238679382,
      "loss": 1.1482,
      "step": 2980
    },
    {
      "epoch": 1.173354262729324,
      "grad_norm": 0.6953125,
      "learning_rate": 0.00013905967728160368,
      "loss": 1.2595,
      "step": 2990
    },
    {
      "epoch": 1.1772785244775827,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00013866925084614503,
      "loss": 0.9483,
      "step": 3000
    },
    {
      "epoch": 1.1812027862258412,
      "grad_norm": 0.52734375,
      "learning_rate": 0.00013827813009066202,
      "loss": 1.3469,
      "step": 3010
    },
    {
      "epoch": 1.1851270479740998,
      "grad_norm": 0.82421875,
      "learning_rate": 0.00013788632203786565,
      "loss": 1.0815,
      "step": 3020
    },
    {
      "epoch": 1.1890513097223585,
      "grad_norm": 0.515625,
      "learning_rate": 0.0001374938337228076,
      "loss": 1.151,
      "step": 3030
    },
    {
      "epoch": 1.1929755714706172,
      "grad_norm": 0.6953125,
      "learning_rate": 0.00013710067219275382,
      "loss": 1.2609,
      "step": 3040
    },
    {
      "epoch": 1.1968998332188756,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00013670684450705812,
      "loss": 0.9252,
      "step": 3050
    },
    {
      "epoch": 1.2008240949671343,
      "grad_norm": 0.57421875,
      "learning_rate": 0.00013631235773703534,
      "loss": 1.3517,
      "step": 3060
    },
    {
      "epoch": 1.204748356715393,
      "grad_norm": 0.796875,
      "learning_rate": 0.00013591721896583453,
      "loss": 1.0971,
      "step": 3070
    },
    {
      "epoch": 1.2086726184636516,
      "grad_norm": 0.494140625,
      "learning_rate": 0.0001355214352883115,
      "loss": 1.142,
      "step": 3080
    },
    {
      "epoch": 1.21259688021191,
      "grad_norm": 0.6484375,
      "learning_rate": 0.00013512501381090158,
      "loss": 1.255,
      "step": 3090
    },
    {
      "epoch": 1.2165211419601687,
      "grad_norm": 0.5546875,
      "learning_rate": 0.00013472796165149217,
      "loss": 0.9205,
      "step": 3100
    },
    {
      "epoch": 1.2204454037084274,
      "grad_norm": 0.56640625,
      "learning_rate": 0.00013433028593929467,
      "loss": 1.3415,
      "step": 3110
    },
    {
      "epoch": 1.224369665456686,
      "grad_norm": 0.79296875,
      "learning_rate": 0.00013393199381471656,
      "loss": 1.0508,
      "step": 3120
    },
    {
      "epoch": 1.2282939272049447,
      "grad_norm": 0.5078125,
      "learning_rate": 0.00013353309242923334,
      "loss": 1.1486,
      "step": 3130
    },
    {
      "epoch": 1.2322181889532031,
      "grad_norm": 0.6796875,
      "learning_rate": 0.00013313358894525995,
      "loss": 1.2524,
      "step": 3140
    },
    {
      "epoch": 1.2361424507014618,
      "grad_norm": 0.5625,
      "learning_rate": 0.0001327334905360222,
      "loss": 0.8998,
      "step": 3150
    },
    {
      "epoch": 1.2400667124497204,
      "grad_norm": 0.5625,
      "learning_rate": 0.00013233280438542796,
      "loss": 1.3702,
      "step": 3160
    },
    {
      "epoch": 1.2439909741979789,
      "grad_norm": 0.80078125,
      "learning_rate": 0.0001319315376879383,
      "loss": 1.0534,
      "step": 3170
    },
    {
      "epoch": 1.2479152359462375,
      "grad_norm": 0.50390625,
      "learning_rate": 0.00013152969764843812,
      "loss": 1.1494,
      "step": 3180
    },
    {
      "epoch": 1.2518394976944962,
      "grad_norm": 0.703125,
      "learning_rate": 0.00013112729148210693,
      "loss": 1.2633,
      "step": 3190
    },
    {
      "epoch": 1.2557637594427549,
      "grad_norm": 0.49609375,
      "learning_rate": 0.00013072432641428932,
      "loss": 0.9218,
      "step": 3200
    },
    {
      "epoch": 1.2596880211910135,
      "grad_norm": 0.55078125,
      "learning_rate": 0.00013032080968036498,
      "loss": 1.3504,
      "step": 3210
    },
    {
      "epoch": 1.263612282939272,
      "grad_norm": 0.77734375,
      "learning_rate": 0.00012991674852561903,
      "loss": 1.0805,
      "step": 3220
    },
    {
      "epoch": 1.2675365446875306,
      "grad_norm": 0.462890625,
      "learning_rate": 0.00012951215020511197,
      "loss": 1.1366,
      "step": 3230
    },
    {
      "epoch": 1.2714608064357893,
      "grad_norm": 0.67578125,
      "learning_rate": 0.00012910702198354915,
      "loss": 1.2441,
      "step": 3240
    },
    {
      "epoch": 1.275385068184048,
      "grad_norm": 0.51953125,
      "learning_rate": 0.00012870137113515053,
      "loss": 0.9222,
      "step": 3250
    },
    {
      "epoch": 1.2793093299323064,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00012829520494352003,
      "loss": 1.3293,
      "step": 3260
    },
    {
      "epoch": 1.283233591680565,
      "grad_norm": 0.76953125,
      "learning_rate": 0.00012788853070151476,
      "loss": 1.0367,
      "step": 3270
    },
    {
      "epoch": 1.2871578534288237,
      "grad_norm": 0.466796875,
      "learning_rate": 0.00012748135571111403,
      "loss": 1.1422,
      "step": 3280
    },
    {
      "epoch": 1.2910821151770824,
      "grad_norm": 0.671875,
      "learning_rate": 0.00012707368728328826,
      "loss": 1.2239,
      "step": 3290
    },
    {
      "epoch": 1.295006376925341,
      "grad_norm": 0.546875,
      "learning_rate": 0.0001266655327378677,
      "loss": 0.9134,
      "step": 3300
    },
    {
      "epoch": 1.2989306386735995,
      "grad_norm": 0.578125,
      "learning_rate": 0.000126256899403411,
      "loss": 1.3148,
      "step": 3310
    },
    {
      "epoch": 1.3028549004218581,
      "grad_norm": 0.84765625,
      "learning_rate": 0.00012584779461707375,
      "loss": 1.0748,
      "step": 3320
    },
    {
      "epoch": 1.3067791621701168,
      "grad_norm": 0.51171875,
      "learning_rate": 0.0001254382257244765,
      "loss": 1.1198,
      "step": 3330
    },
    {
      "epoch": 1.3107034239183752,
      "grad_norm": 0.66796875,
      "learning_rate": 0.000125028200079573,
      "loss": 1.2346,
      "step": 3340
    },
    {
      "epoch": 1.314627685666634,
      "grad_norm": 0.4765625,
      "learning_rate": 0.0001246177250445182,
      "loss": 0.9289,
      "step": 3350
    },
    {
      "epoch": 1.3185519474148926,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00012420680798953603,
      "loss": 1.3461,
      "step": 3360
    },
    {
      "epoch": 1.3224762091631512,
      "grad_norm": 0.79296875,
      "learning_rate": 0.00012379545629278692,
      "loss": 1.0657,
      "step": 3370
    },
    {
      "epoch": 1.32640047091141,
      "grad_norm": 0.482421875,
      "learning_rate": 0.0001233836773402356,
      "loss": 1.1301,
      "step": 3380
    },
    {
      "epoch": 1.3303247326596683,
      "grad_norm": 0.6953125,
      "learning_rate": 0.00012297147852551821,
      "loss": 1.2465,
      "step": 3390
    },
    {
      "epoch": 1.334248994407927,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00012255886724980974,
      "loss": 0.9078,
      "step": 3400
    },
    {
      "epoch": 1.3381732561561857,
      "grad_norm": 0.5546875,
      "learning_rate": 0.000122145850921691,
      "loss": 1.295,
      "step": 3410
    },
    {
      "epoch": 1.342097517904444,
      "grad_norm": 0.8046875,
      "learning_rate": 0.00012173243695701576,
      "loss": 1.0651,
      "step": 3420
    },
    {
      "epoch": 1.3460217796527028,
      "grad_norm": 0.486328125,
      "learning_rate": 0.00012131863277877731,
      "loss": 1.1366,
      "step": 3430
    },
    {
      "epoch": 1.3499460414009614,
      "grad_norm": 0.6796875,
      "learning_rate": 0.0001209044458169756,
      "loss": 1.2487,
      "step": 3440
    },
    {
      "epoch": 1.35387030314922,
      "grad_norm": 0.443359375,
      "learning_rate": 0.00012048988350848338,
      "loss": 0.9414,
      "step": 3450
    },
    {
      "epoch": 1.3577945648974787,
      "grad_norm": 0.54296875,
      "learning_rate": 0.00012007495329691301,
      "loss": 1.3313,
      "step": 3460
    },
    {
      "epoch": 1.3617188266457374,
      "grad_norm": 0.8203125,
      "learning_rate": 0.00011965966263248267,
      "loss": 1.1166,
      "step": 3470
    },
    {
      "epoch": 1.3656430883939958,
      "grad_norm": 0.49609375,
      "learning_rate": 0.00011924401897188248,
      "loss": 1.1513,
      "step": 3480
    },
    {
      "epoch": 1.3695673501422545,
      "grad_norm": 0.71875,
      "learning_rate": 0.00011882802977814091,
      "loss": 1.2217,
      "step": 3490
    },
    {
      "epoch": 1.3734916118905132,
      "grad_norm": 0.494140625,
      "learning_rate": 0.0001184117025204905,
      "loss": 0.8819,
      "step": 3500
    },
    {
      "epoch": 1.3774158736387716,
      "grad_norm": 0.57421875,
      "learning_rate": 0.0001179950446742338,
      "loss": 1.349,
      "step": 3510
    },
    {
      "epoch": 1.3813401353870303,
      "grad_norm": 0.82421875,
      "learning_rate": 0.00011757806372060933,
      "loss": 1.0624,
      "step": 3520
    },
    {
      "epoch": 1.385264397135289,
      "grad_norm": 0.4765625,
      "learning_rate": 0.000117160767146657,
      "loss": 1.1026,
      "step": 3530
    },
    {
      "epoch": 1.3891886588835476,
      "grad_norm": 0.68359375,
      "learning_rate": 0.00011674316244508381,
      "loss": 1.2261,
      "step": 3540
    },
    {
      "epoch": 1.3931129206318063,
      "grad_norm": 0.43359375,
      "learning_rate": 0.00011632525711412937,
      "loss": 0.9299,
      "step": 3550
    },
    {
      "epoch": 1.3970371823800647,
      "grad_norm": 0.55078125,
      "learning_rate": 0.00011590705865743107,
      "loss": 1.3135,
      "step": 3560
    },
    {
      "epoch": 1.4009614441283234,
      "grad_norm": 0.77734375,
      "learning_rate": 0.00011548857458388967,
      "loss": 1.0513,
      "step": 3570
    },
    {
      "epoch": 1.404885705876582,
      "grad_norm": 0.486328125,
      "learning_rate": 0.00011506981240753405,
      "loss": 1.1313,
      "step": 3580
    },
    {
      "epoch": 1.4088099676248405,
      "grad_norm": 0.703125,
      "learning_rate": 0.00011465077964738673,
      "loss": 1.2166,
      "step": 3590
    },
    {
      "epoch": 1.4127342293730991,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00011423148382732853,
      "loss": 0.912,
      "step": 3600
    },
    {
      "epoch": 1.4166584911213578,
      "grad_norm": 0.57421875,
      "learning_rate": 0.00011381193247596364,
      "loss": 1.3163,
      "step": 3610
    },
    {
      "epoch": 1.4205827528696164,
      "grad_norm": 0.79296875,
      "learning_rate": 0.0001133921331264844,
      "loss": 1.0537,
      "step": 3620
    },
    {
      "epoch": 1.424507014617875,
      "grad_norm": 0.453125,
      "learning_rate": 0.00011297209331653606,
      "loss": 1.1087,
      "step": 3630
    },
    {
      "epoch": 1.4284312763661335,
      "grad_norm": 0.671875,
      "learning_rate": 0.00011255182058808143,
      "loss": 1.2389,
      "step": 3640
    },
    {
      "epoch": 1.4323555381143922,
      "grad_norm": 0.46484375,
      "learning_rate": 0.00011213132248726541,
      "loss": 0.9407,
      "step": 3650
    },
    {
      "epoch": 1.4362797998626509,
      "grad_norm": 0.55859375,
      "learning_rate": 0.00011171060656427956,
      "loss": 1.3161,
      "step": 3660
    },
    {
      "epoch": 1.4402040616109095,
      "grad_norm": 0.83984375,
      "learning_rate": 0.00011128968037322653,
      "loss": 1.0669,
      "step": 3670
    },
    {
      "epoch": 1.444128323359168,
      "grad_norm": 0.490234375,
      "learning_rate": 0.00011086855147198441,
      "loss": 1.1376,
      "step": 3680
    },
    {
      "epoch": 1.4480525851074266,
      "grad_norm": 0.6796875,
      "learning_rate": 0.00011044722742207101,
      "loss": 1.2273,
      "step": 3690
    },
    {
      "epoch": 1.4519768468556853,
      "grad_norm": 0.484375,
      "learning_rate": 0.00011002571578850806,
      "loss": 0.8897,
      "step": 3700
    },
    {
      "epoch": 1.455901108603944,
      "grad_norm": 0.55859375,
      "learning_rate": 0.0001096040241396855,
      "loss": 1.3295,
      "step": 3710
    },
    {
      "epoch": 1.4598253703522026,
      "grad_norm": 0.81640625,
      "learning_rate": 0.0001091821600472255,
      "loss": 1.0385,
      "step": 3720
    },
    {
      "epoch": 1.463749632100461,
      "grad_norm": 0.46875,
      "learning_rate": 0.00010876013108584644,
      "loss": 1.1092,
      "step": 3730
    },
    {
      "epoch": 1.4676738938487197,
      "grad_norm": 0.734375,
      "learning_rate": 0.00010833794483322699,
      "loss": 1.2404,
      "step": 3740
    },
    {
      "epoch": 1.4715981555969784,
      "grad_norm": 0.47265625,
      "learning_rate": 0.00010791560886987016,
      "loss": 0.8964,
      "step": 3750
    },
    {
      "epoch": 1.4755224173452368,
      "grad_norm": 0.5859375,
      "learning_rate": 0.00010749313077896697,
      "loss": 1.3215,
      "step": 3760
    },
    {
      "epoch": 1.4794466790934955,
      "grad_norm": 0.79296875,
      "learning_rate": 0.00010707051814626034,
      "loss": 1.0667,
      "step": 3770
    },
    {
      "epoch": 1.4833709408417541,
      "grad_norm": 0.451171875,
      "learning_rate": 0.00010664777855990909,
      "loss": 1.0913,
      "step": 3780
    },
    {
      "epoch": 1.4872952025900128,
      "grad_norm": 0.703125,
      "learning_rate": 0.00010622491961035139,
      "loss": 1.2477,
      "step": 3790
    },
    {
      "epoch": 1.4912194643382715,
      "grad_norm": 0.4375,
      "learning_rate": 0.00010580194889016865,
      "loss": 0.9113,
      "step": 3800
    },
    {
      "epoch": 1.49514372608653,
      "grad_norm": 0.58984375,
      "learning_rate": 0.00010537887399394925,
      "loss": 1.2986,
      "step": 3810
    },
    {
      "epoch": 1.4990679878347886,
      "grad_norm": 0.796875,
      "learning_rate": 0.00010495570251815204,
      "loss": 1.0299,
      "step": 3820
    },
    {
      "epoch": 1.5029922495830472,
      "grad_norm": 0.451171875,
      "learning_rate": 0.00010453244206096992,
      "loss": 1.1155,
      "step": 3830
    },
    {
      "epoch": 1.5069165113313057,
      "grad_norm": 0.6640625,
      "learning_rate": 0.00010410910022219356,
      "loss": 1.2115,
      "step": 3840
    },
    {
      "epoch": 1.5108407730795643,
      "grad_norm": 0.42578125,
      "learning_rate": 0.00010368568460307481,
      "loss": 0.9153,
      "step": 3850
    },
    {
      "epoch": 1.514765034827823,
      "grad_norm": 0.58203125,
      "learning_rate": 0.00010326220280619037,
      "loss": 1.3031,
      "step": 3860
    },
    {
      "epoch": 1.5186892965760816,
      "grad_norm": 0.796875,
      "learning_rate": 0.00010283866243530505,
      "loss": 1.0198,
      "step": 3870
    },
    {
      "epoch": 1.5226135583243403,
      "grad_norm": 0.466796875,
      "learning_rate": 0.00010241507109523552,
      "loss": 1.1138,
      "step": 3880
    },
    {
      "epoch": 1.526537820072599,
      "grad_norm": 0.67578125,
      "learning_rate": 0.00010199143639171349,
      "loss": 1.2407,
      "step": 3890
    },
    {
      "epoch": 1.5304620818208574,
      "grad_norm": 0.4453125,
      "learning_rate": 0.00010156776593124933,
      "loss": 0.9041,
      "step": 3900
    },
    {
      "epoch": 1.534386343569116,
      "grad_norm": 0.5546875,
      "learning_rate": 0.00010114406732099548,
      "loss": 1.3171,
      "step": 3910
    },
    {
      "epoch": 1.5383106053173745,
      "grad_norm": 0.84375,
      "learning_rate": 0.00010072034816860979,
      "loss": 1.0573,
      "step": 3920
    },
    {
      "epoch": 1.5422348670656332,
      "grad_norm": 0.462890625,
      "learning_rate": 0.00010029661608211885,
      "loss": 1.0974,
      "step": 3930
    },
    {
      "epoch": 1.5461591288138918,
      "grad_norm": 0.6953125,
      "learning_rate": 9.987287866978169e-05,
      "loss": 1.2192,
      "step": 3940
    },
    {
      "epoch": 1.5500833905621505,
      "grad_norm": 0.37890625,
      "learning_rate": 9.944914353995277e-05,
      "loss": 0.9216,
      "step": 3950
    },
    {
      "epoch": 1.5540076523104092,
      "grad_norm": 0.5546875,
      "learning_rate": 9.902541830094569e-05,
      "loss": 1.3008,
      "step": 3960
    },
    {
      "epoch": 1.5579319140586678,
      "grad_norm": 0.7890625,
      "learning_rate": 9.860171056089646e-05,
      "loss": 1.0243,
      "step": 3970
    },
    {
      "epoch": 1.5618561758069265,
      "grad_norm": 0.439453125,
      "learning_rate": 9.817802792762676e-05,
      "loss": 1.0857,
      "step": 3980
    },
    {
      "epoch": 1.565780437555185,
      "grad_norm": 0.66796875,
      "learning_rate": 9.775437800850763e-05,
      "loss": 1.2254,
      "step": 3990
    },
    {
      "epoch": 1.5697046993034436,
      "grad_norm": 0.43359375,
      "learning_rate": 9.73307684103226e-05,
      "loss": 0.9229,
      "step": 4000
    },
    {
      "epoch": 1.573628961051702,
      "grad_norm": 0.6171875,
      "learning_rate": 9.690720673913133e-05,
      "loss": 1.2679,
      "step": 4010
    },
    {
      "epoch": 1.5775532227999607,
      "grad_norm": 0.8125,
      "learning_rate": 9.648370060013279e-05,
      "loss": 1.0309,
      "step": 4020
    },
    {
      "epoch": 1.5814774845482193,
      "grad_norm": 0.455078125,
      "learning_rate": 9.606025759752895e-05,
      "loss": 1.0931,
      "step": 4030
    },
    {
      "epoch": 1.585401746296478,
      "grad_norm": 0.6953125,
      "learning_rate": 9.563688533438819e-05,
      "loss": 1.2299,
      "step": 4040
    },
    {
      "epoch": 1.5893260080447367,
      "grad_norm": 0.431640625,
      "learning_rate": 9.521359141250859e-05,
      "loss": 0.8948,
      "step": 4050
    },
    {
      "epoch": 1.5932502697929953,
      "grad_norm": 0.56640625,
      "learning_rate": 9.479038343228173e-05,
      "loss": 1.3212,
      "step": 4060
    },
    {
      "epoch": 1.5971745315412538,
      "grad_norm": 0.77734375,
      "learning_rate": 9.436726899255597e-05,
      "loss": 1.0682,
      "step": 4070
    },
    {
      "epoch": 1.6010987932895124,
      "grad_norm": 0.466796875,
      "learning_rate": 9.394425569050017e-05,
      "loss": 1.1235,
      "step": 4080
    },
    {
      "epoch": 1.6050230550377709,
      "grad_norm": 0.70703125,
      "learning_rate": 9.352135112146726e-05,
      "loss": 1.2034,
      "step": 4090
    },
    {
      "epoch": 1.6089473167860295,
      "grad_norm": 0.396484375,
      "learning_rate": 9.309856287885774e-05,
      "loss": 0.9005,
      "step": 4100
    },
    {
      "epoch": 1.6128715785342882,
      "grad_norm": 0.56640625,
      "learning_rate": 9.267589855398356e-05,
      "loss": 1.2964,
      "step": 4110
    },
    {
      "epoch": 1.6167958402825469,
      "grad_norm": 0.859375,
      "learning_rate": 9.225336573593149e-05,
      "loss": 1.0545,
      "step": 4120
    },
    {
      "epoch": 1.6207201020308055,
      "grad_norm": 0.47265625,
      "learning_rate": 9.183097201142722e-05,
      "loss": 1.0947,
      "step": 4130
    },
    {
      "epoch": 1.6246443637790642,
      "grad_norm": 0.69140625,
      "learning_rate": 9.14087249646989e-05,
      "loss": 1.205,
      "step": 4140
    },
    {
      "epoch": 1.6285686255273226,
      "grad_norm": 0.40625,
      "learning_rate": 9.098663217734102e-05,
      "loss": 0.8922,
      "step": 4150
    },
    {
      "epoch": 1.6324928872755813,
      "grad_norm": 0.56640625,
      "learning_rate": 9.056470122817836e-05,
      "loss": 1.295,
      "step": 4160
    },
    {
      "epoch": 1.63641714902384,
      "grad_norm": 0.79296875,
      "learning_rate": 9.01429396931297e-05,
      "loss": 1.048,
      "step": 4170
    },
    {
      "epoch": 1.6403414107720984,
      "grad_norm": 0.486328125,
      "learning_rate": 8.972135514507211e-05,
      "loss": 1.0945,
      "step": 4180
    },
    {
      "epoch": 1.644265672520357,
      "grad_norm": 0.69140625,
      "learning_rate": 8.92999551537046e-05,
      "loss": 1.2191,
      "step": 4190
    },
    {
      "epoch": 1.6481899342686157,
      "grad_norm": 0.39453125,
      "learning_rate": 8.887874728541258e-05,
      "loss": 0.8925,
      "step": 4200
    },
    {
      "epoch": 1.6521141960168744,
      "grad_norm": 0.578125,
      "learning_rate": 8.845773910313168e-05,
      "loss": 1.2614,
      "step": 4210
    },
    {
      "epoch": 1.656038457765133,
      "grad_norm": 0.8203125,
      "learning_rate": 8.803693816621217e-05,
      "loss": 1.0209,
      "step": 4220
    },
    {
      "epoch": 1.6599627195133917,
      "grad_norm": 0.474609375,
      "learning_rate": 8.761635203028318e-05,
      "loss": 1.081,
      "step": 4230
    },
    {
      "epoch": 1.6638869812616501,
      "grad_norm": 0.7109375,
      "learning_rate": 8.719598824711693e-05,
      "loss": 1.2477,
      "step": 4240
    },
    {
      "epoch": 1.6678112430099088,
      "grad_norm": 0.3671875,
      "learning_rate": 8.677585436449333e-05,
      "loss": 0.8814,
      "step": 4250
    },
    {
      "epoch": 1.6717355047581672,
      "grad_norm": 0.5703125,
      "learning_rate": 8.635595792606419e-05,
      "loss": 1.2822,
      "step": 4260
    },
    {
      "epoch": 1.675659766506426,
      "grad_norm": 0.8203125,
      "learning_rate": 8.593630647121809e-05,
      "loss": 1.0122,
      "step": 4270
    },
    {
      "epoch": 1.6795840282546846,
      "grad_norm": 0.453125,
      "learning_rate": 8.551690753494476e-05,
      "loss": 1.0888,
      "step": 4280
    },
    {
      "epoch": 1.6835082900029432,
      "grad_norm": 0.70703125,
      "learning_rate": 8.509776864769981e-05,
      "loss": 1.2059,
      "step": 4290
    },
    {
      "epoch": 1.6874325517512019,
      "grad_norm": 0.412109375,
      "learning_rate": 8.467889733526977e-05,
      "loss": 0.8872,
      "step": 4300
    },
    {
      "epoch": 1.6913568134994605,
      "grad_norm": 0.59765625,
      "learning_rate": 8.426030111863653e-05,
      "loss": 1.3049,
      "step": 4310
    },
    {
      "epoch": 1.695281075247719,
      "grad_norm": 0.828125,
      "learning_rate": 8.384198751384272e-05,
      "loss": 1.0193,
      "step": 4320
    },
    {
      "epoch": 1.6992053369959776,
      "grad_norm": 0.447265625,
      "learning_rate": 8.342396403185648e-05,
      "loss": 1.0903,
      "step": 4330
    },
    {
      "epoch": 1.703129598744236,
      "grad_norm": 0.7109375,
      "learning_rate": 8.300623817843673e-05,
      "loss": 1.1745,
      "step": 4340
    },
    {
      "epoch": 1.7070538604924947,
      "grad_norm": 0.3671875,
      "learning_rate": 8.258881745399837e-05,
      "loss": 0.8747,
      "step": 4350
    },
    {
      "epoch": 1.7109781222407534,
      "grad_norm": 0.5859375,
      "learning_rate": 8.217170935347756e-05,
      "loss": 1.2523,
      "step": 4360
    },
    {
      "epoch": 1.714902383989012,
      "grad_norm": 0.79296875,
      "learning_rate": 8.175492136619729e-05,
      "loss": 1.0037,
      "step": 4370
    },
    {
      "epoch": 1.7188266457372707,
      "grad_norm": 0.455078125,
      "learning_rate": 8.133846097573263e-05,
      "loss": 1.0986,
      "step": 4380
    },
    {
      "epoch": 1.7227509074855294,
      "grad_norm": 0.703125,
      "learning_rate": 8.092233565977669e-05,
      "loss": 1.1954,
      "step": 4390
    },
    {
      "epoch": 1.726675169233788,
      "grad_norm": 0.376953125,
      "learning_rate": 8.050655289000612e-05,
      "loss": 0.8793,
      "step": 4400
    },
    {
      "epoch": 1.7305994309820465,
      "grad_norm": 0.5859375,
      "learning_rate": 8.009112013194706e-05,
      "loss": 1.2952,
      "step": 4410
    },
    {
      "epoch": 1.7345236927303052,
      "grad_norm": 0.8515625,
      "learning_rate": 7.96760448448411e-05,
      "loss": 0.9848,
      "step": 4420
    },
    {
      "epoch": 1.7384479544785636,
      "grad_norm": 0.470703125,
      "learning_rate": 7.926133448151121e-05,
      "loss": 1.0929,
      "step": 4430
    },
    {
      "epoch": 1.7423722162268223,
      "grad_norm": 0.6875,
      "learning_rate": 7.884699648822815e-05,
      "loss": 1.2004,
      "step": 4440
    },
    {
      "epoch": 1.746296477975081,
      "grad_norm": 0.349609375,
      "learning_rate": 7.843303830457655e-05,
      "loss": 0.8528,
      "step": 4450
    },
    {
      "epoch": 1.7502207397233396,
      "grad_norm": 0.57421875,
      "learning_rate": 7.801946736332144e-05,
      "loss": 1.2763,
      "step": 4460
    },
    {
      "epoch": 1.7541450014715982,
      "grad_norm": 0.83984375,
      "learning_rate": 7.760629109027488e-05,
      "loss": 0.9997,
      "step": 4470
    },
    {
      "epoch": 1.758069263219857,
      "grad_norm": 0.447265625,
      "learning_rate": 7.719351690416233e-05,
      "loss": 1.0847,
      "step": 4480
    },
    {
      "epoch": 1.7619935249681153,
      "grad_norm": 0.6875,
      "learning_rate": 7.678115221648983e-05,
      "loss": 1.173,
      "step": 4490
    },
    {
      "epoch": 1.765917786716374,
      "grad_norm": 0.35546875,
      "learning_rate": 7.636920443141057e-05,
      "loss": 0.8901,
      "step": 4500
    },
    {
      "epoch": 1.7698420484646324,
      "grad_norm": 0.56640625,
      "learning_rate": 7.595768094559226e-05,
      "loss": 1.2692,
      "step": 4510
    },
    {
      "epoch": 1.773766310212891,
      "grad_norm": 0.828125,
      "learning_rate": 7.554658914808404e-05,
      "loss": 1.0238,
      "step": 4520
    },
    {
      "epoch": 1.7776905719611498,
      "grad_norm": 0.439453125,
      "learning_rate": 7.513593642018398e-05,
      "loss": 1.0963,
      "step": 4530
    },
    {
      "epoch": 1.7816148337094084,
      "grad_norm": 0.71875,
      "learning_rate": 7.472573013530657e-05,
      "loss": 1.182,
      "step": 4540
    },
    {
      "epoch": 1.785539095457667,
      "grad_norm": 0.3359375,
      "learning_rate": 7.431597765885013e-05,
      "loss": 0.9033,
      "step": 4550
    },
    {
      "epoch": 1.7894633572059258,
      "grad_norm": 0.57421875,
      "learning_rate": 7.39066863480648e-05,
      "loss": 1.2733,
      "step": 4560
    },
    {
      "epoch": 1.7933876189541842,
      "grad_norm": 0.81640625,
      "learning_rate": 7.349786355192023e-05,
      "loss": 1.0028,
      "step": 4570
    },
    {
      "epoch": 1.7973118807024429,
      "grad_norm": 0.419921875,
      "learning_rate": 7.308951661097379e-05,
      "loss": 1.049,
      "step": 4580
    },
    {
      "epoch": 1.8012361424507015,
      "grad_norm": 0.68359375,
      "learning_rate": 7.268165285723875e-05,
      "loss": 1.2096,
      "step": 4590
    },
    {
      "epoch": 1.80516040419896,
      "grad_norm": 0.32421875,
      "learning_rate": 7.227427961405245e-05,
      "loss": 0.8845,
      "step": 4600
    },
    {
      "epoch": 1.8090846659472186,
      "grad_norm": 0.578125,
      "learning_rate": 7.186740419594505e-05,
      "loss": 1.2801,
      "step": 4610
    },
    {
      "epoch": 1.8130089276954773,
      "grad_norm": 0.84765625,
      "learning_rate": 7.1461033908508e-05,
      "loss": 1.0108,
      "step": 4620
    },
    {
      "epoch": 1.816933189443736,
      "grad_norm": 0.443359375,
      "learning_rate": 7.105517604826309e-05,
      "loss": 1.0922,
      "step": 4630
    },
    {
      "epoch": 1.8208574511919946,
      "grad_norm": 0.6796875,
      "learning_rate": 7.064983790253108e-05,
      "loss": 1.1807,
      "step": 4640
    },
    {
      "epoch": 1.8247817129402533,
      "grad_norm": 0.33984375,
      "learning_rate": 7.024502674930132e-05,
      "loss": 0.8823,
      "step": 4650
    },
    {
      "epoch": 1.8287059746885117,
      "grad_norm": 0.5546875,
      "learning_rate": 6.984074985710067e-05,
      "loss": 1.2632,
      "step": 4660
    },
    {
      "epoch": 1.8326302364367704,
      "grad_norm": 0.81640625,
      "learning_rate": 6.943701448486313e-05,
      "loss": 1.0365,
      "step": 4670
    },
    {
      "epoch": 1.8365544981850288,
      "grad_norm": 0.4296875,
      "learning_rate": 6.903382788179961e-05,
      "loss": 1.1024,
      "step": 4680
    },
    {
      "epoch": 1.8404787599332875,
      "grad_norm": 0.6875,
      "learning_rate": 6.863119728726763e-05,
      "loss": 1.1877,
      "step": 4690
    },
    {
      "epoch": 1.8444030216815461,
      "grad_norm": 0.330078125,
      "learning_rate": 6.82291299306414e-05,
      "loss": 0.8796,
      "step": 4700
    },
    {
      "epoch": 1.8483272834298048,
      "grad_norm": 0.5703125,
      "learning_rate": 6.782763303118193e-05,
      "loss": 1.3008,
      "step": 4710
    },
    {
      "epoch": 1.8522515451780635,
      "grad_norm": 0.81640625,
      "learning_rate": 6.742671379790756e-05,
      "loss": 0.9914,
      "step": 4720
    },
    {
      "epoch": 1.8561758069263221,
      "grad_norm": 0.42578125,
      "learning_rate": 6.702637942946442e-05,
      "loss": 1.073,
      "step": 4730
    },
    {
      "epoch": 1.8601000686745806,
      "grad_norm": 0.6796875,
      "learning_rate": 6.662663711399705e-05,
      "loss": 1.1615,
      "step": 4740
    },
    {
      "epoch": 1.8640243304228392,
      "grad_norm": 0.349609375,
      "learning_rate": 6.622749402901971e-05,
      "loss": 0.8775,
      "step": 4750
    },
    {
      "epoch": 1.8679485921710977,
      "grad_norm": 0.55078125,
      "learning_rate": 6.582895734128702e-05,
      "loss": 1.2686,
      "step": 4760
    },
    {
      "epoch": 1.8718728539193563,
      "grad_norm": 0.83984375,
      "learning_rate": 6.543103420666569e-05,
      "loss": 1.0347,
      "step": 4770
    },
    {
      "epoch": 1.875797115667615,
      "grad_norm": 0.40234375,
      "learning_rate": 6.503373177000581e-05,
      "loss": 1.0777,
      "step": 4780
    },
    {
      "epoch": 1.8797213774158736,
      "grad_norm": 0.68359375,
      "learning_rate": 6.46370571650126e-05,
      "loss": 1.2004,
      "step": 4790
    },
    {
      "epoch": 1.8836456391641323,
      "grad_norm": 0.3359375,
      "learning_rate": 6.424101751411842e-05,
      "loss": 0.8658,
      "step": 4800
    },
    {
      "epoch": 1.887569900912391,
      "grad_norm": 0.59765625,
      "learning_rate": 6.384561992835467e-05,
      "loss": 1.2496,
      "step": 4810
    },
    {
      "epoch": 1.8914941626606496,
      "grad_norm": 0.82421875,
      "learning_rate": 6.34508715072244e-05,
      "loss": 0.9922,
      "step": 4820
    },
    {
      "epoch": 1.895418424408908,
      "grad_norm": 0.4140625,
      "learning_rate": 6.305677933857455e-05,
      "loss": 1.0738,
      "step": 4830
    },
    {
      "epoch": 1.8993426861571667,
      "grad_norm": 0.6796875,
      "learning_rate": 6.266335049846885e-05,
      "loss": 1.1636,
      "step": 4840
    },
    {
      "epoch": 1.9032669479054252,
      "grad_norm": 0.318359375,
      "learning_rate": 6.227059205106084e-05,
      "loss": 0.8669,
      "step": 4850
    },
    {
      "epoch": 1.9071912096536838,
      "grad_norm": 0.59765625,
      "learning_rate": 6.187851104846676e-05,
      "loss": 1.2582,
      "step": 4860
    },
    {
      "epoch": 1.9111154714019425,
      "grad_norm": 0.8046875,
      "learning_rate": 6.148711453063921e-05,
      "loss": 1.0103,
      "step": 4870
    },
    {
      "epoch": 1.9150397331502012,
      "grad_norm": 0.400390625,
      "learning_rate": 6.109640952524051e-05,
      "loss": 1.0755,
      "step": 4880
    },
    {
      "epoch": 1.9189639948984598,
      "grad_norm": 0.66015625,
      "learning_rate": 6.0706403047516766e-05,
      "loss": 1.1949,
      "step": 4890
    },
    {
      "epoch": 1.9228882566467185,
      "grad_norm": 0.30859375,
      "learning_rate": 6.0317102100171706e-05,
      "loss": 0.873,
      "step": 4900
    },
    {
      "epoch": 1.926812518394977,
      "grad_norm": 0.58203125,
      "learning_rate": 5.992851367324097e-05,
      "loss": 1.2465,
      "step": 4910
    },
    {
      "epoch": 1.9307367801432356,
      "grad_norm": 0.80859375,
      "learning_rate": 5.954064474396674e-05,
      "loss": 1.0336,
      "step": 4920
    },
    {
      "epoch": 1.934661041891494,
      "grad_norm": 0.392578125,
      "learning_rate": 5.915350227667225e-05,
      "loss": 1.0674,
      "step": 4930
    },
    {
      "epoch": 1.9385853036397527,
      "grad_norm": 0.70703125,
      "learning_rate": 5.876709322263696e-05,
      "loss": 1.1995,
      "step": 4940
    },
    {
      "epoch": 1.9425095653880113,
      "grad_norm": 0.31640625,
      "learning_rate": 5.838142451997155e-05,
      "loss": 0.8616,
      "step": 4950
    },
    {
      "epoch": 1.94643382713627,
      "grad_norm": 0.52734375,
      "learning_rate": 5.799650309349348e-05,
      "loss": 1.2762,
      "step": 4960
    },
    {
      "epoch": 1.9503580888845287,
      "grad_norm": 0.81640625,
      "learning_rate": 5.761233585460265e-05,
      "loss": 1.0103,
      "step": 4970
    },
    {
      "epoch": 1.9542823506327873,
      "grad_norm": 0.392578125,
      "learning_rate": 5.722892970115712e-05,
      "loss": 1.0486,
      "step": 4980
    },
    {
      "epoch": 1.9582066123810458,
      "grad_norm": 0.69921875,
      "learning_rate": 5.684629151734948e-05,
      "loss": 1.1864,
      "step": 4990
    },
    {
      "epoch": 1.9621308741293044,
      "grad_norm": 0.298828125,
      "learning_rate": 5.646442817358317e-05,
      "loss": 0.8509,
      "step": 5000
    },
    {
      "epoch": 1.966055135877563,
      "grad_norm": 0.5546875,
      "learning_rate": 5.608334652634913e-05,
      "loss": 1.2297,
      "step": 5010
    },
    {
      "epoch": 1.9699793976258215,
      "grad_norm": 0.90625,
      "learning_rate": 5.5703053418102514e-05,
      "loss": 1.0207,
      "step": 5020
    },
    {
      "epoch": 1.9739036593740802,
      "grad_norm": 0.416015625,
      "learning_rate": 5.5323555677140124e-05,
      "loss": 1.0675,
      "step": 5030
    },
    {
      "epoch": 1.9778279211223389,
      "grad_norm": 0.6875,
      "learning_rate": 5.4944860117477615e-05,
      "loss": 1.1624,
      "step": 5040
    },
    {
      "epoch": 1.9817521828705975,
      "grad_norm": 0.306640625,
      "learning_rate": 5.456697353872722e-05,
      "loss": 0.8661,
      "step": 5050
    },
    {
      "epoch": 1.9856764446188562,
      "grad_norm": 0.53515625,
      "learning_rate": 5.418990272597561e-05,
      "loss": 1.2655,
      "step": 5060
    },
    {
      "epoch": 1.9896007063671148,
      "grad_norm": 0.890625,
      "learning_rate": 5.381365444966204e-05,
      "loss": 0.9878,
      "step": 5070
    },
    {
      "epoch": 1.9935249681153733,
      "grad_norm": 0.46875,
      "learning_rate": 5.343823546545692e-05,
      "loss": 1.0445,
      "step": 5080
    },
    {
      "epoch": 1.997449229863632,
      "grad_norm": 0.74609375,
      "learning_rate": 5.306365251414043e-05,
      "loss": 1.1299,
      "step": 5090
    },
    {
      "epoch": 1.999803786912587,
      "eval_loss": 1.0273644924163818,
      "eval_runtime": 530.8878,
      "eval_samples_per_second": 47.091,
      "eval_steps_per_second": 11.773,
      "step": 5096
    },
    {
      "epoch": 2.0013734916118904,
      "grad_norm": 0.306640625,
      "learning_rate": 5.268991232148137e-05,
      "loss": 0.8913,
      "step": 5100
    },
    {
      "epoch": 2.005297753360149,
      "grad_norm": 0.62109375,
      "learning_rate": 5.2317021598116634e-05,
      "loss": 1.1946,
      "step": 5110
    },
    {
      "epoch": 2.0092220151084077,
      "grad_norm": 0.6875,
      "learning_rate": 5.194498703943054e-05,
      "loss": 0.6512,
      "step": 5120
    },
    {
      "epoch": 2.0131462768566664,
      "grad_norm": 0.56640625,
      "learning_rate": 5.1573815325434726e-05,
      "loss": 1.0757,
      "step": 5130
    },
    {
      "epoch": 2.017070538604925,
      "grad_norm": 0.78515625,
      "learning_rate": 5.120351312064802e-05,
      "loss": 0.902,
      "step": 5140
    },
    {
      "epoch": 2.0209948003531837,
      "grad_norm": 0.41796875,
      "learning_rate": 5.083408707397703e-05,
      "loss": 0.7074,
      "step": 5150
    },
    {
      "epoch": 2.0249190621014423,
      "grad_norm": 0.671875,
      "learning_rate": 5.0465543818596626e-05,
      "loss": 1.202,
      "step": 5160
    },
    {
      "epoch": 2.0288433238497006,
      "grad_norm": 0.7421875,
      "learning_rate": 5.009788997183073e-05,
      "loss": 0.665,
      "step": 5170
    },
    {
      "epoch": 2.0327675855979592,
      "grad_norm": 0.58203125,
      "learning_rate": 4.9731132135033786e-05,
      "loss": 1.0946,
      "step": 5180
    },
    {
      "epoch": 2.036691847346218,
      "grad_norm": 0.7421875,
      "learning_rate": 4.936527689347194e-05,
      "loss": 0.9398,
      "step": 5190
    },
    {
      "epoch": 2.0406161090944765,
      "grad_norm": 0.328125,
      "learning_rate": 4.9000330816204995e-05,
      "loss": 0.6837,
      "step": 5200
    },
    {
      "epoch": 2.044540370842735,
      "grad_norm": 0.65625,
      "learning_rate": 4.863630045596837e-05,
      "loss": 1.1604,
      "step": 5210
    },
    {
      "epoch": 2.048464632590994,
      "grad_norm": 0.71484375,
      "learning_rate": 4.827319234905541e-05,
      "loss": 0.6389,
      "step": 5220
    },
    {
      "epoch": 2.0523888943392525,
      "grad_norm": 0.50390625,
      "learning_rate": 4.791101301520016e-05,
      "loss": 1.0757,
      "step": 5230
    },
    {
      "epoch": 2.056313156087511,
      "grad_norm": 0.71484375,
      "learning_rate": 4.754976895746007e-05,
      "loss": 0.9377,
      "step": 5240
    },
    {
      "epoch": 2.06023741783577,
      "grad_norm": 0.3203125,
      "learning_rate": 4.7189466662099654e-05,
      "loss": 0.7264,
      "step": 5250
    },
    {
      "epoch": 2.064161679584028,
      "grad_norm": 0.6640625,
      "learning_rate": 4.683011259847345e-05,
      "loss": 1.1635,
      "step": 5260
    },
    {
      "epoch": 2.0680859413322867,
      "grad_norm": 0.7265625,
      "learning_rate": 4.647171321891034e-05,
      "loss": 0.6629,
      "step": 5270
    },
    {
      "epoch": 2.0720102030805454,
      "grad_norm": 0.50390625,
      "learning_rate": 4.61142749585975e-05,
      "loss": 1.0713,
      "step": 5280
    },
    {
      "epoch": 2.075934464828804,
      "grad_norm": 0.7109375,
      "learning_rate": 4.575780423546476e-05,
      "loss": 0.9147,
      "step": 5290
    },
    {
      "epoch": 2.0798587265770627,
      "grad_norm": 0.326171875,
      "learning_rate": 4.540230745006961e-05,
      "loss": 0.687,
      "step": 5300
    },
    {
      "epoch": 2.0837829883253214,
      "grad_norm": 0.66015625,
      "learning_rate": 4.504779098548209e-05,
      "loss": 1.1454,
      "step": 5310
    },
    {
      "epoch": 2.08770725007358,
      "grad_norm": 0.734375,
      "learning_rate": 4.469426120717025e-05,
      "loss": 0.6386,
      "step": 5320
    },
    {
      "epoch": 2.0916315118218387,
      "grad_norm": 0.5,
      "learning_rate": 4.4341724462885794e-05,
      "loss": 1.0515,
      "step": 5330
    },
    {
      "epoch": 2.095555773570097,
      "grad_norm": 0.75,
      "learning_rate": 4.399018708255018e-05,
      "loss": 0.9298,
      "step": 5340
    },
    {
      "epoch": 2.0994800353183556,
      "grad_norm": 0.310546875,
      "learning_rate": 4.363965537814102e-05,
      "loss": 0.6958,
      "step": 5350
    },
    {
      "epoch": 2.1034042970666142,
      "grad_norm": 0.640625,
      "learning_rate": 4.3290135643578476e-05,
      "loss": 1.1748,
      "step": 5360
    },
    {
      "epoch": 2.107328558814873,
      "grad_norm": 0.70703125,
      "learning_rate": 4.294163415461258e-05,
      "loss": 0.6699,
      "step": 5370
    },
    {
      "epoch": 2.1112528205631316,
      "grad_norm": 0.51953125,
      "learning_rate": 4.259415716871037e-05,
      "loss": 1.0746,
      "step": 5380
    },
    {
      "epoch": 2.1151770823113902,
      "grad_norm": 0.7421875,
      "learning_rate": 4.224771092494355e-05,
      "loss": 0.9284,
      "step": 5390
    },
    {
      "epoch": 2.119101344059649,
      "grad_norm": 0.330078125,
      "learning_rate": 4.190230164387655e-05,
      "loss": 0.7028,
      "step": 5400
    },
    {
      "epoch": 2.1230256058079076,
      "grad_norm": 0.65234375,
      "learning_rate": 4.155793552745465e-05,
      "loss": 1.1528,
      "step": 5410
    },
    {
      "epoch": 2.126949867556166,
      "grad_norm": 0.7421875,
      "learning_rate": 4.1214618758892866e-05,
      "loss": 0.6642,
      "step": 5420
    },
    {
      "epoch": 2.1308741293044244,
      "grad_norm": 0.5,
      "learning_rate": 4.0872357502564686e-05,
      "loss": 1.086,
      "step": 5430
    },
    {
      "epoch": 2.134798391052683,
      "grad_norm": 0.73828125,
      "learning_rate": 4.053115790389159e-05,
      "loss": 0.9343,
      "step": 5440
    },
    {
      "epoch": 2.1387226528009418,
      "grad_norm": 0.3203125,
      "learning_rate": 4.019102608923262e-05,
      "loss": 0.7032,
      "step": 5450
    },
    {
      "epoch": 2.1426469145492004,
      "grad_norm": 0.6640625,
      "learning_rate": 3.985196816577433e-05,
      "loss": 1.1612,
      "step": 5460
    },
    {
      "epoch": 2.146571176297459,
      "grad_norm": 0.69140625,
      "learning_rate": 3.951399022142127e-05,
      "loss": 0.6389,
      "step": 5470
    },
    {
      "epoch": 2.1504954380457177,
      "grad_norm": 0.494140625,
      "learning_rate": 3.9177098324686413e-05,
      "loss": 1.0929,
      "step": 5480
    },
    {
      "epoch": 2.1544196997939764,
      "grad_norm": 0.734375,
      "learning_rate": 3.884129852458254e-05,
      "loss": 0.9421,
      "step": 5490
    },
    {
      "epoch": 2.158343961542235,
      "grad_norm": 0.330078125,
      "learning_rate": 3.8506596850513355e-05,
      "loss": 0.7117,
      "step": 5500
    },
    {
      "epoch": 2.1622682232904933,
      "grad_norm": 0.6640625,
      "learning_rate": 3.817299931216537e-05,
      "loss": 1.1313,
      "step": 5510
    },
    {
      "epoch": 2.166192485038752,
      "grad_norm": 0.7734375,
      "learning_rate": 3.7840511899399954e-05,
      "loss": 0.6632,
      "step": 5520
    },
    {
      "epoch": 2.1701167467870106,
      "grad_norm": 0.5078125,
      "learning_rate": 3.7509140582145706e-05,
      "loss": 1.0829,
      "step": 5530
    },
    {
      "epoch": 2.1740410085352693,
      "grad_norm": 0.7265625,
      "learning_rate": 3.7178891310291444e-05,
      "loss": 0.9173,
      "step": 5540
    },
    {
      "epoch": 2.177965270283528,
      "grad_norm": 0.322265625,
      "learning_rate": 3.684977001357913e-05,
      "loss": 0.7305,
      "step": 5550
    },
    {
      "epoch": 2.1818895320317866,
      "grad_norm": 0.6484375,
      "learning_rate": 3.6521782601497677e-05,
      "loss": 1.1576,
      "step": 5560
    },
    {
      "epoch": 2.1858137937800453,
      "grad_norm": 0.76171875,
      "learning_rate": 3.619493496317662e-05,
      "loss": 0.6324,
      "step": 5570
    },
    {
      "epoch": 2.189738055528304,
      "grad_norm": 0.478515625,
      "learning_rate": 3.586923296728046e-05,
      "loss": 1.0634,
      "step": 5580
    },
    {
      "epoch": 2.193662317276562,
      "grad_norm": 0.74609375,
      "learning_rate": 3.5544682461903366e-05,
      "loss": 0.9577,
      "step": 5590
    },
    {
      "epoch": 2.197586579024821,
      "grad_norm": 0.314453125,
      "learning_rate": 3.522128927446392e-05,
      "loss": 0.7106,
      "step": 5600
    },
    {
      "epoch": 2.2015108407730795,
      "grad_norm": 0.66796875,
      "learning_rate": 3.4899059211600825e-05,
      "loss": 1.1586,
      "step": 5610
    },
    {
      "epoch": 2.205435102521338,
      "grad_norm": 0.73046875,
      "learning_rate": 3.457799805906835e-05,
      "loss": 0.6468,
      "step": 5620
    },
    {
      "epoch": 2.209359364269597,
      "grad_norm": 0.482421875,
      "learning_rate": 3.4258111581632636e-05,
      "loss": 1.0982,
      "step": 5630
    },
    {
      "epoch": 2.2132836260178554,
      "grad_norm": 0.7578125,
      "learning_rate": 3.3939405522968105e-05,
      "loss": 0.9259,
      "step": 5640
    },
    {
      "epoch": 2.217207887766114,
      "grad_norm": 0.30859375,
      "learning_rate": 3.362188560555434e-05,
      "loss": 0.6994,
      "step": 5650
    },
    {
      "epoch": 2.2211321495143728,
      "grad_norm": 0.65625,
      "learning_rate": 3.330555753057336e-05,
      "loss": 1.1708,
      "step": 5660
    },
    {
      "epoch": 2.2250564112626314,
      "grad_norm": 0.73046875,
      "learning_rate": 3.2990426977807156e-05,
      "loss": 0.652,
      "step": 5670
    },
    {
      "epoch": 2.2289806730108896,
      "grad_norm": 0.482421875,
      "learning_rate": 3.2676499605535914e-05,
      "loss": 1.0557,
      "step": 5680
    },
    {
      "epoch": 2.2329049347591483,
      "grad_norm": 0.734375,
      "learning_rate": 3.23637810504361e-05,
      "loss": 0.9274,
      "step": 5690
    },
    {
      "epoch": 2.236829196507407,
      "grad_norm": 0.31640625,
      "learning_rate": 3.205227692747968e-05,
      "loss": 0.6796,
      "step": 5700
    },
    {
      "epoch": 2.2407534582556656,
      "grad_norm": 0.65625,
      "learning_rate": 3.174199282983292e-05,
      "loss": 1.1646,
      "step": 5710
    },
    {
      "epoch": 2.2446777200039243,
      "grad_norm": 0.70703125,
      "learning_rate": 3.143293432875607e-05,
      "loss": 0.6551,
      "step": 5720
    },
    {
      "epoch": 2.248601981752183,
      "grad_norm": 0.482421875,
      "learning_rate": 3.1125106973503485e-05,
      "loss": 1.1028,
      "step": 5730
    },
    {
      "epoch": 2.2525262435004416,
      "grad_norm": 0.734375,
      "learning_rate": 3.081851629122372e-05,
      "loss": 0.9458,
      "step": 5740
    },
    {
      "epoch": 2.2564505052487003,
      "grad_norm": 0.310546875,
      "learning_rate": 3.051316778686055e-05,
      "loss": 0.6966,
      "step": 5750
    },
    {
      "epoch": 2.260374766996959,
      "grad_norm": 0.6640625,
      "learning_rate": 3.0209066943053943e-05,
      "loss": 1.167,
      "step": 5760
    },
    {
      "epoch": 2.264299028745217,
      "grad_norm": 0.73828125,
      "learning_rate": 2.990621922004172e-05,
      "loss": 0.6117,
      "step": 5770
    },
    {
      "epoch": 2.268223290493476,
      "grad_norm": 0.5078125,
      "learning_rate": 2.960463005556149e-05,
      "loss": 1.0999,
      "step": 5780
    },
    {
      "epoch": 2.2721475522417345,
      "grad_norm": 0.75,
      "learning_rate": 2.9304304864752884e-05,
      "loss": 0.9232,
      "step": 5790
    },
    {
      "epoch": 2.276071813989993,
      "grad_norm": 0.306640625,
      "learning_rate": 2.9005249040060612e-05,
      "loss": 0.6874,
      "step": 5800
    },
    {
      "epoch": 2.279996075738252,
      "grad_norm": 0.67578125,
      "learning_rate": 2.87074679511373e-05,
      "loss": 1.1354,
      "step": 5810
    },
    {
      "epoch": 2.2839203374865105,
      "grad_norm": 0.75,
      "learning_rate": 2.8410966944747376e-05,
      "loss": 0.6271,
      "step": 5820
    },
    {
      "epoch": 2.287844599234769,
      "grad_norm": 0.5078125,
      "learning_rate": 2.811575134467086e-05,
      "loss": 1.0869,
      "step": 5830
    },
    {
      "epoch": 2.2917688609830273,
      "grad_norm": 0.7890625,
      "learning_rate": 2.782182645160789e-05,
      "loss": 0.9089,
      "step": 5840
    },
    {
      "epoch": 2.295693122731286,
      "grad_norm": 0.310546875,
      "learning_rate": 2.7529197543083508e-05,
      "loss": 0.6925,
      "step": 5850
    },
    {
      "epoch": 2.2996173844795447,
      "grad_norm": 0.671875,
      "learning_rate": 2.7237869873352827e-05,
      "loss": 1.1549,
      "step": 5860
    },
    {
      "epoch": 2.3035416462278033,
      "grad_norm": 0.72265625,
      "learning_rate": 2.6947848673306853e-05,
      "loss": 0.6687,
      "step": 5870
    },
    {
      "epoch": 2.307465907976062,
      "grad_norm": 0.49609375,
      "learning_rate": 2.6659139150378377e-05,
      "loss": 1.0728,
      "step": 5880
    },
    {
      "epoch": 2.3113901697243207,
      "grad_norm": 0.73828125,
      "learning_rate": 2.6371746488448613e-05,
      "loss": 0.9141,
      "step": 5890
    },
    {
      "epoch": 2.3153144314725793,
      "grad_norm": 0.296875,
      "learning_rate": 2.6085675847754153e-05,
      "loss": 0.6683,
      "step": 5900
    },
    {
      "epoch": 2.319238693220838,
      "grad_norm": 0.67578125,
      "learning_rate": 2.580093236479406e-05,
      "loss": 1.1323,
      "step": 5910
    },
    {
      "epoch": 2.3231629549690966,
      "grad_norm": 0.72265625,
      "learning_rate": 2.5517521152237965e-05,
      "loss": 0.642,
      "step": 5920
    },
    {
      "epoch": 2.327087216717355,
      "grad_norm": 0.50390625,
      "learning_rate": 2.5235447298834004e-05,
      "loss": 1.0879,
      "step": 5930
    },
    {
      "epoch": 2.3310114784656135,
      "grad_norm": 0.7734375,
      "learning_rate": 2.4954715869317603e-05,
      "loss": 0.9193,
      "step": 5940
    },
    {
      "epoch": 2.334935740213872,
      "grad_norm": 0.318359375,
      "learning_rate": 2.4675331904320532e-05,
      "loss": 0.6856,
      "step": 5950
    },
    {
      "epoch": 2.338860001962131,
      "grad_norm": 0.65625,
      "learning_rate": 2.43973004202803e-05,
      "loss": 1.167,
      "step": 5960
    },
    {
      "epoch": 2.3427842637103895,
      "grad_norm": 0.7734375,
      "learning_rate": 2.4120626409350212e-05,
      "loss": 0.6359,
      "step": 5970
    },
    {
      "epoch": 2.346708525458648,
      "grad_norm": 0.52734375,
      "learning_rate": 2.3845314839309564e-05,
      "loss": 1.0877,
      "step": 5980
    },
    {
      "epoch": 2.350632787206907,
      "grad_norm": 0.73828125,
      "learning_rate": 2.3571370653474657e-05,
      "loss": 0.9256,
      "step": 5990
    },
    {
      "epoch": 2.3545570489551655,
      "grad_norm": 0.3046875,
      "learning_rate": 2.329879877060981e-05,
      "loss": 0.6817,
      "step": 6000
    },
    {
      "epoch": 2.358481310703424,
      "grad_norm": 0.66796875,
      "learning_rate": 2.302760408483926e-05,
      "loss": 1.1628,
      "step": 6010
    },
    {
      "epoch": 2.3624055724516824,
      "grad_norm": 0.73046875,
      "learning_rate": 2.2757791465559153e-05,
      "loss": 0.6344,
      "step": 6020
    },
    {
      "epoch": 2.366329834199941,
      "grad_norm": 0.490234375,
      "learning_rate": 2.2489365757350135e-05,
      "loss": 1.0605,
      "step": 6030
    },
    {
      "epoch": 2.3702540959481997,
      "grad_norm": 0.73046875,
      "learning_rate": 2.2222331779890394e-05,
      "loss": 0.9443,
      "step": 6040
    },
    {
      "epoch": 2.3741783576964584,
      "grad_norm": 0.29296875,
      "learning_rate": 2.1956694327869042e-05,
      "loss": 0.7066,
      "step": 6050
    },
    {
      "epoch": 2.378102619444717,
      "grad_norm": 0.66796875,
      "learning_rate": 2.1692458170900198e-05,
      "loss": 1.1249,
      "step": 6060
    },
    {
      "epoch": 2.3820268811929757,
      "grad_norm": 0.71484375,
      "learning_rate": 2.1429628053437078e-05,
      "loss": 0.6377,
      "step": 6070
    },
    {
      "epoch": 2.3859511429412343,
      "grad_norm": 0.486328125,
      "learning_rate": 2.116820869468711e-05,
      "loss": 1.1068,
      "step": 6080
    },
    {
      "epoch": 2.3898754046894926,
      "grad_norm": 0.78125,
      "learning_rate": 2.0908204788526964e-05,
      "loss": 0.9715,
      "step": 6090
    },
    {
      "epoch": 2.393799666437751,
      "grad_norm": 0.314453125,
      "learning_rate": 2.064962100341842e-05,
      "loss": 0.6953,
      "step": 6100
    },
    {
      "epoch": 2.39772392818601,
      "grad_norm": 0.65625,
      "learning_rate": 2.0392461982324463e-05,
      "loss": 1.1269,
      "step": 6110
    },
    {
      "epoch": 2.4016481899342685,
      "grad_norm": 0.73046875,
      "learning_rate": 2.0136732342625875e-05,
      "loss": 0.6428,
      "step": 6120
    },
    {
      "epoch": 2.405572451682527,
      "grad_norm": 0.4765625,
      "learning_rate": 1.9882436676038472e-05,
      "loss": 1.0982,
      "step": 6130
    },
    {
      "epoch": 2.409496713430786,
      "grad_norm": 0.75390625,
      "learning_rate": 1.9629579548530553e-05,
      "loss": 0.946,
      "step": 6140
    },
    {
      "epoch": 2.4134209751790445,
      "grad_norm": 0.3046875,
      "learning_rate": 1.9378165500240943e-05,
      "loss": 0.7233,
      "step": 6150
    },
    {
      "epoch": 2.417345236927303,
      "grad_norm": 0.640625,
      "learning_rate": 1.9128199045397488e-05,
      "loss": 1.1507,
      "step": 6160
    },
    {
      "epoch": 2.421269498675562,
      "grad_norm": 0.73828125,
      "learning_rate": 1.887968467223591e-05,
      "loss": 0.6497,
      "step": 6170
    },
    {
      "epoch": 2.42519376042382,
      "grad_norm": 0.5,
      "learning_rate": 1.86326268429194e-05,
      "loss": 1.071,
      "step": 6180
    },
    {
      "epoch": 2.4291180221720787,
      "grad_norm": 0.77734375,
      "learning_rate": 1.838702999345827e-05,
      "loss": 0.9367,
      "step": 6190
    },
    {
      "epoch": 2.4330422839203374,
      "grad_norm": 0.3125,
      "learning_rate": 1.8142898533630536e-05,
      "loss": 0.7125,
      "step": 6200
    },
    {
      "epoch": 2.436966545668596,
      "grad_norm": 0.66796875,
      "learning_rate": 1.7900236846902572e-05,
      "loss": 1.1568,
      "step": 6210
    },
    {
      "epoch": 2.4408908074168547,
      "grad_norm": 0.75390625,
      "learning_rate": 1.765904929035046e-05,
      "loss": 0.6499,
      "step": 6220
    },
    {
      "epoch": 2.4448150691651134,
      "grad_norm": 0.52734375,
      "learning_rate": 1.74193401945818e-05,
      "loss": 1.0884,
      "step": 6230
    },
    {
      "epoch": 2.448739330913372,
      "grad_norm": 0.7578125,
      "learning_rate": 1.7181113863657805e-05,
      "loss": 0.9254,
      "step": 6240
    },
    {
      "epoch": 2.4526635926616307,
      "grad_norm": 0.318359375,
      "learning_rate": 1.6944374575016253e-05,
      "loss": 0.7223,
      "step": 6250
    },
    {
      "epoch": 2.4565878544098894,
      "grad_norm": 0.6796875,
      "learning_rate": 1.670912657939443e-05,
      "loss": 1.1667,
      "step": 6260
    },
    {
      "epoch": 2.4605121161581476,
      "grad_norm": 0.71484375,
      "learning_rate": 1.6475374100753018e-05,
      "loss": 0.6277,
      "step": 6270
    },
    {
      "epoch": 2.4644363779064062,
      "grad_norm": 0.498046875,
      "learning_rate": 1.6243121336200127e-05,
      "loss": 1.078,
      "step": 6280
    },
    {
      "epoch": 2.468360639654665,
      "grad_norm": 0.78515625,
      "learning_rate": 1.601237245591599e-05,
      "loss": 0.9466,
      "step": 6290
    },
    {
      "epoch": 2.4722849014029236,
      "grad_norm": 0.310546875,
      "learning_rate": 1.5783131603078083e-05,
      "loss": 0.7033,
      "step": 6300
    },
    {
      "epoch": 2.4762091631511822,
      "grad_norm": 0.640625,
      "learning_rate": 1.555540289378663e-05,
      "loss": 1.1623,
      "step": 6310
    },
    {
      "epoch": 2.480133424899441,
      "grad_norm": 0.7578125,
      "learning_rate": 1.532919041699089e-05,
      "loss": 0.6573,
      "step": 6320
    },
    {
      "epoch": 2.4840576866476995,
      "grad_norm": 0.5,
      "learning_rate": 1.510449823441561e-05,
      "loss": 1.0979,
      "step": 6330
    },
    {
      "epoch": 2.4879819483959578,
      "grad_norm": 0.75390625,
      "learning_rate": 1.4881330380488013e-05,
      "loss": 0.9151,
      "step": 6340
    },
    {
      "epoch": 2.4919062101442164,
      "grad_norm": 0.306640625,
      "learning_rate": 1.4659690862265673e-05,
      "loss": 0.6961,
      "step": 6350
    },
    {
      "epoch": 2.495830471892475,
      "grad_norm": 0.62890625,
      "learning_rate": 1.4439583659364153e-05,
      "loss": 1.1481,
      "step": 6360
    },
    {
      "epoch": 2.4997547336407338,
      "grad_norm": 0.7109375,
      "learning_rate": 1.4221012723885874e-05,
      "loss": 0.664,
      "step": 6370
    },
    {
      "epoch": 2.5036789953889924,
      "grad_norm": 0.5234375,
      "learning_rate": 1.4003981980348969e-05,
      "loss": 1.0948,
      "step": 6380
    },
    {
      "epoch": 2.507603257137251,
      "grad_norm": 0.78125,
      "learning_rate": 1.3788495325616912e-05,
      "loss": 0.9448,
      "step": 6390
    },
    {
      "epoch": 2.5115275188855097,
      "grad_norm": 0.30078125,
      "learning_rate": 1.357455662882855e-05,
      "loss": 0.7031,
      "step": 6400
    },
    {
      "epoch": 2.5154517806337684,
      "grad_norm": 0.66015625,
      "learning_rate": 1.3362169731328534e-05,
      "loss": 1.1394,
      "step": 6410
    },
    {
      "epoch": 2.519376042382027,
      "grad_norm": 0.7578125,
      "learning_rate": 1.3151338446598482e-05,
      "loss": 0.6406,
      "step": 6420
    },
    {
      "epoch": 2.5233003041302853,
      "grad_norm": 0.51171875,
      "learning_rate": 1.2942066560188348e-05,
      "loss": 1.0769,
      "step": 6430
    },
    {
      "epoch": 2.527224565878544,
      "grad_norm": 0.74609375,
      "learning_rate": 1.2734357829648625e-05,
      "loss": 0.8985,
      "step": 6440
    },
    {
      "epoch": 2.5311488276268026,
      "grad_norm": 0.3046875,
      "learning_rate": 1.2528215984462765e-05,
      "loss": 0.6953,
      "step": 6450
    },
    {
      "epoch": 2.5350730893750613,
      "grad_norm": 0.6484375,
      "learning_rate": 1.23236447259802e-05,
      "loss": 1.1671,
      "step": 6460
    },
    {
      "epoch": 2.53899735112332,
      "grad_norm": 0.671875,
      "learning_rate": 1.2120647727349976e-05,
      "loss": 0.6413,
      "step": 6470
    },
    {
      "epoch": 2.5429216128715786,
      "grad_norm": 0.478515625,
      "learning_rate": 1.1919228633454737e-05,
      "loss": 1.1075,
      "step": 6480
    },
    {
      "epoch": 2.5468458746198372,
      "grad_norm": 0.73046875,
      "learning_rate": 1.1719391060845298e-05,
      "loss": 0.9472,
      "step": 6490
    },
    {
      "epoch": 2.550770136368096,
      "grad_norm": 0.30078125,
      "learning_rate": 1.1521138597675651e-05,
      "loss": 0.6991,
      "step": 6500
    },
    {
      "epoch": 2.5546943981163546,
      "grad_norm": 0.6484375,
      "learning_rate": 1.1324474803638652e-05,
      "loss": 1.1961,
      "step": 6510
    },
    {
      "epoch": 2.558618659864613,
      "grad_norm": 0.734375,
      "learning_rate": 1.1129403209902034e-05,
      "loss": 0.6599,
      "step": 6520
    },
    {
      "epoch": 2.5625429216128714,
      "grad_norm": 0.5,
      "learning_rate": 1.0935927319044958e-05,
      "loss": 1.0651,
      "step": 6530
    },
    {
      "epoch": 2.56646718336113,
      "grad_norm": 0.78125,
      "learning_rate": 1.0744050604995237e-05,
      "loss": 0.9095,
      "step": 6540
    },
    {
      "epoch": 2.5703914451093888,
      "grad_norm": 0.30859375,
      "learning_rate": 1.0553776512966885e-05,
      "loss": 0.6786,
      "step": 6550
    },
    {
      "epoch": 2.5743157068576474,
      "grad_norm": 0.65234375,
      "learning_rate": 1.0365108459398277e-05,
      "loss": 1.1456,
      "step": 6560
    },
    {
      "epoch": 2.578239968605906,
      "grad_norm": 0.69921875,
      "learning_rate": 1.0178049831890768e-05,
      "loss": 0.6548,
      "step": 6570
    },
    {
      "epoch": 2.5821642303541648,
      "grad_norm": 0.51171875,
      "learning_rate": 9.992603989147942e-06,
      "loss": 1.1119,
      "step": 6580
    },
    {
      "epoch": 2.586088492102423,
      "grad_norm": 0.75390625,
      "learning_rate": 9.808774260915244e-06,
      "loss": 0.9635,
      "step": 6590
    },
    {
      "epoch": 2.590012753850682,
      "grad_norm": 0.3125,
      "learning_rate": 9.62656394792023e-06,
      "loss": 0.6739,
      "step": 6600
    },
    {
      "epoch": 2.5939370155989403,
      "grad_norm": 0.6640625,
      "learning_rate": 9.445976321813277e-06,
      "loss": 1.1048,
      "step": 6610
    },
    {
      "epoch": 2.597861277347199,
      "grad_norm": 0.70703125,
      "learning_rate": 9.267014625108805e-06,
      "loss": 0.6407,
      "step": 6620
    },
    {
      "epoch": 2.6017855390954576,
      "grad_norm": 0.5,
      "learning_rate": 9.08968207112717e-06,
      "loss": 1.1126,
      "step": 6630
    },
    {
      "epoch": 2.6057098008437163,
      "grad_norm": 0.765625,
      "learning_rate": 8.91398184393687e-06,
      "loss": 0.9357,
      "step": 6640
    },
    {
      "epoch": 2.609634062591975,
      "grad_norm": 0.3046875,
      "learning_rate": 8.739917098297356e-06,
      "loss": 0.7173,
      "step": 6650
    },
    {
      "epoch": 2.6135583243402336,
      "grad_norm": 0.6796875,
      "learning_rate": 8.56749095960251e-06,
      "loss": 1.1604,
      "step": 6660
    },
    {
      "epoch": 2.6174825860884923,
      "grad_norm": 0.73046875,
      "learning_rate": 8.396706523824372e-06,
      "loss": 0.644,
      "step": 6670
    },
    {
      "epoch": 2.6214068478367505,
      "grad_norm": 0.4765625,
      "learning_rate": 8.227566857457703e-06,
      "loss": 1.0806,
      "step": 6680
    },
    {
      "epoch": 2.6253311095850096,
      "grad_norm": 0.73046875,
      "learning_rate": 8.060074997464772e-06,
      "loss": 0.95,
      "step": 6690
    },
    {
      "epoch": 2.629255371333268,
      "grad_norm": 0.30859375,
      "learning_rate": 7.894233951220953e-06,
      "loss": 0.7049,
      "step": 6700
    },
    {
      "epoch": 2.6331796330815265,
      "grad_norm": 0.671875,
      "learning_rate": 7.73004669646069e-06,
      "loss": 1.1413,
      "step": 6710
    },
    {
      "epoch": 2.637103894829785,
      "grad_norm": 0.6953125,
      "learning_rate": 7.567516181223966e-06,
      "loss": 0.6413,
      "step": 6720
    },
    {
      "epoch": 2.641028156578044,
      "grad_norm": 0.482421875,
      "learning_rate": 7.406645323803463e-06,
      "loss": 1.0811,
      "step": 6730
    },
    {
      "epoch": 2.6449524183263025,
      "grad_norm": 0.7578125,
      "learning_rate": 7.247437012692104e-06,
      "loss": 0.9361,
      "step": 6740
    },
    {
      "epoch": 2.648876680074561,
      "grad_norm": 0.31640625,
      "learning_rate": 7.089894106531214e-06,
      "loss": 0.7131,
      "step": 6750
    },
    {
      "epoch": 2.65280094182282,
      "grad_norm": 0.68359375,
      "learning_rate": 6.9340194340592135e-06,
      "loss": 1.1594,
      "step": 6760
    },
    {
      "epoch": 2.656725203571078,
      "grad_norm": 0.72265625,
      "learning_rate": 6.779815794060718e-06,
      "loss": 0.6191,
      "step": 6770
    },
    {
      "epoch": 2.6606494653193367,
      "grad_norm": 0.498046875,
      "learning_rate": 6.627285955316475e-06,
      "loss": 1.082,
      "step": 6780
    },
    {
      "epoch": 2.6645737270675953,
      "grad_norm": 0.78125,
      "learning_rate": 6.476432656553411e-06,
      "loss": 0.935,
      "step": 6790
    },
    {
      "epoch": 2.668497988815854,
      "grad_norm": 0.3125,
      "learning_rate": 6.327258606395736e-06,
      "loss": 0.6897,
      "step": 6800
    },
    {
      "epoch": 2.6724222505641126,
      "grad_norm": 0.68359375,
      "learning_rate": 6.179766483316041e-06,
      "loss": 1.1265,
      "step": 6810
    },
    {
      "epoch": 2.6763465123123713,
      "grad_norm": 0.71484375,
      "learning_rate": 6.033958935587369e-06,
      "loss": 0.6411,
      "step": 6820
    },
    {
      "epoch": 2.68027077406063,
      "grad_norm": 0.5078125,
      "learning_rate": 5.88983858123564e-06,
      "loss": 1.0963,
      "step": 6830
    },
    {
      "epoch": 2.684195035808888,
      "grad_norm": 0.7734375,
      "learning_rate": 5.747408007992572e-06,
      "loss": 0.93,
      "step": 6840
    },
    {
      "epoch": 2.6881192975571473,
      "grad_norm": 0.30859375,
      "learning_rate": 5.606669773249296e-06,
      "loss": 0.7206,
      "step": 6850
    },
    {
      "epoch": 2.6920435593054055,
      "grad_norm": 0.65625,
      "learning_rate": 5.467626404010406e-06,
      "loss": 1.1426,
      "step": 6860
    },
    {
      "epoch": 2.695967821053664,
      "grad_norm": 0.7734375,
      "learning_rate": 5.3302803968486195e-06,
      "loss": 0.6473,
      "step": 6870
    },
    {
      "epoch": 2.699892082801923,
      "grad_norm": 0.50390625,
      "learning_rate": 5.19463421785985e-06,
      "loss": 1.0892,
      "step": 6880
    },
    {
      "epoch": 2.7038163445501815,
      "grad_norm": 0.7421875,
      "learning_rate": 5.0606903026190535e-06,
      "loss": 0.9335,
      "step": 6890
    },
    {
      "epoch": 2.70774060629844,
      "grad_norm": 0.302734375,
      "learning_rate": 4.928451056136441e-06,
      "loss": 0.6924,
      "step": 6900
    },
    {
      "epoch": 2.711664868046699,
      "grad_norm": 0.66015625,
      "learning_rate": 4.797918852814254e-06,
      "loss": 1.156,
      "step": 6910
    },
    {
      "epoch": 2.7155891297949575,
      "grad_norm": 0.7421875,
      "learning_rate": 4.669096036404197e-06,
      "loss": 0.623,
      "step": 6920
    },
    {
      "epoch": 2.7195133915432157,
      "grad_norm": 0.49609375,
      "learning_rate": 4.541984919965336e-06,
      "loss": 1.076,
      "step": 6930
    },
    {
      "epoch": 2.723437653291475,
      "grad_norm": 0.734375,
      "learning_rate": 4.416587785822568e-06,
      "loss": 0.9313,
      "step": 6940
    },
    {
      "epoch": 2.727361915039733,
      "grad_norm": 0.3125,
      "learning_rate": 4.2929068855256275e-06,
      "loss": 0.7028,
      "step": 6950
    },
    {
      "epoch": 2.7312861767879917,
      "grad_norm": 0.6484375,
      "learning_rate": 4.170944439808622e-06,
      "loss": 1.1536,
      "step": 6960
    },
    {
      "epoch": 2.7352104385362503,
      "grad_norm": 0.75390625,
      "learning_rate": 4.050702638550275e-06,
      "loss": 0.653,
      "step": 6970
    },
    {
      "epoch": 2.739134700284509,
      "grad_norm": 0.486328125,
      "learning_rate": 3.932183640734466e-06,
      "loss": 1.0827,
      "step": 6980
    },
    {
      "epoch": 2.7430589620327677,
      "grad_norm": 0.76171875,
      "learning_rate": 3.815389574411576e-06,
      "loss": 0.9324,
      "step": 6990
    },
    {
      "epoch": 2.7469832237810263,
      "grad_norm": 0.30078125,
      "learning_rate": 3.700322536660228e-06,
      "loss": 0.6899,
      "step": 7000
    },
    {
      "epoch": 2.750907485529285,
      "grad_norm": 0.66796875,
      "learning_rate": 3.586984593549614e-06,
      "loss": 1.1497,
      "step": 7010
    },
    {
      "epoch": 2.754831747277543,
      "grad_norm": 0.73046875,
      "learning_rate": 3.4753777801024513e-06,
      "loss": 0.6323,
      "step": 7020
    },
    {
      "epoch": 2.758756009025802,
      "grad_norm": 0.5078125,
      "learning_rate": 3.3655041002583987e-06,
      "loss": 1.0728,
      "step": 7030
    },
    {
      "epoch": 2.7626802707740605,
      "grad_norm": 0.76171875,
      "learning_rate": 3.2573655268380743e-06,
      "loss": 0.9067,
      "step": 7040
    },
    {
      "epoch": 2.766604532522319,
      "grad_norm": 0.310546875,
      "learning_rate": 3.1509640015076948e-06,
      "loss": 0.7041,
      "step": 7050
    },
    {
      "epoch": 2.770528794270578,
      "grad_norm": 0.64453125,
      "learning_rate": 3.0463014347441255e-06,
      "loss": 1.157,
      "step": 7060
    },
    {
      "epoch": 2.7744530560188365,
      "grad_norm": 0.7421875,
      "learning_rate": 2.94337970580062e-06,
      "loss": 0.6522,
      "step": 7070
    },
    {
      "epoch": 2.778377317767095,
      "grad_norm": 0.490234375,
      "learning_rate": 2.8422006626731114e-06,
      "loss": 1.0915,
      "step": 7080
    },
    {
      "epoch": 2.782301579515354,
      "grad_norm": 0.7421875,
      "learning_rate": 2.7427661220669532e-06,
      "loss": 0.9352,
      "step": 7090
    },
    {
      "epoch": 2.7862258412636125,
      "grad_norm": 0.296875,
      "learning_rate": 2.645077869364354e-06,
      "loss": 0.6797,
      "step": 7100
    },
    {
      "epoch": 2.7901501030118707,
      "grad_norm": 0.66015625,
      "learning_rate": 2.5491376585923266e-06,
      "loss": 1.148,
      "step": 7110
    },
    {
      "epoch": 2.7940743647601294,
      "grad_norm": 0.6796875,
      "learning_rate": 2.4549472123911566e-06,
      "loss": 0.6476,
      "step": 7120
    },
    {
      "epoch": 2.797998626508388,
      "grad_norm": 0.5234375,
      "learning_rate": 2.362508221983484e-06,
      "loss": 1.0701,
      "step": 7130
    },
    {
      "epoch": 2.8019228882566467,
      "grad_norm": 0.7578125,
      "learning_rate": 2.2718223471439815e-06,
      "loss": 0.9238,
      "step": 7140
    },
    {
      "epoch": 2.8058471500049054,
      "grad_norm": 0.294921875,
      "learning_rate": 2.182891216169447e-06,
      "loss": 0.6908,
      "step": 7150
    },
    {
      "epoch": 2.809771411753164,
      "grad_norm": 0.6796875,
      "learning_rate": 2.095716425849703e-06,
      "loss": 1.1427,
      "step": 7160
    },
    {
      "epoch": 2.8136956735014227,
      "grad_norm": 0.7421875,
      "learning_rate": 2.010299541438798e-06,
      "loss": 0.6392,
      "step": 7170
    },
    {
      "epoch": 2.817619935249681,
      "grad_norm": 0.50390625,
      "learning_rate": 1.9266420966270184e-06,
      "loss": 1.0614,
      "step": 7180
    },
    {
      "epoch": 2.82154419699794,
      "grad_norm": 0.76171875,
      "learning_rate": 1.8447455935132418e-06,
      "loss": 0.9082,
      "step": 7190
    },
    {
      "epoch": 2.8254684587461982,
      "grad_norm": 0.3125,
      "learning_rate": 1.7646115025780507e-06,
      "loss": 0.6934,
      "step": 7200
    },
    {
      "epoch": 2.829392720494457,
      "grad_norm": 0.6484375,
      "learning_rate": 1.6862412626572842e-06,
      "loss": 1.158,
      "step": 7210
    },
    {
      "epoch": 2.8333169822427156,
      "grad_norm": 0.76171875,
      "learning_rate": 1.6096362809162047e-06,
      "loss": 0.6727,
      "step": 7220
    },
    {
      "epoch": 2.837241243990974,
      "grad_norm": 0.48828125,
      "learning_rate": 1.5347979328242611e-06,
      "loss": 1.0736,
      "step": 7230
    },
    {
      "epoch": 2.841165505739233,
      "grad_norm": 0.74609375,
      "learning_rate": 1.461727562130344e-06,
      "loss": 0.9245,
      "step": 7240
    },
    {
      "epoch": 2.8450897674874915,
      "grad_norm": 0.302734375,
      "learning_rate": 1.3904264808387246e-06,
      "loss": 0.6848,
      "step": 7250
    },
    {
      "epoch": 2.84901402923575,
      "grad_norm": 0.65234375,
      "learning_rate": 1.3208959691854539e-06,
      "loss": 1.1484,
      "step": 7260
    },
    {
      "epoch": 2.8529382909840084,
      "grad_norm": 0.7734375,
      "learning_rate": 1.2531372756153458e-06,
      "loss": 0.618,
      "step": 7270
    },
    {
      "epoch": 2.856862552732267,
      "grad_norm": 0.486328125,
      "learning_rate": 1.1871516167596186e-06,
      "loss": 1.0914,
      "step": 7280
    },
    {
      "epoch": 2.8607868144805257,
      "grad_norm": 0.75,
      "learning_rate": 1.1229401774140448e-06,
      "loss": 0.9386,
      "step": 7290
    },
    {
      "epoch": 2.8647110762287844,
      "grad_norm": 0.294921875,
      "learning_rate": 1.0605041105176129e-06,
      "loss": 0.6919,
      "step": 7300
    },
    {
      "epoch": 2.868635337977043,
      "grad_norm": 0.63671875,
      "learning_rate": 9.998445371319332e-07,
      "loss": 1.132,
      "step": 7310
    },
    {
      "epoch": 2.8725595997253017,
      "grad_norm": 0.70703125,
      "learning_rate": 9.409625464210092e-07,
      "loss": 0.6276,
      "step": 7320
    },
    {
      "epoch": 2.8764838614735604,
      "grad_norm": 0.5,
      "learning_rate": 8.838591956317311e-07,
      "loss": 1.0811,
      "step": 7330
    },
    {
      "epoch": 2.880408123221819,
      "grad_norm": 0.76171875,
      "learning_rate": 8.285355100748904e-07,
      "loss": 0.9206,
      "step": 7340
    },
    {
      "epoch": 2.8843323849700777,
      "grad_norm": 0.3046875,
      "learning_rate": 7.749924831067401e-07,
      "loss": 0.7131,
      "step": 7350
    },
    {
      "epoch": 2.888256646718336,
      "grad_norm": 0.6796875,
      "learning_rate": 7.232310761112082e-07,
      "loss": 1.1284,
      "step": 7360
    },
    {
      "epoch": 2.8921809084665946,
      "grad_norm": 0.71484375,
      "learning_rate": 6.732522184825896e-07,
      "loss": 0.6367,
      "step": 7370
    },
    {
      "epoch": 2.8961051702148533,
      "grad_norm": 0.515625,
      "learning_rate": 6.250568076088815e-07,
      "loss": 1.0708,
      "step": 7380
    },
    {
      "epoch": 2.900029431963112,
      "grad_norm": 0.80078125,
      "learning_rate": 5.78645708855674e-07,
      "loss": 0.9427,
      "step": 7390
    },
    {
      "epoch": 2.9039536937113706,
      "grad_norm": 0.3046875,
      "learning_rate": 5.340197555505966e-07,
      "loss": 0.6974,
      "step": 7400
    },
    {
      "epoch": 2.9078779554596292,
      "grad_norm": 0.6875,
      "learning_rate": 4.911797489683733e-07,
      "loss": 1.1502,
      "step": 7410
    },
    {
      "epoch": 2.911802217207888,
      "grad_norm": 0.71484375,
      "learning_rate": 4.501264583164022e-07,
      "loss": 0.6293,
      "step": 7420
    },
    {
      "epoch": 2.915726478956146,
      "grad_norm": 0.474609375,
      "learning_rate": 4.108606207209875e-07,
      "loss": 1.0858,
      "step": 7430
    },
    {
      "epoch": 2.9196507407044052,
      "grad_norm": 0.7578125,
      "learning_rate": 3.7338294121407323e-07,
      "loss": 0.9514,
      "step": 7440
    },
    {
      "epoch": 2.9235750024526634,
      "grad_norm": 0.310546875,
      "learning_rate": 3.376940927206196e-07,
      "loss": 0.6791,
      "step": 7450
    },
    {
      "epoch": 2.927499264200922,
      "grad_norm": 0.6640625,
      "learning_rate": 3.0379471604645717e-07,
      "loss": 1.1273,
      "step": 7460
    },
    {
      "epoch": 2.9314235259491808,
      "grad_norm": 0.73828125,
      "learning_rate": 2.716854198668517e-07,
      "loss": 0.6277,
      "step": 7470
    },
    {
      "epoch": 2.9353477876974394,
      "grad_norm": 0.53515625,
      "learning_rate": 2.41366780715524e-07,
      "loss": 1.0748,
      "step": 7480
    },
    {
      "epoch": 2.939272049445698,
      "grad_norm": 0.78125,
      "learning_rate": 2.1283934297432473e-07,
      "loss": 0.9025,
      "step": 7490
    },
    {
      "epoch": 2.9431963111939567,
      "grad_norm": 0.306640625,
      "learning_rate": 1.861036188634424e-07,
      "loss": 0.688,
      "step": 7500
    },
    {
      "epoch": 2.9471205729422154,
      "grad_norm": 0.64453125,
      "learning_rate": 1.6116008843224395e-07,
      "loss": 1.1647,
      "step": 7510
    },
    {
      "epoch": 2.9510448346904736,
      "grad_norm": 0.70703125,
      "learning_rate": 1.3800919955058167e-07,
      "loss": 0.6272,
      "step": 7520
    },
    {
      "epoch": 2.9549690964387327,
      "grad_norm": 0.50390625,
      "learning_rate": 1.1665136790084407e-07,
      "loss": 1.0956,
      "step": 7530
    },
    {
      "epoch": 2.958893358186991,
      "grad_norm": 0.7890625,
      "learning_rate": 9.708697697040636e-08,
      "loss": 0.9263,
      "step": 7540
    },
    {
      "epoch": 2.9628176199352496,
      "grad_norm": 0.298828125,
      "learning_rate": 7.931637804481362e-08,
      "loss": 0.6904,
      "step": 7550
    },
    {
      "epoch": 2.9667418816835083,
      "grad_norm": 0.66015625,
      "learning_rate": 6.333989020143039e-08,
      "loss": 1.1335,
      "step": 7560
    },
    {
      "epoch": 2.970666143431767,
      "grad_norm": 0.73046875,
      "learning_rate": 4.9157800303722965e-08,
      "loss": 0.6351,
      "step": 7570
    },
    {
      "epoch": 2.9745904051800256,
      "grad_norm": 0.5078125,
      "learning_rate": 3.6770362996108033e-08,
      "loss": 1.0857,
      "step": 7580
    },
    {
      "epoch": 2.9785146669282843,
      "grad_norm": 0.76171875,
      "learning_rate": 2.617780069940068e-08,
      "loss": 0.9432,
      "step": 7590
    },
    {
      "epoch": 2.982438928676543,
      "grad_norm": 0.3125,
      "learning_rate": 1.738030360677323e-08,
      "loss": 0.6816,
      "step": 7600
    },
    {
      "epoch": 2.986363190424801,
      "grad_norm": 0.63671875,
      "learning_rate": 1.0378029680391255e-08,
      "loss": 1.1422,
      "step": 7610
    },
    {
      "epoch": 2.99028745217306,
      "grad_norm": 0.70703125,
      "learning_rate": 5.171104648549196e-09,
      "loss": 0.6828,
      "step": 7620
    },
    {
      "epoch": 2.9942117139213185,
      "grad_norm": 0.55859375,
      "learning_rate": 1.7596220034277188e-09,
      "loss": 1.0756,
      "step": 7630
    },
    {
      "epoch": 2.998135975669577,
      "grad_norm": 0.7890625,
      "learning_rate": 1.436429993950661e-10,
      "loss": 0.8489,
      "step": 7640
    },
    {
      "epoch": 2.999705680368881,
      "eval_loss": 0.9989476799964905,
      "eval_runtime": 529.9443,
      "eval_samples_per_second": 47.175,
      "eval_steps_per_second": 11.794,
      "step": 7644
    },
    {
      "epoch": 2.999705680368881,
      "step": 7644,
      "total_flos": 4.155662038275719e+18,
      "train_loss": 1.1789655896583071,
      "train_runtime": 38325.8603,
      "train_samples_per_second": 25.53,
      "train_steps_per_second": 0.199
    }
  ],
  "logging_steps": 10,
  "max_steps": 7644,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 100,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 4.155662038275719e+18,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}