{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 4.977777777777778,
  "eval_steps": 500,
  "global_step": 560,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.044444444444444446,
      "grad_norm": 81.70519960215101,
      "learning_rate": 4.4642857142857147e-07,
      "loss": 2.6592,
      "step": 5
    },
    {
      "epoch": 0.08888888888888889,
      "grad_norm": 65.23941572305874,
      "learning_rate": 8.928571428571429e-07,
      "loss": 2.4229,
      "step": 10
    },
    {
      "epoch": 0.13333333333333333,
      "grad_norm": 23.030445392122235,
      "learning_rate": 1.3392857142857143e-06,
      "loss": 2.3129,
      "step": 15
    },
    {
      "epoch": 0.17777777777777778,
      "grad_norm": 15.10886078640394,
      "learning_rate": 1.7857142857142859e-06,
      "loss": 2.1464,
      "step": 20
    },
    {
      "epoch": 0.2222222222222222,
      "grad_norm": 15.158918433245649,
      "learning_rate": 2.2321428571428573e-06,
      "loss": 1.9653,
      "step": 25
    },
    {
      "epoch": 0.26666666666666666,
      "grad_norm": 10.452753266560764,
      "learning_rate": 2.6785714285714285e-06,
      "loss": 1.9509,
      "step": 30
    },
    {
      "epoch": 0.3111111111111111,
      "grad_norm": 9.403469397121878,
      "learning_rate": 3.125e-06,
      "loss": 1.939,
      "step": 35
    },
    {
      "epoch": 0.35555555555555557,
      "grad_norm": 11.056967394101143,
      "learning_rate": 3.5714285714285718e-06,
      "loss": 1.8875,
      "step": 40
    },
    {
      "epoch": 0.4,
      "grad_norm": 9.328316923554103,
      "learning_rate": 4.017857142857143e-06,
      "loss": 1.8571,
      "step": 45
    },
    {
      "epoch": 0.4444444444444444,
      "grad_norm": 8.513590804020774,
      "learning_rate": 4.464285714285715e-06,
      "loss": 1.8091,
      "step": 50
    },
    {
      "epoch": 0.4888888888888889,
      "grad_norm": 8.195744354512959,
      "learning_rate": 4.910714285714286e-06,
      "loss": 1.7601,
      "step": 55
    },
    {
      "epoch": 0.5333333333333333,
      "grad_norm": 8.438758464497496,
      "learning_rate": 5.357142857142857e-06,
      "loss": 1.7015,
      "step": 60
    },
    {
      "epoch": 0.5777777777777777,
      "grad_norm": 7.858160572018814,
      "learning_rate": 5.8035714285714295e-06,
      "loss": 1.7021,
      "step": 65
    },
    {
      "epoch": 0.6222222222222222,
      "grad_norm": 8.910231103765815,
      "learning_rate": 6.25e-06,
      "loss": 1.6429,
      "step": 70
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 7.384754032842493,
      "learning_rate": 6.696428571428571e-06,
      "loss": 1.6268,
      "step": 75
    },
    {
      "epoch": 0.7111111111111111,
      "grad_norm": 7.692581371932647,
      "learning_rate": 7.1428571428571436e-06,
      "loss": 1.6956,
      "step": 80
    },
    {
      "epoch": 0.7555555555555555,
      "grad_norm": 7.8951737417986445,
      "learning_rate": 7.589285714285714e-06,
      "loss": 1.6613,
      "step": 85
    },
    {
      "epoch": 0.8,
      "grad_norm": 7.678660178384809,
      "learning_rate": 8.035714285714286e-06,
      "loss": 1.6367,
      "step": 90
    },
    {
      "epoch": 0.8444444444444444,
      "grad_norm": 7.9498820978188895,
      "learning_rate": 8.482142857142858e-06,
      "loss": 1.6568,
      "step": 95
    },
    {
      "epoch": 0.8888888888888888,
      "grad_norm": 7.48336511323642,
      "learning_rate": 8.92857142857143e-06,
      "loss": 1.6309,
      "step": 100
    },
    {
      "epoch": 0.9333333333333333,
      "grad_norm": 7.481491041447778,
      "learning_rate": 9.375000000000001e-06,
      "loss": 1.6108,
      "step": 105
    },
    {
      "epoch": 0.9777777777777777,
      "grad_norm": 7.720239899968915,
      "learning_rate": 9.821428571428573e-06,
      "loss": 1.5646,
      "step": 110
    },
    {
      "epoch": 1.0222222222222221,
      "grad_norm": 7.0981539773430615,
      "learning_rate": 9.933035714285715e-06,
      "loss": 1.4956,
      "step": 115
    },
    {
      "epoch": 1.0666666666666667,
      "grad_norm": 7.510833590279441,
      "learning_rate": 9.821428571428573e-06,
      "loss": 1.3929,
      "step": 120
    },
    {
      "epoch": 1.1111111111111112,
      "grad_norm": 7.248134123936718,
      "learning_rate": 9.70982142857143e-06,
      "loss": 1.4206,
      "step": 125
    },
    {
      "epoch": 1.1555555555555554,
      "grad_norm": 8.174657015727915,
      "learning_rate": 9.598214285714287e-06,
      "loss": 1.3426,
      "step": 130
    },
    {
      "epoch": 1.2,
      "grad_norm": 7.352102037038091,
      "learning_rate": 9.486607142857144e-06,
      "loss": 1.3171,
      "step": 135
    },
    {
      "epoch": 1.2444444444444445,
      "grad_norm": 7.5635904551535225,
      "learning_rate": 9.375000000000001e-06,
      "loss": 1.3779,
      "step": 140
    },
    {
      "epoch": 1.2888888888888888,
      "grad_norm": 9.033956503305323,
      "learning_rate": 9.263392857142858e-06,
      "loss": 1.345,
      "step": 145
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 8.224641097275477,
      "learning_rate": 9.151785714285715e-06,
      "loss": 1.4234,
      "step": 150
    },
    {
      "epoch": 1.3777777777777778,
      "grad_norm": 8.830468339631146,
      "learning_rate": 9.040178571428572e-06,
      "loss": 1.2946,
      "step": 155
    },
    {
      "epoch": 1.4222222222222223,
      "grad_norm": 8.443856362876113,
      "learning_rate": 8.92857142857143e-06,
      "loss": 1.3646,
      "step": 160
    },
    {
      "epoch": 1.4666666666666668,
      "grad_norm": 8.043363545508665,
      "learning_rate": 8.816964285714286e-06,
      "loss": 1.3936,
      "step": 165
    },
    {
      "epoch": 1.511111111111111,
      "grad_norm": 8.616005497888816,
      "learning_rate": 8.705357142857143e-06,
      "loss": 1.3557,
      "step": 170
    },
    {
      "epoch": 1.5555555555555556,
      "grad_norm": 8.255699646794918,
      "learning_rate": 8.59375e-06,
      "loss": 1.3142,
      "step": 175
    },
    {
      "epoch": 1.6,
      "grad_norm": 8.215553491915397,
      "learning_rate": 8.482142857142858e-06,
      "loss": 1.3464,
      "step": 180
    },
    {
      "epoch": 1.6444444444444444,
      "grad_norm": 7.885356619483398,
      "learning_rate": 8.370535714285715e-06,
      "loss": 1.3553,
      "step": 185
    },
    {
      "epoch": 1.6888888888888889,
      "grad_norm": 8.12112906938901,
      "learning_rate": 8.258928571428572e-06,
      "loss": 1.3527,
      "step": 190
    },
    {
      "epoch": 1.7333333333333334,
      "grad_norm": 8.350294666975472,
      "learning_rate": 8.147321428571429e-06,
      "loss": 1.2518,
      "step": 195
    },
    {
      "epoch": 1.7777777777777777,
      "grad_norm": 7.905524590675295,
      "learning_rate": 8.035714285714286e-06,
      "loss": 1.3491,
      "step": 200
    },
    {
      "epoch": 1.8222222222222222,
      "grad_norm": 7.871873965564558,
      "learning_rate": 7.924107142857143e-06,
      "loss": 1.382,
      "step": 205
    },
    {
      "epoch": 1.8666666666666667,
      "grad_norm": 8.12507505022624,
      "learning_rate": 7.8125e-06,
      "loss": 1.3336,
      "step": 210
    },
    {
      "epoch": 1.911111111111111,
      "grad_norm": 8.058616824355166,
      "learning_rate": 7.700892857142857e-06,
      "loss": 1.3512,
      "step": 215
    },
    {
      "epoch": 1.9555555555555557,
      "grad_norm": 7.59358946949836,
      "learning_rate": 7.589285714285714e-06,
      "loss": 1.3093,
      "step": 220
    },
    {
      "epoch": 2.0,
      "grad_norm": 8.88478802727675,
      "learning_rate": 7.4776785714285714e-06,
      "loss": 1.3261,
      "step": 225
    },
    {
      "epoch": 2.0444444444444443,
      "grad_norm": 7.6072417424325,
      "learning_rate": 7.366071428571429e-06,
      "loss": 0.9253,
      "step": 230
    },
    {
      "epoch": 2.088888888888889,
      "grad_norm": 8.772960686537047,
      "learning_rate": 7.2544642857142865e-06,
      "loss": 0.8591,
      "step": 235
    },
    {
      "epoch": 2.1333333333333333,
      "grad_norm": 9.452930658317475,
      "learning_rate": 7.1428571428571436e-06,
      "loss": 0.8771,
      "step": 240
    },
    {
      "epoch": 2.1777777777777776,
      "grad_norm": 9.3220762052475,
      "learning_rate": 7.031250000000001e-06,
      "loss": 0.8593,
      "step": 245
    },
    {
      "epoch": 2.2222222222222223,
      "grad_norm": 8.32864095740941,
      "learning_rate": 6.919642857142858e-06,
      "loss": 0.9117,
      "step": 250
    },
    {
      "epoch": 2.2666666666666666,
      "grad_norm": 8.792144175107566,
      "learning_rate": 6.808035714285715e-06,
      "loss": 0.8772,
      "step": 255
    },
    {
      "epoch": 2.311111111111111,
      "grad_norm": 9.98800755135626,
      "learning_rate": 6.696428571428571e-06,
      "loss": 0.858,
      "step": 260
    },
    {
      "epoch": 2.3555555555555556,
      "grad_norm": 9.018987249678533,
      "learning_rate": 6.58482142857143e-06,
      "loss": 0.8708,
      "step": 265
    },
    {
      "epoch": 2.4,
      "grad_norm": 8.547777859748455,
      "learning_rate": 6.473214285714287e-06,
      "loss": 0.872,
      "step": 270
    },
    {
      "epoch": 2.4444444444444446,
      "grad_norm": 8.771861798383114,
      "learning_rate": 6.361607142857143e-06,
      "loss": 0.8603,
      "step": 275
    },
    {
      "epoch": 2.488888888888889,
      "grad_norm": 10.206792686412927,
      "learning_rate": 6.25e-06,
      "loss": 0.9321,
      "step": 280
    },
    {
      "epoch": 2.533333333333333,
      "grad_norm": 8.15300521664367,
      "learning_rate": 6.138392857142857e-06,
      "loss": 0.7993,
      "step": 285
    },
    {
      "epoch": 2.5777777777777775,
      "grad_norm": 8.916029898752333,
      "learning_rate": 6.0267857142857145e-06,
      "loss": 0.9229,
      "step": 290
    },
    {
      "epoch": 2.6222222222222222,
      "grad_norm": 9.688980370039602,
      "learning_rate": 5.9151785714285716e-06,
      "loss": 0.8954,
      "step": 295
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 8.749780229882957,
      "learning_rate": 5.8035714285714295e-06,
      "loss": 0.8473,
      "step": 300
    },
    {
      "epoch": 2.7111111111111112,
      "grad_norm": 9.261115491971523,
      "learning_rate": 5.691964285714287e-06,
      "loss": 0.8772,
      "step": 305
    },
    {
      "epoch": 2.7555555555555555,
      "grad_norm": 9.010412339583716,
      "learning_rate": 5.580357142857144e-06,
      "loss": 0.8786,
      "step": 310
    },
    {
      "epoch": 2.8,
      "grad_norm": 9.17920752101898,
      "learning_rate": 5.468750000000001e-06,
      "loss": 0.851,
      "step": 315
    },
    {
      "epoch": 2.8444444444444446,
      "grad_norm": 8.254891628439802,
      "learning_rate": 5.357142857142857e-06,
      "loss": 0.8981,
      "step": 320
    },
    {
      "epoch": 2.888888888888889,
      "grad_norm": 8.642637449945827,
      "learning_rate": 5.245535714285714e-06,
      "loss": 0.8717,
      "step": 325
    },
    {
      "epoch": 2.9333333333333336,
      "grad_norm": 8.815270770583437,
      "learning_rate": 5.133928571428571e-06,
      "loss": 0.8865,
      "step": 330
    },
    {
      "epoch": 2.977777777777778,
      "grad_norm": 9.062651499166742,
      "learning_rate": 5.022321428571429e-06,
      "loss": 0.8776,
      "step": 335
    },
    {
      "epoch": 3.022222222222222,
      "grad_norm": 7.480764915704419,
      "learning_rate": 4.910714285714286e-06,
      "loss": 0.7016,
      "step": 340
    },
    {
      "epoch": 3.066666666666667,
      "grad_norm": 8.739944237211384,
      "learning_rate": 4.799107142857143e-06,
      "loss": 0.5105,
      "step": 345
    },
    {
      "epoch": 3.111111111111111,
      "grad_norm": 11.051330955646318,
      "learning_rate": 4.6875000000000004e-06,
      "loss": 0.5077,
      "step": 350
    },
    {
      "epoch": 3.1555555555555554,
      "grad_norm": 8.402494171987351,
      "learning_rate": 4.5758928571428575e-06,
      "loss": 0.4504,
      "step": 355
    },
    {
      "epoch": 3.2,
      "grad_norm": 9.554895055203646,
      "learning_rate": 4.464285714285715e-06,
      "loss": 0.4525,
      "step": 360
    },
    {
      "epoch": 3.2444444444444445,
      "grad_norm": 9.849412176242506,
      "learning_rate": 4.352678571428572e-06,
      "loss": 0.438,
      "step": 365
    },
    {
      "epoch": 3.2888888888888888,
      "grad_norm": 9.879820678510647,
      "learning_rate": 4.241071428571429e-06,
      "loss": 0.4634,
      "step": 370
    },
    {
      "epoch": 3.3333333333333335,
      "grad_norm": 9.560743040510404,
      "learning_rate": 4.129464285714286e-06,
      "loss": 0.4804,
      "step": 375
    },
    {
      "epoch": 3.3777777777777778,
      "grad_norm": 9.122163123715797,
      "learning_rate": 4.017857142857143e-06,
      "loss": 0.4346,
      "step": 380
    },
    {
      "epoch": 3.422222222222222,
      "grad_norm": 9.57226826749949,
      "learning_rate": 3.90625e-06,
      "loss": 0.4483,
      "step": 385
    },
    {
      "epoch": 3.466666666666667,
      "grad_norm": 10.306235127578079,
      "learning_rate": 3.794642857142857e-06,
      "loss": 0.4713,
      "step": 390
    },
    {
      "epoch": 3.511111111111111,
      "grad_norm": 9.64313411436284,
      "learning_rate": 3.6830357142857147e-06,
      "loss": 0.4359,
      "step": 395
    },
    {
      "epoch": 3.5555555555555554,
      "grad_norm": 10.420285208662374,
      "learning_rate": 3.5714285714285718e-06,
      "loss": 0.451,
      "step": 400
    },
    {
      "epoch": 3.6,
      "grad_norm": 9.87846382809359,
      "learning_rate": 3.459821428571429e-06,
      "loss": 0.4763,
      "step": 405
    },
    {
      "epoch": 3.6444444444444444,
      "grad_norm": 9.876467528663943,
      "learning_rate": 3.3482142857142855e-06,
      "loss": 0.4537,
      "step": 410
    },
    {
      "epoch": 3.688888888888889,
      "grad_norm": 8.675841435317817,
      "learning_rate": 3.2366071428571435e-06,
      "loss": 0.437,
      "step": 415
    },
    {
      "epoch": 3.7333333333333334,
      "grad_norm": 9.325507180126642,
      "learning_rate": 3.125e-06,
      "loss": 0.4497,
      "step": 420
    },
    {
      "epoch": 3.7777777777777777,
      "grad_norm": 10.153198500901361,
      "learning_rate": 3.0133928571428572e-06,
      "loss": 0.4441,
      "step": 425
    },
    {
      "epoch": 3.822222222222222,
      "grad_norm": 10.645689697884485,
      "learning_rate": 2.9017857142857148e-06,
      "loss": 0.4706,
      "step": 430
    },
    {
      "epoch": 3.8666666666666667,
      "grad_norm": 9.668794366955733,
      "learning_rate": 2.790178571428572e-06,
      "loss": 0.4576,
      "step": 435
    },
    {
      "epoch": 3.911111111111111,
      "grad_norm": 8.850840067616335,
      "learning_rate": 2.6785714285714285e-06,
      "loss": 0.4327,
      "step": 440
    },
    {
      "epoch": 3.9555555555555557,
      "grad_norm": 9.06809450411661,
      "learning_rate": 2.5669642857142856e-06,
      "loss": 0.489,
      "step": 445
    },
    {
      "epoch": 4.0,
      "grad_norm": 9.875435470206298,
      "learning_rate": 2.455357142857143e-06,
      "loss": 0.4454,
      "step": 450
    },
    {
      "epoch": 4.044444444444444,
      "grad_norm": 14.31106135217529,
      "learning_rate": 2.3437500000000002e-06,
      "loss": 0.2196,
      "step": 455
    },
    {
      "epoch": 4.088888888888889,
      "grad_norm": 8.609859535415907,
      "learning_rate": 2.2321428571428573e-06,
      "loss": 0.2121,
      "step": 460
    },
    {
      "epoch": 4.133333333333334,
      "grad_norm": 13.280395274888972,
      "learning_rate": 2.1205357142857144e-06,
      "loss": 0.2035,
      "step": 465
    },
    {
      "epoch": 4.177777777777778,
      "grad_norm": 8.75211366566662,
      "learning_rate": 2.0089285714285715e-06,
      "loss": 0.1911,
      "step": 470
    },
    {
      "epoch": 4.222222222222222,
      "grad_norm": 7.090600469273079,
      "learning_rate": 1.8973214285714286e-06,
      "loss": 0.1923,
      "step": 475
    },
    {
      "epoch": 4.266666666666667,
      "grad_norm": 6.939677578044171,
      "learning_rate": 1.7857142857142859e-06,
      "loss": 0.1895,
      "step": 480
    },
    {
      "epoch": 4.311111111111111,
      "grad_norm": 7.862142692655875,
      "learning_rate": 1.6741071428571428e-06,
      "loss": 0.1977,
      "step": 485
    },
    {
      "epoch": 4.355555555555555,
      "grad_norm": 7.583820440823111,
      "learning_rate": 1.5625e-06,
      "loss": 0.1751,
      "step": 490
    },
    {
      "epoch": 4.4,
      "grad_norm": 8.947913314563216,
      "learning_rate": 1.4508928571428574e-06,
      "loss": 0.1838,
      "step": 495
    },
    {
      "epoch": 4.444444444444445,
      "grad_norm": 7.658722193186348,
      "learning_rate": 1.3392857142857143e-06,
      "loss": 0.2006,
      "step": 500
    },
    {
      "epoch": 4.488888888888889,
      "grad_norm": 8.207622619508404,
      "learning_rate": 1.2276785714285716e-06,
      "loss": 0.186,
      "step": 505
    },
    {
      "epoch": 4.533333333333333,
      "grad_norm": 8.120063690560118,
      "learning_rate": 1.1160714285714287e-06,
      "loss": 0.2101,
      "step": 510
    },
    {
      "epoch": 4.5777777777777775,
      "grad_norm": 7.850134563130463,
      "learning_rate": 1.0044642857142857e-06,
      "loss": 0.1954,
      "step": 515
    },
    {
      "epoch": 4.622222222222222,
      "grad_norm": 10.423756147314887,
      "learning_rate": 8.928571428571429e-07,
      "loss": 0.1877,
      "step": 520
    },
    {
      "epoch": 4.666666666666667,
      "grad_norm": 7.945666215863678,
      "learning_rate": 7.8125e-07,
      "loss": 0.1988,
      "step": 525
    },
    {
      "epoch": 4.711111111111111,
      "grad_norm": 8.106328872459784,
      "learning_rate": 6.696428571428571e-07,
      "loss": 0.2015,
      "step": 530
    },
    {
      "epoch": 4.7555555555555555,
      "grad_norm": 8.767972813276023,
      "learning_rate": 5.580357142857143e-07,
      "loss": 0.188,
      "step": 535
    },
    {
      "epoch": 4.8,
      "grad_norm": 9.369887507108524,
      "learning_rate": 4.4642857142857147e-07,
      "loss": 0.1859,
      "step": 540
    },
    {
      "epoch": 4.844444444444444,
      "grad_norm": 8.892651474952375,
      "learning_rate": 3.3482142857142856e-07,
      "loss": 0.1826,
      "step": 545
    },
    {
      "epoch": 4.888888888888889,
      "grad_norm": 7.261334614879305,
      "learning_rate": 2.2321428571428574e-07,
      "loss": 0.1852,
      "step": 550
    },
    {
      "epoch": 4.933333333333334,
      "grad_norm": 9.36687860609368,
      "learning_rate": 1.1160714285714287e-07,
      "loss": 0.1889,
      "step": 555
    },
    {
      "epoch": 4.977777777777778,
      "grad_norm": 10.485619564355174,
      "learning_rate": 0.0,
      "loss": 0.1811,
      "step": 560
    },
    {
      "epoch": 4.977777777777778,
      "step": 560,
      "total_flos": 770201321472.0,
      "train_loss": 0.950020797018494,
      "train_runtime": 694.3412,
      "train_samples_per_second": 25.924,
      "train_steps_per_second": 0.807
    }
  ],
  "logging_steps": 5,
  "max_steps": 560,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": false,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 770201321472.0,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}