{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 5.734767025089606,
  "eval_steps": 2000,
  "global_step": 8000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.07,
      "grad_norm": 2.5333876609802246,
      "learning_rate": 9.9e-07,
      "loss": 0.9981,
      "step": 100
    },
    {
      "epoch": 0.14,
      "grad_norm": 2.1556735038757324,
      "learning_rate": 9.9e-07,
      "loss": 0.7461,
      "step": 200
    },
    {
      "epoch": 0.22,
      "grad_norm": 2.5188281536102295,
      "learning_rate": 9.7989898989899e-07,
      "loss": 0.6707,
      "step": 300
    },
    {
      "epoch": 0.29,
      "grad_norm": 2.559631824493408,
      "learning_rate": 9.697979797979798e-07,
      "loss": 0.6281,
      "step": 400
    },
    {
      "epoch": 0.36,
      "grad_norm": 3.3940675258636475,
      "learning_rate": 9.596969696969696e-07,
      "loss": 0.6437,
      "step": 500
    },
    {
      "epoch": 0.43,
      "grad_norm": 2.0749378204345703,
      "learning_rate": 9.495959595959595e-07,
      "loss": 0.6319,
      "step": 600
    },
    {
      "epoch": 0.5,
      "grad_norm": 2.3027114868164062,
      "learning_rate": 9.394949494949495e-07,
      "loss": 0.6477,
      "step": 700
    },
    {
      "epoch": 0.57,
      "grad_norm": 1.6429717540740967,
      "learning_rate": 9.293939393939394e-07,
      "loss": 0.6564,
      "step": 800
    },
    {
      "epoch": 0.65,
      "grad_norm": 2.391716718673706,
      "learning_rate": 9.192929292929292e-07,
      "loss": 0.6263,
      "step": 900
    },
    {
      "epoch": 0.72,
      "grad_norm": 2.9794795513153076,
      "learning_rate": 9.091919191919192e-07,
      "loss": 0.6422,
      "step": 1000
    },
    {
      "epoch": 0.79,
      "grad_norm": 2.853337049484253,
      "learning_rate": 8.99090909090909e-07,
      "loss": 0.6043,
      "step": 1100
    },
    {
      "epoch": 0.86,
      "grad_norm": 2.453305244445801,
      "learning_rate": 8.88989898989899e-07,
      "loss": 0.5914,
      "step": 1200
    },
    {
      "epoch": 0.93,
      "grad_norm": 2.768860340118408,
      "learning_rate": 8.788888888888889e-07,
      "loss": 0.6194,
      "step": 1300
    },
    {
      "epoch": 1.0,
      "grad_norm": 4.065279483795166,
      "learning_rate": 8.687878787878787e-07,
      "loss": 0.6038,
      "step": 1400
    },
    {
      "epoch": 1.08,
      "grad_norm": 3.8275232315063477,
      "learning_rate": 8.586868686868687e-07,
      "loss": 0.6038,
      "step": 1500
    },
    {
      "epoch": 1.15,
      "grad_norm": 2.6090586185455322,
      "learning_rate": 8.485858585858586e-07,
      "loss": 0.592,
      "step": 1600
    },
    {
      "epoch": 1.22,
      "grad_norm": 1.9870526790618896,
      "learning_rate": 8.384848484848484e-07,
      "loss": 0.5885,
      "step": 1700
    },
    {
      "epoch": 1.29,
      "grad_norm": 2.2942471504211426,
      "learning_rate": 8.283838383838383e-07,
      "loss": 0.622,
      "step": 1800
    },
    {
      "epoch": 1.36,
      "grad_norm": 2.0904691219329834,
      "learning_rate": 8.182828282828283e-07,
      "loss": 0.5857,
      "step": 1900
    },
    {
      "epoch": 1.43,
      "grad_norm": 2.909604072570801,
      "learning_rate": 8.081818181818182e-07,
      "loss": 0.5841,
      "step": 2000
    },
    {
      "epoch": 1.43,
      "eval_loss": 0.7236512303352356,
      "eval_runtime": 192.8239,
      "eval_samples_per_second": 5.186,
      "eval_steps_per_second": 0.648,
      "step": 2000
    },
    {
      "epoch": 1.51,
      "grad_norm": 2.645550012588501,
      "learning_rate": 7.98080808080808e-07,
      "loss": 0.6013,
      "step": 2100
    },
    {
      "epoch": 1.58,
      "grad_norm": 2.5540435314178467,
      "learning_rate": 7.88080808080808e-07,
      "loss": 0.6049,
      "step": 2200
    },
    {
      "epoch": 1.65,
      "grad_norm": 2.3379604816436768,
      "learning_rate": 7.779797979797979e-07,
      "loss": 0.5794,
      "step": 2300
    },
    {
      "epoch": 1.72,
      "grad_norm": 2.4884698390960693,
      "learning_rate": 7.678787878787878e-07,
      "loss": 0.5832,
      "step": 2400
    },
    {
      "epoch": 1.79,
      "grad_norm": 2.4760003089904785,
      "learning_rate": 7.577777777777778e-07,
      "loss": 0.5852,
      "step": 2500
    },
    {
      "epoch": 1.86,
      "grad_norm": 2.334510087966919,
      "learning_rate": 7.476767676767676e-07,
      "loss": 0.5825,
      "step": 2600
    },
    {
      "epoch": 1.94,
      "grad_norm": 3.3062024116516113,
      "learning_rate": 7.375757575757575e-07,
      "loss": 0.5739,
      "step": 2700
    },
    {
      "epoch": 2.01,
      "grad_norm": 3.0897557735443115,
      "learning_rate": 7.274747474747475e-07,
      "loss": 0.5823,
      "step": 2800
    },
    {
      "epoch": 2.08,
      "grad_norm": 3.0633504390716553,
      "learning_rate": 7.173737373737373e-07,
      "loss": 0.5625,
      "step": 2900
    },
    {
      "epoch": 2.15,
      "grad_norm": 3.3426527976989746,
      "learning_rate": 7.072727272727272e-07,
      "loss": 0.5484,
      "step": 3000
    },
    {
      "epoch": 2.22,
      "grad_norm": 2.7486729621887207,
      "learning_rate": 6.971717171717172e-07,
      "loss": 0.5853,
      "step": 3100
    },
    {
      "epoch": 2.29,
      "grad_norm": 2.5889084339141846,
      "learning_rate": 6.87070707070707e-07,
      "loss": 0.568,
      "step": 3200
    },
    {
      "epoch": 2.37,
      "grad_norm": 2.474393606185913,
      "learning_rate": 6.76969696969697e-07,
      "loss": 0.5699,
      "step": 3300
    },
    {
      "epoch": 2.44,
      "grad_norm": 2.99786114692688,
      "learning_rate": 6.668686868686869e-07,
      "loss": 0.5723,
      "step": 3400
    },
    {
      "epoch": 2.51,
      "grad_norm": 2.888652801513672,
      "learning_rate": 6.567676767676767e-07,
      "loss": 0.5942,
      "step": 3500
    },
    {
      "epoch": 2.58,
      "grad_norm": 3.20930552482605,
      "learning_rate": 6.466666666666666e-07,
      "loss": 0.5827,
      "step": 3600
    },
    {
      "epoch": 2.65,
      "grad_norm": 2.312199592590332,
      "learning_rate": 6.365656565656565e-07,
      "loss": 0.5644,
      "step": 3700
    },
    {
      "epoch": 2.72,
      "grad_norm": 3.1894469261169434,
      "learning_rate": 6.264646464646465e-07,
      "loss": 0.5409,
      "step": 3800
    },
    {
      "epoch": 2.8,
      "grad_norm": 2.740865468978882,
      "learning_rate": 6.163636363636363e-07,
      "loss": 0.5518,
      "step": 3900
    },
    {
      "epoch": 2.87,
      "grad_norm": 4.003296375274658,
      "learning_rate": 6.062626262626262e-07,
      "loss": 0.5796,
      "step": 4000
    },
    {
      "epoch": 2.87,
      "eval_loss": 0.7056564688682556,
      "eval_runtime": 192.8797,
      "eval_samples_per_second": 5.185,
      "eval_steps_per_second": 0.648,
      "step": 4000
    },
    {
      "epoch": 2.94,
      "grad_norm": 3.06562876701355,
      "learning_rate": 5.961616161616161e-07,
      "loss": 0.58,
      "step": 4100
    },
    {
      "epoch": 3.01,
      "grad_norm": 8.636664390563965,
      "learning_rate": 5.860606060606061e-07,
      "loss": 0.552,
      "step": 4200
    },
    {
      "epoch": 3.08,
      "grad_norm": 3.2252180576324463,
      "learning_rate": 5.759595959595959e-07,
      "loss": 0.5515,
      "step": 4300
    },
    {
      "epoch": 3.15,
      "grad_norm": 3.46083664894104,
      "learning_rate": 5.658585858585858e-07,
      "loss": 0.55,
      "step": 4400
    },
    {
      "epoch": 3.23,
      "grad_norm": 2.788499116897583,
      "learning_rate": 5.558585858585858e-07,
      "loss": 0.561,
      "step": 4500
    },
    {
      "epoch": 3.3,
      "grad_norm": 4.106410026550293,
      "learning_rate": 5.457575757575758e-07,
      "loss": 0.546,
      "step": 4600
    },
    {
      "epoch": 3.37,
      "grad_norm": 2.6162779331207275,
      "learning_rate": 5.356565656565656e-07,
      "loss": 0.556,
      "step": 4700
    },
    {
      "epoch": 3.44,
      "grad_norm": 3.473173141479492,
      "learning_rate": 5.255555555555555e-07,
      "loss": 0.552,
      "step": 4800
    },
    {
      "epoch": 3.51,
      "grad_norm": 3.0796926021575928,
      "learning_rate": 5.154545454545455e-07,
      "loss": 0.538,
      "step": 4900
    },
    {
      "epoch": 3.58,
      "grad_norm": 2.6169357299804688,
      "learning_rate": 5.053535353535353e-07,
      "loss": 0.5527,
      "step": 5000
    },
    {
      "epoch": 3.66,
      "grad_norm": 2.964797019958496,
      "learning_rate": 4.952525252525253e-07,
      "loss": 0.5603,
      "step": 5100
    },
    {
      "epoch": 3.73,
      "grad_norm": 3.1007957458496094,
      "learning_rate": 4.851515151515152e-07,
      "loss": 0.5326,
      "step": 5200
    },
    {
      "epoch": 3.8,
      "grad_norm": 2.996995449066162,
      "learning_rate": 4.75050505050505e-07,
      "loss": 0.5446,
      "step": 5300
    },
    {
      "epoch": 3.87,
      "grad_norm": 3.0833990573883057,
      "learning_rate": 4.6494949494949493e-07,
      "loss": 0.5612,
      "step": 5400
    },
    {
      "epoch": 3.94,
      "grad_norm": 3.361348867416382,
      "learning_rate": 4.548484848484848e-07,
      "loss": 0.5771,
      "step": 5500
    },
    {
      "epoch": 4.01,
      "grad_norm": 3.4631500244140625,
      "learning_rate": 4.447474747474747e-07,
      "loss": 0.5531,
      "step": 5600
    },
    {
      "epoch": 4.09,
      "grad_norm": 3.6133594512939453,
      "learning_rate": 4.3464646464646463e-07,
      "loss": 0.529,
      "step": 5700
    },
    {
      "epoch": 4.16,
      "grad_norm": 3.092081308364868,
      "learning_rate": 4.2454545454545454e-07,
      "loss": 0.5346,
      "step": 5800
    },
    {
      "epoch": 4.23,
      "grad_norm": 3.862914800643921,
      "learning_rate": 4.144444444444444e-07,
      "loss": 0.5271,
      "step": 5900
    },
    {
      "epoch": 4.3,
      "grad_norm": 3.39168643951416,
      "learning_rate": 4.043434343434343e-07,
      "loss": 0.5466,
      "step": 6000
    },
    {
      "epoch": 4.3,
      "eval_loss": 0.7162824869155884,
      "eval_runtime": 192.3954,
      "eval_samples_per_second": 5.198,
      "eval_steps_per_second": 0.65,
      "step": 6000
    },
    {
      "epoch": 4.37,
      "grad_norm": 2.3356008529663086,
      "learning_rate": 3.942424242424242e-07,
      "loss": 0.5182,
      "step": 6100
    },
    {
      "epoch": 4.44,
      "grad_norm": 4.625268459320068,
      "learning_rate": 3.841414141414141e-07,
      "loss": 0.5124,
      "step": 6200
    },
    {
      "epoch": 4.52,
      "grad_norm": 3.602734327316284,
      "learning_rate": 3.7404040404040407e-07,
      "loss": 0.5356,
      "step": 6300
    },
    {
      "epoch": 4.59,
      "grad_norm": 3.235563039779663,
      "learning_rate": 3.6393939393939393e-07,
      "loss": 0.5481,
      "step": 6400
    },
    {
      "epoch": 4.66,
      "grad_norm": 3.244750738143921,
      "learning_rate": 3.539393939393939e-07,
      "loss": 0.5375,
      "step": 6500
    },
    {
      "epoch": 4.73,
      "grad_norm": 4.223140716552734,
      "learning_rate": 3.4383838383838384e-07,
      "loss": 0.5418,
      "step": 6600
    },
    {
      "epoch": 4.8,
      "grad_norm": 3.9079456329345703,
      "learning_rate": 3.337373737373737e-07,
      "loss": 0.5575,
      "step": 6700
    },
    {
      "epoch": 4.87,
      "grad_norm": 3.0635576248168945,
      "learning_rate": 3.236363636363636e-07,
      "loss": 0.5314,
      "step": 6800
    },
    {
      "epoch": 4.95,
      "grad_norm": 3.8982629776000977,
      "learning_rate": 3.1353535353535353e-07,
      "loss": 0.5493,
      "step": 6900
    },
    {
      "epoch": 5.02,
      "grad_norm": 3.750946521759033,
      "learning_rate": 3.0343434343434345e-07,
      "loss": 0.5356,
      "step": 7000
    },
    {
      "epoch": 5.09,
      "grad_norm": 3.491605520248413,
      "learning_rate": 2.933333333333333e-07,
      "loss": 0.5302,
      "step": 7100
    },
    {
      "epoch": 5.16,
      "grad_norm": 2.702629566192627,
      "learning_rate": 2.8323232323232323e-07,
      "loss": 0.5144,
      "step": 7200
    },
    {
      "epoch": 5.23,
      "grad_norm": 4.217553615570068,
      "learning_rate": 2.731313131313131e-07,
      "loss": 0.5309,
      "step": 7300
    },
    {
      "epoch": 5.3,
      "grad_norm": 2.7974092960357666,
      "learning_rate": 2.63030303030303e-07,
      "loss": 0.519,
      "step": 7400
    },
    {
      "epoch": 5.38,
      "grad_norm": 4.0323686599731445,
      "learning_rate": 2.529292929292929e-07,
      "loss": 0.5513,
      "step": 7500
    },
    {
      "epoch": 5.45,
      "grad_norm": 4.013027191162109,
      "learning_rate": 2.4282828282828284e-07,
      "loss": 0.5331,
      "step": 7600
    },
    {
      "epoch": 5.52,
      "grad_norm": 3.470280647277832,
      "learning_rate": 2.3272727272727273e-07,
      "loss": 0.5252,
      "step": 7700
    },
    {
      "epoch": 5.59,
      "grad_norm": 3.5019471645355225,
      "learning_rate": 2.2262626262626262e-07,
      "loss": 0.5203,
      "step": 7800
    },
    {
      "epoch": 5.66,
      "grad_norm": 3.3975048065185547,
      "learning_rate": 2.1252525252525254e-07,
      "loss": 0.5255,
      "step": 7900
    },
    {
      "epoch": 5.73,
      "grad_norm": 3.226830244064331,
      "learning_rate": 2.0242424242424242e-07,
      "loss": 0.5264,
      "step": 8000
    },
    {
      "epoch": 5.73,
      "eval_loss": 0.7162704467773438,
      "eval_runtime": 209.7748,
      "eval_samples_per_second": 4.767,
      "eval_steps_per_second": 0.596,
      "step": 8000
    }
  ],
  "logging_steps": 100,
  "max_steps": 10000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 8,
  "save_steps": 2000,
  "total_flos": 1.507806463005819e+18,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}