{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.7172314864622557,
  "eval_steps": 2000,
  "global_step": 2000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "grad_norm": 0.9492120146751404,
      "learning_rate": 1.0000000000000002e-06,
      "loss": 0.9769,
      "step": 10
    },
    {
      "epoch": 0.01,
      "grad_norm": 1.344916582107544,
      "learning_rate": 2.0000000000000003e-06,
      "loss": 1.0718,
      "step": 20
    },
    {
      "epoch": 0.01,
      "grad_norm": 0.6680417060852051,
      "learning_rate": 3e-06,
      "loss": 0.8901,
      "step": 30
    },
    {
      "epoch": 0.01,
      "grad_norm": 0.8369797468185425,
      "learning_rate": 4.000000000000001e-06,
      "loss": 0.9398,
      "step": 40
    },
    {
      "epoch": 0.02,
      "grad_norm": 0.5738756656646729,
      "learning_rate": 5e-06,
      "loss": 0.9152,
      "step": 50
    },
    {
      "epoch": 0.02,
      "grad_norm": 1.028475284576416,
      "learning_rate": 6e-06,
      "loss": 0.849,
      "step": 60
    },
    {
      "epoch": 0.03,
      "grad_norm": 1.2893372774124146,
      "learning_rate": 7e-06,
      "loss": 0.7312,
      "step": 70
    },
    {
      "epoch": 0.03,
      "grad_norm": 0.8779547810554504,
      "learning_rate": 8.000000000000001e-06,
      "loss": 0.6707,
      "step": 80
    },
    {
      "epoch": 0.03,
      "grad_norm": 0.8972748517990112,
      "learning_rate": 9e-06,
      "loss": 0.6413,
      "step": 90
    },
    {
      "epoch": 0.04,
      "grad_norm": 1.505399227142334,
      "learning_rate": 1e-05,
      "loss": 0.6129,
      "step": 100
    },
    {
      "epoch": 0.04,
      "grad_norm": 0.6442121267318726,
      "learning_rate": 9.989898989898991e-06,
      "loss": 0.5579,
      "step": 110
    },
    {
      "epoch": 0.04,
      "grad_norm": 0.623290479183197,
      "learning_rate": 9.97979797979798e-06,
      "loss": 0.6022,
      "step": 120
    },
    {
      "epoch": 0.05,
      "grad_norm": 0.7216657996177673,
      "learning_rate": 9.96969696969697e-06,
      "loss": 0.5481,
      "step": 130
    },
    {
      "epoch": 0.05,
      "grad_norm": 0.7031328678131104,
      "learning_rate": 9.95959595959596e-06,
      "loss": 0.5594,
      "step": 140
    },
    {
      "epoch": 0.05,
      "grad_norm": 0.7411965727806091,
      "learning_rate": 9.94949494949495e-06,
      "loss": 0.6241,
      "step": 150
    },
    {
      "epoch": 0.06,
      "grad_norm": 0.8735277652740479,
      "learning_rate": 9.939393939393939e-06,
      "loss": 0.498,
      "step": 160
    },
    {
      "epoch": 0.06,
      "grad_norm": 1.1062073707580566,
      "learning_rate": 9.92929292929293e-06,
      "loss": 0.6463,
      "step": 170
    },
    {
      "epoch": 0.06,
      "grad_norm": 0.7930333614349365,
      "learning_rate": 9.91919191919192e-06,
      "loss": 0.6231,
      "step": 180
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.8774802684783936,
      "learning_rate": 9.90909090909091e-06,
      "loss": 0.5895,
      "step": 190
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.929542064666748,
      "learning_rate": 9.8989898989899e-06,
      "loss": 0.632,
      "step": 200
    },
    {
      "epoch": 0.08,
      "grad_norm": 0.6024735569953918,
      "learning_rate": 9.88888888888889e-06,
      "loss": 0.5682,
      "step": 210
    },
    {
      "epoch": 0.08,
      "grad_norm": 0.5887441635131836,
      "learning_rate": 9.87878787878788e-06,
      "loss": 0.5192,
      "step": 220
    },
    {
      "epoch": 0.08,
      "grad_norm": 0.7813496589660645,
      "learning_rate": 9.86868686868687e-06,
      "loss": 0.5841,
      "step": 230
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.5949407815933228,
      "learning_rate": 9.85858585858586e-06,
      "loss": 0.5263,
      "step": 240
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.8114556670188904,
      "learning_rate": 9.84848484848485e-06,
      "loss": 0.5513,
      "step": 250
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.7692855596542358,
      "learning_rate": 9.838383838383839e-06,
      "loss": 0.5217,
      "step": 260
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.8319826126098633,
      "learning_rate": 9.828282828282829e-06,
      "loss": 0.4386,
      "step": 270
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.6725042462348938,
      "learning_rate": 9.81818181818182e-06,
      "loss": 0.4919,
      "step": 280
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.779315173625946,
      "learning_rate": 9.80808080808081e-06,
      "loss": 0.5281,
      "step": 290
    },
    {
      "epoch": 0.11,
      "grad_norm": 0.8005223274230957,
      "learning_rate": 9.797979797979798e-06,
      "loss": 0.5206,
      "step": 300
    },
    {
      "epoch": 0.11,
      "grad_norm": 0.6395801901817322,
      "learning_rate": 9.787878787878788e-06,
      "loss": 0.474,
      "step": 310
    },
    {
      "epoch": 0.11,
      "grad_norm": 1.1016992330551147,
      "learning_rate": 9.777777777777779e-06,
      "loss": 0.5403,
      "step": 320
    },
    {
      "epoch": 0.12,
      "grad_norm": 1.395189881324768,
      "learning_rate": 9.767676767676767e-06,
      "loss": 0.5181,
      "step": 330
    },
    {
      "epoch": 0.12,
      "grad_norm": 0.801499605178833,
      "learning_rate": 9.757575757575758e-06,
      "loss": 0.5296,
      "step": 340
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.7606889605522156,
      "learning_rate": 9.747474747474748e-06,
      "loss": 0.5406,
      "step": 350
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.580170750617981,
      "learning_rate": 9.737373737373738e-06,
      "loss": 0.541,
      "step": 360
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.6730871796607971,
      "learning_rate": 9.727272727272728e-06,
      "loss": 0.5665,
      "step": 370
    },
    {
      "epoch": 0.14,
      "grad_norm": 1.240430474281311,
      "learning_rate": 9.717171717171719e-06,
      "loss": 0.5148,
      "step": 380
    },
    {
      "epoch": 0.14,
      "grad_norm": 0.9439683556556702,
      "learning_rate": 9.707070707070709e-06,
      "loss": 0.513,
      "step": 390
    },
    {
      "epoch": 0.14,
      "grad_norm": 0.6674547791481018,
      "learning_rate": 9.696969696969698e-06,
      "loss": 0.5202,
      "step": 400
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.8516045808792114,
      "learning_rate": 9.686868686868688e-06,
      "loss": 0.5162,
      "step": 410
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.8700432181358337,
      "learning_rate": 9.676767676767678e-06,
      "loss": 0.5392,
      "step": 420
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.5687388777732849,
      "learning_rate": 9.666666666666667e-06,
      "loss": 0.5106,
      "step": 430
    },
    {
      "epoch": 0.16,
      "grad_norm": 1.2382631301879883,
      "learning_rate": 9.656565656565657e-06,
      "loss": 0.5074,
      "step": 440
    },
    {
      "epoch": 0.16,
      "grad_norm": 0.8921974301338196,
      "learning_rate": 9.646464646464647e-06,
      "loss": 0.561,
      "step": 450
    },
    {
      "epoch": 0.16,
      "grad_norm": 1.3508048057556152,
      "learning_rate": 9.636363636363638e-06,
      "loss": 0.5484,
      "step": 460
    },
    {
      "epoch": 0.17,
      "grad_norm": 1.1822205781936646,
      "learning_rate": 9.626262626262626e-06,
      "loss": 0.5719,
      "step": 470
    },
    {
      "epoch": 0.17,
      "grad_norm": 0.6617271304130554,
      "learning_rate": 9.616161616161616e-06,
      "loss": 0.535,
      "step": 480
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.6614571809768677,
      "learning_rate": 9.606060606060607e-06,
      "loss": 0.4549,
      "step": 490
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.8642953038215637,
      "learning_rate": 9.595959595959597e-06,
      "loss": 0.4789,
      "step": 500
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.614743173122406,
      "learning_rate": 9.585858585858586e-06,
      "loss": 0.4854,
      "step": 510
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.970829427242279,
      "learning_rate": 9.575757575757576e-06,
      "loss": 0.5196,
      "step": 520
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.7311980128288269,
      "learning_rate": 9.565656565656566e-06,
      "loss": 0.5106,
      "step": 530
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.765849769115448,
      "learning_rate": 9.555555555555556e-06,
      "loss": 0.5782,
      "step": 540
    },
    {
      "epoch": 0.2,
      "grad_norm": 1.0889312028884888,
      "learning_rate": 9.545454545454547e-06,
      "loss": 0.5824,
      "step": 550
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.7402384877204895,
      "learning_rate": 9.535353535353537e-06,
      "loss": 0.5005,
      "step": 560
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.707028329372406,
      "learning_rate": 9.525252525252526e-06,
      "loss": 0.5233,
      "step": 570
    },
    {
      "epoch": 0.21,
      "grad_norm": 0.8338315486907959,
      "learning_rate": 9.515151515151516e-06,
      "loss": 0.4694,
      "step": 580
    },
    {
      "epoch": 0.21,
      "grad_norm": 0.7450662851333618,
      "learning_rate": 9.505050505050506e-06,
      "loss": 0.4762,
      "step": 590
    },
    {
      "epoch": 0.22,
      "grad_norm": 0.7595840692520142,
      "learning_rate": 9.494949494949497e-06,
      "loss": 0.5018,
      "step": 600
    },
    {
      "epoch": 0.22,
      "grad_norm": 0.5880123376846313,
      "learning_rate": 9.484848484848485e-06,
      "loss": 0.5228,
      "step": 610
    },
    {
      "epoch": 0.22,
      "grad_norm": 0.9635146260261536,
      "learning_rate": 9.474747474747475e-06,
      "loss": 0.4987,
      "step": 620
    },
    {
      "epoch": 0.23,
      "grad_norm": 1.1274373531341553,
      "learning_rate": 9.464646464646466e-06,
      "loss": 0.5085,
      "step": 630
    },
    {
      "epoch": 0.23,
      "grad_norm": 1.1324831247329712,
      "learning_rate": 9.454545454545456e-06,
      "loss": 0.504,
      "step": 640
    },
    {
      "epoch": 0.23,
      "grad_norm": 0.5410157442092896,
      "learning_rate": 9.444444444444445e-06,
      "loss": 0.4619,
      "step": 650
    },
    {
      "epoch": 0.24,
      "grad_norm": 0.7583281993865967,
      "learning_rate": 9.434343434343435e-06,
      "loss": 0.5162,
      "step": 660
    },
    {
      "epoch": 0.24,
      "grad_norm": 0.6546668410301208,
      "learning_rate": 9.424242424242425e-06,
      "loss": 0.4969,
      "step": 670
    },
    {
      "epoch": 0.24,
      "grad_norm": 0.6070376634597778,
      "learning_rate": 9.414141414141414e-06,
      "loss": 0.4805,
      "step": 680
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.0108693838119507,
      "learning_rate": 9.404040404040404e-06,
      "loss": 0.4808,
      "step": 690
    },
    {
      "epoch": 0.25,
      "grad_norm": 0.8799183368682861,
      "learning_rate": 9.393939393939396e-06,
      "loss": 0.5082,
      "step": 700
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.3432070016860962,
      "learning_rate": 9.383838383838385e-06,
      "loss": 0.4353,
      "step": 710
    },
    {
      "epoch": 0.26,
      "grad_norm": 0.6518195271492004,
      "learning_rate": 9.373737373737375e-06,
      "loss": 0.4933,
      "step": 720
    },
    {
      "epoch": 0.26,
      "grad_norm": 0.6736329793930054,
      "learning_rate": 9.363636363636365e-06,
      "loss": 0.5342,
      "step": 730
    },
    {
      "epoch": 0.27,
      "grad_norm": 0.776785135269165,
      "learning_rate": 9.353535353535354e-06,
      "loss": 0.5162,
      "step": 740
    },
    {
      "epoch": 0.27,
      "grad_norm": 0.9443957805633545,
      "learning_rate": 9.343434343434344e-06,
      "loss": 0.4486,
      "step": 750
    },
    {
      "epoch": 0.27,
      "grad_norm": 0.8882728815078735,
      "learning_rate": 9.333333333333334e-06,
      "loss": 0.4774,
      "step": 760
    },
    {
      "epoch": 0.28,
      "grad_norm": 0.819239616394043,
      "learning_rate": 9.323232323232325e-06,
      "loss": 0.5,
      "step": 770
    },
    {
      "epoch": 0.28,
      "grad_norm": 0.8829065561294556,
      "learning_rate": 9.313131313131313e-06,
      "loss": 0.4655,
      "step": 780
    },
    {
      "epoch": 0.28,
      "grad_norm": 1.1993345022201538,
      "learning_rate": 9.303030303030303e-06,
      "loss": 0.5392,
      "step": 790
    },
    {
      "epoch": 0.29,
      "grad_norm": 0.681409478187561,
      "learning_rate": 9.292929292929294e-06,
      "loss": 0.5076,
      "step": 800
    },
    {
      "epoch": 0.29,
      "grad_norm": 1.075088381767273,
      "learning_rate": 9.282828282828284e-06,
      "loss": 0.4953,
      "step": 810
    },
    {
      "epoch": 0.29,
      "grad_norm": 0.7040195465087891,
      "learning_rate": 9.272727272727273e-06,
      "loss": 0.5596,
      "step": 820
    },
    {
      "epoch": 0.3,
      "grad_norm": 1.1210192441940308,
      "learning_rate": 9.262626262626263e-06,
      "loss": 0.5299,
      "step": 830
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.583011269569397,
      "learning_rate": 9.252525252525253e-06,
      "loss": 0.497,
      "step": 840
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.6583887338638306,
      "learning_rate": 9.242424242424244e-06,
      "loss": 0.5556,
      "step": 850
    },
    {
      "epoch": 0.31,
      "grad_norm": 0.8040810227394104,
      "learning_rate": 9.232323232323232e-06,
      "loss": 0.5257,
      "step": 860
    },
    {
      "epoch": 0.31,
      "grad_norm": 0.9269919991493225,
      "learning_rate": 9.222222222222224e-06,
      "loss": 0.4421,
      "step": 870
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.9947918653488159,
      "learning_rate": 9.212121212121213e-06,
      "loss": 0.5297,
      "step": 880
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.6900811791419983,
      "learning_rate": 9.202020202020203e-06,
      "loss": 0.4833,
      "step": 890
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.8033557534217834,
      "learning_rate": 9.191919191919193e-06,
      "loss": 0.4894,
      "step": 900
    },
    {
      "epoch": 0.33,
      "grad_norm": 0.635124683380127,
      "learning_rate": 9.181818181818184e-06,
      "loss": 0.4554,
      "step": 910
    },
    {
      "epoch": 0.33,
      "grad_norm": 0.7293840646743774,
      "learning_rate": 9.171717171717172e-06,
      "loss": 0.4693,
      "step": 920
    },
    {
      "epoch": 0.33,
      "grad_norm": 0.7628031373023987,
      "learning_rate": 9.161616161616162e-06,
      "loss": 0.5181,
      "step": 930
    },
    {
      "epoch": 0.34,
      "grad_norm": 1.0783181190490723,
      "learning_rate": 9.151515151515153e-06,
      "loss": 0.4632,
      "step": 940
    },
    {
      "epoch": 0.34,
      "grad_norm": 0.5340379476547241,
      "learning_rate": 9.141414141414143e-06,
      "loss": 0.4664,
      "step": 950
    },
    {
      "epoch": 0.34,
      "grad_norm": 0.9029551148414612,
      "learning_rate": 9.131313131313132e-06,
      "loss": 0.5333,
      "step": 960
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.7257616519927979,
      "learning_rate": 9.121212121212122e-06,
      "loss": 0.5168,
      "step": 970
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.761325478553772,
      "learning_rate": 9.111111111111112e-06,
      "loss": 0.5606,
      "step": 980
    },
    {
      "epoch": 0.36,
      "grad_norm": 0.8582245707511902,
      "learning_rate": 9.1010101010101e-06,
      "loss": 0.4332,
      "step": 990
    },
    {
      "epoch": 0.36,
      "grad_norm": 0.8598415851593018,
      "learning_rate": 9.090909090909091e-06,
      "loss": 0.5884,
      "step": 1000
    },
    {
      "epoch": 0.36,
      "grad_norm": 0.8292351365089417,
      "learning_rate": 9.080808080808081e-06,
      "loss": 0.4848,
      "step": 1010
    },
    {
      "epoch": 0.37,
      "grad_norm": 1.0559266805648804,
      "learning_rate": 9.070707070707072e-06,
      "loss": 0.4588,
      "step": 1020
    },
    {
      "epoch": 0.37,
      "grad_norm": 0.6693033576011658,
      "learning_rate": 9.06060606060606e-06,
      "loss": 0.5333,
      "step": 1030
    },
    {
      "epoch": 0.37,
      "grad_norm": 0.8114706873893738,
      "learning_rate": 9.050505050505052e-06,
      "loss": 0.5166,
      "step": 1040
    },
    {
      "epoch": 0.38,
      "grad_norm": 0.8659316301345825,
      "learning_rate": 9.040404040404042e-06,
      "loss": 0.4504,
      "step": 1050
    },
    {
      "epoch": 0.38,
      "grad_norm": 0.9083582758903503,
      "learning_rate": 9.030303030303031e-06,
      "loss": 0.5611,
      "step": 1060
    },
    {
      "epoch": 0.38,
      "grad_norm": 0.6691566109657288,
      "learning_rate": 9.020202020202021e-06,
      "loss": 0.5192,
      "step": 1070
    },
    {
      "epoch": 0.39,
      "grad_norm": 0.5889317989349365,
      "learning_rate": 9.010101010101012e-06,
      "loss": 0.4515,
      "step": 1080
    },
    {
      "epoch": 0.39,
      "grad_norm": 0.9215373992919922,
      "learning_rate": 9e-06,
      "loss": 0.4776,
      "step": 1090
    },
    {
      "epoch": 0.39,
      "grad_norm": 0.7439729571342468,
      "learning_rate": 8.98989898989899e-06,
      "loss": 0.4656,
      "step": 1100
    },
    {
      "epoch": 0.4,
      "grad_norm": 1.1780657768249512,
      "learning_rate": 8.97979797979798e-06,
      "loss": 0.4933,
      "step": 1110
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.9686077833175659,
      "learning_rate": 8.969696969696971e-06,
      "loss": 0.5167,
      "step": 1120
    },
    {
      "epoch": 0.41,
      "grad_norm": 0.829994261264801,
      "learning_rate": 8.95959595959596e-06,
      "loss": 0.491,
      "step": 1130
    },
    {
      "epoch": 0.41,
      "grad_norm": 0.6313827633857727,
      "learning_rate": 8.94949494949495e-06,
      "loss": 0.4864,
      "step": 1140
    },
    {
      "epoch": 0.41,
      "grad_norm": 0.596537709236145,
      "learning_rate": 8.93939393939394e-06,
      "loss": 0.4807,
      "step": 1150
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.8714896440505981,
      "learning_rate": 8.92929292929293e-06,
      "loss": 0.512,
      "step": 1160
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.9466399550437927,
      "learning_rate": 8.919191919191919e-06,
      "loss": 0.4883,
      "step": 1170
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.7337993383407593,
      "learning_rate": 8.90909090909091e-06,
      "loss": 0.4757,
      "step": 1180
    },
    {
      "epoch": 0.43,
      "grad_norm": 0.7684504985809326,
      "learning_rate": 8.8989898989899e-06,
      "loss": 0.5224,
      "step": 1190
    },
    {
      "epoch": 0.43,
      "grad_norm": 0.6455299854278564,
      "learning_rate": 8.888888888888888e-06,
      "loss": 0.5346,
      "step": 1200
    },
    {
      "epoch": 0.43,
      "grad_norm": 0.7279661893844604,
      "learning_rate": 8.87878787878788e-06,
      "loss": 0.4845,
      "step": 1210
    },
    {
      "epoch": 0.44,
      "grad_norm": 0.7996425032615662,
      "learning_rate": 8.86868686868687e-06,
      "loss": 0.4812,
      "step": 1220
    },
    {
      "epoch": 0.44,
      "grad_norm": 0.7299336791038513,
      "learning_rate": 8.85858585858586e-06,
      "loss": 0.4985,
      "step": 1230
    },
    {
      "epoch": 0.44,
      "grad_norm": 0.7462379932403564,
      "learning_rate": 8.84848484848485e-06,
      "loss": 0.5742,
      "step": 1240
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.7165307998657227,
      "learning_rate": 8.83838383838384e-06,
      "loss": 0.4627,
      "step": 1250
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.7239411473274231,
      "learning_rate": 8.82828282828283e-06,
      "loss": 0.5469,
      "step": 1260
    },
    {
      "epoch": 0.46,
      "grad_norm": 0.5761345028877258,
      "learning_rate": 8.818181818181819e-06,
      "loss": 0.4391,
      "step": 1270
    },
    {
      "epoch": 0.46,
      "grad_norm": 0.8207817077636719,
      "learning_rate": 8.808080808080809e-06,
      "loss": 0.5477,
      "step": 1280
    },
    {
      "epoch": 0.46,
      "grad_norm": 0.9331930875778198,
      "learning_rate": 8.7979797979798e-06,
      "loss": 0.4975,
      "step": 1290
    },
    {
      "epoch": 0.47,
      "grad_norm": 1.0401968955993652,
      "learning_rate": 8.787878787878788e-06,
      "loss": 0.5149,
      "step": 1300
    },
    {
      "epoch": 0.47,
      "grad_norm": 0.7848596572875977,
      "learning_rate": 8.777777777777778e-06,
      "loss": 0.5448,
      "step": 1310
    },
    {
      "epoch": 0.47,
      "grad_norm": 1.5979989767074585,
      "learning_rate": 8.767676767676768e-06,
      "loss": 0.4861,
      "step": 1320
    },
    {
      "epoch": 0.48,
      "grad_norm": 0.7734145522117615,
      "learning_rate": 8.757575757575759e-06,
      "loss": 0.5955,
      "step": 1330
    },
    {
      "epoch": 0.48,
      "grad_norm": 0.9506089687347412,
      "learning_rate": 8.747474747474747e-06,
      "loss": 0.5212,
      "step": 1340
    },
    {
      "epoch": 0.48,
      "grad_norm": 0.7118907570838928,
      "learning_rate": 8.737373737373738e-06,
      "loss": 0.474,
      "step": 1350
    },
    {
      "epoch": 0.49,
      "grad_norm": 1.0045005083084106,
      "learning_rate": 8.727272727272728e-06,
      "loss": 0.509,
      "step": 1360
    },
    {
      "epoch": 0.49,
      "grad_norm": 1.1283302307128906,
      "learning_rate": 8.717171717171718e-06,
      "loss": 0.5096,
      "step": 1370
    },
    {
      "epoch": 0.49,
      "grad_norm": 1.3884085416793823,
      "learning_rate": 8.707070707070707e-06,
      "loss": 0.5008,
      "step": 1380
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.7452566027641296,
      "learning_rate": 8.696969696969699e-06,
      "loss": 0.4792,
      "step": 1390
    },
    {
      "epoch": 0.5,
      "grad_norm": 1.3434414863586426,
      "learning_rate": 8.686868686868687e-06,
      "loss": 0.5385,
      "step": 1400
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.9600369930267334,
      "learning_rate": 8.676767676767678e-06,
      "loss": 0.4743,
      "step": 1410
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.6895599961280823,
      "learning_rate": 8.666666666666668e-06,
      "loss": 0.5639,
      "step": 1420
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.9460670948028564,
      "learning_rate": 8.656565656565658e-06,
      "loss": 0.5957,
      "step": 1430
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.6181186437606812,
      "learning_rate": 8.646464646464647e-06,
      "loss": 0.5032,
      "step": 1440
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.6992838382720947,
      "learning_rate": 8.636363636363637e-06,
      "loss": 0.5371,
      "step": 1450
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.6007195711135864,
      "learning_rate": 8.626262626262627e-06,
      "loss": 0.4446,
      "step": 1460
    },
    {
      "epoch": 0.53,
      "grad_norm": 0.8171485662460327,
      "learning_rate": 8.616161616161618e-06,
      "loss": 0.4835,
      "step": 1470
    },
    {
      "epoch": 0.53,
      "grad_norm": 0.9369902610778809,
      "learning_rate": 8.606060606060606e-06,
      "loss": 0.5437,
      "step": 1480
    },
    {
      "epoch": 0.53,
      "grad_norm": 1.1894652843475342,
      "learning_rate": 8.595959595959596e-06,
      "loss": 0.5151,
      "step": 1490
    },
    {
      "epoch": 0.54,
      "grad_norm": 0.694837212562561,
      "learning_rate": 8.585858585858587e-06,
      "loss": 0.488,
      "step": 1500
    },
    {
      "epoch": 0.54,
      "grad_norm": 0.8813522458076477,
      "learning_rate": 8.575757575757575e-06,
      "loss": 0.5129,
      "step": 1510
    },
    {
      "epoch": 0.55,
      "grad_norm": 1.0121350288391113,
      "learning_rate": 8.565656565656566e-06,
      "loss": 0.5165,
      "step": 1520
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.7935439944267273,
      "learning_rate": 8.555555555555556e-06,
      "loss": 0.4736,
      "step": 1530
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.7670463919639587,
      "learning_rate": 8.545454545454546e-06,
      "loss": 0.5214,
      "step": 1540
    },
    {
      "epoch": 0.56,
      "grad_norm": 1.212927222251892,
      "learning_rate": 8.535353535353535e-06,
      "loss": 0.5125,
      "step": 1550
    },
    {
      "epoch": 0.56,
      "grad_norm": 0.7966919541358948,
      "learning_rate": 8.525252525252527e-06,
      "loss": 0.4823,
      "step": 1560
    },
    {
      "epoch": 0.56,
      "grad_norm": 1.0880494117736816,
      "learning_rate": 8.515151515151517e-06,
      "loss": 0.478,
      "step": 1570
    },
    {
      "epoch": 0.57,
      "grad_norm": 1.0308737754821777,
      "learning_rate": 8.505050505050506e-06,
      "loss": 0.5368,
      "step": 1580
    },
    {
      "epoch": 0.57,
      "grad_norm": 0.7291275262832642,
      "learning_rate": 8.494949494949496e-06,
      "loss": 0.4838,
      "step": 1590
    },
    {
      "epoch": 0.57,
      "grad_norm": 0.6764214038848877,
      "learning_rate": 8.484848484848486e-06,
      "loss": 0.4882,
      "step": 1600
    },
    {
      "epoch": 0.58,
      "grad_norm": 1.051628828048706,
      "learning_rate": 8.474747474747475e-06,
      "loss": 0.4564,
      "step": 1610
    },
    {
      "epoch": 0.58,
      "grad_norm": 0.8614614605903625,
      "learning_rate": 8.464646464646465e-06,
      "loss": 0.5632,
      "step": 1620
    },
    {
      "epoch": 0.58,
      "grad_norm": 1.1045228242874146,
      "learning_rate": 8.454545454545455e-06,
      "loss": 0.4535,
      "step": 1630
    },
    {
      "epoch": 0.59,
      "grad_norm": 0.8160364031791687,
      "learning_rate": 8.444444444444446e-06,
      "loss": 0.4964,
      "step": 1640
    },
    {
      "epoch": 0.59,
      "grad_norm": 0.7776429653167725,
      "learning_rate": 8.434343434343434e-06,
      "loss": 0.4554,
      "step": 1650
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.7589672207832336,
      "learning_rate": 8.424242424242425e-06,
      "loss": 0.4458,
      "step": 1660
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.825233519077301,
      "learning_rate": 8.414141414141415e-06,
      "loss": 0.4814,
      "step": 1670
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.8226912617683411,
      "learning_rate": 8.404040404040405e-06,
      "loss": 0.4892,
      "step": 1680
    },
    {
      "epoch": 0.61,
      "grad_norm": 0.9273412823677063,
      "learning_rate": 8.393939393939394e-06,
      "loss": 0.5143,
      "step": 1690
    },
    {
      "epoch": 0.61,
      "grad_norm": 0.9043828248977661,
      "learning_rate": 8.383838383838384e-06,
      "loss": 0.4598,
      "step": 1700
    },
    {
      "epoch": 0.61,
      "grad_norm": 2.2596805095672607,
      "learning_rate": 8.373737373737374e-06,
      "loss": 0.5415,
      "step": 1710
    },
    {
      "epoch": 0.62,
      "grad_norm": 1.0041202306747437,
      "learning_rate": 8.363636363636365e-06,
      "loss": 0.5899,
      "step": 1720
    },
    {
      "epoch": 0.62,
      "grad_norm": 0.9188370704650879,
      "learning_rate": 8.353535353535355e-06,
      "loss": 0.5119,
      "step": 1730
    },
    {
      "epoch": 0.62,
      "grad_norm": 0.7778961062431335,
      "learning_rate": 8.343434343434345e-06,
      "loss": 0.5237,
      "step": 1740
    },
    {
      "epoch": 0.63,
      "grad_norm": 0.7438649535179138,
      "learning_rate": 8.333333333333334e-06,
      "loss": 0.4999,
      "step": 1750
    },
    {
      "epoch": 0.63,
      "grad_norm": 0.5649489760398865,
      "learning_rate": 8.323232323232324e-06,
      "loss": 0.552,
      "step": 1760
    },
    {
      "epoch": 0.63,
      "grad_norm": 0.5625451803207397,
      "learning_rate": 8.313131313131314e-06,
      "loss": 0.4549,
      "step": 1770
    },
    {
      "epoch": 0.64,
      "grad_norm": 1.3711755275726318,
      "learning_rate": 8.303030303030305e-06,
      "loss": 0.445,
      "step": 1780
    },
    {
      "epoch": 0.64,
      "grad_norm": 1.4339165687561035,
      "learning_rate": 8.292929292929293e-06,
      "loss": 0.4975,
      "step": 1790
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.8113200068473816,
      "learning_rate": 8.282828282828283e-06,
      "loss": 0.5288,
      "step": 1800
    },
    {
      "epoch": 0.65,
      "grad_norm": 1.1567124128341675,
      "learning_rate": 8.272727272727274e-06,
      "loss": 0.4669,
      "step": 1810
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.7966761589050293,
      "learning_rate": 8.262626262626264e-06,
      "loss": 0.4856,
      "step": 1820
    },
    {
      "epoch": 0.66,
      "grad_norm": 1.0181186199188232,
      "learning_rate": 8.252525252525253e-06,
      "loss": 0.4576,
      "step": 1830
    },
    {
      "epoch": 0.66,
      "grad_norm": 0.611566960811615,
      "learning_rate": 8.242424242424243e-06,
      "loss": 0.496,
      "step": 1840
    },
    {
      "epoch": 0.66,
      "grad_norm": 0.6482832431793213,
      "learning_rate": 8.232323232323233e-06,
      "loss": 0.4601,
      "step": 1850
    },
    {
      "epoch": 0.67,
      "grad_norm": 0.7550622820854187,
      "learning_rate": 8.222222222222222e-06,
      "loss": 0.5036,
      "step": 1860
    },
    {
      "epoch": 0.67,
      "grad_norm": 0.7835694551467896,
      "learning_rate": 8.212121212121212e-06,
      "loss": 0.5617,
      "step": 1870
    },
    {
      "epoch": 0.67,
      "grad_norm": 0.7926068305969238,
      "learning_rate": 8.202020202020202e-06,
      "loss": 0.4327,
      "step": 1880
    },
    {
      "epoch": 0.68,
      "grad_norm": 0.786851167678833,
      "learning_rate": 8.191919191919193e-06,
      "loss": 0.4654,
      "step": 1890
    },
    {
      "epoch": 0.68,
      "grad_norm": 0.9023171663284302,
      "learning_rate": 8.181818181818183e-06,
      "loss": 0.5426,
      "step": 1900
    },
    {
      "epoch": 0.68,
      "grad_norm": 1.0345401763916016,
      "learning_rate": 8.171717171717173e-06,
      "loss": 0.52,
      "step": 1910
    },
    {
      "epoch": 0.69,
      "grad_norm": 0.945004940032959,
      "learning_rate": 8.161616161616162e-06,
      "loss": 0.5512,
      "step": 1920
    },
    {
      "epoch": 0.69,
      "grad_norm": 0.658362090587616,
      "learning_rate": 8.151515151515152e-06,
      "loss": 0.513,
      "step": 1930
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.6390058398246765,
      "learning_rate": 8.141414141414142e-06,
      "loss": 0.5652,
      "step": 1940
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.7705880403518677,
      "learning_rate": 8.131313131313133e-06,
      "loss": 0.4929,
      "step": 1950
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.5400047302246094,
      "learning_rate": 8.121212121212121e-06,
      "loss": 0.4932,
      "step": 1960
    },
    {
      "epoch": 0.71,
      "grad_norm": 0.9128320217132568,
      "learning_rate": 8.111111111111112e-06,
      "loss": 0.5085,
      "step": 1970
    },
    {
      "epoch": 0.71,
      "grad_norm": 1.0019017457962036,
      "learning_rate": 8.101010101010102e-06,
      "loss": 0.4552,
      "step": 1980
    },
    {
      "epoch": 0.71,
      "grad_norm": 0.818148136138916,
      "learning_rate": 8.090909090909092e-06,
      "loss": 0.5517,
      "step": 1990
    },
    {
      "epoch": 0.72,
      "grad_norm": 0.8848174810409546,
      "learning_rate": 8.08080808080808e-06,
      "loss": 0.5094,
      "step": 2000
    },
    {
      "epoch": 0.72,
      "eval_loss": 0.6485620141029358,
      "eval_runtime": 340.5095,
      "eval_samples_per_second": 2.937,
      "eval_steps_per_second": 2.937,
      "step": 2000
    }
  ],
  "logging_steps": 10,
  "max_steps": 10000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 4,
  "save_steps": 2000,
  "total_flos": 1.63205502468096e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}