{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.997593524392912,
  "eval_steps": 500,
  "global_step": 3426,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "eval_loss": 2.1641297340393066,
      "eval_runtime": 4.2967,
      "eval_samples_per_second": 4.655,
      "eval_steps_per_second": 0.465,
      "step": 0
    },
    {
      "epoch": 0.0008750820389411508,
      "grad_norm": 6.37307071685791,
      "learning_rate": 2.915451895043732e-07,
      "loss": 1.6434,
      "step": 1
    },
    {
      "epoch": 0.008750820389411508,
      "grad_norm": 4.0545759201049805,
      "learning_rate": 2.915451895043732e-06,
      "loss": 1.6965,
      "step": 10
    },
    {
      "epoch": 0.017501640778823015,
      "grad_norm": 2.363191604614258,
      "learning_rate": 5.830903790087464e-06,
      "loss": 1.5272,
      "step": 20
    },
    {
      "epoch": 0.02625246116823452,
      "grad_norm": 1.6348848342895508,
      "learning_rate": 8.746355685131195e-06,
      "loss": 1.389,
      "step": 30
    },
    {
      "epoch": 0.03500328155764603,
      "grad_norm": 1.6716902256011963,
      "learning_rate": 1.1661807580174927e-05,
      "loss": 1.348,
      "step": 40
    },
    {
      "epoch": 0.04375410194705753,
      "grad_norm": 1.510228157043457,
      "learning_rate": 1.457725947521866e-05,
      "loss": 1.2997,
      "step": 50
    },
    {
      "epoch": 0.05250492233646904,
      "grad_norm": 1.6153247356414795,
      "learning_rate": 1.749271137026239e-05,
      "loss": 1.2665,
      "step": 60
    },
    {
      "epoch": 0.06125574272588055,
      "grad_norm": 1.446387767791748,
      "learning_rate": 2.0408163265306123e-05,
      "loss": 1.237,
      "step": 70
    },
    {
      "epoch": 0.07000656311529206,
      "grad_norm": 1.3039863109588623,
      "learning_rate": 2.3323615160349855e-05,
      "loss": 1.2341,
      "step": 80
    },
    {
      "epoch": 0.07875738350470357,
      "grad_norm": 1.4866383075714111,
      "learning_rate": 2.6239067055393584e-05,
      "loss": 1.2338,
      "step": 90
    },
    {
      "epoch": 0.08750820389411507,
      "grad_norm": 1.358859658241272,
      "learning_rate": 2.915451895043732e-05,
      "loss": 1.2331,
      "step": 100
    },
    {
      "epoch": 0.09625902428352658,
      "grad_norm": 1.847506046295166,
      "learning_rate": 3.206997084548105e-05,
      "loss": 1.1628,
      "step": 110
    },
    {
      "epoch": 0.10500984467293809,
      "grad_norm": 1.7742356061935425,
      "learning_rate": 3.498542274052478e-05,
      "loss": 1.1846,
      "step": 120
    },
    {
      "epoch": 0.1137606650623496,
      "grad_norm": 1.3415714502334595,
      "learning_rate": 3.790087463556852e-05,
      "loss": 1.1699,
      "step": 130
    },
    {
      "epoch": 0.1225114854517611,
      "grad_norm": 1.5319396257400513,
      "learning_rate": 4.0816326530612245e-05,
      "loss": 1.181,
      "step": 140
    },
    {
      "epoch": 0.1312623058411726,
      "grad_norm": 1.3915642499923706,
      "learning_rate": 4.373177842565598e-05,
      "loss": 1.189,
      "step": 150
    },
    {
      "epoch": 0.14001312623058412,
      "grad_norm": 1.175917625427246,
      "learning_rate": 4.664723032069971e-05,
      "loss": 1.1605,
      "step": 160
    },
    {
      "epoch": 0.14876394661999562,
      "grad_norm": 1.3125519752502441,
      "learning_rate": 4.956268221574344e-05,
      "loss": 1.1734,
      "step": 170
    },
    {
      "epoch": 0.15751476700940714,
      "grad_norm": 1.2966923713684082,
      "learning_rate": 5.247813411078717e-05,
      "loss": 1.2075,
      "step": 180
    },
    {
      "epoch": 0.16626558739881864,
      "grad_norm": 1.2630099058151245,
      "learning_rate": 5.539358600583091e-05,
      "loss": 1.1791,
      "step": 190
    },
    {
      "epoch": 0.17501640778823013,
      "grad_norm": 1.306665301322937,
      "learning_rate": 5.830903790087464e-05,
      "loss": 1.1469,
      "step": 200
    },
    {
      "epoch": 0.18376722817764166,
      "grad_norm": 1.19467031955719,
      "learning_rate": 6.122448979591838e-05,
      "loss": 1.185,
      "step": 210
    },
    {
      "epoch": 0.19251804856705315,
      "grad_norm": 1.416387677192688,
      "learning_rate": 6.41399416909621e-05,
      "loss": 1.2061,
      "step": 220
    },
    {
      "epoch": 0.20126886895646467,
      "grad_norm": 1.3546264171600342,
      "learning_rate": 6.705539358600583e-05,
      "loss": 1.2016,
      "step": 230
    },
    {
      "epoch": 0.21001968934587617,
      "grad_norm": 1.2417850494384766,
      "learning_rate": 6.997084548104956e-05,
      "loss": 1.1797,
      "step": 240
    },
    {
      "epoch": 0.2187705097352877,
      "grad_norm": 1.2703570127487183,
      "learning_rate": 7.28862973760933e-05,
      "loss": 1.2059,
      "step": 250
    },
    {
      "epoch": 0.2275213301246992,
      "grad_norm": 1.2479429244995117,
      "learning_rate": 7.580174927113704e-05,
      "loss": 1.1958,
      "step": 260
    },
    {
      "epoch": 0.23627215051411068,
      "grad_norm": 1.507601261138916,
      "learning_rate": 7.871720116618077e-05,
      "loss": 1.2151,
      "step": 270
    },
    {
      "epoch": 0.2450229709035222,
      "grad_norm": 1.2083854675292969,
      "learning_rate": 8.163265306122449e-05,
      "loss": 1.2202,
      "step": 280
    },
    {
      "epoch": 0.2537737912929337,
      "grad_norm": 1.2939566373825073,
      "learning_rate": 8.454810495626822e-05,
      "loss": 1.1916,
      "step": 290
    },
    {
      "epoch": 0.2625246116823452,
      "grad_norm": 1.2871382236480713,
      "learning_rate": 8.746355685131195e-05,
      "loss": 1.222,
      "step": 300
    },
    {
      "epoch": 0.27127543207175675,
      "grad_norm": 1.1558256149291992,
      "learning_rate": 9.037900874635569e-05,
      "loss": 1.2094,
      "step": 310
    },
    {
      "epoch": 0.28002625246116825,
      "grad_norm": 1.0638883113861084,
      "learning_rate": 9.329446064139942e-05,
      "loss": 1.2,
      "step": 320
    },
    {
      "epoch": 0.28877707285057974,
      "grad_norm": 1.1072458028793335,
      "learning_rate": 9.620991253644317e-05,
      "loss": 1.224,
      "step": 330
    },
    {
      "epoch": 0.29752789323999124,
      "grad_norm": 1.1332110166549683,
      "learning_rate": 9.912536443148688e-05,
      "loss": 1.2266,
      "step": 340
    },
    {
      "epoch": 0.30627871362940273,
      "grad_norm": 1.0537033081054688,
      "learning_rate": 9.999872800049233e-05,
      "loss": 1.2173,
      "step": 350
    },
    {
      "epoch": 0.3150295340188143,
      "grad_norm": 1.090928316116333,
      "learning_rate": 9.999249795462235e-05,
      "loss": 1.2416,
      "step": 360
    },
    {
      "epoch": 0.3237803544082258,
      "grad_norm": 1.0084835290908813,
      "learning_rate": 9.998107687592326e-05,
      "loss": 1.2094,
      "step": 370
    },
    {
      "epoch": 0.3325311747976373,
      "grad_norm": 1.82576584815979,
      "learning_rate": 9.996446595031691e-05,
      "loss": 1.2246,
      "step": 380
    },
    {
      "epoch": 0.34128199518704877,
      "grad_norm": 1.2305560111999512,
      "learning_rate": 9.994266690261928e-05,
      "loss": 1.2647,
      "step": 390
    },
    {
      "epoch": 0.35003281557646027,
      "grad_norm": 1.170067548751831,
      "learning_rate": 9.991568199636148e-05,
      "loss": 1.1829,
      "step": 400
    },
    {
      "epoch": 0.3587836359658718,
      "grad_norm": 1.0299819707870483,
      "learning_rate": 9.988351403355454e-05,
      "loss": 1.2471,
      "step": 410
    },
    {
      "epoch": 0.3675344563552833,
      "grad_norm": 0.9742593765258789,
      "learning_rate": 9.98461663543987e-05,
      "loss": 1.2049,
      "step": 420
    },
    {
      "epoch": 0.3762852767446948,
      "grad_norm": 0.9377045631408691,
      "learning_rate": 9.980364283693638e-05,
      "loss": 1.1963,
      "step": 430
    },
    {
      "epoch": 0.3850360971341063,
      "grad_norm": 1.0425899028778076,
      "learning_rate": 9.97559478966496e-05,
      "loss": 1.2167,
      "step": 440
    },
    {
      "epoch": 0.39378691752351785,
      "grad_norm": 0.9749513864517212,
      "learning_rate": 9.97030864860014e-05,
      "loss": 1.2166,
      "step": 450
    },
    {
      "epoch": 0.40253773791292935,
      "grad_norm": 1.0552780628204346,
      "learning_rate": 9.964506409392176e-05,
      "loss": 1.2228,
      "step": 460
    },
    {
      "epoch": 0.41128855830234085,
      "grad_norm": 1.0261808633804321,
      "learning_rate": 9.958188674523748e-05,
      "loss": 1.2252,
      "step": 470
    },
    {
      "epoch": 0.42003937869175234,
      "grad_norm": 0.9530945420265198,
      "learning_rate": 9.951356100004665e-05,
      "loss": 1.1845,
      "step": 480
    },
    {
      "epoch": 0.42879019908116384,
      "grad_norm": 0.8990481495857239,
      "learning_rate": 9.94400939530375e-05,
      "loss": 1.2283,
      "step": 490
    },
    {
      "epoch": 0.4375410194705754,
      "grad_norm": 0.9713592529296875,
      "learning_rate": 9.936149323275169e-05,
      "loss": 1.1945,
      "step": 500
    },
    {
      "epoch": 0.4462918398599869,
      "grad_norm": 1.003854513168335,
      "learning_rate": 9.927776700079216e-05,
      "loss": 1.2322,
      "step": 510
    },
    {
      "epoch": 0.4550426602493984,
      "grad_norm": 0.935400664806366,
      "learning_rate": 9.918892395097572e-05,
      "loss": 1.2272,
      "step": 520
    },
    {
      "epoch": 0.4637934806388099,
      "grad_norm": 1.075519323348999,
      "learning_rate": 9.90949733084303e-05,
      "loss": 1.2069,
      "step": 530
    },
    {
      "epoch": 0.47254430102822137,
      "grad_norm": 0.9949712157249451,
      "learning_rate": 9.8995924828637e-05,
      "loss": 1.2083,
      "step": 540
    },
    {
      "epoch": 0.4812951214176329,
      "grad_norm": 0.8942195773124695,
      "learning_rate": 9.889178879641717e-05,
      "loss": 1.2137,
      "step": 550
    },
    {
      "epoch": 0.4900459418070444,
      "grad_norm": 0.9525908827781677,
      "learning_rate": 9.878257602486445e-05,
      "loss": 1.2348,
      "step": 560
    },
    {
      "epoch": 0.4987967621964559,
      "grad_norm": 0.895167350769043,
      "learning_rate": 9.8668297854222e-05,
      "loss": 1.2034,
      "step": 570
    },
    {
      "epoch": 0.5075475825858674,
      "grad_norm": 1.0244312286376953,
      "learning_rate": 9.854896615070495e-05,
      "loss": 1.1942,
      "step": 580
    },
    {
      "epoch": 0.516298402975279,
      "grad_norm": 0.9835163950920105,
      "learning_rate": 9.842459330526826e-05,
      "loss": 1.2182,
      "step": 590
    },
    {
      "epoch": 0.5250492233646904,
      "grad_norm": 0.9238565564155579,
      "learning_rate": 9.829519223232006e-05,
      "loss": 1.1992,
      "step": 600
    },
    {
      "epoch": 0.533800043754102,
      "grad_norm": 0.9063233733177185,
      "learning_rate": 9.81607763683808e-05,
      "loss": 1.2135,
      "step": 610
    },
    {
      "epoch": 0.5425508641435135,
      "grad_norm": 0.9810764193534851,
      "learning_rate": 9.802135967068782e-05,
      "loss": 1.1909,
      "step": 620
    },
    {
      "epoch": 0.5513016845329249,
      "grad_norm": 0.9388274550437927,
      "learning_rate": 9.787695661574632e-05,
      "loss": 1.1693,
      "step": 630
    },
    {
      "epoch": 0.5600525049223365,
      "grad_norm": 0.933283805847168,
      "learning_rate": 9.772758219782601e-05,
      "loss": 1.1857,
      "step": 640
    },
    {
      "epoch": 0.5688033253117479,
      "grad_norm": 0.947989284992218,
      "learning_rate": 9.757325192740425e-05,
      "loss": 1.1901,
      "step": 650
    },
    {
      "epoch": 0.5775541457011595,
      "grad_norm": 0.8272686004638672,
      "learning_rate": 9.741398182955548e-05,
      "loss": 1.2087,
      "step": 660
    },
    {
      "epoch": 0.586304966090571,
      "grad_norm": 0.8907813429832458,
      "learning_rate": 9.724978844228717e-05,
      "loss": 1.201,
      "step": 670
    },
    {
      "epoch": 0.5950557864799825,
      "grad_norm": 0.8374966382980347,
      "learning_rate": 9.708068881482271e-05,
      "loss": 1.2028,
      "step": 680
    },
    {
      "epoch": 0.603806606869394,
      "grad_norm": 0.9283761382102966,
      "learning_rate": 9.690670050583095e-05,
      "loss": 1.1995,
      "step": 690
    },
    {
      "epoch": 0.6125574272588055,
      "grad_norm": 0.8573452234268188,
      "learning_rate": 9.672784158160307e-05,
      "loss": 1.1372,
      "step": 700
    },
    {
      "epoch": 0.621308247648217,
      "grad_norm": 0.8831506371498108,
      "learning_rate": 9.654413061417655e-05,
      "loss": 1.1626,
      "step": 710
    },
    {
      "epoch": 0.6300590680376286,
      "grad_norm": 0.9180812239646912,
      "learning_rate": 9.635558667940683e-05,
      "loss": 1.1466,
      "step": 720
    },
    {
      "epoch": 0.63880988842704,
      "grad_norm": 0.8657775521278381,
      "learning_rate": 9.616222935498646e-05,
      "loss": 1.2052,
      "step": 730
    },
    {
      "epoch": 0.6475607088164516,
      "grad_norm": 0.9488115310668945,
      "learning_rate": 9.596407871841229e-05,
      "loss": 1.1651,
      "step": 740
    },
    {
      "epoch": 0.656311529205863,
      "grad_norm": 0.8592752814292908,
      "learning_rate": 9.57611553449006e-05,
      "loss": 1.1953,
      "step": 750
    },
    {
      "epoch": 0.6650623495952745,
      "grad_norm": 0.889445960521698,
      "learning_rate": 9.555348030525076e-05,
      "loss": 1.1647,
      "step": 760
    },
    {
      "epoch": 0.6738131699846861,
      "grad_norm": 0.8997176885604858,
      "learning_rate": 9.534107516365725e-05,
      "loss": 1.167,
      "step": 770
    },
    {
      "epoch": 0.6825639903740975,
      "grad_norm": 0.8208065032958984,
      "learning_rate": 9.512396197547053e-05,
      "loss": 1.1631,
      "step": 780
    },
    {
      "epoch": 0.6913148107635091,
      "grad_norm": 0.8419785499572754,
      "learning_rate": 9.490216328490692e-05,
      "loss": 1.1857,
      "step": 790
    },
    {
      "epoch": 0.7000656311529205,
      "grad_norm": 0.8125676512718201,
      "learning_rate": 9.467570212270767e-05,
      "loss": 1.2426,
      "step": 800
    },
    {
      "epoch": 0.7088164515423321,
      "grad_norm": 0.9222508072853088,
      "learning_rate": 9.444460200374748e-05,
      "loss": 1.1452,
      "step": 810
    },
    {
      "epoch": 0.7175672719317436,
      "grad_norm": 0.8932722210884094,
      "learning_rate": 9.420888692459291e-05,
      "loss": 1.1647,
      "step": 820
    },
    {
      "epoch": 0.7263180923211551,
      "grad_norm": 0.817932665348053,
      "learning_rate": 9.396858136101066e-05,
      "loss": 1.1618,
      "step": 830
    },
    {
      "epoch": 0.7350689127105666,
      "grad_norm": 0.8717408180236816,
      "learning_rate": 9.372371026542595e-05,
      "loss": 1.1517,
      "step": 840
    },
    {
      "epoch": 0.7438197330999782,
      "grad_norm": 0.8578230142593384,
      "learning_rate": 9.347429906433174e-05,
      "loss": 1.2009,
      "step": 850
    },
    {
      "epoch": 0.7525705534893896,
      "grad_norm": 0.8276301622390747,
      "learning_rate": 9.322037365564845e-05,
      "loss": 1.1627,
      "step": 860
    },
    {
      "epoch": 0.7613213738788012,
      "grad_norm": 0.7934173941612244,
      "learning_rate": 9.296196040603475e-05,
      "loss": 1.1735,
      "step": 870
    },
    {
      "epoch": 0.7700721942682126,
      "grad_norm": 0.8270857930183411,
      "learning_rate": 9.269908614814995e-05,
      "loss": 1.149,
      "step": 880
    },
    {
      "epoch": 0.7788230146576242,
      "grad_norm": 1.0632795095443726,
      "learning_rate": 9.243177817786757e-05,
      "loss": 1.1937,
      "step": 890
    },
    {
      "epoch": 0.7875738350470357,
      "grad_norm": 0.8449203968048096,
      "learning_rate": 9.216006425144118e-05,
      "loss": 1.1787,
      "step": 900
    },
    {
      "epoch": 0.7963246554364471,
      "grad_norm": 0.7634311318397522,
      "learning_rate": 9.188397258262221e-05,
      "loss": 1.1445,
      "step": 910
    },
    {
      "epoch": 0.8050754758258587,
      "grad_norm": 0.7856956720352173,
      "learning_rate": 9.160353183973046e-05,
      "loss": 1.1519,
      "step": 920
    },
    {
      "epoch": 0.8138262962152701,
      "grad_norm": 0.8162206411361694,
      "learning_rate": 9.131877114267711e-05,
      "loss": 1.1282,
      "step": 930
    },
    {
      "epoch": 0.8225771166046817,
      "grad_norm": 0.8263719081878662,
      "learning_rate": 9.102972005994122e-05,
      "loss": 1.1127,
      "step": 940
    },
    {
      "epoch": 0.8313279369940932,
      "grad_norm": 0.831969141960144,
      "learning_rate": 9.073640860549931e-05,
      "loss": 1.1446,
      "step": 950
    },
    {
      "epoch": 0.8400787573835047,
      "grad_norm": 0.8127765655517578,
      "learning_rate": 9.043886723570884e-05,
      "loss": 1.1649,
      "step": 960
    },
    {
      "epoch": 0.8488295777729162,
      "grad_norm": 0.7642777562141418,
      "learning_rate": 9.013712684614582e-05,
      "loss": 1.1656,
      "step": 970
    },
    {
      "epoch": 0.8575803981623277,
      "grad_norm": 0.8071634769439697,
      "learning_rate": 8.983121876839657e-05,
      "loss": 1.1414,
      "step": 980
    },
    {
      "epoch": 0.8663312185517392,
      "grad_norm": 0.7789682149887085,
      "learning_rate": 8.95211747668046e-05,
      "loss": 1.1127,
      "step": 990
    },
    {
      "epoch": 0.8750820389411508,
      "grad_norm": 0.7899104356765747,
      "learning_rate": 8.920702703517204e-05,
      "loss": 1.1218,
      "step": 1000
    },
    {
      "epoch": 0.8838328593305622,
      "grad_norm": 0.726874053478241,
      "learning_rate": 8.888880819341702e-05,
      "loss": 1.1345,
      "step": 1010
    },
    {
      "epoch": 0.8925836797199738,
      "grad_norm": 0.8020316958427429,
      "learning_rate": 8.856655128418637e-05,
      "loss": 1.1229,
      "step": 1020
    },
    {
      "epoch": 0.9013345001093852,
      "grad_norm": 0.8466740250587463,
      "learning_rate": 8.824028976942465e-05,
      "loss": 1.1451,
      "step": 1030
    },
    {
      "epoch": 0.9100853204987968,
      "grad_norm": 0.7729008793830872,
      "learning_rate": 8.791005752689958e-05,
      "loss": 1.1204,
      "step": 1040
    },
    {
      "epoch": 0.9188361408882083,
      "grad_norm": 0.8055347204208374,
      "learning_rate": 8.757588884668434e-05,
      "loss": 1.1372,
      "step": 1050
    },
    {
      "epoch": 0.9275869612776197,
      "grad_norm": 0.8180489540100098,
      "learning_rate": 8.723781842759695e-05,
      "loss": 1.1632,
      "step": 1060
    },
    {
      "epoch": 0.9363377816670313,
      "grad_norm": 0.816849946975708,
      "learning_rate": 8.689588137359732e-05,
      "loss": 1.1226,
      "step": 1070
    },
    {
      "epoch": 0.9450886020564427,
      "grad_norm": 0.8028609752655029,
      "learning_rate": 8.655011319014216e-05,
      "loss": 1.1559,
      "step": 1080
    },
    {
      "epoch": 0.9538394224458543,
      "grad_norm": 0.8719303011894226,
      "learning_rate": 8.620054978049829e-05,
      "loss": 1.1522,
      "step": 1090
    },
    {
      "epoch": 0.9625902428352658,
      "grad_norm": 0.766177237033844,
      "learning_rate": 8.584722744201447e-05,
      "loss": 1.1186,
      "step": 1100
    },
    {
      "epoch": 0.9713410632246773,
      "grad_norm": 0.7790910005569458,
      "learning_rate": 8.549018286235255e-05,
      "loss": 1.1227,
      "step": 1110
    },
    {
      "epoch": 0.9800918836140888,
      "grad_norm": 0.7803717255592346,
      "learning_rate": 8.512945311567786e-05,
      "loss": 1.1186,
      "step": 1120
    },
    {
      "epoch": 0.9888427040035004,
      "grad_norm": 0.7084935307502747,
      "learning_rate": 8.476507565880957e-05,
      "loss": 1.1085,
      "step": 1130
    },
    {
      "epoch": 0.9975935243929118,
      "grad_norm": 0.7696670889854431,
      "learning_rate": 8.439708832733142e-05,
      "loss": 1.1123,
      "step": 1140
    },
    {
      "epoch": 1.006125574272588,
      "grad_norm": 0.9948705434799194,
      "learning_rate": 8.402552933166288e-05,
      "loss": 0.9984,
      "step": 1150
    },
    {
      "epoch": 1.0148763946619996,
      "grad_norm": 0.8488800525665283,
      "learning_rate": 8.36504372530916e-05,
      "loss": 0.9603,
      "step": 1160
    },
    {
      "epoch": 1.023627215051411,
      "grad_norm": 0.8748326897621155,
      "learning_rate": 8.32718510397673e-05,
      "loss": 0.9535,
      "step": 1170
    },
    {
      "epoch": 1.0323780354408225,
      "grad_norm": 0.8636799454689026,
      "learning_rate": 8.288981000265748e-05,
      "loss": 0.9705,
      "step": 1180
    },
    {
      "epoch": 1.041128855830234,
      "grad_norm": 0.7722444534301758,
      "learning_rate": 8.250435381146549e-05,
      "loss": 0.9411,
      "step": 1190
    },
    {
      "epoch": 1.0498796762196456,
      "grad_norm": 0.792931318283081,
      "learning_rate": 8.211552249051149e-05,
      "loss": 0.9455,
      "step": 1200
    },
    {
      "epoch": 1.0586304966090572,
      "grad_norm": 0.8179608583450317,
      "learning_rate": 8.17233564145764e-05,
      "loss": 0.9418,
      "step": 1210
    },
    {
      "epoch": 1.0673813169984685,
      "grad_norm": 0.8081513047218323,
      "learning_rate": 8.132789630470952e-05,
      "loss": 0.9451,
      "step": 1220
    },
    {
      "epoch": 1.07613213738788,
      "grad_norm": 0.8941613435745239,
      "learning_rate": 8.092918322400027e-05,
      "loss": 0.9336,
      "step": 1230
    },
    {
      "epoch": 1.0848829577772916,
      "grad_norm": 0.8505945801734924,
      "learning_rate": 8.052725857331429e-05,
      "loss": 0.9432,
      "step": 1240
    },
    {
      "epoch": 1.0936337781667032,
      "grad_norm": 0.7953508496284485,
      "learning_rate": 8.012216408699455e-05,
      "loss": 0.9608,
      "step": 1250
    },
    {
      "epoch": 1.1023845985561147,
      "grad_norm": 0.9065775871276855,
      "learning_rate": 7.971394182852789e-05,
      "loss": 0.9583,
      "step": 1260
    },
    {
      "epoch": 1.1111354189455263,
      "grad_norm": 0.8752979040145874,
      "learning_rate": 7.930263418617713e-05,
      "loss": 0.9752,
      "step": 1270
    },
    {
      "epoch": 1.1198862393349376,
      "grad_norm": 0.8173210024833679,
      "learning_rate": 7.888828386857985e-05,
      "loss": 0.9505,
      "step": 1280
    },
    {
      "epoch": 1.1286370597243491,
      "grad_norm": 0.7964232563972473,
      "learning_rate": 7.84709339003135e-05,
      "loss": 0.9346,
      "step": 1290
    },
    {
      "epoch": 1.1373878801137607,
      "grad_norm": 0.8364039063453674,
      "learning_rate": 7.805062761742799e-05,
      "loss": 0.9606,
      "step": 1300
    },
    {
      "epoch": 1.1461387005031722,
      "grad_norm": 0.8122743964195251,
      "learning_rate": 7.76274086629458e-05,
      "loss": 0.9455,
      "step": 1310
    },
    {
      "epoch": 1.1548895208925836,
      "grad_norm": 0.8229530453681946,
      "learning_rate": 7.720132098233031e-05,
      "loss": 0.9465,
      "step": 1320
    },
    {
      "epoch": 1.1636403412819951,
      "grad_norm": 0.7859545946121216,
      "learning_rate": 7.677240881892258e-05,
      "loss": 0.9421,
      "step": 1330
    },
    {
      "epoch": 1.1723911616714067,
      "grad_norm": 0.8277194499969482,
      "learning_rate": 7.634071670934734e-05,
      "loss": 0.9541,
      "step": 1340
    },
    {
      "epoch": 1.1811419820608182,
      "grad_norm": 0.7695391774177551,
      "learning_rate": 7.590628947888847e-05,
      "loss": 0.9433,
      "step": 1350
    },
    {
      "epoch": 1.1898928024502298,
      "grad_norm": 0.82956862449646,
      "learning_rate": 7.546917223683453e-05,
      "loss": 0.9395,
      "step": 1360
    },
    {
      "epoch": 1.1986436228396413,
      "grad_norm": 0.8398957252502441,
      "learning_rate": 7.502941037179474e-05,
      "loss": 0.9432,
      "step": 1370
    },
    {
      "epoch": 1.2073944432290526,
      "grad_norm": 0.8232219219207764,
      "learning_rate": 7.4587049546986e-05,
      "loss": 0.9887,
      "step": 1380
    },
    {
      "epoch": 1.2161452636184642,
      "grad_norm": 0.7886401414871216,
      "learning_rate": 7.414213569549145e-05,
      "loss": 0.959,
      "step": 1390
    },
    {
      "epoch": 1.2248960840078758,
      "grad_norm": 0.7588061690330505,
      "learning_rate": 7.369471501549087e-05,
      "loss": 0.9651,
      "step": 1400
    },
    {
      "epoch": 1.2336469043972873,
      "grad_norm": 0.8234547972679138,
      "learning_rate": 7.324483396546371e-05,
      "loss": 0.9935,
      "step": 1410
    },
    {
      "epoch": 1.2423977247866986,
      "grad_norm": 0.7623049020767212,
      "learning_rate": 7.279253925936498e-05,
      "loss": 0.95,
      "step": 1420
    },
    {
      "epoch": 1.2511485451761102,
      "grad_norm": 0.7172533869743347,
      "learning_rate": 7.233787786177464e-05,
      "loss": 0.9441,
      "step": 1430
    },
    {
      "epoch": 1.2598993655655217,
      "grad_norm": 0.7973024249076843,
      "learning_rate": 7.188089698302098e-05,
      "loss": 0.9682,
      "step": 1440
    },
    {
      "epoch": 1.2686501859549333,
      "grad_norm": 0.7969169020652771,
      "learning_rate": 7.14216440742785e-05,
      "loss": 0.9541,
      "step": 1450
    },
    {
      "epoch": 1.2774010063443448,
      "grad_norm": 0.8296008110046387,
      "learning_rate": 7.096016682264074e-05,
      "loss": 0.9597,
      "step": 1460
    },
    {
      "epoch": 1.2861518267337564,
      "grad_norm": 0.8538773655891418,
      "learning_rate": 7.049651314616862e-05,
      "loss": 0.9225,
      "step": 1470
    },
    {
      "epoch": 1.2949026471231677,
      "grad_norm": 0.8068443536758423,
      "learning_rate": 7.003073118891487e-05,
      "loss": 0.9282,
      "step": 1480
    },
    {
      "epoch": 1.3036534675125793,
      "grad_norm": 0.8286350965499878,
      "learning_rate": 6.956286931592481e-05,
      "loss": 0.9645,
      "step": 1490
    },
    {
      "epoch": 1.3124042879019908,
      "grad_norm": 0.796650767326355,
      "learning_rate": 6.909297610821443e-05,
      "loss": 0.934,
      "step": 1500
    },
    {
      "epoch": 1.3211551082914024,
      "grad_norm": 0.7479636073112488,
      "learning_rate": 6.862110035772589e-05,
      "loss": 0.9319,
      "step": 1510
    },
    {
      "epoch": 1.3299059286808137,
      "grad_norm": 0.7734546065330505,
      "learning_rate": 6.81472910622611e-05,
      "loss": 0.9577,
      "step": 1520
    },
    {
      "epoch": 1.3386567490702252,
      "grad_norm": 0.821553647518158,
      "learning_rate": 6.7671597420394e-05,
      "loss": 0.966,
      "step": 1530
    },
    {
      "epoch": 1.3474075694596368,
      "grad_norm": 0.8141497373580933,
      "learning_rate": 6.719406882636196e-05,
      "loss": 0.9593,
      "step": 1540
    },
    {
      "epoch": 1.3561583898490484,
      "grad_norm": 0.7738481163978577,
      "learning_rate": 6.671475486493691e-05,
      "loss": 0.9474,
      "step": 1550
    },
    {
      "epoch": 1.36490921023846,
      "grad_norm": 0.7933306694030762,
      "learning_rate": 6.62337053062766e-05,
      "loss": 0.9297,
      "step": 1560
    },
    {
      "epoch": 1.3736600306278715,
      "grad_norm": 0.7921035885810852,
      "learning_rate": 6.575097010075664e-05,
      "loss": 0.9283,
      "step": 1570
    },
    {
      "epoch": 1.382410851017283,
      "grad_norm": 0.7895304560661316,
      "learning_rate": 6.526659937378392e-05,
      "loss": 0.9559,
      "step": 1580
    },
    {
      "epoch": 1.3911616714066943,
      "grad_norm": 0.8518770933151245,
      "learning_rate": 6.478064342059164e-05,
      "loss": 0.9571,
      "step": 1590
    },
    {
      "epoch": 1.3999124917961059,
      "grad_norm": 0.8479484915733337,
      "learning_rate": 6.429315270101701e-05,
      "loss": 0.9578,
      "step": 1600
    },
    {
      "epoch": 1.4086633121855174,
      "grad_norm": 0.763520359992981,
      "learning_rate": 6.380417783426153e-05,
      "loss": 0.9304,
      "step": 1610
    },
    {
      "epoch": 1.4174141325749288,
      "grad_norm": 0.7657537460327148,
      "learning_rate": 6.3313769593635e-05,
      "loss": 0.9541,
      "step": 1620
    },
    {
      "epoch": 1.4261649529643403,
      "grad_norm": 0.7461577653884888,
      "learning_rate": 6.282197890128338e-05,
      "loss": 0.9331,
      "step": 1630
    },
    {
      "epoch": 1.4349157733537519,
      "grad_norm": 0.7814643383026123,
      "learning_rate": 6.23288568229012e-05,
      "loss": 0.9431,
      "step": 1640
    },
    {
      "epoch": 1.4436665937431634,
      "grad_norm": 0.7600048184394836,
      "learning_rate": 6.183445456242904e-05,
      "loss": 0.9549,
      "step": 1650
    },
    {
      "epoch": 1.452417414132575,
      "grad_norm": 0.7684770226478577,
      "learning_rate": 6.13388234567369e-05,
      "loss": 0.9074,
      "step": 1660
    },
    {
      "epoch": 1.4611682345219865,
      "grad_norm": 0.7696465849876404,
      "learning_rate": 6.084201497029333e-05,
      "loss": 0.9449,
      "step": 1670
    },
    {
      "epoch": 1.469919054911398,
      "grad_norm": 0.818397045135498,
      "learning_rate": 6.034408068982172e-05,
      "loss": 0.9396,
      "step": 1680
    },
    {
      "epoch": 1.4786698753008094,
      "grad_norm": 0.7443113923072815,
      "learning_rate": 5.984507231894364e-05,
      "loss": 0.928,
      "step": 1690
    },
    {
      "epoch": 1.487420695690221,
      "grad_norm": 0.8166981339454651,
      "learning_rate": 5.934504167281022e-05,
      "loss": 0.9323,
      "step": 1700
    },
    {
      "epoch": 1.4961715160796325,
      "grad_norm": 0.8331442475318909,
      "learning_rate": 5.884404067272174e-05,
      "loss": 0.9458,
      "step": 1710
    },
    {
      "epoch": 1.5049223364690438,
      "grad_norm": 0.7990359663963318,
      "learning_rate": 5.834212134073644e-05,
      "loss": 0.9094,
      "step": 1720
    },
    {
      "epoch": 1.5136731568584554,
      "grad_norm": 0.768055260181427,
      "learning_rate": 5.7839335794268666e-05,
      "loss": 0.9385,
      "step": 1730
    },
    {
      "epoch": 1.522423977247867,
      "grad_norm": 0.7425386309623718,
      "learning_rate": 5.7335736240677174e-05,
      "loss": 0.9446,
      "step": 1740
    },
    {
      "epoch": 1.5311747976372785,
      "grad_norm": 0.7698354721069336,
      "learning_rate": 5.683137497184424e-05,
      "loss": 0.9057,
      "step": 1750
    },
    {
      "epoch": 1.53992561802669,
      "grad_norm": 0.8186885118484497,
      "learning_rate": 5.6326304358745694e-05,
      "loss": 0.9758,
      "step": 1760
    },
    {
      "epoch": 1.5486764384161016,
      "grad_norm": 0.8006178736686707,
      "learning_rate": 5.5820576846013086e-05,
      "loss": 0.9424,
      "step": 1770
    },
    {
      "epoch": 1.5574272588055131,
      "grad_norm": 0.7572160363197327,
      "learning_rate": 5.531424494648789e-05,
      "loss": 0.945,
      "step": 1780
    },
    {
      "epoch": 1.5661780791949247,
      "grad_norm": 0.8019212484359741,
      "learning_rate": 5.480736123576886e-05,
      "loss": 0.9528,
      "step": 1790
    },
    {
      "epoch": 1.574928899584336,
      "grad_norm": 0.7867367267608643,
      "learning_rate": 5.4299978346752756e-05,
      "loss": 0.9499,
      "step": 1800
    },
    {
      "epoch": 1.5836797199737476,
      "grad_norm": 0.7404739856719971,
      "learning_rate": 5.379214896416907e-05,
      "loss": 0.9177,
      "step": 1810
    },
    {
      "epoch": 1.592430540363159,
      "grad_norm": 0.712374746799469,
      "learning_rate": 5.328392581910961e-05,
      "loss": 0.9212,
      "step": 1820
    },
    {
      "epoch": 1.6011813607525704,
      "grad_norm": 0.7724602222442627,
      "learning_rate": 5.277536168355293e-05,
      "loss": 0.9402,
      "step": 1830
    },
    {
      "epoch": 1.609932181141982,
      "grad_norm": 0.8452624678611755,
      "learning_rate": 5.226650936488474e-05,
      "loss": 0.9193,
      "step": 1840
    },
    {
      "epoch": 1.6186830015313936,
      "grad_norm": 0.7993892431259155,
      "learning_rate": 5.1757421700414645e-05,
      "loss": 0.9328,
      "step": 1850
    },
    {
      "epoch": 1.627433821920805,
      "grad_norm": 0.7931053042411804,
      "learning_rate": 5.124815155188966e-05,
      "loss": 0.9759,
      "step": 1860
    },
    {
      "epoch": 1.6361846423102167,
      "grad_norm": 0.7516879439353943,
      "learning_rate": 5.0738751800005226e-05,
      "loss": 0.9229,
      "step": 1870
    },
    {
      "epoch": 1.6449354626996282,
      "grad_norm": 0.6935803890228271,
      "learning_rate": 5.022927533891434e-05,
      "loss": 0.9234,
      "step": 1880
    },
    {
      "epoch": 1.6536862830890398,
      "grad_norm": 0.8120819926261902,
      "learning_rate": 4.9719775070735196e-05,
      "loss": 0.9582,
      "step": 1890
    },
    {
      "epoch": 1.662437103478451,
      "grad_norm": 0.781491219997406,
      "learning_rate": 4.921030390005802e-05,
      "loss": 0.9177,
      "step": 1900
    },
    {
      "epoch": 1.6711879238678626,
      "grad_norm": 0.7723197340965271,
      "learning_rate": 4.870091472845166e-05,
      "loss": 0.9382,
      "step": 1910
    },
    {
      "epoch": 1.679938744257274,
      "grad_norm": 0.751848578453064,
      "learning_rate": 4.819166044897047e-05,
      "loss": 0.9148,
      "step": 1920
    },
    {
      "epoch": 1.6886895646466855,
      "grad_norm": 0.8265687823295593,
      "learning_rate": 4.768259394066216e-05,
      "loss": 0.9386,
      "step": 1930
    },
    {
      "epoch": 1.697440385036097,
      "grad_norm": 0.7489150762557983,
      "learning_rate": 4.7173768063076956e-05,
      "loss": 0.9337,
      "step": 1940
    },
    {
      "epoch": 1.7061912054255086,
      "grad_norm": 0.8112419247627258,
      "learning_rate": 4.666523565077889e-05,
      "loss": 0.917,
      "step": 1950
    },
    {
      "epoch": 1.7149420258149202,
      "grad_norm": 0.7611903548240662,
      "learning_rate": 4.615704950785965e-05,
      "loss": 0.9338,
      "step": 1960
    },
    {
      "epoch": 1.7236928462043317,
      "grad_norm": 0.7738552093505859,
      "learning_rate": 4.5649262402455704e-05,
      "loss": 0.9359,
      "step": 1970
    },
    {
      "epoch": 1.7324436665937433,
      "grad_norm": 0.7741129994392395,
      "learning_rate": 4.5141927061268844e-05,
      "loss": 0.9209,
      "step": 1980
    },
    {
      "epoch": 1.7411944869831548,
      "grad_norm": 0.7815123200416565,
      "learning_rate": 4.463509616409144e-05,
      "loss": 0.916,
      "step": 1990
    },
    {
      "epoch": 1.7499453073725662,
      "grad_norm": 0.6793907284736633,
      "learning_rate": 4.412882233833629e-05,
      "loss": 0.9042,
      "step": 2000
    },
    {
      "epoch": 1.7586961277619777,
      "grad_norm": 0.711877167224884,
      "learning_rate": 4.362315815357197e-05,
      "loss": 0.9072,
      "step": 2010
    },
    {
      "epoch": 1.767446948151389,
      "grad_norm": 0.7671152353286743,
      "learning_rate": 4.311815611606423e-05,
      "loss": 0.9226,
      "step": 2020
    },
    {
      "epoch": 1.7761977685408006,
      "grad_norm": 0.7221760153770447,
      "learning_rate": 4.261386866332397e-05,
      "loss": 0.9077,
      "step": 2030
    },
    {
      "epoch": 1.7849485889302121,
      "grad_norm": 0.7419185042381287,
      "learning_rate": 4.211034815866228e-05,
      "loss": 0.9017,
      "step": 2040
    },
    {
      "epoch": 1.7936994093196237,
      "grad_norm": 0.740652859210968,
      "learning_rate": 4.1607646885753206e-05,
      "loss": 0.9159,
      "step": 2050
    },
    {
      "epoch": 1.8024502297090352,
      "grad_norm": 0.7923645377159119,
      "learning_rate": 4.110581704320495e-05,
      "loss": 0.9135,
      "step": 2060
    },
    {
      "epoch": 1.8112010500984468,
      "grad_norm": 0.7528581023216248,
      "learning_rate": 4.060491073913957e-05,
      "loss": 0.9291,
      "step": 2070
    },
    {
      "epoch": 1.8199518704878583,
      "grad_norm": 0.7974929809570312,
      "learning_rate": 4.010497998578239e-05,
      "loss": 0.9208,
      "step": 2080
    },
    {
      "epoch": 1.8287026908772699,
      "grad_norm": 0.7388021349906921,
      "learning_rate": 3.960607669406129e-05,
      "loss": 0.9155,
      "step": 2090
    },
    {
      "epoch": 1.8374535112666812,
      "grad_norm": 0.7927631735801697,
      "learning_rate": 3.910825266821629e-05,
      "loss": 0.9156,
      "step": 2100
    },
    {
      "epoch": 1.8462043316560928,
      "grad_norm": 0.7545862793922424,
      "learning_rate": 3.861155960042057e-05,
      "loss": 0.877,
      "step": 2110
    },
    {
      "epoch": 1.8549551520455043,
      "grad_norm": 0.8433863520622253,
      "learning_rate": 3.8116049065412864e-05,
      "loss": 0.9107,
      "step": 2120
    },
    {
      "epoch": 1.8637059724349156,
      "grad_norm": 0.7610001564025879,
      "learning_rate": 3.762177251514218e-05,
      "loss": 0.9068,
      "step": 2130
    },
    {
      "epoch": 1.8724567928243272,
      "grad_norm": 0.7452635765075684,
      "learning_rate": 3.712878127342515e-05,
      "loss": 0.9113,
      "step": 2140
    },
    {
      "epoch": 1.8812076132137387,
      "grad_norm": 0.7530847787857056,
      "learning_rate": 3.663712653061682e-05,
      "loss": 0.8881,
      "step": 2150
    },
    {
      "epoch": 1.8899584336031503,
      "grad_norm": 0.7940835356712341,
      "learning_rate": 3.614685933829525e-05,
      "loss": 0.8807,
      "step": 2160
    },
    {
      "epoch": 1.8987092539925619,
      "grad_norm": 0.7291942834854126,
      "learning_rate": 3.5658030603960436e-05,
      "loss": 0.9112,
      "step": 2170
    },
    {
      "epoch": 1.9074600743819734,
      "grad_norm": 0.7321741580963135,
      "learning_rate": 3.5170691085748324e-05,
      "loss": 0.8835,
      "step": 2180
    },
    {
      "epoch": 1.916210894771385,
      "grad_norm": 0.7166718244552612,
      "learning_rate": 3.468489138716029e-05,
      "loss": 0.9115,
      "step": 2190
    },
    {
      "epoch": 1.9249617151607963,
      "grad_norm": 0.741908848285675,
      "learning_rate": 3.42006819518086e-05,
      "loss": 0.9198,
      "step": 2200
    },
    {
      "epoch": 1.9337125355502078,
      "grad_norm": 0.8034552931785583,
      "learning_rate": 3.371811305817858e-05,
      "loss": 0.8952,
      "step": 2210
    },
    {
      "epoch": 1.9424633559396194,
      "grad_norm": 0.6716712713241577,
      "learning_rate": 3.32372348144079e-05,
      "loss": 0.9197,
      "step": 2220
    },
    {
      "epoch": 1.9512141763290307,
      "grad_norm": 0.6731935739517212,
      "learning_rate": 3.275809715308349e-05,
      "loss": 0.8949,
      "step": 2230
    },
    {
      "epoch": 1.9599649967184423,
      "grad_norm": 0.7521949410438538,
      "learning_rate": 3.228074982605673e-05,
      "loss": 0.8949,
      "step": 2240
    },
    {
      "epoch": 1.9687158171078538,
      "grad_norm": 0.6889147162437439,
      "learning_rate": 3.180524239927749e-05,
      "loss": 0.8941,
      "step": 2250
    },
    {
      "epoch": 1.9774666374972654,
      "grad_norm": 0.7169541716575623,
      "learning_rate": 3.1331624247647276e-05,
      "loss": 0.8922,
      "step": 2260
    },
    {
      "epoch": 1.986217457886677,
      "grad_norm": 0.7812347412109375,
      "learning_rate": 3.0859944549892325e-05,
      "loss": 0.8915,
      "step": 2270
    },
    {
      "epoch": 1.9949682782760885,
      "grad_norm": 0.7609400153160095,
      "learning_rate": 3.039025228345725e-05,
      "loss": 0.9227,
      "step": 2280
    },
    {
      "epoch": 2.0035003281557646,
      "grad_norm": 0.7317432165145874,
      "learning_rate": 2.9922596219419158e-05,
      "loss": 0.8076,
      "step": 2290
    },
    {
      "epoch": 2.012251148545176,
      "grad_norm": 0.8811588883399963,
      "learning_rate": 2.9457024917423593e-05,
      "loss": 0.6705,
      "step": 2300
    },
    {
      "epoch": 2.0210019689345877,
      "grad_norm": 0.856757640838623,
      "learning_rate": 2.8993586720642307e-05,
      "loss": 0.6602,
      "step": 2310
    },
    {
      "epoch": 2.0297527893239993,
      "grad_norm": 0.791458249092102,
      "learning_rate": 2.85323297507534e-05,
      "loss": 0.6417,
      "step": 2320
    },
    {
      "epoch": 2.038503609713411,
      "grad_norm": 0.8533851504325867,
      "learning_rate": 2.8073301902944515e-05,
      "loss": 0.6673,
      "step": 2330
    },
    {
      "epoch": 2.047254430102822,
      "grad_norm": 0.8373380899429321,
      "learning_rate": 2.7616550840939757e-05,
      "loss": 0.6674,
      "step": 2340
    },
    {
      "epoch": 2.0560052504922335,
      "grad_norm": 0.8470285534858704,
      "learning_rate": 2.7162123992050316e-05,
      "loss": 0.6611,
      "step": 2350
    },
    {
      "epoch": 2.064756070881645,
      "grad_norm": 0.8354354500770569,
      "learning_rate": 2.671006854224982e-05,
      "loss": 0.6449,
      "step": 2360
    },
    {
      "epoch": 2.0735068912710566,
      "grad_norm": 0.8518164157867432,
      "learning_rate": 2.626043143127477e-05,
      "loss": 0.6387,
      "step": 2370
    },
    {
      "epoch": 2.082257711660468,
      "grad_norm": 0.9259853959083557,
      "learning_rate": 2.5813259347750447e-05,
      "loss": 0.6864,
      "step": 2380
    },
    {
      "epoch": 2.0910085320498797,
      "grad_norm": 0.8142094612121582,
      "learning_rate": 2.5368598724342974e-05,
      "loss": 0.6668,
      "step": 2390
    },
    {
      "epoch": 2.0997593524392912,
      "grad_norm": 0.8513460755348206,
      "learning_rate": 2.4926495732937876e-05,
      "loss": 0.6588,
      "step": 2400
    },
    {
      "epoch": 2.108510172828703,
      "grad_norm": 0.8699485063552856,
      "learning_rate": 2.4486996279845862e-05,
      "loss": 0.681,
      "step": 2410
    },
    {
      "epoch": 2.1172609932181143,
      "grad_norm": 0.8531279563903809,
      "learning_rate": 2.4050146001035983e-05,
      "loss": 0.6548,
      "step": 2420
    },
    {
      "epoch": 2.126011813607526,
      "grad_norm": 0.8686200976371765,
      "learning_rate": 2.3615990257396946e-05,
      "loss": 0.6546,
      "step": 2430
    },
    {
      "epoch": 2.134762633996937,
      "grad_norm": 0.8861080408096313,
      "learning_rate": 2.318457413002718e-05,
      "loss": 0.655,
      "step": 2440
    },
    {
      "epoch": 2.1435134543863485,
      "grad_norm": 0.8589378595352173,
      "learning_rate": 2.2755942415553593e-05,
      "loss": 0.6359,
      "step": 2450
    },
    {
      "epoch": 2.15226427477576,
      "grad_norm": 0.784140944480896,
      "learning_rate": 2.233013962148016e-05,
      "loss": 0.6206,
      "step": 2460
    },
    {
      "epoch": 2.1610150951651717,
      "grad_norm": 0.9516744613647461,
      "learning_rate": 2.1907209961566415e-05,
      "loss": 0.671,
      "step": 2470
    },
    {
      "epoch": 2.169765915554583,
      "grad_norm": 0.8857921957969666,
      "learning_rate": 2.1487197351236443e-05,
      "loss": 0.6435,
      "step": 2480
    },
    {
      "epoch": 2.1785167359439948,
      "grad_norm": 0.8351215124130249,
      "learning_rate": 2.107014540301886e-05,
      "loss": 0.6536,
      "step": 2490
    },
    {
      "epoch": 2.1872675563334063,
      "grad_norm": 0.8609489798545837,
      "learning_rate": 2.0656097422018346e-05,
      "loss": 0.652,
      "step": 2500
    },
    {
      "epoch": 2.196018376722818,
      "grad_norm": 0.8463150858879089,
      "learning_rate": 2.024509640141885e-05,
      "loss": 0.6601,
      "step": 2510
    },
    {
      "epoch": 2.2047691971122294,
      "grad_norm": 0.8469570279121399,
      "learning_rate": 1.9837185018019467e-05,
      "loss": 0.6646,
      "step": 2520
    },
    {
      "epoch": 2.213520017501641,
      "grad_norm": 0.870857298374176,
      "learning_rate": 1.9432405627802963e-05,
      "loss": 0.6429,
      "step": 2530
    },
    {
      "epoch": 2.2222708378910525,
      "grad_norm": 0.865593671798706,
      "learning_rate": 1.903080026153774e-05,
      "loss": 0.649,
      "step": 2540
    },
    {
      "epoch": 2.2310216582804636,
      "grad_norm": 0.8608151078224182,
      "learning_rate": 1.8632410620413503e-05,
      "loss": 0.6564,
      "step": 2550
    },
    {
      "epoch": 2.239772478669875,
      "grad_norm": 0.8044232726097107,
      "learning_rate": 1.823727807171115e-05,
      "loss": 0.6786,
      "step": 2560
    },
    {
      "epoch": 2.2485232990592867,
      "grad_norm": 0.8390374779701233,
      "learning_rate": 1.7845443644507354e-05,
      "loss": 0.6385,
      "step": 2570
    },
    {
      "epoch": 2.2572741194486983,
      "grad_norm": 0.8899725079536438,
      "learning_rate": 1.745694802541429e-05,
      "loss": 0.6674,
      "step": 2580
    },
    {
      "epoch": 2.26602493983811,
      "grad_norm": 0.8680664300918579,
      "learning_rate": 1.7071831554354823e-05,
      "loss": 0.6337,
      "step": 2590
    },
    {
      "epoch": 2.2747757602275214,
      "grad_norm": 0.843752384185791,
      "learning_rate": 1.6690134220373864e-05,
      "loss": 0.6582,
      "step": 2600
    },
    {
      "epoch": 2.283526580616933,
      "grad_norm": 0.8479620814323425,
      "learning_rate": 1.631189565748596e-05,
      "loss": 0.6531,
      "step": 2610
    },
    {
      "epoch": 2.2922774010063445,
      "grad_norm": 0.8540963530540466,
      "learning_rate": 1.5937155140559884e-05,
      "loss": 0.6386,
      "step": 2620
    },
    {
      "epoch": 2.301028221395756,
      "grad_norm": 0.8910626769065857,
      "learning_rate": 1.556595158124049e-05,
      "loss": 0.6533,
      "step": 2630
    },
    {
      "epoch": 2.309779041785167,
      "grad_norm": 0.8980609178543091,
      "learning_rate": 1.5198323523908265e-05,
      "loss": 0.6487,
      "step": 2640
    },
    {
      "epoch": 2.3185298621745787,
      "grad_norm": 0.8651220202445984,
      "learning_rate": 1.483430914167701e-05,
      "loss": 0.6598,
      "step": 2650
    },
    {
      "epoch": 2.3272806825639902,
      "grad_norm": 0.9363301396369934,
      "learning_rate": 1.447394623243018e-05,
      "loss": 0.678,
      "step": 2660
    },
    {
      "epoch": 2.336031502953402,
      "grad_norm": 0.8213711977005005,
      "learning_rate": 1.4117272214895899e-05,
      "loss": 0.6359,
      "step": 2670
    },
    {
      "epoch": 2.3447823233428133,
      "grad_norm": 0.8319981694221497,
      "learning_rate": 1.3764324124761735e-05,
      "loss": 0.6698,
      "step": 2680
    },
    {
      "epoch": 2.353533143732225,
      "grad_norm": 0.8644684553146362,
      "learning_rate": 1.3415138610828988e-05,
      "loss": 0.6499,
      "step": 2690
    },
    {
      "epoch": 2.3622839641216364,
      "grad_norm": 0.8030415773391724,
      "learning_rate": 1.3069751931207164e-05,
      "loss": 0.6219,
      "step": 2700
    },
    {
      "epoch": 2.371034784511048,
      "grad_norm": 0.717867374420166,
      "learning_rate": 1.2728199949549142e-05,
      "loss": 0.6315,
      "step": 2710
    },
    {
      "epoch": 2.3797856049004595,
      "grad_norm": 0.790641188621521,
      "learning_rate": 1.2390518131327172e-05,
      "loss": 0.6273,
      "step": 2720
    },
    {
      "epoch": 2.388536425289871,
      "grad_norm": 0.7971424460411072,
      "learning_rate": 1.2056741540150328e-05,
      "loss": 0.6415,
      "step": 2730
    },
    {
      "epoch": 2.3972872456792826,
      "grad_norm": 0.8711332678794861,
      "learning_rate": 1.1726904834123581e-05,
      "loss": 0.6474,
      "step": 2740
    },
    {
      "epoch": 2.4060380660686937,
      "grad_norm": 0.9122375249862671,
      "learning_rate": 1.1401042262249057e-05,
      "loss": 0.656,
      "step": 2750
    },
    {
      "epoch": 2.4147888864581053,
      "grad_norm": 0.9008005261421204,
      "learning_rate": 1.10791876608698e-05,
      "loss": 0.6204,
      "step": 2760
    },
    {
      "epoch": 2.423539706847517,
      "grad_norm": 0.7929515838623047,
      "learning_rate": 1.0761374450156202e-05,
      "loss": 0.6271,
      "step": 2770
    },
    {
      "epoch": 2.4322905272369284,
      "grad_norm": 0.8115386962890625,
      "learning_rate": 1.04476356306359e-05,
      "loss": 0.63,
      "step": 2780
    },
    {
      "epoch": 2.44104134762634,
      "grad_norm": 0.8349243402481079,
      "learning_rate": 1.0138003779767103e-05,
      "loss": 0.6557,
      "step": 2790
    },
    {
      "epoch": 2.4497921680157515,
      "grad_norm": 0.8012370467185974,
      "learning_rate": 9.832511048555826e-06,
      "loss": 0.6668,
      "step": 2800
    },
    {
      "epoch": 2.458542988405163,
      "grad_norm": 0.885390043258667,
      "learning_rate": 9.53118915821748e-06,
      "loss": 0.6331,
      "step": 2810
    },
    {
      "epoch": 2.4672938087945746,
      "grad_norm": 0.8025451302528381,
      "learning_rate": 9.234069396883066e-06,
      "loss": 0.6539,
      "step": 2820
    },
    {
      "epoch": 2.476044629183986,
      "grad_norm": 0.8048675060272217,
      "learning_rate": 8.941182616350318e-06,
      "loss": 0.6396,
      "step": 2830
    },
    {
      "epoch": 2.4847954495733973,
      "grad_norm": 0.8225316405296326,
      "learning_rate": 8.65255922888017e-06,
      "loss": 0.6491,
      "step": 2840
    },
    {
      "epoch": 2.493546269962809,
      "grad_norm": 0.8796355128288269,
      "learning_rate": 8.368229204038897e-06,
      "loss": 0.665,
      "step": 2850
    },
    {
      "epoch": 2.5022970903522204,
      "grad_norm": 0.8243616223335266,
      "learning_rate": 8.088222065586132e-06,
      "loss": 0.6448,
      "step": 2860
    },
    {
      "epoch": 2.511047910741632,
      "grad_norm": 0.873279333114624,
      "learning_rate": 7.812566888409228e-06,
      "loss": 0.6534,
      "step": 2870
    },
    {
      "epoch": 2.5197987311310435,
      "grad_norm": 0.9333387017250061,
      "learning_rate": 7.541292295504327e-06,
      "loss": 0.6393,
      "step": 2880
    },
    {
      "epoch": 2.528549551520455,
      "grad_norm": 0.800175666809082,
      "learning_rate": 7.274426455004129e-06,
      "loss": 0.6389,
      "step": 2890
    },
    {
      "epoch": 2.5373003719098666,
      "grad_norm": 0.8389214873313904,
      "learning_rate": 7.011997077253113e-06,
      "loss": 0.6337,
      "step": 2900
    },
    {
      "epoch": 2.546051192299278,
      "grad_norm": 0.8694761395454407,
      "learning_rate": 6.754031411930139e-06,
      "loss": 0.6512,
      "step": 2910
    },
    {
      "epoch": 2.5548020126886897,
      "grad_norm": 0.8349895477294922,
      "learning_rate": 6.5005562452189865e-06,
      "loss": 0.6452,
      "step": 2920
    },
    {
      "epoch": 2.5635528330781012,
      "grad_norm": 0.8583366274833679,
      "learning_rate": 6.251597897026951e-06,
      "loss": 0.6462,
      "step": 2930
    },
    {
      "epoch": 2.5723036534675128,
      "grad_norm": 0.7806413173675537,
      "learning_rate": 6.007182218251883e-06,
      "loss": 0.6168,
      "step": 2940
    },
    {
      "epoch": 2.5810544738569243,
      "grad_norm": 0.8689520955085754,
      "learning_rate": 5.7673345880979775e-06,
      "loss": 0.6336,
      "step": 2950
    },
    {
      "epoch": 2.5898052942463354,
      "grad_norm": 0.7911117672920227,
      "learning_rate": 5.532079911440419e-06,
      "loss": 0.6244,
      "step": 2960
    },
    {
      "epoch": 2.598556114635747,
      "grad_norm": 0.8084582686424255,
      "learning_rate": 5.301442616239372e-06,
      "loss": 0.6187,
      "step": 2970
    },
    {
      "epoch": 2.6073069350251585,
      "grad_norm": 0.8228966593742371,
      "learning_rate": 5.075446651003535e-06,
      "loss": 0.6187,
      "step": 2980
    },
    {
      "epoch": 2.61605775541457,
      "grad_norm": 0.8767485022544861,
      "learning_rate": 4.854115482303345e-06,
      "loss": 0.617,
      "step": 2990
    },
    {
      "epoch": 2.6248085758039816,
      "grad_norm": 0.8234701752662659,
      "learning_rate": 4.637472092334327e-06,
      "loss": 0.653,
      "step": 3000
    },
    {
      "epoch": 2.633559396193393,
      "grad_norm": 0.8716809749603271,
      "learning_rate": 4.425538976530708e-06,
      "loss": 0.6382,
      "step": 3010
    },
    {
      "epoch": 2.6423102165828047,
      "grad_norm": 0.8220137357711792,
      "learning_rate": 4.218338141229566e-06,
      "loss": 0.6273,
      "step": 3020
    },
    {
      "epoch": 2.6510610369722163,
      "grad_norm": 0.7726868987083435,
      "learning_rate": 4.015891101385794e-06,
      "loss": 0.631,
      "step": 3030
    },
    {
      "epoch": 2.6598118573616274,
      "grad_norm": 0.8567168116569519,
      "learning_rate": 3.818218878338054e-06,
      "loss": 0.6436,
      "step": 3040
    },
    {
      "epoch": 2.668562677751039,
      "grad_norm": 0.8137094378471375,
      "learning_rate": 3.6253419976260027e-06,
      "loss": 0.6439,
      "step": 3050
    },
    {
      "epoch": 2.6773134981404505,
      "grad_norm": 0.835558295249939,
      "learning_rate": 3.437280486858996e-06,
      "loss": 0.6324,
      "step": 3060
    },
    {
      "epoch": 2.686064318529862,
      "grad_norm": 0.8575820326805115,
      "learning_rate": 3.254053873636498e-06,
      "loss": 0.6339,
      "step": 3070
    },
    {
      "epoch": 2.6948151389192736,
      "grad_norm": 0.8318324685096741,
      "learning_rate": 3.075681183520407e-06,
      "loss": 0.6386,
      "step": 3080
    },
    {
      "epoch": 2.703565959308685,
      "grad_norm": 0.8100464940071106,
      "learning_rate": 2.9021809380595066e-06,
      "loss": 0.6362,
      "step": 3090
    },
    {
      "epoch": 2.7123167796980967,
      "grad_norm": 0.7721264362335205,
      "learning_rate": 2.7335711528663046e-06,
      "loss": 0.639,
      "step": 3100
    },
    {
      "epoch": 2.7210676000875083,
      "grad_norm": 0.8360996842384338,
      "learning_rate": 2.5698693357462665e-06,
      "loss": 0.625,
      "step": 3110
    },
    {
      "epoch": 2.72981842047692,
      "grad_norm": 0.859639048576355,
      "learning_rate": 2.4110924848799566e-06,
      "loss": 0.6159,
      "step": 3120
    },
    {
      "epoch": 2.7385692408663314,
      "grad_norm": 0.9522019624710083,
      "learning_rate": 2.2572570870579848e-06,
      "loss": 0.6437,
      "step": 3130
    },
    {
      "epoch": 2.747320061255743,
      "grad_norm": 0.8836742639541626,
      "learning_rate": 2.1083791159690613e-06,
      "loss": 0.6294,
      "step": 3140
    },
    {
      "epoch": 2.7560708816451545,
      "grad_norm": 0.901431143283844,
      "learning_rate": 1.9644740305413633e-06,
      "loss": 0.6387,
      "step": 3150
    },
    {
      "epoch": 2.764821702034566,
      "grad_norm": 0.8418229818344116,
      "learning_rate": 1.825556773337339e-06,
      "loss": 0.6355,
      "step": 3160
    },
    {
      "epoch": 2.773572522423977,
      "grad_norm": 0.854288637638092,
      "learning_rate": 1.691641769002139e-06,
      "loss": 0.6357,
      "step": 3170
    },
    {
      "epoch": 2.7823233428133887,
      "grad_norm": 0.897578239440918,
      "learning_rate": 1.5627429227657863e-06,
      "loss": 0.6584,
      "step": 3180
    },
    {
      "epoch": 2.7910741632028,
      "grad_norm": 0.94583660364151,
      "learning_rate": 1.4388736189993258e-06,
      "loss": 0.6333,
      "step": 3190
    },
    {
      "epoch": 2.7998249835922118,
      "grad_norm": 0.8198401927947998,
      "learning_rate": 1.3200467198250688e-06,
      "loss": 0.6343,
      "step": 3200
    },
    {
      "epoch": 2.8085758039816233,
      "grad_norm": 0.8177739381790161,
      "learning_rate": 1.2062745637809736e-06,
      "loss": 0.6351,
      "step": 3210
    },
    {
      "epoch": 2.817326624371035,
      "grad_norm": 0.8060468435287476,
      "learning_rate": 1.0975689645395193e-06,
      "loss": 0.616,
      "step": 3220
    },
    {
      "epoch": 2.8260774447604464,
      "grad_norm": 0.8761562705039978,
      "learning_rate": 9.939412096809986e-07,
      "loss": 0.6426,
      "step": 3230
    },
    {
      "epoch": 2.8348282651498575,
      "grad_norm": 0.8060517311096191,
      "learning_rate": 8.954020595214441e-07,
      "loss": 0.6432,
      "step": 3240
    },
    {
      "epoch": 2.843579085539269,
      "grad_norm": 0.8677119016647339,
      "learning_rate": 8.019617459953332e-07,
      "loss": 0.6246,
      "step": 3250
    },
    {
      "epoch": 2.8523299059286806,
      "grad_norm": 0.8648504018783569,
      "learning_rate": 7.136299715931438e-07,
      "loss": 0.6246,
      "step": 3260
    },
    {
      "epoch": 2.861080726318092,
      "grad_norm": 0.8173497319221497,
      "learning_rate": 6.304159083538652e-07,
      "loss": 0.6188,
      "step": 3270
    },
    {
      "epoch": 2.8698315467075037,
      "grad_norm": 0.8254984617233276,
      "learning_rate": 5.523281969126271e-07,
      "loss": 0.6365,
      "step": 3280
    },
    {
      "epoch": 2.8785823670969153,
      "grad_norm": 0.8288102746009827,
      "learning_rate": 4.793749456034946e-07,
      "loss": 0.6199,
      "step": 3290
    },
    {
      "epoch": 2.887333187486327,
      "grad_norm": 0.8244780898094177,
      "learning_rate": 4.115637296174979e-07,
      "loss": 0.6438,
      "step": 3300
    },
    {
      "epoch": 2.8960840078757384,
      "grad_norm": 0.8321549296379089,
      "learning_rate": 3.4890159021608303e-07,
      "loss": 0.623,
      "step": 3310
    },
    {
      "epoch": 2.90483482826515,
      "grad_norm": 0.748741865158081,
      "learning_rate": 2.913950339999638e-07,
      "loss": 0.6297,
      "step": 3320
    },
    {
      "epoch": 2.9135856486545615,
      "grad_norm": 0.8655825853347778,
      "learning_rate": 2.390500322334954e-07,
      "loss": 0.6257,
      "step": 3330
    },
    {
      "epoch": 2.922336469043973,
      "grad_norm": 0.8270664215087891,
      "learning_rate": 1.9187202022464823e-07,
      "loss": 0.6356,
      "step": 3340
    },
    {
      "epoch": 2.9310872894333846,
      "grad_norm": 0.892652153968811,
      "learning_rate": 1.4986589676063145e-07,
      "loss": 0.6228,
      "step": 3350
    },
    {
      "epoch": 2.939838109822796,
      "grad_norm": 0.9109422564506531,
      "learning_rate": 1.1303602359919452e-07,
      "loss": 0.6493,
      "step": 3360
    },
    {
      "epoch": 2.9485889302122072,
      "grad_norm": 0.820085883140564,
      "learning_rate": 8.138622501575598e-08,
      "loss": 0.5908,
      "step": 3370
    },
    {
      "epoch": 2.957339750601619,
      "grad_norm": 0.8783090114593506,
      "learning_rate": 5.491978740627679e-08,
      "loss": 0.6305,
      "step": 3380
    },
    {
      "epoch": 2.9660905709910304,
      "grad_norm": 0.8442906141281128,
      "learning_rate": 3.36394589460276e-08,
      "loss": 0.6265,
      "step": 3390
    },
    {
      "epoch": 2.974841391380442,
      "grad_norm": 0.8543626666069031,
      "learning_rate": 1.7547449304222695e-08,
      "loss": 0.627,
      "step": 3400
    },
    {
      "epoch": 2.9835922117698535,
      "grad_norm": 0.8204004168510437,
      "learning_rate": 6.645429414570137e-09,
      "loss": 0.6266,
      "step": 3410
    },
    {
      "epoch": 2.992343032159265,
      "grad_norm": 0.804568350315094,
      "learning_rate": 9.34531301793884e-10,
      "loss": 0.6332,
      "step": 3420
    }
  ],
  "logging_steps": 10,
  "max_steps": 3426,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500.0,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 5.3232347449707725e+17,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}