{
  "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": 242.0,
      "learning_rate": 8.695652173913044e-06,
      "loss": 4.3494,
      "step": 10
    },
    {
      "epoch": 0.007848523496517218,
      "grad_norm": 410.0,
      "learning_rate": 1.739130434782609e-05,
      "loss": 4.9023,
      "step": 20
    },
    {
      "epoch": 0.011772785244775827,
      "grad_norm": 156.0,
      "learning_rate": 2.608695652173913e-05,
      "loss": 4.6498,
      "step": 30
    },
    {
      "epoch": 0.015697046993034435,
      "grad_norm": 336.0,
      "learning_rate": 3.478260869565218e-05,
      "loss": 4.403,
      "step": 40
    },
    {
      "epoch": 0.019621308741293045,
      "grad_norm": 482.0,
      "learning_rate": 4.347826086956522e-05,
      "loss": 4.6505,
      "step": 50
    },
    {
      "epoch": 0.023545570489551654,
      "grad_norm": 280.0,
      "learning_rate": 5.217391304347826e-05,
      "loss": 4.0398,
      "step": 60
    },
    {
      "epoch": 0.02746983223781026,
      "grad_norm": 466.0,
      "learning_rate": 6.086956521739131e-05,
      "loss": 4.4326,
      "step": 70
    },
    {
      "epoch": 0.03139409398606887,
      "grad_norm": 215.0,
      "learning_rate": 6.956521739130436e-05,
      "loss": 4.0737,
      "step": 80
    },
    {
      "epoch": 0.035318355734327476,
      "grad_norm": 494.0,
      "learning_rate": 7.82608695652174e-05,
      "loss": 4.4089,
      "step": 90
    },
    {
      "epoch": 0.03924261748258609,
      "grad_norm": 748.0,
      "learning_rate": 8.695652173913044e-05,
      "loss": 4.3768,
      "step": 100
    },
    {
      "epoch": 0.043166879230844696,
      "grad_norm": 342.0,
      "learning_rate": 9.565217391304348e-05,
      "loss": 4.2063,
      "step": 110
    },
    {
      "epoch": 0.04709114097910331,
      "grad_norm": 692.0,
      "learning_rate": 0.00010434782608695653,
      "loss": 4.4437,
      "step": 120
    },
    {
      "epoch": 0.051015402727361915,
      "grad_norm": 208.0,
      "learning_rate": 0.00011304347826086956,
      "loss": 4.131,
      "step": 130
    },
    {
      "epoch": 0.05493966447562052,
      "grad_norm": 298.0,
      "learning_rate": 0.00012173913043478263,
      "loss": 4.5403,
      "step": 140
    },
    {
      "epoch": 0.058863926223879134,
      "grad_norm": 368.0,
      "learning_rate": 0.00013043478260869567,
      "loss": 4.559,
      "step": 150
    },
    {
      "epoch": 0.06278818797213774,
      "grad_norm": 163.0,
      "learning_rate": 0.0001391304347826087,
      "loss": 4.8419,
      "step": 160
    },
    {
      "epoch": 0.06671244972039635,
      "grad_norm": 268.0,
      "learning_rate": 0.00014782608695652173,
      "loss": 5.0152,
      "step": 170
    },
    {
      "epoch": 0.07063671146865495,
      "grad_norm": 156.0,
      "learning_rate": 0.0001565217391304348,
      "loss": 4.6886,
      "step": 180
    },
    {
      "epoch": 0.07456097321691357,
      "grad_norm": 171.0,
      "learning_rate": 0.00016521739130434784,
      "loss": 5.1737,
      "step": 190
    },
    {
      "epoch": 0.07848523496517218,
      "grad_norm": 392.0,
      "learning_rate": 0.00017391304347826088,
      "loss": 4.8763,
      "step": 200
    },
    {
      "epoch": 0.08240949671343079,
      "grad_norm": 163.0,
      "learning_rate": 0.00018260869565217392,
      "loss": 5.124,
      "step": 210
    },
    {
      "epoch": 0.08633375846168939,
      "grad_norm": 456.0,
      "learning_rate": 0.00019130434782608697,
      "loss": 5.0675,
      "step": 220
    },
    {
      "epoch": 0.090258020209948,
      "grad_norm": 376.0,
      "learning_rate": 0.0002,
      "loss": 10.7444,
      "step": 230
    },
    {
      "epoch": 0.09418228195820662,
      "grad_norm": 158.0,
      "learning_rate": 0.00019999910223238217,
      "loss": 13.2814,
      "step": 240
    },
    {
      "epoch": 0.09810654370646522,
      "grad_norm": 221.0,
      "learning_rate": 0.00019999640894564831,
      "loss": 6.4384,
      "step": 250
    },
    {
      "epoch": 0.10203080545472383,
      "grad_norm": 54.25,
      "learning_rate": 0.00019999192018815736,
      "loss": 7.3422,
      "step": 260
    },
    {
      "epoch": 0.10595506720298244,
      "grad_norm": 84.5,
      "learning_rate": 0.0001999856360405066,
      "loss": 6.1521,
      "step": 270
    },
    {
      "epoch": 0.10987932895124104,
      "grad_norm": 28.75,
      "learning_rate": 0.00019997755661553006,
      "loss": 5.4241,
      "step": 280
    },
    {
      "epoch": 0.11380359069949966,
      "grad_norm": 65.0,
      "learning_rate": 0.00019996768205829665,
      "loss": 5.6644,
      "step": 290
    },
    {
      "epoch": 0.11772785244775827,
      "grad_norm": 97.5,
      "learning_rate": 0.00019995601254610757,
      "loss": 4.9921,
      "step": 300
    },
    {
      "epoch": 0.12165211419601688,
      "grad_norm": 43.5,
      "learning_rate": 0.000199942548288493,
      "loss": 5.2209,
      "step": 310
    },
    {
      "epoch": 0.12557637594427548,
      "grad_norm": 93.5,
      "learning_rate": 0.0001999272895272084,
      "loss": 5.1219,
      "step": 320
    },
    {
      "epoch": 0.1295006376925341,
      "grad_norm": 36.75,
      "learning_rate": 0.0001999102365362303,
      "loss": 4.7911,
      "step": 330
    },
    {
      "epoch": 0.1334248994407927,
      "grad_norm": 28.0,
      "learning_rate": 0.00019989138962175104,
      "loss": 5.1623,
      "step": 340
    },
    {
      "epoch": 0.1373491611890513,
      "grad_norm": 74.0,
      "learning_rate": 0.00019987074912217369,
      "loss": 4.8023,
      "step": 350
    },
    {
      "epoch": 0.1412734229373099,
      "grad_norm": 50.5,
      "learning_rate": 0.00019984831540810567,
      "loss": 5.0909,
      "step": 360
    },
    {
      "epoch": 0.14519768468556854,
      "grad_norm": 84.5,
      "learning_rate": 0.00019982408888235225,
      "loss": 5.1888,
      "step": 370
    },
    {
      "epoch": 0.14912194643382715,
      "grad_norm": 30.75,
      "learning_rate": 0.00019979806997990918,
      "loss": 4.3684,
      "step": 380
    },
    {
      "epoch": 0.15304620818208575,
      "grad_norm": 69.5,
      "learning_rate": 0.00019977025916795502,
      "loss": 5.0842,
      "step": 390
    },
    {
      "epoch": 0.15697046993034436,
      "grad_norm": 94.5,
      "learning_rate": 0.00019974065694584279,
      "loss": 4.5769,
      "step": 400
    },
    {
      "epoch": 0.16089473167860296,
      "grad_norm": 39.0,
      "learning_rate": 0.00019970926384509072,
      "loss": 4.7384,
      "step": 410
    },
    {
      "epoch": 0.16481899342686157,
      "grad_norm": 77.0,
      "learning_rate": 0.00019967608042937302,
      "loss": 4.9524,
      "step": 420
    },
    {
      "epoch": 0.16874325517512018,
      "grad_norm": 43.5,
      "learning_rate": 0.00019964110729450964,
      "loss": 4.2478,
      "step": 430
    },
    {
      "epoch": 0.17266751692337878,
      "grad_norm": 69.5,
      "learning_rate": 0.00019960434506845553,
      "loss": 4.9383,
      "step": 440
    },
    {
      "epoch": 0.1765917786716374,
      "grad_norm": 142.0,
      "learning_rate": 0.00019956579441128943,
      "loss": 4.5544,
      "step": 450
    },
    {
      "epoch": 0.180516040419896,
      "grad_norm": 58.5,
      "learning_rate": 0.0001995254560152019,
      "loss": 4.8197,
      "step": 460
    },
    {
      "epoch": 0.18444030216815463,
      "grad_norm": 85.5,
      "learning_rate": 0.0001994833306044831,
      "loss": 4.8595,
      "step": 470
    },
    {
      "epoch": 0.18836456391641324,
      "grad_norm": 24.25,
      "learning_rate": 0.00019943941893550967,
      "loss": 4.1595,
      "step": 480
    },
    {
      "epoch": 0.19228882566467184,
      "grad_norm": 74.0,
      "learning_rate": 0.00019939372179673105,
      "loss": 4.941,
      "step": 490
    },
    {
      "epoch": 0.19621308741293045,
      "grad_norm": 95.0,
      "learning_rate": 0.00019934624000865543,
      "loss": 4.4576,
      "step": 500
    },
    {
      "epoch": 0.20013734916118905,
      "grad_norm": 41.5,
      "learning_rate": 0.00019929697442383512,
      "loss": 4.5666,
      "step": 510
    },
    {
      "epoch": 0.20406161090944766,
      "grad_norm": 49.5,
      "learning_rate": 0.00019924592592685103,
      "loss": 4.8266,
      "step": 520
    },
    {
      "epoch": 0.20798587265770627,
      "grad_norm": 20.75,
      "learning_rate": 0.0001991930954342969,
      "loss": 4.1289,
      "step": 530
    },
    {
      "epoch": 0.21191013440596487,
      "grad_norm": 61.5,
      "learning_rate": 0.00019913848389476285,
      "loss": 4.8051,
      "step": 540
    },
    {
      "epoch": 0.21583439615422348,
      "grad_norm": 105.0,
      "learning_rate": 0.00019908209228881827,
      "loss": 4.4487,
      "step": 550
    },
    {
      "epoch": 0.21975865790248209,
      "grad_norm": 45.75,
      "learning_rate": 0.0001990239216289944,
      "loss": 4.6318,
      "step": 560
    },
    {
      "epoch": 0.2236829196507407,
      "grad_norm": 119.0,
      "learning_rate": 0.00019896397295976584,
      "loss": 4.7547,
      "step": 570
    },
    {
      "epoch": 0.22760718139899933,
      "grad_norm": 20.375,
      "learning_rate": 0.00019890224735753215,
      "loss": 4.0032,
      "step": 580
    },
    {
      "epoch": 0.23153144314725793,
      "grad_norm": 40.5,
      "learning_rate": 0.00019883874593059825,
      "loss": 4.7809,
      "step": 590
    },
    {
      "epoch": 0.23545570489551654,
      "grad_norm": 77.5,
      "learning_rate": 0.0001987734698191546,
      "loss": 4.3707,
      "step": 600
    },
    {
      "epoch": 0.23937996664377514,
      "grad_norm": 38.75,
      "learning_rate": 0.00019870642019525678,
      "loss": 4.7742,
      "step": 610
    },
    {
      "epoch": 0.24330422839203375,
      "grad_norm": 47.0,
      "learning_rate": 0.00019863759826280447,
      "loss": 4.8384,
      "step": 620
    },
    {
      "epoch": 0.24722849014029236,
      "grad_norm": 14.625,
      "learning_rate": 0.00019856700525751965,
      "loss": 4.1704,
      "step": 630
    },
    {
      "epoch": 0.25115275188855096,
      "grad_norm": 33.75,
      "learning_rate": 0.00019849464244692467,
      "loss": 4.8077,
      "step": 640
    },
    {
      "epoch": 0.25507701363680957,
      "grad_norm": 60.5,
      "learning_rate": 0.00019842051113031924,
      "loss": 4.2266,
      "step": 650
    },
    {
      "epoch": 0.2590012753850682,
      "grad_norm": 25.5,
      "learning_rate": 0.00019834461263875728,
      "loss": 4.5856,
      "step": 660
    },
    {
      "epoch": 0.2629255371333268,
      "grad_norm": 42.75,
      "learning_rate": 0.00019826694833502295,
      "loss": 4.7427,
      "step": 670
    },
    {
      "epoch": 0.2668497988815854,
      "grad_norm": 10.9375,
      "learning_rate": 0.00019818751961360622,
      "loss": 4.0552,
      "step": 680
    },
    {
      "epoch": 0.270774060629844,
      "grad_norm": 38.0,
      "learning_rate": 0.00019810632790067772,
      "loss": 4.7338,
      "step": 690
    },
    {
      "epoch": 0.2746983223781026,
      "grad_norm": 38.75,
      "learning_rate": 0.0001980233746540633,
      "loss": 4.2693,
      "step": 700
    },
    {
      "epoch": 0.2786225841263612,
      "grad_norm": 18.875,
      "learning_rate": 0.00019793866136321773,
      "loss": 4.5473,
      "step": 710
    },
    {
      "epoch": 0.2825468458746198,
      "grad_norm": 33.25,
      "learning_rate": 0.00019785218954919797,
      "loss": 4.6076,
      "step": 720
    },
    {
      "epoch": 0.2864711076228785,
      "grad_norm": 13.25,
      "learning_rate": 0.00019776396076463595,
      "loss": 4.0571,
      "step": 730
    },
    {
      "epoch": 0.2903953693711371,
      "grad_norm": 17.0,
      "learning_rate": 0.00019767397659371057,
      "loss": 4.7611,
      "step": 740
    },
    {
      "epoch": 0.2943196311193957,
      "grad_norm": 32.75,
      "learning_rate": 0.0001975822386521193,
      "loss": 4.2498,
      "step": 750
    },
    {
      "epoch": 0.2982438928676543,
      "grad_norm": 15.875,
      "learning_rate": 0.00019748874858704923,
      "loss": 4.3607,
      "step": 760
    },
    {
      "epoch": 0.3021681546159129,
      "grad_norm": 15.375,
      "learning_rate": 0.0001973935080771474,
      "loss": 4.4836,
      "step": 770
    },
    {
      "epoch": 0.3060924163641715,
      "grad_norm": 23.0,
      "learning_rate": 0.00019729651883249074,
      "loss": 4.0879,
      "step": 780
    },
    {
      "epoch": 0.3100166781124301,
      "grad_norm": 14.6875,
      "learning_rate": 0.00019719778259455532,
      "loss": 4.5268,
      "step": 790
    },
    {
      "epoch": 0.3139409398606887,
      "grad_norm": 45.0,
      "learning_rate": 0.00019709730113618508,
      "loss": 3.95,
      "step": 800
    },
    {
      "epoch": 0.3178652016089473,
      "grad_norm": 21.25,
      "learning_rate": 0.00019699507626156,
      "loss": 4.2036,
      "step": 810
    },
    {
      "epoch": 0.32178946335720593,
      "grad_norm": 23.0,
      "learning_rate": 0.0001968911098061637,
      "loss": 4.3519,
      "step": 820
    },
    {
      "epoch": 0.32571372510546454,
      "grad_norm": 17.75,
      "learning_rate": 0.0001967854036367506,
      "loss": 3.6939,
      "step": 830
    },
    {
      "epoch": 0.32963798685372314,
      "grad_norm": 28.875,
      "learning_rate": 0.00019667795965131214,
      "loss": 4.3679,
      "step": 840
    },
    {
      "epoch": 0.33356224860198175,
      "grad_norm": 74.5,
      "learning_rate": 0.00019656877977904298,
      "loss": 3.856,
      "step": 850
    },
    {
      "epoch": 0.33748651035024035,
      "grad_norm": 11.875,
      "learning_rate": 0.00019645786598030616,
      "loss": 4.0697,
      "step": 860
    },
    {
      "epoch": 0.34141077209849896,
      "grad_norm": 29.0,
      "learning_rate": 0.00019634522024659804,
      "loss": 4.2,
      "step": 870
    },
    {
      "epoch": 0.34533503384675757,
      "grad_norm": 8.625,
      "learning_rate": 0.00019623084460051248,
      "loss": 3.4584,
      "step": 880
    },
    {
      "epoch": 0.3492592955950162,
      "grad_norm": 17.375,
      "learning_rate": 0.00019611474109570444,
      "loss": 4.1994,
      "step": 890
    },
    {
      "epoch": 0.3531835573432748,
      "grad_norm": 27.875,
      "learning_rate": 0.00019599691181685334,
      "loss": 3.7362,
      "step": 900
    },
    {
      "epoch": 0.3571078190915334,
      "grad_norm": 22.0,
      "learning_rate": 0.00019587735887962533,
      "loss": 4.0137,
      "step": 910
    },
    {
      "epoch": 0.361032080839792,
      "grad_norm": 25.375,
      "learning_rate": 0.00019575608443063558,
      "loss": 4.201,
      "step": 920
    },
    {
      "epoch": 0.3649563425880506,
      "grad_norm": 11.8125,
      "learning_rate": 0.00019563309064740954,
      "loss": 3.5118,
      "step": 930
    },
    {
      "epoch": 0.36888060433630926,
      "grad_norm": 19.125,
      "learning_rate": 0.0001955083797383439,
      "loss": 4.1748,
      "step": 940
    },
    {
      "epoch": 0.37280486608456787,
      "grad_norm": 25.5,
      "learning_rate": 0.000195381953942667,
      "loss": 3.6698,
      "step": 950
    },
    {
      "epoch": 0.37672912783282647,
      "grad_norm": 18.25,
      "learning_rate": 0.00019525381553039852,
      "loss": 3.9182,
      "step": 960
    },
    {
      "epoch": 0.3806533895810851,
      "grad_norm": 40.0,
      "learning_rate": 0.00019512396680230882,
      "loss": 4.1465,
      "step": 970
    },
    {
      "epoch": 0.3845776513293437,
      "grad_norm": 11.3125,
      "learning_rate": 0.0001949924100898776,
      "loss": 3.4983,
      "step": 980
    },
    {
      "epoch": 0.3885019130776023,
      "grad_norm": 23.875,
      "learning_rate": 0.00019485914775525193,
      "loss": 4.1647,
      "step": 990
    },
    {
      "epoch": 0.3924261748258609,
      "grad_norm": 42.0,
      "learning_rate": 0.000194724182191204,
      "loss": 3.6669,
      "step": 1000
    },
    {
      "epoch": 0.3963504365741195,
      "grad_norm": 11.5625,
      "learning_rate": 0.0001945875158210881,
      "loss": 3.8869,
      "step": 1010
    },
    {
      "epoch": 0.4002746983223781,
      "grad_norm": 22.375,
      "learning_rate": 0.00019444915109879704,
      "loss": 4.1022,
      "step": 1020
    },
    {
      "epoch": 0.4041989600706367,
      "grad_norm": 8.0625,
      "learning_rate": 0.00019430909050871815,
      "loss": 3.4939,
      "step": 1030
    },
    {
      "epoch": 0.4081232218188953,
      "grad_norm": 12.875,
      "learning_rate": 0.00019416733656568868,
      "loss": 4.2053,
      "step": 1040
    },
    {
      "epoch": 0.4120474835671539,
      "grad_norm": 19.5,
      "learning_rate": 0.00019402389181495062,
      "loss": 3.5654,
      "step": 1050
    },
    {
      "epoch": 0.41597174531541253,
      "grad_norm": 10.25,
      "learning_rate": 0.00019387875883210507,
      "loss": 3.9361,
      "step": 1060
    },
    {
      "epoch": 0.41989600706367114,
      "grad_norm": 13.0,
      "learning_rate": 0.00019373194022306576,
      "loss": 3.9866,
      "step": 1070
    },
    {
      "epoch": 0.42382026881192975,
      "grad_norm": 7.84375,
      "learning_rate": 0.0001935834386240127,
      "loss": 3.4053,
      "step": 1080
    },
    {
      "epoch": 0.42774453056018835,
      "grad_norm": 11.6875,
      "learning_rate": 0.00019343325670134428,
      "loss": 4.1227,
      "step": 1090
    },
    {
      "epoch": 0.43166879230844696,
      "grad_norm": 38.0,
      "learning_rate": 0.00019328139715162995,
      "loss": 3.6222,
      "step": 1100
    },
    {
      "epoch": 0.43559305405670556,
      "grad_norm": 14.875,
      "learning_rate": 0.00019312786270156136,
      "loss": 3.8306,
      "step": 1110
    },
    {
      "epoch": 0.43951731580496417,
      "grad_norm": 22.75,
      "learning_rate": 0.0001929726561079037,
      "loss": 4.0498,
      "step": 1120
    },
    {
      "epoch": 0.4434415775532228,
      "grad_norm": 6.3125,
      "learning_rate": 0.00019281578015744604,
      "loss": 3.4176,
      "step": 1130
    },
    {
      "epoch": 0.4473658393014814,
      "grad_norm": 11.875,
      "learning_rate": 0.00019265723766695133,
      "loss": 4.0281,
      "step": 1140
    },
    {
      "epoch": 0.45129010104974004,
      "grad_norm": 53.25,
      "learning_rate": 0.00019249703148310586,
      "loss": 3.5335,
      "step": 1150
    },
    {
      "epoch": 0.45521436279799865,
      "grad_norm": 10.125,
      "learning_rate": 0.00019233516448246814,
      "loss": 3.8918,
      "step": 1160
    },
    {
      "epoch": 0.45913862454625726,
      "grad_norm": 13.625,
      "learning_rate": 0.00019217163957141716,
      "loss": 3.9509,
      "step": 1170
    },
    {
      "epoch": 0.46306288629451586,
      "grad_norm": 7.0,
      "learning_rate": 0.00019200645968610035,
      "loss": 3.2483,
      "step": 1180
    },
    {
      "epoch": 0.46698714804277447,
      "grad_norm": 12.4375,
      "learning_rate": 0.0001918396277923807,
      "loss": 4.0194,
      "step": 1190
    },
    {
      "epoch": 0.4709114097910331,
      "grad_norm": 34.25,
      "learning_rate": 0.0001916711468857837,
      "loss": 3.475,
      "step": 1200
    },
    {
      "epoch": 0.4748356715392917,
      "grad_norm": 9.0,
      "learning_rate": 0.00019150101999144338,
      "loss": 3.8296,
      "step": 1210
    },
    {
      "epoch": 0.4787599332875503,
      "grad_norm": 24.875,
      "learning_rate": 0.00019132925016404804,
      "loss": 3.8997,
      "step": 1220
    },
    {
      "epoch": 0.4826841950358089,
      "grad_norm": 7.03125,
      "learning_rate": 0.0001911558404877855,
      "loss": 3.3315,
      "step": 1230
    },
    {
      "epoch": 0.4866084567840675,
      "grad_norm": 12.0625,
      "learning_rate": 0.0001909807940762876,
      "loss": 3.9563,
      "step": 1240
    },
    {
      "epoch": 0.4905327185323261,
      "grad_norm": 24.875,
      "learning_rate": 0.00019080411407257431,
      "loss": 3.4103,
      "step": 1250
    },
    {
      "epoch": 0.4944569802805847,
      "grad_norm": 10.4375,
      "learning_rate": 0.00019062580364899732,
      "loss": 3.8053,
      "step": 1260
    },
    {
      "epoch": 0.4983812420288433,
      "grad_norm": 24.625,
      "learning_rate": 0.00019044586600718322,
      "loss": 3.9799,
      "step": 1270
    },
    {
      "epoch": 0.5023055037771019,
      "grad_norm": 11.5625,
      "learning_rate": 0.00019026430437797567,
      "loss": 3.2287,
      "step": 1280
    },
    {
      "epoch": 0.5062297655253606,
      "grad_norm": 20.5,
      "learning_rate": 0.00019008112202137777,
      "loss": 3.9963,
      "step": 1290
    },
    {
      "epoch": 0.5101540272736191,
      "grad_norm": 43.5,
      "learning_rate": 0.0001898963222264932,
      "loss": 3.4738,
      "step": 1300
    },
    {
      "epoch": 0.5140782890218778,
      "grad_norm": 14.625,
      "learning_rate": 0.00018970990831146744,
      "loss": 3.7787,
      "step": 1310
    },
    {
      "epoch": 0.5180025507701363,
      "grad_norm": 20.75,
      "learning_rate": 0.00018952188362342804,
      "loss": 3.9506,
      "step": 1320
    },
    {
      "epoch": 0.521926812518395,
      "grad_norm": 7.15625,
      "learning_rate": 0.00018933225153842446,
      "loss": 3.2765,
      "step": 1330
    },
    {
      "epoch": 0.5258510742666536,
      "grad_norm": 10.0625,
      "learning_rate": 0.00018914101546136766,
      "loss": 4.0051,
      "step": 1340
    },
    {
      "epoch": 0.5297753360149122,
      "grad_norm": 16.875,
      "learning_rate": 0.00018894817882596877,
      "loss": 3.4501,
      "step": 1350
    },
    {
      "epoch": 0.5336995977631708,
      "grad_norm": 8.375,
      "learning_rate": 0.00018875374509467755,
      "loss": 3.8363,
      "step": 1360
    },
    {
      "epoch": 0.5376238595114294,
      "grad_norm": 9.9375,
      "learning_rate": 0.00018855771775862012,
      "loss": 3.9397,
      "step": 1370
    },
    {
      "epoch": 0.541548121259688,
      "grad_norm": 6.65625,
      "learning_rate": 0.00018836010033753637,
      "loss": 3.2016,
      "step": 1380
    },
    {
      "epoch": 0.5454723830079466,
      "grad_norm": 9.625,
      "learning_rate": 0.00018816089637971675,
      "loss": 3.9679,
      "step": 1390
    },
    {
      "epoch": 0.5493966447562052,
      "grad_norm": 15.625,
      "learning_rate": 0.0001879601094619385,
      "loss": 3.4429,
      "step": 1400
    },
    {
      "epoch": 0.5533209065044639,
      "grad_norm": 6.65625,
      "learning_rate": 0.0001877577431894015,
      "loss": 3.7048,
      "step": 1410
    },
    {
      "epoch": 0.5572451682527224,
      "grad_norm": 10.4375,
      "learning_rate": 0.00018755380119566341,
      "loss": 3.8671,
      "step": 1420
    },
    {
      "epoch": 0.5611694300009811,
      "grad_norm": 4.46875,
      "learning_rate": 0.0001873482871425747,
      "loss": 3.227,
      "step": 1430
    },
    {
      "epoch": 0.5650936917492396,
      "grad_norm": 7.21875,
      "learning_rate": 0.0001871412047202125,
      "loss": 3.8601,
      "step": 1440
    },
    {
      "epoch": 0.5690179534974983,
      "grad_norm": 18.5,
      "learning_rate": 0.00018693255764681474,
      "loss": 3.3434,
      "step": 1450
    },
    {
      "epoch": 0.572942215245757,
      "grad_norm": 10.6875,
      "learning_rate": 0.0001867223496687131,
      "loss": 3.6944,
      "step": 1460
    },
    {
      "epoch": 0.5768664769940155,
      "grad_norm": 8.625,
      "learning_rate": 0.00018651058456026592,
      "loss": 3.7045,
      "step": 1470
    },
    {
      "epoch": 0.5807907387422742,
      "grad_norm": 5.625,
      "learning_rate": 0.00018629726612379033,
      "loss": 3.1084,
      "step": 1480
    },
    {
      "epoch": 0.5847150004905327,
      "grad_norm": 6.53125,
      "learning_rate": 0.000186082398189494,
      "loss": 3.8749,
      "step": 1490
    },
    {
      "epoch": 0.5886392622387914,
      "grad_norm": 16.125,
      "learning_rate": 0.00018586598461540644,
      "loss": 3.212,
      "step": 1500
    },
    {
      "epoch": 0.5925635239870499,
      "grad_norm": 8.5,
      "learning_rate": 0.00018564802928730965,
      "loss": 3.6172,
      "step": 1510
    },
    {
      "epoch": 0.5964877857353086,
      "grad_norm": 12.875,
      "learning_rate": 0.00018542853611866825,
      "loss": 3.7353,
      "step": 1520
    },
    {
      "epoch": 0.6004120474835671,
      "grad_norm": 6.375,
      "learning_rate": 0.00018520750905055947,
      "loss": 3.0673,
      "step": 1530
    },
    {
      "epoch": 0.6043363092318258,
      "grad_norm": 9.8125,
      "learning_rate": 0.00018498495205160226,
      "loss": 3.7441,
      "step": 1540
    },
    {
      "epoch": 0.6082605709800843,
      "grad_norm": 17.375,
      "learning_rate": 0.00018476086911788585,
      "loss": 3.257,
      "step": 1550
    },
    {
      "epoch": 0.612184832728343,
      "grad_norm": 7.96875,
      "learning_rate": 0.00018453526427289835,
      "loss": 3.5339,
      "step": 1560
    },
    {
      "epoch": 0.6161090944766016,
      "grad_norm": 16.375,
      "learning_rate": 0.0001843081415674542,
      "loss": 3.6882,
      "step": 1570
    },
    {
      "epoch": 0.6200333562248602,
      "grad_norm": 5.5,
      "learning_rate": 0.00018407950507962165,
      "loss": 3.0389,
      "step": 1580
    },
    {
      "epoch": 0.6239576179731188,
      "grad_norm": 9.4375,
      "learning_rate": 0.0001838493589146494,
      "loss": 3.7312,
      "step": 1590
    },
    {
      "epoch": 0.6278818797213774,
      "grad_norm": 21.75,
      "learning_rate": 0.00018361770720489286,
      "loss": 3.1317,
      "step": 1600
    },
    {
      "epoch": 0.631806141469636,
      "grad_norm": 6.75,
      "learning_rate": 0.0001833845541097402,
      "loss": 3.5398,
      "step": 1610
    },
    {
      "epoch": 0.6357304032178946,
      "grad_norm": 10.5625,
      "learning_rate": 0.00018314990381553727,
      "loss": 3.6186,
      "step": 1620
    },
    {
      "epoch": 0.6396546649661532,
      "grad_norm": 4.875,
      "learning_rate": 0.00018291376053551292,
      "loss": 3.0659,
      "step": 1630
    },
    {
      "epoch": 0.6435789267144119,
      "grad_norm": 7.65625,
      "learning_rate": 0.0001826761285097029,
      "loss": 3.7513,
      "step": 1640
    },
    {
      "epoch": 0.6475031884626704,
      "grad_norm": 19.0,
      "learning_rate": 0.0001824370120048739,
      "loss": 3.1782,
      "step": 1650
    },
    {
      "epoch": 0.6514274502109291,
      "grad_norm": 7.375,
      "learning_rate": 0.00018219641531444714,
      "loss": 3.4937,
      "step": 1660
    },
    {
      "epoch": 0.6553517119591877,
      "grad_norm": 10.125,
      "learning_rate": 0.0001819543427584209,
      "loss": 3.6165,
      "step": 1670
    },
    {
      "epoch": 0.6592759737074463,
      "grad_norm": 5.75,
      "learning_rate": 0.0001817107986832932,
      "loss": 2.9775,
      "step": 1680
    },
    {
      "epoch": 0.663200235455705,
      "grad_norm": 15.4375,
      "learning_rate": 0.00018146578746198376,
      "loss": 3.6877,
      "step": 1690
    },
    {
      "epoch": 0.6671244972039635,
      "grad_norm": 18.625,
      "learning_rate": 0.00018121931349375538,
      "loss": 3.19,
      "step": 1700
    },
    {
      "epoch": 0.6710487589522222,
      "grad_norm": 14.5,
      "learning_rate": 0.000180971381204135,
      "loss": 3.4588,
      "step": 1710
    },
    {
      "epoch": 0.6749730207004807,
      "grad_norm": 10.0,
      "learning_rate": 0.00018072199504483427,
      "loss": 3.5459,
      "step": 1720
    },
    {
      "epoch": 0.6788972824487394,
      "grad_norm": 5.25,
      "learning_rate": 0.00018047115949366952,
      "loss": 2.9997,
      "step": 1730
    },
    {
      "epoch": 0.6828215441969979,
      "grad_norm": 7.5625,
      "learning_rate": 0.00018021887905448147,
      "loss": 3.6714,
      "step": 1740
    },
    {
      "epoch": 0.6867458059452566,
      "grad_norm": 19.25,
      "learning_rate": 0.0001799651582570543,
      "loss": 3.1299,
      "step": 1750
    },
    {
      "epoch": 0.6906700676935151,
      "grad_norm": 6.3125,
      "learning_rate": 0.00017971000165703434,
      "loss": 3.5023,
      "step": 1760
    },
    {
      "epoch": 0.6945943294417738,
      "grad_norm": 11.6875,
      "learning_rate": 0.00017945341383584818,
      "loss": 3.571,
      "step": 1770
    },
    {
      "epoch": 0.6985185911900323,
      "grad_norm": 6.09375,
      "learning_rate": 0.00017919539940062067,
      "loss": 2.9268,
      "step": 1780
    },
    {
      "epoch": 0.702442852938291,
      "grad_norm": 8.375,
      "learning_rate": 0.00017893596298409182,
      "loss": 3.7299,
      "step": 1790
    },
    {
      "epoch": 0.7063671146865496,
      "grad_norm": 17.875,
      "learning_rate": 0.00017867510924453393,
      "loss": 3.0783,
      "step": 1800
    },
    {
      "epoch": 0.7102913764348082,
      "grad_norm": 8.3125,
      "learning_rate": 0.00017841284286566783,
      "loss": 3.4935,
      "step": 1810
    },
    {
      "epoch": 0.7142156381830668,
      "grad_norm": 11.5625,
      "learning_rate": 0.00017814916855657872,
      "loss": 3.5949,
      "step": 1820
    },
    {
      "epoch": 0.7181398999313254,
      "grad_norm": 4.96875,
      "learning_rate": 0.00017788409105163176,
      "loss": 3.0166,
      "step": 1830
    },
    {
      "epoch": 0.722064161679584,
      "grad_norm": 8.875,
      "learning_rate": 0.00017761761511038695,
      "loss": 3.6276,
      "step": 1840
    },
    {
      "epoch": 0.7259884234278426,
      "grad_norm": 15.6875,
      "learning_rate": 0.0001773497455175137,
      "loss": 3.1449,
      "step": 1850
    },
    {
      "epoch": 0.7299126851761012,
      "grad_norm": 7.59375,
      "learning_rate": 0.00017708048708270496,
      "loss": 3.496,
      "step": 1860
    },
    {
      "epoch": 0.7338369469243599,
      "grad_norm": 10.0625,
      "learning_rate": 0.00017680984464059078,
      "loss": 3.6332,
      "step": 1870
    },
    {
      "epoch": 0.7377612086726185,
      "grad_norm": 5.59375,
      "learning_rate": 0.00017653782305065158,
      "loss": 2.9672,
      "step": 1880
    },
    {
      "epoch": 0.7416854704208771,
      "grad_norm": 7.71875,
      "learning_rate": 0.00017626442719713084,
      "loss": 3.7048,
      "step": 1890
    },
    {
      "epoch": 0.7456097321691357,
      "grad_norm": 15.0625,
      "learning_rate": 0.00017598966198894745,
      "loss": 3.103,
      "step": 1900
    },
    {
      "epoch": 0.7495339939173943,
      "grad_norm": 6.375,
      "learning_rate": 0.00017571353235960752,
      "loss": 3.4647,
      "step": 1910
    },
    {
      "epoch": 0.7534582556656529,
      "grad_norm": 11.625,
      "learning_rate": 0.00017543604326711592,
      "loss": 3.5593,
      "step": 1920
    },
    {
      "epoch": 0.7573825174139115,
      "grad_norm": 5.5625,
      "learning_rate": 0.00017515719969388697,
      "loss": 3.011,
      "step": 1930
    },
    {
      "epoch": 0.7613067791621702,
      "grad_norm": 8.25,
      "learning_rate": 0.00017487700664665535,
      "loss": 3.6346,
      "step": 1940
    },
    {
      "epoch": 0.7652310409104287,
      "grad_norm": 14.8125,
      "learning_rate": 0.00017459546915638593,
      "loss": 3.1225,
      "step": 1950
    },
    {
      "epoch": 0.7691553026586874,
      "grad_norm": 5.625,
      "learning_rate": 0.0001743125922781836,
      "loss": 3.4616,
      "step": 1960
    },
    {
      "epoch": 0.7730795644069459,
      "grad_norm": 9.0,
      "learning_rate": 0.00017402838109120228,
      "loss": 3.5482,
      "step": 1970
    },
    {
      "epoch": 0.7770038261552046,
      "grad_norm": 4.15625,
      "learning_rate": 0.0001737428406985541,
      "loss": 2.8644,
      "step": 1980
    },
    {
      "epoch": 0.7809280879034631,
      "grad_norm": 5.90625,
      "learning_rate": 0.00017345597622721727,
      "loss": 3.5951,
      "step": 1990
    },
    {
      "epoch": 0.7848523496517218,
      "grad_norm": 18.875,
      "learning_rate": 0.00017316779282794458,
      "loss": 3.0467,
      "step": 2000
    },
    {
      "epoch": 0.7887766113999803,
      "grad_norm": 6.03125,
      "learning_rate": 0.00017287829567517047,
      "loss": 3.413,
      "step": 2010
    },
    {
      "epoch": 0.792700873148239,
      "grad_norm": 9.0,
      "learning_rate": 0.00017258748996691828,
      "loss": 3.4455,
      "step": 2020
    },
    {
      "epoch": 0.7966251348964976,
      "grad_norm": 5.40625,
      "learning_rate": 0.00017229538092470708,
      "loss": 2.9249,
      "step": 2030
    },
    {
      "epoch": 0.8005493966447562,
      "grad_norm": 11.375,
      "learning_rate": 0.00017200197379345751,
      "loss": 3.5993,
      "step": 2040
    },
    {
      "epoch": 0.8044736583930148,
      "grad_norm": 33.75,
      "learning_rate": 0.00017170727384139807,
      "loss": 3.0903,
      "step": 2050
    },
    {
      "epoch": 0.8083979201412734,
      "grad_norm": 8.8125,
      "learning_rate": 0.00017141128635997027,
      "loss": 3.423,
      "step": 2060
    },
    {
      "epoch": 0.812322181889532,
      "grad_norm": 9.4375,
      "learning_rate": 0.00017111401666373356,
      "loss": 3.5083,
      "step": 2070
    },
    {
      "epoch": 0.8162464436377906,
      "grad_norm": 4.5625,
      "learning_rate": 0.00017081547009027012,
      "loss": 2.9129,
      "step": 2080
    },
    {
      "epoch": 0.8201707053860493,
      "grad_norm": 8.8125,
      "learning_rate": 0.00017051565200008886,
      "loss": 3.5944,
      "step": 2090
    },
    {
      "epoch": 0.8240949671343079,
      "grad_norm": 15.5,
      "learning_rate": 0.00017021456777652926,
      "loss": 3.0865,
      "step": 2100
    },
    {
      "epoch": 0.8280192288825665,
      "grad_norm": 6.0,
      "learning_rate": 0.00016991222282566464,
      "loss": 3.3838,
      "step": 2110
    },
    {
      "epoch": 0.8319434906308251,
      "grad_norm": 10.9375,
      "learning_rate": 0.00016960862257620507,
      "loss": 3.5384,
      "step": 2120
    },
    {
      "epoch": 0.8358677523790837,
      "grad_norm": 3.890625,
      "learning_rate": 0.00016930377247940004,
      "loss": 2.9262,
      "step": 2130
    },
    {
      "epoch": 0.8397920141273423,
      "grad_norm": 9.25,
      "learning_rate": 0.0001689976780089405,
      "loss": 3.5506,
      "step": 2140
    },
    {
      "epoch": 0.8437162758756009,
      "grad_norm": 14.875,
      "learning_rate": 0.00016869034466086044,
      "loss": 3.0486,
      "step": 2150
    },
    {
      "epoch": 0.8476405376238595,
      "grad_norm": 4.8125,
      "learning_rate": 0.00016838177795343846,
      "loss": 3.4115,
      "step": 2160
    },
    {
      "epoch": 0.8515647993721182,
      "grad_norm": 7.4375,
      "learning_rate": 0.00016807198342709857,
      "loss": 3.5092,
      "step": 2170
    },
    {
      "epoch": 0.8554890611203767,
      "grad_norm": 7.84375,
      "learning_rate": 0.0001677609666443105,
      "loss": 2.8772,
      "step": 2180
    },
    {
      "epoch": 0.8594133228686354,
      "grad_norm": 5.84375,
      "learning_rate": 0.00016744873318949032,
      "loss": 3.6786,
      "step": 2190
    },
    {
      "epoch": 0.8633375846168939,
      "grad_norm": 11.3125,
      "learning_rate": 0.00016713528866889966,
      "loss": 3.1029,
      "step": 2200
    },
    {
      "epoch": 0.8672618463651526,
      "grad_norm": 5.5625,
      "learning_rate": 0.00016682063871054534,
      "loss": 3.4294,
      "step": 2210
    },
    {
      "epoch": 0.8711861081134111,
      "grad_norm": 12.75,
      "learning_rate": 0.00016650478896407825,
      "loss": 3.4854,
      "step": 2220
    },
    {
      "epoch": 0.8751103698616698,
      "grad_norm": 5.34375,
      "learning_rate": 0.00016618774510069185,
      "loss": 2.9029,
      "step": 2230
    },
    {
      "epoch": 0.8790346316099283,
      "grad_norm": 7.65625,
      "learning_rate": 0.00016586951281302045,
      "loss": 3.6094,
      "step": 2240
    },
    {
      "epoch": 0.882958893358187,
      "grad_norm": 17.5,
      "learning_rate": 0.0001655500978150369,
      "loss": 3.0472,
      "step": 2250
    },
    {
      "epoch": 0.8868831551064456,
      "grad_norm": 6.375,
      "learning_rate": 0.00016522950584195,
      "loss": 3.394,
      "step": 2260
    },
    {
      "epoch": 0.8908074168547042,
      "grad_norm": 11.25,
      "learning_rate": 0.00016490774265010168,
      "loss": 3.4595,
      "step": 2270
    },
    {
      "epoch": 0.8947316786029628,
      "grad_norm": 5.8125,
      "learning_rate": 0.0001645848140168633,
      "loss": 2.8604,
      "step": 2280
    },
    {
      "epoch": 0.8986559403512214,
      "grad_norm": 7.1875,
      "learning_rate": 0.00016426072574053237,
      "loss": 3.5617,
      "step": 2290
    },
    {
      "epoch": 0.9025802020994801,
      "grad_norm": 20.0,
      "learning_rate": 0.00016393548364022803,
      "loss": 3.021,
      "step": 2300
    },
    {
      "epoch": 0.9065044638477386,
      "grad_norm": 4.8125,
      "learning_rate": 0.0001636090935557868,
      "loss": 3.3993,
      "step": 2310
    },
    {
      "epoch": 0.9104287255959973,
      "grad_norm": 8.0625,
      "learning_rate": 0.00016328156134765763,
      "loss": 3.4679,
      "step": 2320
    },
    {
      "epoch": 0.9143529873442559,
      "grad_norm": 4.65625,
      "learning_rate": 0.00016295289289679674,
      "loss": 2.909,
      "step": 2330
    },
    {
      "epoch": 0.9182772490925145,
      "grad_norm": 7.03125,
      "learning_rate": 0.000162623094104562,
      "loss": 3.5442,
      "step": 2340
    },
    {
      "epoch": 0.9222015108407731,
      "grad_norm": 21.625,
      "learning_rate": 0.00016229217089260692,
      "loss": 3.0159,
      "step": 2350
    },
    {
      "epoch": 0.9261257725890317,
      "grad_norm": 9.625,
      "learning_rate": 0.00016196012920277438,
      "loss": 3.4266,
      "step": 2360
    },
    {
      "epoch": 0.9300500343372903,
      "grad_norm": 10.75,
      "learning_rate": 0.00016162697499698986,
      "loss": 3.4957,
      "step": 2370
    },
    {
      "epoch": 0.9339742960855489,
      "grad_norm": 4.65625,
      "learning_rate": 0.0001612927142571546,
      "loss": 2.8601,
      "step": 2380
    },
    {
      "epoch": 0.9378985578338075,
      "grad_norm": 9.4375,
      "learning_rate": 0.00016095735298503788,
      "loss": 3.5581,
      "step": 2390
    },
    {
      "epoch": 0.9418228195820662,
      "grad_norm": 24.875,
      "learning_rate": 0.00016062089720216956,
      "loss": 3.0244,
      "step": 2400
    },
    {
      "epoch": 0.9457470813303247,
      "grad_norm": 9.75,
      "learning_rate": 0.0001602833529497318,
      "loss": 3.4227,
      "step": 2410
    },
    {
      "epoch": 0.9496713430785834,
      "grad_norm": 13.4375,
      "learning_rate": 0.00015994472628845055,
      "loss": 3.4701,
      "step": 2420
    },
    {
      "epoch": 0.9535956048268419,
      "grad_norm": 6.09375,
      "learning_rate": 0.0001596050232984868,
      "loss": 2.8914,
      "step": 2430
    },
    {
      "epoch": 0.9575198665751006,
      "grad_norm": 6.34375,
      "learning_rate": 0.00015926425007932747,
      "loss": 3.5191,
      "step": 2440
    },
    {
      "epoch": 0.9614441283233591,
      "grad_norm": 13.75,
      "learning_rate": 0.00015892241274967578,
      "loss": 3.0005,
      "step": 2450
    },
    {
      "epoch": 0.9653683900716178,
      "grad_norm": 4.25,
      "learning_rate": 0.00015857951744734146,
      "loss": 3.4518,
      "step": 2460
    },
    {
      "epoch": 0.9692926518198763,
      "grad_norm": 7.90625,
      "learning_rate": 0.00015823557032913044,
      "loss": 3.4465,
      "step": 2470
    },
    {
      "epoch": 0.973216913568135,
      "grad_norm": 4.15625,
      "learning_rate": 0.00015789057757073445,
      "loss": 2.8706,
      "step": 2480
    },
    {
      "epoch": 0.9771411753163936,
      "grad_norm": 5.375,
      "learning_rate": 0.00015754454536662,
      "loss": 3.5373,
      "step": 2490
    },
    {
      "epoch": 0.9810654370646522,
      "grad_norm": 10.9375,
      "learning_rate": 0.0001571974799299172,
      "loss": 3.034,
      "step": 2500
    },
    {
      "epoch": 0.9849896988129109,
      "grad_norm": 4.84375,
      "learning_rate": 0.0001568493874923084,
      "loss": 3.3803,
      "step": 2510
    },
    {
      "epoch": 0.9889139605611694,
      "grad_norm": 6.8125,
      "learning_rate": 0.00015650027430391585,
      "loss": 3.4196,
      "step": 2520
    },
    {
      "epoch": 0.9928382223094281,
      "grad_norm": 3.4375,
      "learning_rate": 0.0001561501466331899,
      "loss": 2.8777,
      "step": 2530
    },
    {
      "epoch": 0.9967624840576866,
      "grad_norm": 4.96875,
      "learning_rate": 0.00015579901076679623,
      "loss": 3.5175,
      "step": 2540
    },
    {
      "epoch": 0.9999018934562935,
      "eval_loss": 3.3654825687408447,
      "eval_runtime": 90.0166,
      "eval_samples_per_second": 277.726,
      "eval_steps_per_second": 69.432,
      "step": 2548
    },
    {
      "epoch": 1.0006867458059452,
      "grad_norm": 5.5,
      "learning_rate": 0.00015544687300950307,
      "loss": 2.9753,
      "step": 2550
    },
    {
      "epoch": 1.0046110075542038,
      "grad_norm": 4.90625,
      "learning_rate": 0.0001550937396840679,
      "loss": 3.4558,
      "step": 2560
    },
    {
      "epoch": 1.0085352693024625,
      "grad_norm": 7.25,
      "learning_rate": 0.00015473961713112405,
      "loss": 3.2725,
      "step": 2570
    },
    {
      "epoch": 1.0124595310507212,
      "grad_norm": 4.90625,
      "learning_rate": 0.00015438451170906673,
      "loss": 2.9956,
      "step": 2580
    },
    {
      "epoch": 1.0163837927989796,
      "grad_norm": 5.75,
      "learning_rate": 0.00015402842979393884,
      "loss": 3.4295,
      "step": 2590
    },
    {
      "epoch": 1.0203080545472383,
      "grad_norm": 7.125,
      "learning_rate": 0.00015367137777931673,
      "loss": 2.9263,
      "step": 2600
    },
    {
      "epoch": 1.024232316295497,
      "grad_norm": 4.90625,
      "learning_rate": 0.00015331336207619506,
      "loss": 3.4202,
      "step": 2610
    },
    {
      "epoch": 1.0281565780437556,
      "grad_norm": 6.875,
      "learning_rate": 0.000152954389112872,
      "loss": 3.3213,
      "step": 2620
    },
    {
      "epoch": 1.032080839792014,
      "grad_norm": 3.671875,
      "learning_rate": 0.00015259446533483356,
      "loss": 2.9749,
      "step": 2630
    },
    {
      "epoch": 1.0360051015402727,
      "grad_norm": 6.0625,
      "learning_rate": 0.00015223359720463795,
      "loss": 3.4695,
      "step": 2640
    },
    {
      "epoch": 1.0399293632885314,
      "grad_norm": 6.3125,
      "learning_rate": 0.0001518717912017997,
      "loss": 2.8989,
      "step": 2650
    },
    {
      "epoch": 1.04385362503679,
      "grad_norm": 5.96875,
      "learning_rate": 0.00015150905382267296,
      "loss": 3.407,
      "step": 2660
    },
    {
      "epoch": 1.0477778867850485,
      "grad_norm": 8.375,
      "learning_rate": 0.00015114539158033529,
      "loss": 3.3359,
      "step": 2670
    },
    {
      "epoch": 1.0517021485333071,
      "grad_norm": 4.90625,
      "learning_rate": 0.00015078081100447036,
      "loss": 2.9954,
      "step": 2680
    },
    {
      "epoch": 1.0556264102815658,
      "grad_norm": 6.875,
      "learning_rate": 0.00015041531864125082,
      "loss": 3.452,
      "step": 2690
    },
    {
      "epoch": 1.0595506720298244,
      "grad_norm": 5.1875,
      "learning_rate": 0.0001500489210532209,
      "loss": 2.8374,
      "step": 2700
    },
    {
      "epoch": 1.063474933778083,
      "grad_norm": 5.15625,
      "learning_rate": 0.00014968162481917836,
      "loss": 3.4162,
      "step": 2710
    },
    {
      "epoch": 1.0673991955263415,
      "grad_norm": 12.5625,
      "learning_rate": 0.00014931343653405653,
      "loss": 3.3368,
      "step": 2720
    },
    {
      "epoch": 1.0713234572746002,
      "grad_norm": 6.0,
      "learning_rate": 0.00014894436280880576,
      "loss": 2.9601,
      "step": 2730
    },
    {
      "epoch": 1.0752477190228589,
      "grad_norm": 13.5,
      "learning_rate": 0.00014857441027027484,
      "loss": 3.4346,
      "step": 2740
    },
    {
      "epoch": 1.0791719807711175,
      "grad_norm": 4.84375,
      "learning_rate": 0.000148203585561092,
      "loss": 2.8455,
      "step": 2750
    },
    {
      "epoch": 1.083096242519376,
      "grad_norm": 6.375,
      "learning_rate": 0.00014783189533954555,
      "loss": 3.3706,
      "step": 2760
    },
    {
      "epoch": 1.0870205042676346,
      "grad_norm": 8.875,
      "learning_rate": 0.00014745934627946432,
      "loss": 3.3015,
      "step": 2770
    },
    {
      "epoch": 1.0909447660158933,
      "grad_norm": 4.15625,
      "learning_rate": 0.00014708594507009806,
      "loss": 2.9207,
      "step": 2780
    },
    {
      "epoch": 1.094869027764152,
      "grad_norm": 6.1875,
      "learning_rate": 0.00014671169841599694,
      "loss": 3.4934,
      "step": 2790
    },
    {
      "epoch": 1.0987932895124104,
      "grad_norm": 7.6875,
      "learning_rate": 0.00014633661303689157,
      "loss": 2.8739,
      "step": 2800
    },
    {
      "epoch": 1.102717551260669,
      "grad_norm": 8.0,
      "learning_rate": 0.00014596069566757207,
      "loss": 3.4009,
      "step": 2810
    },
    {
      "epoch": 1.1066418130089277,
      "grad_norm": 10.625,
      "learning_rate": 0.0001455839530577673,
      "loss": 3.349,
      "step": 2820
    },
    {
      "epoch": 1.1105660747571864,
      "grad_norm": 7.1875,
      "learning_rate": 0.00014520639197202354,
      "loss": 2.9313,
      "step": 2830
    },
    {
      "epoch": 1.1144903365054448,
      "grad_norm": 19.875,
      "learning_rate": 0.0001448280191895831,
      "loss": 3.4649,
      "step": 2840
    },
    {
      "epoch": 1.1184145982537035,
      "grad_norm": 3.953125,
      "learning_rate": 0.00014444884150426269,
      "loss": 2.8805,
      "step": 2850
    },
    {
      "epoch": 1.1223388600019621,
      "grad_norm": 7.25,
      "learning_rate": 0.00014406886572433113,
      "loss": 3.4173,
      "step": 2860
    },
    {
      "epoch": 1.1262631217502208,
      "grad_norm": 11.875,
      "learning_rate": 0.00014368809867238753,
      "loss": 3.2955,
      "step": 2870
    },
    {
      "epoch": 1.1301873834984795,
      "grad_norm": 6.28125,
      "learning_rate": 0.00014330654718523847,
      "loss": 2.9336,
      "step": 2880
    },
    {
      "epoch": 1.134111645246738,
      "grad_norm": 14.5625,
      "learning_rate": 0.00014292421811377533,
      "loss": 3.4674,
      "step": 2890
    },
    {
      "epoch": 1.1380359069949966,
      "grad_norm": 9.9375,
      "learning_rate": 0.00014254111832285126,
      "loss": 2.9199,
      "step": 2900
    },
    {
      "epoch": 1.1419601687432552,
      "grad_norm": 11.0625,
      "learning_rate": 0.00014215725469115807,
      "loss": 3.3945,
      "step": 2910
    },
    {
      "epoch": 1.1458844304915137,
      "grad_norm": 10.875,
      "learning_rate": 0.0001417726341111025,
      "loss": 3.2816,
      "step": 2920
    },
    {
      "epoch": 1.1498086922397723,
      "grad_norm": 5.96875,
      "learning_rate": 0.00014138726348868249,
      "loss": 2.9691,
      "step": 2930
    },
    {
      "epoch": 1.153732953988031,
      "grad_norm": 10.4375,
      "learning_rate": 0.00014100114974336352,
      "loss": 3.4742,
      "step": 2940
    },
    {
      "epoch": 1.1576572157362897,
      "grad_norm": 6.21875,
      "learning_rate": 0.00014061429980795383,
      "loss": 2.8609,
      "step": 2950
    },
    {
      "epoch": 1.1615814774845483,
      "grad_norm": 10.0625,
      "learning_rate": 0.00014022672062848036,
      "loss": 3.3848,
      "step": 2960
    },
    {
      "epoch": 1.1655057392328068,
      "grad_norm": 11.9375,
      "learning_rate": 0.00013983841916406382,
      "loss": 3.2637,
      "step": 2970
    },
    {
      "epoch": 1.1694300009810654,
      "grad_norm": 11.0625,
      "learning_rate": 0.00013944940238679382,
      "loss": 2.9428,
      "step": 2980
    },
    {
      "epoch": 1.173354262729324,
      "grad_norm": 14.6875,
      "learning_rate": 0.00013905967728160368,
      "loss": 3.4456,
      "step": 2990
    },
    {
      "epoch": 1.1772785244775827,
      "grad_norm": 4.5,
      "learning_rate": 0.00013866925084614503,
      "loss": 2.9011,
      "step": 3000
    },
    {
      "epoch": 1.1812027862258412,
      "grad_norm": 6.125,
      "learning_rate": 0.00013827813009066202,
      "loss": 3.3611,
      "step": 3010
    },
    {
      "epoch": 1.1851270479740998,
      "grad_norm": 8.25,
      "learning_rate": 0.00013788632203786565,
      "loss": 3.3006,
      "step": 3020
    },
    {
      "epoch": 1.1890513097223585,
      "grad_norm": 6.03125,
      "learning_rate": 0.0001374938337228076,
      "loss": 2.978,
      "step": 3030
    },
    {
      "epoch": 1.1929755714706172,
      "grad_norm": 16.75,
      "learning_rate": 0.00013710067219275382,
      "loss": 3.4192,
      "step": 3040
    },
    {
      "epoch": 1.1968998332188756,
      "grad_norm": 5.3125,
      "learning_rate": 0.00013670684450705812,
      "loss": 2.8441,
      "step": 3050
    },
    {
      "epoch": 1.2008240949671343,
      "grad_norm": 7.65625,
      "learning_rate": 0.00013631235773703534,
      "loss": 3.3788,
      "step": 3060
    },
    {
      "epoch": 1.204748356715393,
      "grad_norm": 10.0625,
      "learning_rate": 0.00013591721896583453,
      "loss": 3.2699,
      "step": 3070
    },
    {
      "epoch": 1.2086726184636516,
      "grad_norm": 5.1875,
      "learning_rate": 0.0001355214352883115,
      "loss": 2.9547,
      "step": 3080
    },
    {
      "epoch": 1.21259688021191,
      "grad_norm": 9.3125,
      "learning_rate": 0.00013512501381090158,
      "loss": 3.4273,
      "step": 3090
    },
    {
      "epoch": 1.2165211419601687,
      "grad_norm": 4.84375,
      "learning_rate": 0.00013472796165149217,
      "loss": 2.8588,
      "step": 3100
    },
    {
      "epoch": 1.2204454037084274,
      "grad_norm": 9.4375,
      "learning_rate": 0.00013433028593929467,
      "loss": 3.3623,
      "step": 3110
    },
    {
      "epoch": 1.224369665456686,
      "grad_norm": 10.875,
      "learning_rate": 0.00013393199381471656,
      "loss": 3.1945,
      "step": 3120
    },
    {
      "epoch": 1.2282939272049447,
      "grad_norm": 7.03125,
      "learning_rate": 0.00013353309242923334,
      "loss": 2.9426,
      "step": 3130
    },
    {
      "epoch": 1.2322181889532031,
      "grad_norm": 8.4375,
      "learning_rate": 0.00013313358894525995,
      "loss": 3.4155,
      "step": 3140
    },
    {
      "epoch": 1.2361424507014618,
      "grad_norm": 4.53125,
      "learning_rate": 0.0001327334905360222,
      "loss": 2.8498,
      "step": 3150
    },
    {
      "epoch": 1.2400667124497204,
      "grad_norm": 6.0625,
      "learning_rate": 0.00013233280438542796,
      "loss": 3.3899,
      "step": 3160
    },
    {
      "epoch": 1.2439909741979789,
      "grad_norm": 9.625,
      "learning_rate": 0.0001319315376879383,
      "loss": 3.2313,
      "step": 3170
    },
    {
      "epoch": 1.2479152359462375,
      "grad_norm": 6.5625,
      "learning_rate": 0.00013152969764843812,
      "loss": 2.9613,
      "step": 3180
    },
    {
      "epoch": 1.2518394976944962,
      "grad_norm": 17.5,
      "learning_rate": 0.00013112729148210693,
      "loss": 3.4318,
      "step": 3190
    },
    {
      "epoch": 1.2557637594427549,
      "grad_norm": 4.09375,
      "learning_rate": 0.00013072432641428932,
      "loss": 2.8443,
      "step": 3200
    },
    {
      "epoch": 1.2596880211910135,
      "grad_norm": 6.78125,
      "learning_rate": 0.00013032080968036498,
      "loss": 3.3626,
      "step": 3210
    },
    {
      "epoch": 1.263612282939272,
      "grad_norm": 7.9375,
      "learning_rate": 0.00012991674852561903,
      "loss": 3.2804,
      "step": 3220
    },
    {
      "epoch": 1.2675365446875306,
      "grad_norm": 6.75,
      "learning_rate": 0.00012951215020511197,
      "loss": 2.9222,
      "step": 3230
    },
    {
      "epoch": 1.2714608064357893,
      "grad_norm": 7.75,
      "learning_rate": 0.00012910702198354915,
      "loss": 3.4187,
      "step": 3240
    },
    {
      "epoch": 1.275385068184048,
      "grad_norm": 5.0625,
      "learning_rate": 0.00012870137113515053,
      "loss": 2.8227,
      "step": 3250
    },
    {
      "epoch": 1.2793093299323064,
      "grad_norm": 5.75,
      "learning_rate": 0.00012829520494352003,
      "loss": 3.3366,
      "step": 3260
    },
    {
      "epoch": 1.283233591680565,
      "grad_norm": 8.25,
      "learning_rate": 0.00012788853070151476,
      "loss": 3.1527,
      "step": 3270
    },
    {
      "epoch": 1.2871578534288237,
      "grad_norm": 4.28125,
      "learning_rate": 0.00012748135571111403,
      "loss": 2.9276,
      "step": 3280
    },
    {
      "epoch": 1.2910821151770824,
      "grad_norm": 6.8125,
      "learning_rate": 0.00012707368728328826,
      "loss": 3.357,
      "step": 3290
    },
    {
      "epoch": 1.295006376925341,
      "grad_norm": 3.90625,
      "learning_rate": 0.0001266655327378677,
      "loss": 2.8407,
      "step": 3300
    },
    {
      "epoch": 1.2989306386735995,
      "grad_norm": 5.40625,
      "learning_rate": 0.000126256899403411,
      "loss": 3.3206,
      "step": 3310
    },
    {
      "epoch": 1.3028549004218581,
      "grad_norm": 7.9375,
      "learning_rate": 0.00012584779461707375,
      "loss": 3.2832,
      "step": 3320
    },
    {
      "epoch": 1.3067791621701168,
      "grad_norm": 3.796875,
      "learning_rate": 0.0001254382257244765,
      "loss": 2.8972,
      "step": 3330
    },
    {
      "epoch": 1.3107034239183752,
      "grad_norm": 6.46875,
      "learning_rate": 0.000125028200079573,
      "loss": 3.3673,
      "step": 3340
    },
    {
      "epoch": 1.314627685666634,
      "grad_norm": 3.453125,
      "learning_rate": 0.0001246177250445182,
      "loss": 2.8378,
      "step": 3350
    },
    {
      "epoch": 1.3185519474148926,
      "grad_norm": 5.1875,
      "learning_rate": 0.00012420680798953603,
      "loss": 3.3349,
      "step": 3360
    },
    {
      "epoch": 1.3224762091631512,
      "grad_norm": 9.0,
      "learning_rate": 0.00012379545629278692,
      "loss": 3.2437,
      "step": 3370
    },
    {
      "epoch": 1.32640047091141,
      "grad_norm": 3.640625,
      "learning_rate": 0.0001233836773402356,
      "loss": 2.9013,
      "step": 3380
    },
    {
      "epoch": 1.3303247326596683,
      "grad_norm": 8.25,
      "learning_rate": 0.00012297147852551821,
      "loss": 3.3827,
      "step": 3390
    },
    {
      "epoch": 1.334248994407927,
      "grad_norm": 4.8125,
      "learning_rate": 0.00012255886724980974,
      "loss": 2.8181,
      "step": 3400
    },
    {
      "epoch": 1.3381732561561857,
      "grad_norm": 4.9375,
      "learning_rate": 0.000122145850921691,
      "loss": 3.2365,
      "step": 3410
    },
    {
      "epoch": 1.342097517904444,
      "grad_norm": 8.6875,
      "learning_rate": 0.00012173243695701576,
      "loss": 3.2435,
      "step": 3420
    },
    {
      "epoch": 1.3460217796527028,
      "grad_norm": 4.21875,
      "learning_rate": 0.00012131863277877731,
      "loss": 2.9154,
      "step": 3430
    },
    {
      "epoch": 1.3499460414009614,
      "grad_norm": 6.25,
      "learning_rate": 0.0001209044458169756,
      "loss": 3.3721,
      "step": 3440
    },
    {
      "epoch": 1.35387030314922,
      "grad_norm": 4.0,
      "learning_rate": 0.00012048988350848338,
      "loss": 2.8586,
      "step": 3450
    },
    {
      "epoch": 1.3577945648974787,
      "grad_norm": 4.90625,
      "learning_rate": 0.00012007495329691301,
      "loss": 3.3341,
      "step": 3460
    },
    {
      "epoch": 1.3617188266457374,
      "grad_norm": 9.0625,
      "learning_rate": 0.00011965966263248267,
      "loss": 3.3053,
      "step": 3470
    },
    {
      "epoch": 1.3656430883939958,
      "grad_norm": 4.03125,
      "learning_rate": 0.00011924401897188248,
      "loss": 2.9475,
      "step": 3480
    },
    {
      "epoch": 1.3695673501422545,
      "grad_norm": 8.1875,
      "learning_rate": 0.00011882802977814091,
      "loss": 3.3718,
      "step": 3490
    },
    {
      "epoch": 1.3734916118905132,
      "grad_norm": 3.53125,
      "learning_rate": 0.0001184117025204905,
      "loss": 2.7268,
      "step": 3500
    },
    {
      "epoch": 1.3774158736387716,
      "grad_norm": 5.25,
      "learning_rate": 0.0001179950446742338,
      "loss": 3.3502,
      "step": 3510
    },
    {
      "epoch": 1.3813401353870303,
      "grad_norm": 6.6875,
      "learning_rate": 0.00011757806372060933,
      "loss": 3.2314,
      "step": 3520
    },
    {
      "epoch": 1.385264397135289,
      "grad_norm": 3.453125,
      "learning_rate": 0.000117160767146657,
      "loss": 2.8762,
      "step": 3530
    },
    {
      "epoch": 1.3891886588835476,
      "grad_norm": 6.125,
      "learning_rate": 0.00011674316244508381,
      "loss": 3.3967,
      "step": 3540
    },
    {
      "epoch": 1.3931129206318063,
      "grad_norm": 3.609375,
      "learning_rate": 0.00011632525711412937,
      "loss": 2.8352,
      "step": 3550
    },
    {
      "epoch": 1.3970371823800647,
      "grad_norm": 4.15625,
      "learning_rate": 0.00011590705865743107,
      "loss": 3.2585,
      "step": 3560
    },
    {
      "epoch": 1.4009614441283234,
      "grad_norm": 7.1875,
      "learning_rate": 0.00011548857458388967,
      "loss": 3.192,
      "step": 3570
    },
    {
      "epoch": 1.404885705876582,
      "grad_norm": 4.65625,
      "learning_rate": 0.00011506981240753405,
      "loss": 2.9044,
      "step": 3580
    },
    {
      "epoch": 1.4088099676248405,
      "grad_norm": 5.03125,
      "learning_rate": 0.00011465077964738673,
      "loss": 3.3137,
      "step": 3590
    },
    {
      "epoch": 1.4127342293730991,
      "grad_norm": 4.3125,
      "learning_rate": 0.00011423148382732853,
      "loss": 2.8164,
      "step": 3600
    },
    {
      "epoch": 1.4166584911213578,
      "grad_norm": 5.25,
      "learning_rate": 0.00011381193247596364,
      "loss": 3.2876,
      "step": 3610
    },
    {
      "epoch": 1.4205827528696164,
      "grad_norm": 8.8125,
      "learning_rate": 0.0001133921331264844,
      "loss": 3.1554,
      "step": 3620
    },
    {
      "epoch": 1.424507014617875,
      "grad_norm": 4.1875,
      "learning_rate": 0.00011297209331653606,
      "loss": 2.8988,
      "step": 3630
    },
    {
      "epoch": 1.4284312763661335,
      "grad_norm": 6.40625,
      "learning_rate": 0.00011255182058808143,
      "loss": 3.4016,
      "step": 3640
    },
    {
      "epoch": 1.4323555381143922,
      "grad_norm": 5.46875,
      "learning_rate": 0.00011213132248726541,
      "loss": 2.8671,
      "step": 3650
    },
    {
      "epoch": 1.4362797998626509,
      "grad_norm": 6.15625,
      "learning_rate": 0.00011171060656427956,
      "loss": 3.2767,
      "step": 3660
    },
    {
      "epoch": 1.4402040616109095,
      "grad_norm": 7.4375,
      "learning_rate": 0.00011128968037322653,
      "loss": 3.2419,
      "step": 3670
    },
    {
      "epoch": 1.444128323359168,
      "grad_norm": 4.21875,
      "learning_rate": 0.00011086855147198441,
      "loss": 2.9089,
      "step": 3680
    },
    {
      "epoch": 1.4480525851074266,
      "grad_norm": 6.8125,
      "learning_rate": 0.00011044722742207101,
      "loss": 3.355,
      "step": 3690
    },
    {
      "epoch": 1.4519768468556853,
      "grad_norm": 5.03125,
      "learning_rate": 0.00011002571578850806,
      "loss": 2.7839,
      "step": 3700
    },
    {
      "epoch": 1.455901108603944,
      "grad_norm": 12.875,
      "learning_rate": 0.0001096040241396855,
      "loss": 3.3362,
      "step": 3710
    },
    {
      "epoch": 1.4598253703522026,
      "grad_norm": 6.71875,
      "learning_rate": 0.0001091821600472255,
      "loss": 3.1724,
      "step": 3720
    },
    {
      "epoch": 1.463749632100461,
      "grad_norm": 3.859375,
      "learning_rate": 0.00010876013108584644,
      "loss": 2.8817,
      "step": 3730
    },
    {
      "epoch": 1.4676738938487197,
      "grad_norm": 5.8125,
      "learning_rate": 0.00010833794483322699,
      "loss": 3.3936,
      "step": 3740
    },
    {
      "epoch": 1.4715981555969784,
      "grad_norm": 3.671875,
      "learning_rate": 0.00010791560886987016,
      "loss": 2.8097,
      "step": 3750
    },
    {
      "epoch": 1.4755224173452368,
      "grad_norm": 4.9375,
      "learning_rate": 0.00010749313077896697,
      "loss": 3.3098,
      "step": 3760
    },
    {
      "epoch": 1.4794466790934955,
      "grad_norm": 8.375,
      "learning_rate": 0.00010707051814626034,
      "loss": 3.181,
      "step": 3770
    },
    {
      "epoch": 1.4833709408417541,
      "grad_norm": 3.203125,
      "learning_rate": 0.00010664777855990909,
      "loss": 2.8417,
      "step": 3780
    },
    {
      "epoch": 1.4872952025900128,
      "grad_norm": 5.28125,
      "learning_rate": 0.00010622491961035139,
      "loss": 3.38,
      "step": 3790
    },
    {
      "epoch": 1.4912194643382715,
      "grad_norm": 3.234375,
      "learning_rate": 0.00010580194889016865,
      "loss": 2.8109,
      "step": 3800
    },
    {
      "epoch": 1.49514372608653,
      "grad_norm": 4.4375,
      "learning_rate": 0.00010537887399394925,
      "loss": 3.259,
      "step": 3810
    },
    {
      "epoch": 1.4990679878347886,
      "grad_norm": 7.84375,
      "learning_rate": 0.00010495570251815204,
      "loss": 3.1661,
      "step": 3820
    },
    {
      "epoch": 1.5029922495830472,
      "grad_norm": 4.78125,
      "learning_rate": 0.00010453244206096992,
      "loss": 2.86,
      "step": 3830
    },
    {
      "epoch": 1.5069165113313057,
      "grad_norm": 9.4375,
      "learning_rate": 0.00010410910022219356,
      "loss": 3.3226,
      "step": 3840
    },
    {
      "epoch": 1.5108407730795643,
      "grad_norm": 3.671875,
      "learning_rate": 0.00010368568460307481,
      "loss": 2.8392,
      "step": 3850
    },
    {
      "epoch": 1.514765034827823,
      "grad_norm": 4.5625,
      "learning_rate": 0.00010326220280619037,
      "loss": 3.2903,
      "step": 3860
    },
    {
      "epoch": 1.5186892965760816,
      "grad_norm": 8.25,
      "learning_rate": 0.00010283866243530505,
      "loss": 3.1542,
      "step": 3870
    },
    {
      "epoch": 1.5226135583243403,
      "grad_norm": 4.4375,
      "learning_rate": 0.00010241507109523552,
      "loss": 2.887,
      "step": 3880
    },
    {
      "epoch": 1.526537820072599,
      "grad_norm": 5.28125,
      "learning_rate": 0.00010199143639171349,
      "loss": 3.3735,
      "step": 3890
    },
    {
      "epoch": 1.5304620818208574,
      "grad_norm": 3.296875,
      "learning_rate": 0.00010156776593124933,
      "loss": 2.7965,
      "step": 3900
    },
    {
      "epoch": 1.534386343569116,
      "grad_norm": 4.125,
      "learning_rate": 0.00010114406732099548,
      "loss": 3.2993,
      "step": 3910
    },
    {
      "epoch": 1.5383106053173745,
      "grad_norm": 8.5,
      "learning_rate": 0.00010072034816860979,
      "loss": 3.2196,
      "step": 3920
    },
    {
      "epoch": 1.5422348670656332,
      "grad_norm": 3.5,
      "learning_rate": 0.00010029661608211885,
      "loss": 2.8629,
      "step": 3930
    },
    {
      "epoch": 1.5461591288138918,
      "grad_norm": 5.6875,
      "learning_rate": 9.987287866978169e-05,
      "loss": 3.3406,
      "step": 3940
    },
    {
      "epoch": 1.5500833905621505,
      "grad_norm": 2.96875,
      "learning_rate": 9.944914353995277e-05,
      "loss": 2.8149,
      "step": 3950
    },
    {
      "epoch": 1.5540076523104092,
      "grad_norm": 4.53125,
      "learning_rate": 9.902541830094569e-05,
      "loss": 3.2712,
      "step": 3960
    },
    {
      "epoch": 1.5579319140586678,
      "grad_norm": 6.8125,
      "learning_rate": 9.860171056089646e-05,
      "loss": 3.1448,
      "step": 3970
    },
    {
      "epoch": 1.5618561758069265,
      "grad_norm": 3.53125,
      "learning_rate": 9.817802792762676e-05,
      "loss": 2.8779,
      "step": 3980
    },
    {
      "epoch": 1.565780437555185,
      "grad_norm": 4.84375,
      "learning_rate": 9.775437800850763e-05,
      "loss": 3.3311,
      "step": 3990
    },
    {
      "epoch": 1.5697046993034436,
      "grad_norm": 2.84375,
      "learning_rate": 9.73307684103226e-05,
      "loss": 2.8163,
      "step": 4000
    },
    {
      "epoch": 1.573628961051702,
      "grad_norm": 3.875,
      "learning_rate": 9.690720673913133e-05,
      "loss": 3.2117,
      "step": 4010
    },
    {
      "epoch": 1.5775532227999607,
      "grad_norm": 6.375,
      "learning_rate": 9.648370060013279e-05,
      "loss": 3.169,
      "step": 4020
    },
    {
      "epoch": 1.5814774845482193,
      "grad_norm": 3.421875,
      "learning_rate": 9.606025759752895e-05,
      "loss": 2.8636,
      "step": 4030
    },
    {
      "epoch": 1.585401746296478,
      "grad_norm": 4.78125,
      "learning_rate": 9.563688533438819e-05,
      "loss": 3.3231,
      "step": 4040
    },
    {
      "epoch": 1.5893260080447367,
      "grad_norm": 4.0625,
      "learning_rate": 9.521359141250859e-05,
      "loss": 2.7778,
      "step": 4050
    },
    {
      "epoch": 1.5932502697929953,
      "grad_norm": 4.25,
      "learning_rate": 9.479038343228173e-05,
      "loss": 3.261,
      "step": 4060
    },
    {
      "epoch": 1.5971745315412538,
      "grad_norm": 11.1875,
      "learning_rate": 9.436726899255597e-05,
      "loss": 3.2348,
      "step": 4070
    },
    {
      "epoch": 1.6010987932895124,
      "grad_norm": 3.703125,
      "learning_rate": 9.394425569050017e-05,
      "loss": 2.9147,
      "step": 4080
    },
    {
      "epoch": 1.6050230550377709,
      "grad_norm": 5.1875,
      "learning_rate": 9.352135112146726e-05,
      "loss": 3.3357,
      "step": 4090
    },
    {
      "epoch": 1.6089473167860295,
      "grad_norm": 3.734375,
      "learning_rate": 9.309856287885774e-05,
      "loss": 2.8078,
      "step": 4100
    },
    {
      "epoch": 1.6128715785342882,
      "grad_norm": 5.1875,
      "learning_rate": 9.267589855398356e-05,
      "loss": 3.2358,
      "step": 4110
    },
    {
      "epoch": 1.6167958402825469,
      "grad_norm": 19.875,
      "learning_rate": 9.225336573593149e-05,
      "loss": 3.1905,
      "step": 4120
    },
    {
      "epoch": 1.6207201020308055,
      "grad_norm": 3.3125,
      "learning_rate": 9.183097201142722e-05,
      "loss": 2.8658,
      "step": 4130
    },
    {
      "epoch": 1.6246443637790642,
      "grad_norm": 6.25,
      "learning_rate": 9.14087249646989e-05,
      "loss": 3.3303,
      "step": 4140
    },
    {
      "epoch": 1.6285686255273226,
      "grad_norm": 3.140625,
      "learning_rate": 9.098663217734102e-05,
      "loss": 2.8094,
      "step": 4150
    },
    {
      "epoch": 1.6324928872755813,
      "grad_norm": 4.5625,
      "learning_rate": 9.056470122817836e-05,
      "loss": 3.2288,
      "step": 4160
    },
    {
      "epoch": 1.63641714902384,
      "grad_norm": 7.15625,
      "learning_rate": 9.01429396931297e-05,
      "loss": 3.1835,
      "step": 4170
    },
    {
      "epoch": 1.6403414107720984,
      "grad_norm": 4.34375,
      "learning_rate": 8.972135514507211e-05,
      "loss": 2.8334,
      "step": 4180
    },
    {
      "epoch": 1.644265672520357,
      "grad_norm": 6.375,
      "learning_rate": 8.92999551537046e-05,
      "loss": 3.357,
      "step": 4190
    },
    {
      "epoch": 1.6481899342686157,
      "grad_norm": 3.75,
      "learning_rate": 8.887874728541258e-05,
      "loss": 2.754,
      "step": 4200
    },
    {
      "epoch": 1.6521141960168744,
      "grad_norm": 4.21875,
      "learning_rate": 8.845773910313168e-05,
      "loss": 3.223,
      "step": 4210
    },
    {
      "epoch": 1.656038457765133,
      "grad_norm": 6.78125,
      "learning_rate": 8.803693816621217e-05,
      "loss": 3.1291,
      "step": 4220
    },
    {
      "epoch": 1.6599627195133917,
      "grad_norm": 3.703125,
      "learning_rate": 8.761635203028318e-05,
      "loss": 2.8407,
      "step": 4230
    },
    {
      "epoch": 1.6638869812616501,
      "grad_norm": 6.9375,
      "learning_rate": 8.719598824711693e-05,
      "loss": 3.3596,
      "step": 4240
    },
    {
      "epoch": 1.6678112430099088,
      "grad_norm": 3.171875,
      "learning_rate": 8.677585436449333e-05,
      "loss": 2.7434,
      "step": 4250
    },
    {
      "epoch": 1.6717355047581672,
      "grad_norm": 6.28125,
      "learning_rate": 8.635595792606419e-05,
      "loss": 3.2413,
      "step": 4260
    },
    {
      "epoch": 1.675659766506426,
      "grad_norm": 33.25,
      "learning_rate": 8.593630647121809e-05,
      "loss": 3.1575,
      "step": 4270
    },
    {
      "epoch": 1.6795840282546846,
      "grad_norm": 4.40625,
      "learning_rate": 8.551690753494476e-05,
      "loss": 2.8861,
      "step": 4280
    },
    {
      "epoch": 1.6835082900029432,
      "grad_norm": 7.875,
      "learning_rate": 8.509776864769981e-05,
      "loss": 3.3047,
      "step": 4290
    },
    {
      "epoch": 1.6874325517512019,
      "grad_norm": 3.828125,
      "learning_rate": 8.467889733526977e-05,
      "loss": 2.742,
      "step": 4300
    },
    {
      "epoch": 1.6913568134994605,
      "grad_norm": 7.125,
      "learning_rate": 8.426030111863653e-05,
      "loss": 3.2983,
      "step": 4310
    },
    {
      "epoch": 1.695281075247719,
      "grad_norm": 10.3125,
      "learning_rate": 8.384198751384272e-05,
      "loss": 3.1559,
      "step": 4320
    },
    {
      "epoch": 1.6992053369959776,
      "grad_norm": 4.90625,
      "learning_rate": 8.342396403185648e-05,
      "loss": 2.8285,
      "step": 4330
    },
    {
      "epoch": 1.703129598744236,
      "grad_norm": 6.875,
      "learning_rate": 8.300623817843673e-05,
      "loss": 3.2679,
      "step": 4340
    },
    {
      "epoch": 1.7070538604924947,
      "grad_norm": 3.578125,
      "learning_rate": 8.258881745399837e-05,
      "loss": 2.7309,
      "step": 4350
    },
    {
      "epoch": 1.7109781222407534,
      "grad_norm": 9.8125,
      "learning_rate": 8.217170935347756e-05,
      "loss": 3.2175,
      "step": 4360
    },
    {
      "epoch": 1.714902383989012,
      "grad_norm": 8.125,
      "learning_rate": 8.175492136619729e-05,
      "loss": 3.1113,
      "step": 4370
    },
    {
      "epoch": 1.7188266457372707,
      "grad_norm": 4.34375,
      "learning_rate": 8.133846097573263e-05,
      "loss": 2.8567,
      "step": 4380
    },
    {
      "epoch": 1.7227509074855294,
      "grad_norm": 14.1875,
      "learning_rate": 8.092233565977669e-05,
      "loss": 3.3134,
      "step": 4390
    },
    {
      "epoch": 1.726675169233788,
      "grad_norm": 3.015625,
      "learning_rate": 8.050655289000612e-05,
      "loss": 2.7521,
      "step": 4400
    },
    {
      "epoch": 1.7305994309820465,
      "grad_norm": 4.71875,
      "learning_rate": 8.009112013194706e-05,
      "loss": 3.2908,
      "step": 4410
    },
    {
      "epoch": 1.7345236927303052,
      "grad_norm": 9.6875,
      "learning_rate": 7.96760448448411e-05,
      "loss": 3.1376,
      "step": 4420
    },
    {
      "epoch": 1.7384479544785636,
      "grad_norm": 5.90625,
      "learning_rate": 7.926133448151121e-05,
      "loss": 2.878,
      "step": 4430
    },
    {
      "epoch": 1.7423722162268223,
      "grad_norm": 7.5,
      "learning_rate": 7.884699648822815e-05,
      "loss": 3.3233,
      "step": 4440
    },
    {
      "epoch": 1.746296477975081,
      "grad_norm": 3.546875,
      "learning_rate": 7.843303830457655e-05,
      "loss": 2.7036,
      "step": 4450
    },
    {
      "epoch": 1.7502207397233396,
      "grad_norm": 6.71875,
      "learning_rate": 7.801946736332144e-05,
      "loss": 3.2372,
      "step": 4460
    },
    {
      "epoch": 1.7541450014715982,
      "grad_norm": 9.4375,
      "learning_rate": 7.760629109027488e-05,
      "loss": 3.1136,
      "step": 4470
    },
    {
      "epoch": 1.758069263219857,
      "grad_norm": 4.0625,
      "learning_rate": 7.719351690416233e-05,
      "loss": 2.8625,
      "step": 4480
    },
    {
      "epoch": 1.7619935249681153,
      "grad_norm": 8.5625,
      "learning_rate": 7.678115221648983e-05,
      "loss": 3.2867,
      "step": 4490
    },
    {
      "epoch": 1.765917786716374,
      "grad_norm": 3.9375,
      "learning_rate": 7.636920443141057e-05,
      "loss": 2.7912,
      "step": 4500
    },
    {
      "epoch": 1.7698420484646324,
      "grad_norm": 4.9375,
      "learning_rate": 7.595768094559226e-05,
      "loss": 3.2433,
      "step": 4510
    },
    {
      "epoch": 1.773766310212891,
      "grad_norm": 6.4375,
      "learning_rate": 7.554658914808404e-05,
      "loss": 3.1738,
      "step": 4520
    },
    {
      "epoch": 1.7776905719611498,
      "grad_norm": 3.53125,
      "learning_rate": 7.513593642018398e-05,
      "loss": 2.894,
      "step": 4530
    },
    {
      "epoch": 1.7816148337094084,
      "grad_norm": 5.4375,
      "learning_rate": 7.472573013530657e-05,
      "loss": 3.2958,
      "step": 4540
    },
    {
      "epoch": 1.785539095457667,
      "grad_norm": 3.0625,
      "learning_rate": 7.431597765885013e-05,
      "loss": 2.7814,
      "step": 4550
    },
    {
      "epoch": 1.7894633572059258,
      "grad_norm": 5.3125,
      "learning_rate": 7.39066863480648e-05,
      "loss": 3.2027,
      "step": 4560
    },
    {
      "epoch": 1.7933876189541842,
      "grad_norm": 7.875,
      "learning_rate": 7.349786355192023e-05,
      "loss": 3.1535,
      "step": 4570
    },
    {
      "epoch": 1.7973118807024429,
      "grad_norm": 5.53125,
      "learning_rate": 7.308951661097379e-05,
      "loss": 2.7951,
      "step": 4580
    },
    {
      "epoch": 1.8012361424507015,
      "grad_norm": 8.75,
      "learning_rate": 7.268165285723875e-05,
      "loss": 3.3281,
      "step": 4590
    },
    {
      "epoch": 1.80516040419896,
      "grad_norm": 3.625,
      "learning_rate": 7.227427961405245e-05,
      "loss": 2.7747,
      "step": 4600
    },
    {
      "epoch": 1.8090846659472186,
      "grad_norm": 5.40625,
      "learning_rate": 7.186740419594505e-05,
      "loss": 3.2383,
      "step": 4610
    },
    {
      "epoch": 1.8130089276954773,
      "grad_norm": 7.8125,
      "learning_rate": 7.1461033908508e-05,
      "loss": 3.1483,
      "step": 4620
    },
    {
      "epoch": 1.816933189443736,
      "grad_norm": 3.953125,
      "learning_rate": 7.105517604826309e-05,
      "loss": 2.8681,
      "step": 4630
    },
    {
      "epoch": 1.8208574511919946,
      "grad_norm": 6.03125,
      "learning_rate": 7.064983790253108e-05,
      "loss": 3.2755,
      "step": 4640
    },
    {
      "epoch": 1.8247817129402533,
      "grad_norm": 3.40625,
      "learning_rate": 7.024502674930132e-05,
      "loss": 2.7963,
      "step": 4650
    },
    {
      "epoch": 1.8287059746885117,
      "grad_norm": 5.0625,
      "learning_rate": 6.984074985710067e-05,
      "loss": 3.2172,
      "step": 4660
    },
    {
      "epoch": 1.8326302364367704,
      "grad_norm": 8.0625,
      "learning_rate": 6.943701448486313e-05,
      "loss": 3.1933,
      "step": 4670
    },
    {
      "epoch": 1.8365544981850288,
      "grad_norm": 4.75,
      "learning_rate": 6.903382788179961e-05,
      "loss": 2.9066,
      "step": 4680
    },
    {
      "epoch": 1.8404787599332875,
      "grad_norm": 8.875,
      "learning_rate": 6.863119728726763e-05,
      "loss": 3.3091,
      "step": 4690
    },
    {
      "epoch": 1.8444030216815461,
      "grad_norm": 3.28125,
      "learning_rate": 6.82291299306414e-05,
      "loss": 2.749,
      "step": 4700
    },
    {
      "epoch": 1.8483272834298048,
      "grad_norm": 7.90625,
      "learning_rate": 6.782763303118193e-05,
      "loss": 3.288,
      "step": 4710
    },
    {
      "epoch": 1.8522515451780635,
      "grad_norm": 11.875,
      "learning_rate": 6.742671379790756e-05,
      "loss": 3.1225,
      "step": 4720
    },
    {
      "epoch": 1.8561758069263221,
      "grad_norm": 3.90625,
      "learning_rate": 6.702637942946442e-05,
      "loss": 2.8471,
      "step": 4730
    },
    {
      "epoch": 1.8601000686745806,
      "grad_norm": 13.4375,
      "learning_rate": 6.662663711399705e-05,
      "loss": 3.271,
      "step": 4740
    },
    {
      "epoch": 1.8640243304228392,
      "grad_norm": 3.421875,
      "learning_rate": 6.622749402901971e-05,
      "loss": 2.7424,
      "step": 4750
    },
    {
      "epoch": 1.8679485921710977,
      "grad_norm": 5.34375,
      "learning_rate": 6.582895734128702e-05,
      "loss": 3.2389,
      "step": 4760
    },
    {
      "epoch": 1.8718728539193563,
      "grad_norm": 8.625,
      "learning_rate": 6.543103420666569e-05,
      "loss": 3.1551,
      "step": 4770
    },
    {
      "epoch": 1.875797115667615,
      "grad_norm": 3.875,
      "learning_rate": 6.503373177000581e-05,
      "loss": 2.8416,
      "step": 4780
    },
    {
      "epoch": 1.8797213774158736,
      "grad_norm": 8.0625,
      "learning_rate": 6.46370571650126e-05,
      "loss": 3.3252,
      "step": 4790
    },
    {
      "epoch": 1.8836456391641323,
      "grad_norm": 3.515625,
      "learning_rate": 6.424101751411842e-05,
      "loss": 2.7402,
      "step": 4800
    },
    {
      "epoch": 1.887569900912391,
      "grad_norm": 7.125,
      "learning_rate": 6.384561992835467e-05,
      "loss": 3.2315,
      "step": 4810
    },
    {
      "epoch": 1.8914941626606496,
      "grad_norm": 8.1875,
      "learning_rate": 6.34508715072244e-05,
      "loss": 3.1206,
      "step": 4820
    },
    {
      "epoch": 1.895418424408908,
      "grad_norm": 3.546875,
      "learning_rate": 6.305677933857455e-05,
      "loss": 2.8531,
      "step": 4830
    },
    {
      "epoch": 1.8993426861571667,
      "grad_norm": 6.75,
      "learning_rate": 6.266335049846885e-05,
      "loss": 3.2801,
      "step": 4840
    },
    {
      "epoch": 1.9032669479054252,
      "grad_norm": 2.78125,
      "learning_rate": 6.227059205106084e-05,
      "loss": 2.7515,
      "step": 4850
    },
    {
      "epoch": 1.9071912096536838,
      "grad_norm": 7.9375,
      "learning_rate": 6.187851104846676e-05,
      "loss": 3.2551,
      "step": 4860
    },
    {
      "epoch": 1.9111154714019425,
      "grad_norm": 7.09375,
      "learning_rate": 6.148711453063921e-05,
      "loss": 3.1063,
      "step": 4870
    },
    {
      "epoch": 1.9150397331502012,
      "grad_norm": 3.828125,
      "learning_rate": 6.109640952524051e-05,
      "loss": 2.8671,
      "step": 4880
    },
    {
      "epoch": 1.9189639948984598,
      "grad_norm": 6.5,
      "learning_rate": 6.0706403047516766e-05,
      "loss": 3.3074,
      "step": 4890
    },
    {
      "epoch": 1.9228882566467185,
      "grad_norm": 3.40625,
      "learning_rate": 6.0317102100171706e-05,
      "loss": 2.7798,
      "step": 4900
    },
    {
      "epoch": 1.926812518394977,
      "grad_norm": 5.5,
      "learning_rate": 5.992851367324097e-05,
      "loss": 3.2089,
      "step": 4910
    },
    {
      "epoch": 1.9307367801432356,
      "grad_norm": 10.0,
      "learning_rate": 5.954064474396674e-05,
      "loss": 3.1548,
      "step": 4920
    },
    {
      "epoch": 1.934661041891494,
      "grad_norm": 5.0,
      "learning_rate": 5.915350227667225e-05,
      "loss": 2.8291,
      "step": 4930
    },
    {
      "epoch": 1.9385853036397527,
      "grad_norm": 7.15625,
      "learning_rate": 5.876709322263696e-05,
      "loss": 3.3094,
      "step": 4940
    },
    {
      "epoch": 1.9425095653880113,
      "grad_norm": 3.0,
      "learning_rate": 5.838142451997155e-05,
      "loss": 2.7196,
      "step": 4950
    },
    {
      "epoch": 1.94643382713627,
      "grad_norm": 5.40625,
      "learning_rate": 5.799650309349348e-05,
      "loss": 3.2393,
      "step": 4960
    },
    {
      "epoch": 1.9503580888845287,
      "grad_norm": 12.125,
      "learning_rate": 5.761233585460265e-05,
      "loss": 3.159,
      "step": 4970
    },
    {
      "epoch": 1.9542823506327873,
      "grad_norm": 3.953125,
      "learning_rate": 5.722892970115712e-05,
      "loss": 2.7903,
      "step": 4980
    },
    {
      "epoch": 1.9582066123810458,
      "grad_norm": 6.09375,
      "learning_rate": 5.684629151734948e-05,
      "loss": 3.3124,
      "step": 4990
    },
    {
      "epoch": 1.9621308741293044,
      "grad_norm": 3.171875,
      "learning_rate": 5.646442817358317e-05,
      "loss": 2.727,
      "step": 5000
    },
    {
      "epoch": 1.966055135877563,
      "grad_norm": 7.15625,
      "learning_rate": 5.608334652634913e-05,
      "loss": 3.1912,
      "step": 5010
    },
    {
      "epoch": 1.9699793976258215,
      "grad_norm": 8.25,
      "learning_rate": 5.5703053418102514e-05,
      "loss": 3.1617,
      "step": 5020
    },
    {
      "epoch": 1.9739036593740802,
      "grad_norm": 5.71875,
      "learning_rate": 5.5323555677140124e-05,
      "loss": 2.8438,
      "step": 5030
    },
    {
      "epoch": 1.9778279211223389,
      "grad_norm": 7.8125,
      "learning_rate": 5.4944860117477615e-05,
      "loss": 3.2579,
      "step": 5040
    },
    {
      "epoch": 1.9817521828705975,
      "grad_norm": 2.859375,
      "learning_rate": 5.456697353872722e-05,
      "loss": 2.7288,
      "step": 5050
    },
    {
      "epoch": 1.9856764446188562,
      "grad_norm": 4.78125,
      "learning_rate": 5.418990272597561e-05,
      "loss": 3.2191,
      "step": 5060
    },
    {
      "epoch": 1.9896007063671148,
      "grad_norm": 7.59375,
      "learning_rate": 5.381365444966204e-05,
      "loss": 3.1076,
      "step": 5070
    },
    {
      "epoch": 1.9935249681153733,
      "grad_norm": 4.21875,
      "learning_rate": 5.343823546545692e-05,
      "loss": 2.8043,
      "step": 5080
    },
    {
      "epoch": 1.997449229863632,
      "grad_norm": 6.125,
      "learning_rate": 5.306365251414043e-05,
      "loss": 3.2614,
      "step": 5090
    },
    {
      "epoch": 1.999803786912587,
      "eval_loss": 3.0933620929718018,
      "eval_runtime": 89.9673,
      "eval_samples_per_second": 277.879,
      "eval_steps_per_second": 69.47,
      "step": 5096
    },
    {
      "epoch": 2.0013734916118904,
      "grad_norm": 3.4375,
      "learning_rate": 5.268991232148137e-05,
      "loss": 2.6659,
      "step": 5100
    },
    {
      "epoch": 2.005297753360149,
      "grad_norm": 7.15625,
      "learning_rate": 5.2317021598116634e-05,
      "loss": 3.3594,
      "step": 5110
    },
    {
      "epoch": 2.0092220151084077,
      "grad_norm": 9.8125,
      "learning_rate": 5.194498703943054e-05,
      "loss": 2.9003,
      "step": 5120
    },
    {
      "epoch": 2.0131462768566664,
      "grad_norm": 4.625,
      "learning_rate": 5.1573815325434726e-05,
      "loss": 2.9892,
      "step": 5130
    },
    {
      "epoch": 2.017070538604925,
      "grad_norm": 9.4375,
      "learning_rate": 5.120351312064802e-05,
      "loss": 3.2151,
      "step": 5140
    },
    {
      "epoch": 2.0209948003531837,
      "grad_norm": 3.234375,
      "learning_rate": 5.083408707397703e-05,
      "loss": 2.6459,
      "step": 5150
    },
    {
      "epoch": 2.0249190621014423,
      "grad_norm": 6.75,
      "learning_rate": 5.0465543818596626e-05,
      "loss": 3.3432,
      "step": 5160
    },
    {
      "epoch": 2.0288433238497006,
      "grad_norm": 10.375,
      "learning_rate": 5.009788997183073e-05,
      "loss": 2.9179,
      "step": 5170
    },
    {
      "epoch": 2.0327675855979592,
      "grad_norm": 4.75,
      "learning_rate": 4.9731132135033786e-05,
      "loss": 3.0019,
      "step": 5180
    },
    {
      "epoch": 2.036691847346218,
      "grad_norm": 7.71875,
      "learning_rate": 4.936527689347194e-05,
      "loss": 3.2437,
      "step": 5190
    },
    {
      "epoch": 2.0406161090944765,
      "grad_norm": 2.953125,
      "learning_rate": 4.9000330816204995e-05,
      "loss": 2.6258,
      "step": 5200
    },
    {
      "epoch": 2.044540370842735,
      "grad_norm": 9.3125,
      "learning_rate": 4.863630045596837e-05,
      "loss": 3.2837,
      "step": 5210
    },
    {
      "epoch": 2.048464632590994,
      "grad_norm": 9.0625,
      "learning_rate": 4.827319234905541e-05,
      "loss": 2.9071,
      "step": 5220
    },
    {
      "epoch": 2.0523888943392525,
      "grad_norm": 4.5,
      "learning_rate": 4.791101301520016e-05,
      "loss": 2.9952,
      "step": 5230
    },
    {
      "epoch": 2.056313156087511,
      "grad_norm": 8.5625,
      "learning_rate": 4.754976895746007e-05,
      "loss": 3.2184,
      "step": 5240
    },
    {
      "epoch": 2.06023741783577,
      "grad_norm": 3.21875,
      "learning_rate": 4.7189466662099654e-05,
      "loss": 2.7167,
      "step": 5250
    },
    {
      "epoch": 2.064161679584028,
      "grad_norm": 8.0625,
      "learning_rate": 4.683011259847345e-05,
      "loss": 3.3024,
      "step": 5260
    },
    {
      "epoch": 2.0680859413322867,
      "grad_norm": 11.5,
      "learning_rate": 4.647171321891034e-05,
      "loss": 2.9297,
      "step": 5270
    },
    {
      "epoch": 2.0720102030805454,
      "grad_norm": 4.28125,
      "learning_rate": 4.61142749585975e-05,
      "loss": 2.9569,
      "step": 5280
    },
    {
      "epoch": 2.075934464828804,
      "grad_norm": 12.625,
      "learning_rate": 4.575780423546476e-05,
      "loss": 3.2569,
      "step": 5290
    },
    {
      "epoch": 2.0798587265770627,
      "grad_norm": 2.96875,
      "learning_rate": 4.540230745006961e-05,
      "loss": 2.6422,
      "step": 5300
    },
    {
      "epoch": 2.0837829883253214,
      "grad_norm": 5.5,
      "learning_rate": 4.504779098548209e-05,
      "loss": 3.2815,
      "step": 5310
    },
    {
      "epoch": 2.08770725007358,
      "grad_norm": 8.625,
      "learning_rate": 4.469426120717025e-05,
      "loss": 2.9197,
      "step": 5320
    },
    {
      "epoch": 2.0916315118218387,
      "grad_norm": 4.1875,
      "learning_rate": 4.4341724462885794e-05,
      "loss": 2.95,
      "step": 5330
    },
    {
      "epoch": 2.095555773570097,
      "grad_norm": 8.125,
      "learning_rate": 4.399018708255018e-05,
      "loss": 3.249,
      "step": 5340
    },
    {
      "epoch": 2.0994800353183556,
      "grad_norm": 2.46875,
      "learning_rate": 4.363965537814102e-05,
      "loss": 2.6602,
      "step": 5350
    },
    {
      "epoch": 2.1034042970666142,
      "grad_norm": 6.21875,
      "learning_rate": 4.3290135643578476e-05,
      "loss": 3.341,
      "step": 5360
    },
    {
      "epoch": 2.107328558814873,
      "grad_norm": 8.125,
      "learning_rate": 4.294163415461258e-05,
      "loss": 2.9468,
      "step": 5370
    },
    {
      "epoch": 2.1112528205631316,
      "grad_norm": 3.84375,
      "learning_rate": 4.259415716871037e-05,
      "loss": 2.988,
      "step": 5380
    },
    {
      "epoch": 2.1151770823113902,
      "grad_norm": 19.0,
      "learning_rate": 4.224771092494355e-05,
      "loss": 3.2259,
      "step": 5390
    },
    {
      "epoch": 2.119101344059649,
      "grad_norm": 2.65625,
      "learning_rate": 4.190230164387655e-05,
      "loss": 2.6436,
      "step": 5400
    },
    {
      "epoch": 2.1230256058079076,
      "grad_norm": 4.90625,
      "learning_rate": 4.155793552745465e-05,
      "loss": 3.3128,
      "step": 5410
    },
    {
      "epoch": 2.126949867556166,
      "grad_norm": 10.0,
      "learning_rate": 4.1214618758892866e-05,
      "loss": 2.9354,
      "step": 5420
    },
    {
      "epoch": 2.1308741293044244,
      "grad_norm": 4.28125,
      "learning_rate": 4.0872357502564686e-05,
      "loss": 3.0076,
      "step": 5430
    },
    {
      "epoch": 2.134798391052683,
      "grad_norm": 7.1875,
      "learning_rate": 4.053115790389159e-05,
      "loss": 3.2409,
      "step": 5440
    },
    {
      "epoch": 2.1387226528009418,
      "grad_norm": 2.953125,
      "learning_rate": 4.019102608923262e-05,
      "loss": 2.6429,
      "step": 5450
    },
    {
      "epoch": 2.1426469145492004,
      "grad_norm": 5.75,
      "learning_rate": 3.985196816577433e-05,
      "loss": 3.2948,
      "step": 5460
    },
    {
      "epoch": 2.146571176297459,
      "grad_norm": 8.6875,
      "learning_rate": 3.951399022142127e-05,
      "loss": 2.8924,
      "step": 5470
    },
    {
      "epoch": 2.1504954380457177,
      "grad_norm": 5.0,
      "learning_rate": 3.9177098324686413e-05,
      "loss": 3.004,
      "step": 5480
    },
    {
      "epoch": 2.1544196997939764,
      "grad_norm": 6.65625,
      "learning_rate": 3.884129852458254e-05,
      "loss": 3.2547,
      "step": 5490
    },
    {
      "epoch": 2.158343961542235,
      "grad_norm": 2.875,
      "learning_rate": 3.8506596850513355e-05,
      "loss": 2.7089,
      "step": 5500
    },
    {
      "epoch": 2.1622682232904933,
      "grad_norm": 4.65625,
      "learning_rate": 3.817299931216537e-05,
      "loss": 3.2661,
      "step": 5510
    },
    {
      "epoch": 2.166192485038752,
      "grad_norm": 8.4375,
      "learning_rate": 3.7840511899399954e-05,
      "loss": 2.9323,
      "step": 5520
    },
    {
      "epoch": 2.1701167467870106,
      "grad_norm": 3.953125,
      "learning_rate": 3.7509140582145706e-05,
      "loss": 3.0021,
      "step": 5530
    },
    {
      "epoch": 2.1740410085352693,
      "grad_norm": 6.375,
      "learning_rate": 3.7178891310291444e-05,
      "loss": 3.26,
      "step": 5540
    },
    {
      "epoch": 2.177965270283528,
      "grad_norm": 2.53125,
      "learning_rate": 3.684977001357913e-05,
      "loss": 2.7229,
      "step": 5550
    },
    {
      "epoch": 2.1818895320317866,
      "grad_norm": 6.375,
      "learning_rate": 3.6521782601497677e-05,
      "loss": 3.3369,
      "step": 5560
    },
    {
      "epoch": 2.1858137937800453,
      "grad_norm": 11.0625,
      "learning_rate": 3.619493496317662e-05,
      "loss": 2.8716,
      "step": 5570
    },
    {
      "epoch": 2.189738055528304,
      "grad_norm": 3.96875,
      "learning_rate": 3.586923296728046e-05,
      "loss": 2.9205,
      "step": 5580
    },
    {
      "epoch": 2.193662317276562,
      "grad_norm": 7.5625,
      "learning_rate": 3.5544682461903366e-05,
      "loss": 3.2348,
      "step": 5590
    },
    {
      "epoch": 2.197586579024821,
      "grad_norm": 2.859375,
      "learning_rate": 3.522128927446392e-05,
      "loss": 2.6707,
      "step": 5600
    },
    {
      "epoch": 2.2015108407730795,
      "grad_norm": 5.09375,
      "learning_rate": 3.4899059211600825e-05,
      "loss": 3.297,
      "step": 5610
    },
    {
      "epoch": 2.205435102521338,
      "grad_norm": 13.0625,
      "learning_rate": 3.457799805906835e-05,
      "loss": 2.8501,
      "step": 5620
    },
    {
      "epoch": 2.209359364269597,
      "grad_norm": 4.25,
      "learning_rate": 3.4258111581632636e-05,
      "loss": 3.0116,
      "step": 5630
    },
    {
      "epoch": 2.2132836260178554,
      "grad_norm": 6.84375,
      "learning_rate": 3.3939405522968105e-05,
      "loss": 3.2553,
      "step": 5640
    },
    {
      "epoch": 2.217207887766114,
      "grad_norm": 3.03125,
      "learning_rate": 3.362188560555434e-05,
      "loss": 2.6431,
      "step": 5650
    },
    {
      "epoch": 2.2211321495143728,
      "grad_norm": 4.9375,
      "learning_rate": 3.330555753057336e-05,
      "loss": 3.2988,
      "step": 5660
    },
    {
      "epoch": 2.2250564112626314,
      "grad_norm": 18.375,
      "learning_rate": 3.2990426977807156e-05,
      "loss": 2.9394,
      "step": 5670
    },
    {
      "epoch": 2.2289806730108896,
      "grad_norm": 4.3125,
      "learning_rate": 3.2676499605535914e-05,
      "loss": 2.9451,
      "step": 5680
    },
    {
      "epoch": 2.2329049347591483,
      "grad_norm": 6.28125,
      "learning_rate": 3.23637810504361e-05,
      "loss": 3.2422,
      "step": 5690
    },
    {
      "epoch": 2.236829196507407,
      "grad_norm": 2.796875,
      "learning_rate": 3.205227692747968e-05,
      "loss": 2.638,
      "step": 5700
    },
    {
      "epoch": 2.2407534582556656,
      "grad_norm": 4.65625,
      "learning_rate": 3.174199282983292e-05,
      "loss": 3.3034,
      "step": 5710
    },
    {
      "epoch": 2.2446777200039243,
      "grad_norm": 9.375,
      "learning_rate": 3.143293432875607e-05,
      "loss": 2.9075,
      "step": 5720
    },
    {
      "epoch": 2.248601981752183,
      "grad_norm": 4.65625,
      "learning_rate": 3.1125106973503485e-05,
      "loss": 3.0105,
      "step": 5730
    },
    {
      "epoch": 2.2525262435004416,
      "grad_norm": 9.125,
      "learning_rate": 3.081851629122372e-05,
      "loss": 3.291,
      "step": 5740
    },
    {
      "epoch": 2.2564505052487003,
      "grad_norm": 2.53125,
      "learning_rate": 3.051316778686055e-05,
      "loss": 2.6691,
      "step": 5750
    },
    {
      "epoch": 2.260374766996959,
      "grad_norm": 6.53125,
      "learning_rate": 3.0209066943053943e-05,
      "loss": 3.3347,
      "step": 5760
    },
    {
      "epoch": 2.264299028745217,
      "grad_norm": 8.25,
      "learning_rate": 2.990621922004172e-05,
      "loss": 2.8579,
      "step": 5770
    },
    {
      "epoch": 2.268223290493476,
      "grad_norm": 3.9375,
      "learning_rate": 2.960463005556149e-05,
      "loss": 3.0193,
      "step": 5780
    },
    {
      "epoch": 2.2721475522417345,
      "grad_norm": 8.0625,
      "learning_rate": 2.9304304864752884e-05,
      "loss": 3.2612,
      "step": 5790
    },
    {
      "epoch": 2.276071813989993,
      "grad_norm": 2.96875,
      "learning_rate": 2.9005249040060612e-05,
      "loss": 2.6391,
      "step": 5800
    },
    {
      "epoch": 2.279996075738252,
      "grad_norm": 6.125,
      "learning_rate": 2.87074679511373e-05,
      "loss": 3.2579,
      "step": 5810
    },
    {
      "epoch": 2.2839203374865105,
      "grad_norm": 10.1875,
      "learning_rate": 2.8410966944747376e-05,
      "loss": 2.8872,
      "step": 5820
    },
    {
      "epoch": 2.287844599234769,
      "grad_norm": 3.90625,
      "learning_rate": 2.811575134467086e-05,
      "loss": 2.9771,
      "step": 5830
    },
    {
      "epoch": 2.2917688609830273,
      "grad_norm": 6.3125,
      "learning_rate": 2.782182645160789e-05,
      "loss": 3.2379,
      "step": 5840
    },
    {
      "epoch": 2.295693122731286,
      "grad_norm": 2.46875,
      "learning_rate": 2.7529197543083508e-05,
      "loss": 2.6509,
      "step": 5850
    },
    {
      "epoch": 2.2996173844795447,
      "grad_norm": 6.21875,
      "learning_rate": 2.7237869873352827e-05,
      "loss": 3.298,
      "step": 5860
    },
    {
      "epoch": 2.3035416462278033,
      "grad_norm": 8.5,
      "learning_rate": 2.6947848673306853e-05,
      "loss": 2.9396,
      "step": 5870
    },
    {
      "epoch": 2.307465907976062,
      "grad_norm": 3.890625,
      "learning_rate": 2.6659139150378377e-05,
      "loss": 2.9588,
      "step": 5880
    },
    {
      "epoch": 2.3113901697243207,
      "grad_norm": 7.59375,
      "learning_rate": 2.6371746488448613e-05,
      "loss": 3.2112,
      "step": 5890
    },
    {
      "epoch": 2.3153144314725793,
      "grad_norm": 2.59375,
      "learning_rate": 2.6085675847754153e-05,
      "loss": 2.6169,
      "step": 5900
    },
    {
      "epoch": 2.319238693220838,
      "grad_norm": 4.4375,
      "learning_rate": 2.580093236479406e-05,
      "loss": 3.279,
      "step": 5910
    },
    {
      "epoch": 2.3231629549690966,
      "grad_norm": 8.5625,
      "learning_rate": 2.5517521152237965e-05,
      "loss": 2.8971,
      "step": 5920
    },
    {
      "epoch": 2.327087216717355,
      "grad_norm": 3.8125,
      "learning_rate": 2.5235447298834004e-05,
      "loss": 2.9995,
      "step": 5930
    },
    {
      "epoch": 2.3310114784656135,
      "grad_norm": 6.3125,
      "learning_rate": 2.4954715869317603e-05,
      "loss": 3.2365,
      "step": 5940
    },
    {
      "epoch": 2.334935740213872,
      "grad_norm": 2.546875,
      "learning_rate": 2.4675331904320532e-05,
      "loss": 2.6159,
      "step": 5950
    },
    {
      "epoch": 2.338860001962131,
      "grad_norm": 6.1875,
      "learning_rate": 2.43973004202803e-05,
      "loss": 3.3102,
      "step": 5960
    },
    {
      "epoch": 2.3427842637103895,
      "grad_norm": 9.5625,
      "learning_rate": 2.4120626409350212e-05,
      "loss": 2.8904,
      "step": 5970
    },
    {
      "epoch": 2.346708525458648,
      "grad_norm": 4.71875,
      "learning_rate": 2.3845314839309564e-05,
      "loss": 3.0166,
      "step": 5980
    },
    {
      "epoch": 2.350632787206907,
      "grad_norm": 7.21875,
      "learning_rate": 2.3571370653474657e-05,
      "loss": 3.2609,
      "step": 5990
    },
    {
      "epoch": 2.3545570489551655,
      "grad_norm": 2.765625,
      "learning_rate": 2.329879877060981e-05,
      "loss": 2.6724,
      "step": 6000
    },
    {
      "epoch": 2.358481310703424,
      "grad_norm": 5.875,
      "learning_rate": 2.302760408483926e-05,
      "loss": 3.3243,
      "step": 6010
    },
    {
      "epoch": 2.3624055724516824,
      "grad_norm": 11.4375,
      "learning_rate": 2.2757791465559153e-05,
      "loss": 2.8908,
      "step": 6020
    },
    {
      "epoch": 2.366329834199941,
      "grad_norm": 4.0,
      "learning_rate": 2.2489365757350135e-05,
      "loss": 2.9607,
      "step": 6030
    },
    {
      "epoch": 2.3702540959481997,
      "grad_norm": 6.96875,
      "learning_rate": 2.2222331779890394e-05,
      "loss": 3.2574,
      "step": 6040
    },
    {
      "epoch": 2.3741783576964584,
      "grad_norm": 2.328125,
      "learning_rate": 2.1956694327869042e-05,
      "loss": 2.6913,
      "step": 6050
    },
    {
      "epoch": 2.378102619444717,
      "grad_norm": 5.78125,
      "learning_rate": 2.1692458170900198e-05,
      "loss": 3.2603,
      "step": 6060
    },
    {
      "epoch": 2.3820268811929757,
      "grad_norm": 10.1875,
      "learning_rate": 2.1429628053437078e-05,
      "loss": 2.8675,
      "step": 6070
    },
    {
      "epoch": 2.3859511429412343,
      "grad_norm": 4.15625,
      "learning_rate": 2.116820869468711e-05,
      "loss": 3.0073,
      "step": 6080
    },
    {
      "epoch": 2.3898754046894926,
      "grad_norm": 7.75,
      "learning_rate": 2.0908204788526964e-05,
      "loss": 3.3149,
      "step": 6090
    },
    {
      "epoch": 2.393799666437751,
      "grad_norm": 2.625,
      "learning_rate": 2.064962100341842e-05,
      "loss": 2.6488,
      "step": 6100
    },
    {
      "epoch": 2.39772392818601,
      "grad_norm": 5.875,
      "learning_rate": 2.0392461982324463e-05,
      "loss": 3.2548,
      "step": 6110
    },
    {
      "epoch": 2.4016481899342685,
      "grad_norm": 8.125,
      "learning_rate": 2.0136732342625875e-05,
      "loss": 2.8826,
      "step": 6120
    },
    {
      "epoch": 2.405572451682527,
      "grad_norm": 4.09375,
      "learning_rate": 1.9882436676038472e-05,
      "loss": 2.9756,
      "step": 6130
    },
    {
      "epoch": 2.409496713430786,
      "grad_norm": 6.46875,
      "learning_rate": 1.9629579548530553e-05,
      "loss": 3.2445,
      "step": 6140
    },
    {
      "epoch": 2.4134209751790445,
      "grad_norm": 2.640625,
      "learning_rate": 1.9378165500240943e-05,
      "loss": 2.7284,
      "step": 6150
    },
    {
      "epoch": 2.417345236927303,
      "grad_norm": 5.78125,
      "learning_rate": 1.9128199045397488e-05,
      "loss": 3.2975,
      "step": 6160
    },
    {
      "epoch": 2.421269498675562,
      "grad_norm": 8.25,
      "learning_rate": 1.887968467223591e-05,
      "loss": 2.9231,
      "step": 6170
    },
    {
      "epoch": 2.42519376042382,
      "grad_norm": 4.21875,
      "learning_rate": 1.86326268429194e-05,
      "loss": 2.9953,
      "step": 6180
    },
    {
      "epoch": 2.4291180221720787,
      "grad_norm": 7.84375,
      "learning_rate": 1.838702999345827e-05,
      "loss": 3.2725,
      "step": 6190
    },
    {
      "epoch": 2.4330422839203374,
      "grad_norm": 3.234375,
      "learning_rate": 1.8142898533630536e-05,
      "loss": 2.6703,
      "step": 6200
    },
    {
      "epoch": 2.436966545668596,
      "grad_norm": 5.84375,
      "learning_rate": 1.7900236846902572e-05,
      "loss": 3.2818,
      "step": 6210
    },
    {
      "epoch": 2.4408908074168547,
      "grad_norm": 8.875,
      "learning_rate": 1.765904929035046e-05,
      "loss": 2.9124,
      "step": 6220
    },
    {
      "epoch": 2.4448150691651134,
      "grad_norm": 3.9375,
      "learning_rate": 1.74193401945818e-05,
      "loss": 2.9999,
      "step": 6230
    },
    {
      "epoch": 2.448739330913372,
      "grad_norm": 6.28125,
      "learning_rate": 1.7181113863657805e-05,
      "loss": 3.2456,
      "step": 6240
    },
    {
      "epoch": 2.4526635926616307,
      "grad_norm": 2.421875,
      "learning_rate": 1.6944374575016253e-05,
      "loss": 2.7104,
      "step": 6250
    },
    {
      "epoch": 2.4565878544098894,
      "grad_norm": 5.0,
      "learning_rate": 1.670912657939443e-05,
      "loss": 3.2779,
      "step": 6260
    },
    {
      "epoch": 2.4605121161581476,
      "grad_norm": 8.875,
      "learning_rate": 1.6475374100753018e-05,
      "loss": 2.8773,
      "step": 6270
    },
    {
      "epoch": 2.4644363779064062,
      "grad_norm": 4.21875,
      "learning_rate": 1.6243121336200127e-05,
      "loss": 3.0077,
      "step": 6280
    },
    {
      "epoch": 2.468360639654665,
      "grad_norm": 6.4375,
      "learning_rate": 1.601237245591599e-05,
      "loss": 3.2565,
      "step": 6290
    },
    {
      "epoch": 2.4722849014029236,
      "grad_norm": 2.53125,
      "learning_rate": 1.5783131603078083e-05,
      "loss": 2.7047,
      "step": 6300
    },
    {
      "epoch": 2.4762091631511822,
      "grad_norm": 5.09375,
      "learning_rate": 1.555540289378663e-05,
      "loss": 3.2782,
      "step": 6310
    },
    {
      "epoch": 2.480133424899441,
      "grad_norm": 15.5,
      "learning_rate": 1.532919041699089e-05,
      "loss": 2.9229,
      "step": 6320
    },
    {
      "epoch": 2.4840576866476995,
      "grad_norm": 4.34375,
      "learning_rate": 1.510449823441561e-05,
      "loss": 3.0135,
      "step": 6330
    },
    {
      "epoch": 2.4879819483959578,
      "grad_norm": 10.3125,
      "learning_rate": 1.4881330380488013e-05,
      "loss": 3.2243,
      "step": 6340
    },
    {
      "epoch": 2.4919062101442164,
      "grad_norm": 2.78125,
      "learning_rate": 1.4659690862265673e-05,
      "loss": 2.6542,
      "step": 6350
    },
    {
      "epoch": 2.495830471892475,
      "grad_norm": 4.9375,
      "learning_rate": 1.4439583659364153e-05,
      "loss": 3.2584,
      "step": 6360
    },
    {
      "epoch": 2.4997547336407338,
      "grad_norm": 7.90625,
      "learning_rate": 1.4221012723885874e-05,
      "loss": 2.8858,
      "step": 6370
    },
    {
      "epoch": 2.5036789953889924,
      "grad_norm": 4.15625,
      "learning_rate": 1.4003981980348969e-05,
      "loss": 3.0093,
      "step": 6380
    },
    {
      "epoch": 2.507603257137251,
      "grad_norm": 6.5,
      "learning_rate": 1.3788495325616912e-05,
      "loss": 3.2543,
      "step": 6390
    },
    {
      "epoch": 2.5115275188855097,
      "grad_norm": 2.625,
      "learning_rate": 1.357455662882855e-05,
      "loss": 2.6444,
      "step": 6400
    },
    {
      "epoch": 2.5154517806337684,
      "grad_norm": 13.3125,
      "learning_rate": 1.3362169731328534e-05,
      "loss": 3.2754,
      "step": 6410
    },
    {
      "epoch": 2.519376042382027,
      "grad_norm": 11.625,
      "learning_rate": 1.3151338446598482e-05,
      "loss": 2.9064,
      "step": 6420
    },
    {
      "epoch": 2.5233003041302853,
      "grad_norm": 3.546875,
      "learning_rate": 1.2942066560188348e-05,
      "loss": 2.9891,
      "step": 6430
    },
    {
      "epoch": 2.527224565878544,
      "grad_norm": 9.125,
      "learning_rate": 1.2734357829648625e-05,
      "loss": 3.2099,
      "step": 6440
    },
    {
      "epoch": 2.5311488276268026,
      "grad_norm": 2.78125,
      "learning_rate": 1.2528215984462765e-05,
      "loss": 2.6214,
      "step": 6450
    },
    {
      "epoch": 2.5350730893750613,
      "grad_norm": 5.03125,
      "learning_rate": 1.23236447259802e-05,
      "loss": 3.3063,
      "step": 6460
    },
    {
      "epoch": 2.53899735112332,
      "grad_norm": 9.4375,
      "learning_rate": 1.2120647727349976e-05,
      "loss": 2.8502,
      "step": 6470
    },
    {
      "epoch": 2.5429216128715786,
      "grad_norm": 3.640625,
      "learning_rate": 1.1919228633454737e-05,
      "loss": 3.0197,
      "step": 6480
    },
    {
      "epoch": 2.5468458746198372,
      "grad_norm": 6.4375,
      "learning_rate": 1.1719391060845298e-05,
      "loss": 3.2618,
      "step": 6490
    },
    {
      "epoch": 2.550770136368096,
      "grad_norm": 2.84375,
      "learning_rate": 1.1521138597675651e-05,
      "loss": 2.6511,
      "step": 6500
    },
    {
      "epoch": 2.5546943981163546,
      "grad_norm": 5.03125,
      "learning_rate": 1.1324474803638652e-05,
      "loss": 3.3513,
      "step": 6510
    },
    {
      "epoch": 2.558618659864613,
      "grad_norm": 9.0,
      "learning_rate": 1.1129403209902034e-05,
      "loss": 2.9096,
      "step": 6520
    },
    {
      "epoch": 2.5625429216128714,
      "grad_norm": 4.15625,
      "learning_rate": 1.0935927319044958e-05,
      "loss": 2.9592,
      "step": 6530
    },
    {
      "epoch": 2.56646718336113,
      "grad_norm": 6.875,
      "learning_rate": 1.0744050604995237e-05,
      "loss": 3.2411,
      "step": 6540
    },
    {
      "epoch": 2.5703914451093888,
      "grad_norm": 2.578125,
      "learning_rate": 1.0553776512966885e-05,
      "loss": 2.6071,
      "step": 6550
    },
    {
      "epoch": 2.5743157068576474,
      "grad_norm": 5.40625,
      "learning_rate": 1.0365108459398277e-05,
      "loss": 3.2846,
      "step": 6560
    },
    {
      "epoch": 2.578239968605906,
      "grad_norm": 11.0625,
      "learning_rate": 1.0178049831890768e-05,
      "loss": 2.9241,
      "step": 6570
    },
    {
      "epoch": 2.5821642303541648,
      "grad_norm": 3.90625,
      "learning_rate": 9.992603989147942e-06,
      "loss": 3.0117,
      "step": 6580
    },
    {
      "epoch": 2.586088492102423,
      "grad_norm": 7.59375,
      "learning_rate": 9.808774260915244e-06,
      "loss": 3.2983,
      "step": 6590
    },
    {
      "epoch": 2.590012753850682,
      "grad_norm": 2.53125,
      "learning_rate": 9.62656394792023e-06,
      "loss": 2.6326,
      "step": 6600
    },
    {
      "epoch": 2.5939370155989403,
      "grad_norm": 5.5625,
      "learning_rate": 9.445976321813277e-06,
      "loss": 3.221,
      "step": 6610
    },
    {
      "epoch": 2.597861277347199,
      "grad_norm": 9.9375,
      "learning_rate": 9.267014625108805e-06,
      "loss": 2.8799,
      "step": 6620
    },
    {
      "epoch": 2.6017855390954576,
      "grad_norm": 4.21875,
      "learning_rate": 9.08968207112717e-06,
      "loss": 3.0436,
      "step": 6630
    },
    {
      "epoch": 2.6057098008437163,
      "grad_norm": 6.6875,
      "learning_rate": 8.91398184393687e-06,
      "loss": 3.2728,
      "step": 6640
    },
    {
      "epoch": 2.609634062591975,
      "grad_norm": 2.921875,
      "learning_rate": 8.739917098297356e-06,
      "loss": 2.6928,
      "step": 6650
    },
    {
      "epoch": 2.6135583243402336,
      "grad_norm": 6.40625,
      "learning_rate": 8.56749095960251e-06,
      "loss": 3.2771,
      "step": 6660
    },
    {
      "epoch": 2.6174825860884923,
      "grad_norm": 9.5625,
      "learning_rate": 8.396706523824372e-06,
      "loss": 2.9105,
      "step": 6670
    },
    {
      "epoch": 2.6214068478367505,
      "grad_norm": 3.90625,
      "learning_rate": 8.227566857457703e-06,
      "loss": 2.9854,
      "step": 6680
    },
    {
      "epoch": 2.6253311095850096,
      "grad_norm": 6.65625,
      "learning_rate": 8.060074997464772e-06,
      "loss": 3.2696,
      "step": 6690
    },
    {
      "epoch": 2.629255371333268,
      "grad_norm": 2.6875,
      "learning_rate": 7.894233951220953e-06,
      "loss": 2.6429,
      "step": 6700
    },
    {
      "epoch": 2.6331796330815265,
      "grad_norm": 6.65625,
      "learning_rate": 7.73004669646069e-06,
      "loss": 3.2937,
      "step": 6710
    },
    {
      "epoch": 2.637103894829785,
      "grad_norm": 8.5625,
      "learning_rate": 7.567516181223966e-06,
      "loss": 2.9059,
      "step": 6720
    },
    {
      "epoch": 2.641028156578044,
      "grad_norm": 3.9375,
      "learning_rate": 7.406645323803463e-06,
      "loss": 3.0014,
      "step": 6730
    },
    {
      "epoch": 2.6449524183263025,
      "grad_norm": 6.3125,
      "learning_rate": 7.247437012692104e-06,
      "loss": 3.2418,
      "step": 6740
    },
    {
      "epoch": 2.648876680074561,
      "grad_norm": 2.515625,
      "learning_rate": 7.089894106531214e-06,
      "loss": 2.6491,
      "step": 6750
    },
    {
      "epoch": 2.65280094182282,
      "grad_norm": 5.125,
      "learning_rate": 6.9340194340592135e-06,
      "loss": 3.303,
      "step": 6760
    },
    {
      "epoch": 2.656725203571078,
      "grad_norm": 10.8125,
      "learning_rate": 6.779815794060718e-06,
      "loss": 2.7982,
      "step": 6770
    },
    {
      "epoch": 2.6606494653193367,
      "grad_norm": 3.984375,
      "learning_rate": 6.627285955316475e-06,
      "loss": 2.99,
      "step": 6780
    },
    {
      "epoch": 2.6645737270675953,
      "grad_norm": 7.78125,
      "learning_rate": 6.476432656553411e-06,
      "loss": 3.2882,
      "step": 6790
    },
    {
      "epoch": 2.668497988815854,
      "grad_norm": 2.671875,
      "learning_rate": 6.327258606395736e-06,
      "loss": 2.6036,
      "step": 6800
    },
    {
      "epoch": 2.6724222505641126,
      "grad_norm": 5.53125,
      "learning_rate": 6.179766483316041e-06,
      "loss": 3.2759,
      "step": 6810
    },
    {
      "epoch": 2.6763465123123713,
      "grad_norm": 7.90625,
      "learning_rate": 6.033958935587369e-06,
      "loss": 2.8762,
      "step": 6820
    },
    {
      "epoch": 2.68027077406063,
      "grad_norm": 3.984375,
      "learning_rate": 5.88983858123564e-06,
      "loss": 3.0194,
      "step": 6830
    },
    {
      "epoch": 2.684195035808888,
      "grad_norm": 8.375,
      "learning_rate": 5.747408007992572e-06,
      "loss": 3.2526,
      "step": 6840
    },
    {
      "epoch": 2.6881192975571473,
      "grad_norm": 3.578125,
      "learning_rate": 5.606669773249296e-06,
      "loss": 2.6542,
      "step": 6850
    },
    {
      "epoch": 2.6920435593054055,
      "grad_norm": 5.0,
      "learning_rate": 5.467626404010406e-06,
      "loss": 3.2857,
      "step": 6860
    },
    {
      "epoch": 2.695967821053664,
      "grad_norm": 9.625,
      "learning_rate": 5.3302803968486195e-06,
      "loss": 2.8893,
      "step": 6870
    },
    {
      "epoch": 2.699892082801923,
      "grad_norm": 4.0,
      "learning_rate": 5.19463421785985e-06,
      "loss": 2.9994,
      "step": 6880
    },
    {
      "epoch": 2.7038163445501815,
      "grad_norm": 6.3125,
      "learning_rate": 5.0606903026190535e-06,
      "loss": 3.2525,
      "step": 6890
    },
    {
      "epoch": 2.70774060629844,
      "grad_norm": 2.5625,
      "learning_rate": 4.928451056136441e-06,
      "loss": 2.6296,
      "step": 6900
    },
    {
      "epoch": 2.711664868046699,
      "grad_norm": 5.3125,
      "learning_rate": 4.797918852814254e-06,
      "loss": 3.2911,
      "step": 6910
    },
    {
      "epoch": 2.7155891297949575,
      "grad_norm": 9.8125,
      "learning_rate": 4.669096036404197e-06,
      "loss": 2.8506,
      "step": 6920
    },
    {
      "epoch": 2.7195133915432157,
      "grad_norm": 4.25,
      "learning_rate": 4.541984919965336e-06,
      "loss": 2.9872,
      "step": 6930
    },
    {
      "epoch": 2.723437653291475,
      "grad_norm": 7.53125,
      "learning_rate": 4.416587785822568e-06,
      "loss": 3.2562,
      "step": 6940
    },
    {
      "epoch": 2.727361915039733,
      "grad_norm": 2.9375,
      "learning_rate": 4.2929068855256275e-06,
      "loss": 2.6584,
      "step": 6950
    },
    {
      "epoch": 2.7312861767879917,
      "grad_norm": 5.375,
      "learning_rate": 4.170944439808622e-06,
      "loss": 3.3017,
      "step": 6960
    },
    {
      "epoch": 2.7352104385362503,
      "grad_norm": 10.0625,
      "learning_rate": 4.050702638550275e-06,
      "loss": 2.9594,
      "step": 6970
    },
    {
      "epoch": 2.739134700284509,
      "grad_norm": 3.828125,
      "learning_rate": 3.932183640734466e-06,
      "loss": 2.9776,
      "step": 6980
    },
    {
      "epoch": 2.7430589620327677,
      "grad_norm": 7.5,
      "learning_rate": 3.815389574411576e-06,
      "loss": 3.2975,
      "step": 6990
    },
    {
      "epoch": 2.7469832237810263,
      "grad_norm": 2.546875,
      "learning_rate": 3.700322536660228e-06,
      "loss": 2.6413,
      "step": 7000
    },
    {
      "epoch": 2.750907485529285,
      "grad_norm": 6.75,
      "learning_rate": 3.586984593549614e-06,
      "loss": 3.2786,
      "step": 7010
    },
    {
      "epoch": 2.754831747277543,
      "grad_norm": 9.875,
      "learning_rate": 3.4753777801024513e-06,
      "loss": 2.8895,
      "step": 7020
    },
    {
      "epoch": 2.758756009025802,
      "grad_norm": 5.9375,
      "learning_rate": 3.3655041002583987e-06,
      "loss": 2.9887,
      "step": 7030
    },
    {
      "epoch": 2.7626802707740605,
      "grad_norm": 7.4375,
      "learning_rate": 3.2573655268380743e-06,
      "loss": 3.2071,
      "step": 7040
    },
    {
      "epoch": 2.766604532522319,
      "grad_norm": 3.09375,
      "learning_rate": 3.1509640015076948e-06,
      "loss": 2.6623,
      "step": 7050
    },
    {
      "epoch": 2.770528794270578,
      "grad_norm": 7.59375,
      "learning_rate": 3.0463014347441255e-06,
      "loss": 3.296,
      "step": 7060
    },
    {
      "epoch": 2.7744530560188365,
      "grad_norm": 8.9375,
      "learning_rate": 2.94337970580062e-06,
      "loss": 2.8866,
      "step": 7070
    },
    {
      "epoch": 2.778377317767095,
      "grad_norm": 5.25,
      "learning_rate": 2.8422006626731114e-06,
      "loss": 3.0239,
      "step": 7080
    },
    {
      "epoch": 2.782301579515354,
      "grad_norm": 7.15625,
      "learning_rate": 2.7427661220669532e-06,
      "loss": 3.2212,
      "step": 7090
    },
    {
      "epoch": 2.7862258412636125,
      "grad_norm": 3.03125,
      "learning_rate": 2.645077869364354e-06,
      "loss": 2.624,
      "step": 7100
    },
    {
      "epoch": 2.7901501030118707,
      "grad_norm": 5.3125,
      "learning_rate": 2.5491376585923266e-06,
      "loss": 3.2916,
      "step": 7110
    },
    {
      "epoch": 2.7940743647601294,
      "grad_norm": 23.25,
      "learning_rate": 2.4549472123911566e-06,
      "loss": 2.8916,
      "step": 7120
    },
    {
      "epoch": 2.797998626508388,
      "grad_norm": 4.125,
      "learning_rate": 2.362508221983484e-06,
      "loss": 2.9774,
      "step": 7130
    },
    {
      "epoch": 2.8019228882566467,
      "grad_norm": 8.3125,
      "learning_rate": 2.2718223471439815e-06,
      "loss": 3.2345,
      "step": 7140
    },
    {
      "epoch": 2.8058471500049054,
      "grad_norm": 2.875,
      "learning_rate": 2.182891216169447e-06,
      "loss": 2.6511,
      "step": 7150
    },
    {
      "epoch": 2.809771411753164,
      "grad_norm": 6.6875,
      "learning_rate": 2.095716425849703e-06,
      "loss": 3.3008,
      "step": 7160
    },
    {
      "epoch": 2.8136956735014227,
      "grad_norm": 8.5625,
      "learning_rate": 2.010299541438798e-06,
      "loss": 2.8646,
      "step": 7170
    },
    {
      "epoch": 2.817619935249681,
      "grad_norm": 3.828125,
      "learning_rate": 1.9266420966270184e-06,
      "loss": 2.9769,
      "step": 7180
    },
    {
      "epoch": 2.82154419699794,
      "grad_norm": 13.9375,
      "learning_rate": 1.8447455935132418e-06,
      "loss": 3.2378,
      "step": 7190
    },
    {
      "epoch": 2.8254684587461982,
      "grad_norm": 2.625,
      "learning_rate": 1.7646115025780507e-06,
      "loss": 2.6222,
      "step": 7200
    },
    {
      "epoch": 2.829392720494457,
      "grad_norm": 5.40625,
      "learning_rate": 1.6862412626572842e-06,
      "loss": 3.2937,
      "step": 7210
    },
    {
      "epoch": 2.8333169822427156,
      "grad_norm": 9.6875,
      "learning_rate": 1.6096362809162047e-06,
      "loss": 2.9473,
      "step": 7220
    },
    {
      "epoch": 2.837241243990974,
      "grad_norm": 4.0625,
      "learning_rate": 1.5347979328242611e-06,
      "loss": 2.9966,
      "step": 7230
    },
    {
      "epoch": 2.841165505739233,
      "grad_norm": 5.75,
      "learning_rate": 1.461727562130344e-06,
      "loss": 3.246,
      "step": 7240
    },
    {
      "epoch": 2.8450897674874915,
      "grad_norm": 2.78125,
      "learning_rate": 1.3904264808387246e-06,
      "loss": 2.6215,
      "step": 7250
    },
    {
      "epoch": 2.84901402923575,
      "grad_norm": 5.09375,
      "learning_rate": 1.3208959691854539e-06,
      "loss": 3.263,
      "step": 7260
    },
    {
      "epoch": 2.8529382909840084,
      "grad_norm": 8.3125,
      "learning_rate": 1.2531372756153458e-06,
      "loss": 2.8548,
      "step": 7270
    },
    {
      "epoch": 2.856862552732267,
      "grad_norm": 4.03125,
      "learning_rate": 1.1871516167596186e-06,
      "loss": 3.0046,
      "step": 7280
    },
    {
      "epoch": 2.8607868144805257,
      "grad_norm": 12.0,
      "learning_rate": 1.1229401774140448e-06,
      "loss": 3.2649,
      "step": 7290
    },
    {
      "epoch": 2.8647110762287844,
      "grad_norm": 2.5625,
      "learning_rate": 1.0605041105176129e-06,
      "loss": 2.6403,
      "step": 7300
    },
    {
      "epoch": 2.868635337977043,
      "grad_norm": 5.4375,
      "learning_rate": 9.998445371319332e-07,
      "loss": 3.2856,
      "step": 7310
    },
    {
      "epoch": 2.8725595997253017,
      "grad_norm": 8.25,
      "learning_rate": 9.409625464210092e-07,
      "loss": 2.8989,
      "step": 7320
    },
    {
      "epoch": 2.8764838614735604,
      "grad_norm": 4.09375,
      "learning_rate": 8.838591956317311e-07,
      "loss": 2.9958,
      "step": 7330
    },
    {
      "epoch": 2.880408123221819,
      "grad_norm": 7.4375,
      "learning_rate": 8.285355100748904e-07,
      "loss": 3.2497,
      "step": 7340
    },
    {
      "epoch": 2.8843323849700777,
      "grad_norm": 3.078125,
      "learning_rate": 7.749924831067401e-07,
      "loss": 2.6747,
      "step": 7350
    },
    {
      "epoch": 2.888256646718336,
      "grad_norm": 5.375,
      "learning_rate": 7.232310761112082e-07,
      "loss": 3.2442,
      "step": 7360
    },
    {
      "epoch": 2.8921809084665946,
      "grad_norm": 8.6875,
      "learning_rate": 6.732522184825896e-07,
      "loss": 2.8594,
      "step": 7370
    },
    {
      "epoch": 2.8961051702148533,
      "grad_norm": 4.96875,
      "learning_rate": 6.250568076088815e-07,
      "loss": 2.9698,
      "step": 7380
    },
    {
      "epoch": 2.900029431963112,
      "grad_norm": 9.0625,
      "learning_rate": 5.78645708855674e-07,
      "loss": 3.2714,
      "step": 7390
    },
    {
      "epoch": 2.9039536937113706,
      "grad_norm": 2.484375,
      "learning_rate": 5.340197555505966e-07,
      "loss": 2.6358,
      "step": 7400
    },
    {
      "epoch": 2.9078779554596292,
      "grad_norm": 6.46875,
      "learning_rate": 4.911797489683733e-07,
      "loss": 3.2933,
      "step": 7410
    },
    {
      "epoch": 2.911802217207888,
      "grad_norm": 9.0625,
      "learning_rate": 4.501264583164022e-07,
      "loss": 2.8794,
      "step": 7420
    },
    {
      "epoch": 2.915726478956146,
      "grad_norm": 4.25,
      "learning_rate": 4.108606207209875e-07,
      "loss": 2.9998,
      "step": 7430
    },
    {
      "epoch": 2.9196507407044052,
      "grad_norm": 12.0625,
      "learning_rate": 3.7338294121407323e-07,
      "loss": 3.2604,
      "step": 7440
    },
    {
      "epoch": 2.9235750024526634,
      "grad_norm": 2.703125,
      "learning_rate": 3.376940927206196e-07,
      "loss": 2.6116,
      "step": 7450
    },
    {
      "epoch": 2.927499264200922,
      "grad_norm": 6.15625,
      "learning_rate": 3.0379471604645717e-07,
      "loss": 3.2803,
      "step": 7460
    },
    {
      "epoch": 2.9314235259491808,
      "grad_norm": 9.3125,
      "learning_rate": 2.716854198668517e-07,
      "loss": 2.8721,
      "step": 7470
    },
    {
      "epoch": 2.9353477876974394,
      "grad_norm": 4.40625,
      "learning_rate": 2.41366780715524e-07,
      "loss": 2.9938,
      "step": 7480
    },
    {
      "epoch": 2.939272049445698,
      "grad_norm": 9.5,
      "learning_rate": 2.1283934297432473e-07,
      "loss": 3.2135,
      "step": 7490
    },
    {
      "epoch": 2.9431963111939567,
      "grad_norm": 2.71875,
      "learning_rate": 1.861036188634424e-07,
      "loss": 2.6379,
      "step": 7500
    },
    {
      "epoch": 2.9471205729422154,
      "grad_norm": 6.40625,
      "learning_rate": 1.6116008843224395e-07,
      "loss": 3.2937,
      "step": 7510
    },
    {
      "epoch": 2.9510448346904736,
      "grad_norm": 9.1875,
      "learning_rate": 1.3800919955058167e-07,
      "loss": 2.8557,
      "step": 7520
    },
    {
      "epoch": 2.9549690964387327,
      "grad_norm": 4.375,
      "learning_rate": 1.1665136790084407e-07,
      "loss": 3.0229,
      "step": 7530
    },
    {
      "epoch": 2.958893358186991,
      "grad_norm": 7.8125,
      "learning_rate": 9.708697697040636e-08,
      "loss": 3.2634,
      "step": 7540
    },
    {
      "epoch": 2.9628176199352496,
      "grad_norm": 2.640625,
      "learning_rate": 7.931637804481362e-08,
      "loss": 2.635,
      "step": 7550
    },
    {
      "epoch": 2.9667418816835083,
      "grad_norm": 5.4375,
      "learning_rate": 6.333989020143039e-08,
      "loss": 3.2881,
      "step": 7560
    },
    {
      "epoch": 2.970666143431767,
      "grad_norm": 8.3125,
      "learning_rate": 4.9157800303722965e-08,
      "loss": 2.9015,
      "step": 7570
    },
    {
      "epoch": 2.9745904051800256,
      "grad_norm": 3.890625,
      "learning_rate": 3.6770362996108033e-08,
      "loss": 2.9926,
      "step": 7580
    },
    {
      "epoch": 2.9785146669282843,
      "grad_norm": 6.6875,
      "learning_rate": 2.617780069940068e-08,
      "loss": 3.264,
      "step": 7590
    },
    {
      "epoch": 2.982438928676543,
      "grad_norm": 2.65625,
      "learning_rate": 1.738030360677323e-08,
      "loss": 2.6162,
      "step": 7600
    },
    {
      "epoch": 2.986363190424801,
      "grad_norm": 5.5625,
      "learning_rate": 1.0378029680391255e-08,
      "loss": 3.2659,
      "step": 7610
    },
    {
      "epoch": 2.99028745217306,
      "grad_norm": 15.0625,
      "learning_rate": 5.171104648549196e-09,
      "loss": 2.9792,
      "step": 7620
    },
    {
      "epoch": 2.9942117139213185,
      "grad_norm": 4.625,
      "learning_rate": 1.7596220034277188e-09,
      "loss": 3.045,
      "step": 7630
    },
    {
      "epoch": 2.998135975669577,
      "grad_norm": 12.6875,
      "learning_rate": 1.436429993950661e-10,
      "loss": 3.2092,
      "step": 7640
    },
    {
      "epoch": 2.999705680368881,
      "eval_loss": 3.0855159759521484,
      "eval_runtime": 90.1161,
      "eval_samples_per_second": 277.42,
      "eval_steps_per_second": 69.355,
      "step": 7644
    },
    {
      "epoch": 2.999705680368881,
      "step": 7644,
      "total_flos": 2.04275798866133e+17,
      "train_loss": 3.3574186354516877,
      "train_runtime": 4757.4793,
      "train_samples_per_second": 205.665,
      "train_steps_per_second": 1.607
    }
  ],
  "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": 2.04275798866133e+17,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}