{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 144,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.020833333333333332,
      "grad_norm": 5.9375,
      "learning_rate": 0.0,
      "loss": 1.457,
      "memory/device_reserved (GiB)": 68.17,
      "memory/max_active (GiB)": 62.24,
      "memory/max_allocated (GiB)": 62.24,
      "step": 1,
      "tokens_per_second_per_gpu": 4754.85,
      "total_tokens": 193735
    },
    {
      "epoch": 0.041666666666666664,
      "grad_norm": 6.40625,
      "learning_rate": 1.4285714285714286e-06,
      "loss": 1.5117,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 2,
      "tokens_per_second_per_gpu": 8987.92,
      "total_tokens": 377645
    },
    {
      "epoch": 0.0625,
      "grad_norm": 6.34375,
      "learning_rate": 2.8571428571428573e-06,
      "loss": 1.5352,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 3,
      "tokens_per_second_per_gpu": 10584.88,
      "total_tokens": 572231
    },
    {
      "epoch": 0.08333333333333333,
      "grad_norm": 5.75,
      "learning_rate": 4.2857142857142855e-06,
      "loss": 1.4023,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 4,
      "tokens_per_second_per_gpu": 10641.72,
      "total_tokens": 771980
    },
    {
      "epoch": 0.10416666666666667,
      "grad_norm": 5.8125,
      "learning_rate": 5.7142857142857145e-06,
      "loss": 1.5234,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 5,
      "tokens_per_second_per_gpu": 10136.19,
      "total_tokens": 960629
    },
    {
      "epoch": 0.125,
      "grad_norm": 5.5,
      "learning_rate": 7.1428571428571436e-06,
      "loss": 1.4961,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 6,
      "tokens_per_second_per_gpu": 10066.87,
      "total_tokens": 1145765
    },
    {
      "epoch": 0.14583333333333334,
      "grad_norm": 4.59375,
      "learning_rate": 8.571428571428571e-06,
      "loss": 1.3828,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 7,
      "tokens_per_second_per_gpu": 10793.17,
      "total_tokens": 1339007
    },
    {
      "epoch": 0.16666666666666666,
      "grad_norm": 3.53125,
      "learning_rate": 1e-05,
      "loss": 1.3359,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 8,
      "tokens_per_second_per_gpu": 10042.61,
      "total_tokens": 1533294
    },
    {
      "epoch": 0.1875,
      "grad_norm": 2.625,
      "learning_rate": 1.1428571428571429e-05,
      "loss": 1.3203,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 9,
      "tokens_per_second_per_gpu": 9880.69,
      "total_tokens": 1715421
    },
    {
      "epoch": 0.20833333333333334,
      "grad_norm": 1.859375,
      "learning_rate": 1.2857142857142859e-05,
      "loss": 1.2695,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 10,
      "tokens_per_second_per_gpu": 10012.45,
      "total_tokens": 1899665
    },
    {
      "epoch": 0.22916666666666666,
      "grad_norm": 1.359375,
      "learning_rate": 1.4285714285714287e-05,
      "loss": 1.2539,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 11,
      "tokens_per_second_per_gpu": 10498.2,
      "total_tokens": 2090232
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.359375,
      "learning_rate": 1.5714285714285715e-05,
      "loss": 1.2383,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 12,
      "tokens_per_second_per_gpu": 10504.24,
      "total_tokens": 2277678
    },
    {
      "epoch": 0.2708333333333333,
      "grad_norm": 1.3828125,
      "learning_rate": 1.7142857142857142e-05,
      "loss": 1.2031,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 13,
      "tokens_per_second_per_gpu": 11150.71,
      "total_tokens": 2475912
    },
    {
      "epoch": 0.2916666666666667,
      "grad_norm": 1.3125,
      "learning_rate": 1.8571428571428575e-05,
      "loss": 1.1172,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 14,
      "tokens_per_second_per_gpu": 10549.83,
      "total_tokens": 2677381
    },
    {
      "epoch": 0.3125,
      "grad_norm": 1.3671875,
      "learning_rate": 2e-05,
      "loss": 1.1602,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 15,
      "tokens_per_second_per_gpu": 10901.98,
      "total_tokens": 2876565
    },
    {
      "epoch": 0.3333333333333333,
      "grad_norm": 1.203125,
      "learning_rate": 1.9997080140801932e-05,
      "loss": 1.1445,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 16,
      "tokens_per_second_per_gpu": 10121.73,
      "total_tokens": 3065392
    },
    {
      "epoch": 0.3541666666666667,
      "grad_norm": 1.109375,
      "learning_rate": 1.998832226832327e-05,
      "loss": 1.1016,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 17,
      "tokens_per_second_per_gpu": 10594.37,
      "total_tokens": 3260102
    },
    {
      "epoch": 0.375,
      "grad_norm": 0.93359375,
      "learning_rate": 1.9973731496914914e-05,
      "loss": 1.168,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 18,
      "tokens_per_second_per_gpu": 9736.14,
      "total_tokens": 3442701
    },
    {
      "epoch": 0.3958333333333333,
      "grad_norm": 0.79296875,
      "learning_rate": 1.995331634717649e-05,
      "loss": 1.082,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 19,
      "tokens_per_second_per_gpu": 10128.02,
      "total_tokens": 3638349
    },
    {
      "epoch": 0.4166666666666667,
      "grad_norm": 0.75,
      "learning_rate": 1.992708874098054e-05,
      "loss": 1.1055,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 20,
      "tokens_per_second_per_gpu": 10243.56,
      "total_tokens": 3826742
    },
    {
      "epoch": 0.4375,
      "grad_norm": 0.69140625,
      "learning_rate": 1.9895063994510512e-05,
      "loss": 1.1016,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 21,
      "tokens_per_second_per_gpu": 9967.12,
      "total_tokens": 4014944
    },
    {
      "epoch": 0.4583333333333333,
      "grad_norm": 0.6796875,
      "learning_rate": 1.985726080931651e-05,
      "loss": 1.1211,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 22,
      "tokens_per_second_per_gpu": 10518.66,
      "total_tokens": 4208308
    },
    {
      "epoch": 0.4791666666666667,
      "grad_norm": 0.67578125,
      "learning_rate": 1.9813701261394136e-05,
      "loss": 1.1523,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 23,
      "tokens_per_second_per_gpu": 10033.96,
      "total_tokens": 4394123
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.64453125,
      "learning_rate": 1.9764410788292724e-05,
      "loss": 1.0898,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 24,
      "tokens_per_second_per_gpu": 10669.49,
      "total_tokens": 4588895
    },
    {
      "epoch": 0.5208333333333334,
      "grad_norm": 0.62109375,
      "learning_rate": 1.9709418174260523e-05,
      "loss": 1.1172,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 25,
      "tokens_per_second_per_gpu": 10276.83,
      "total_tokens": 4781600
    },
    {
      "epoch": 0.5416666666666666,
      "grad_norm": 0.6328125,
      "learning_rate": 1.9648755533435517e-05,
      "loss": 1.1094,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 26,
      "tokens_per_second_per_gpu": 9934.02,
      "total_tokens": 4962927
    },
    {
      "epoch": 0.5625,
      "grad_norm": 0.546875,
      "learning_rate": 1.9582458291091664e-05,
      "loss": 1.0898,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 27,
      "tokens_per_second_per_gpu": 10213.04,
      "total_tokens": 5154156
    },
    {
      "epoch": 0.5833333333333334,
      "grad_norm": 0.5390625,
      "learning_rate": 1.9510565162951538e-05,
      "loss": 1.0625,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 28,
      "tokens_per_second_per_gpu": 10650.84,
      "total_tokens": 5346313
    },
    {
      "epoch": 0.6041666666666666,
      "grad_norm": 0.52734375,
      "learning_rate": 1.9433118132577432e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 29,
      "tokens_per_second_per_gpu": 10206.96,
      "total_tokens": 5537564
    },
    {
      "epoch": 0.625,
      "grad_norm": 0.52734375,
      "learning_rate": 1.9350162426854152e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 30,
      "tokens_per_second_per_gpu": 9831.26,
      "total_tokens": 5720388
    },
    {
      "epoch": 0.6458333333333334,
      "grad_norm": 0.54296875,
      "learning_rate": 1.9261746489577767e-05,
      "loss": 1.0781,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 31,
      "tokens_per_second_per_gpu": 10701.1,
      "total_tokens": 5906962
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 0.52734375,
      "learning_rate": 1.9167921953165827e-05,
      "loss": 1.0938,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 32,
      "tokens_per_second_per_gpu": 10732.53,
      "total_tokens": 6104218
    },
    {
      "epoch": 0.6875,
      "grad_norm": 0.52734375,
      "learning_rate": 1.9068743608505454e-05,
      "loss": 1.1133,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 33,
      "tokens_per_second_per_gpu": 10156.96,
      "total_tokens": 6291571
    },
    {
      "epoch": 0.7083333333333334,
      "grad_norm": 0.5,
      "learning_rate": 1.896426937295704e-05,
      "loss": 1.0781,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 34,
      "tokens_per_second_per_gpu": 10651.58,
      "total_tokens": 6476350
    },
    {
      "epoch": 0.7291666666666666,
      "grad_norm": 0.5390625,
      "learning_rate": 1.8854560256532098e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 35,
      "tokens_per_second_per_gpu": 10027.99,
      "total_tokens": 6660433
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.49609375,
      "learning_rate": 1.873968032626518e-05,
      "loss": 1.0625,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 36,
      "tokens_per_second_per_gpu": 10538.25,
      "total_tokens": 6853661
    },
    {
      "epoch": 0.7708333333333334,
      "grad_norm": 0.470703125,
      "learning_rate": 1.8619696668800494e-05,
      "loss": 1.1055,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 37,
      "tokens_per_second_per_gpu": 10499.61,
      "total_tokens": 7041887
    },
    {
      "epoch": 0.7916666666666666,
      "grad_norm": 0.4609375,
      "learning_rate": 1.8494679351215212e-05,
      "loss": 1.0117,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 38,
      "tokens_per_second_per_gpu": 10892.76,
      "total_tokens": 7242129
    },
    {
      "epoch": 0.8125,
      "grad_norm": 0.4765625,
      "learning_rate": 1.8364701380102267e-05,
      "loss": 1.0391,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 39,
      "tokens_per_second_per_gpu": 10826.51,
      "total_tokens": 7442101
    },
    {
      "epoch": 0.8333333333333334,
      "grad_norm": 0.494140625,
      "learning_rate": 1.8229838658936566e-05,
      "loss": 0.998,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 40,
      "tokens_per_second_per_gpu": 9529.96,
      "total_tokens": 7637450
    },
    {
      "epoch": 0.8541666666666666,
      "grad_norm": 0.458984375,
      "learning_rate": 1.8090169943749477e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 41,
      "tokens_per_second_per_gpu": 10530.91,
      "total_tokens": 7830567
    },
    {
      "epoch": 0.875,
      "grad_norm": 0.470703125,
      "learning_rate": 1.7945776797137544e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 42,
      "tokens_per_second_per_gpu": 10621.21,
      "total_tokens": 8025656
    },
    {
      "epoch": 0.8958333333333334,
      "grad_norm": 0.466796875,
      "learning_rate": 1.7796743540632226e-05,
      "loss": 1.0254,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 43,
      "tokens_per_second_per_gpu": 10096.46,
      "total_tokens": 8212765
    },
    {
      "epoch": 0.9166666666666666,
      "grad_norm": 0.447265625,
      "learning_rate": 1.7643157205458483e-05,
      "loss": 1.0332,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 44,
      "tokens_per_second_per_gpu": 10109.75,
      "total_tokens": 8408896
    },
    {
      "epoch": 0.9375,
      "grad_norm": 0.439453125,
      "learning_rate": 1.7485107481711014e-05,
      "loss": 1.002,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 45,
      "tokens_per_second_per_gpu": 10350.81,
      "total_tokens": 8608276
    },
    {
      "epoch": 0.9583333333333334,
      "grad_norm": 0.53125,
      "learning_rate": 1.7322686665977738e-05,
      "loss": 1.1094,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 46,
      "tokens_per_second_per_gpu": 10131.42,
      "total_tokens": 8799852
    },
    {
      "epoch": 0.9791666666666666,
      "grad_norm": 0.435546875,
      "learning_rate": 1.715598960744121e-05,
      "loss": 0.9707,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 47,
      "tokens_per_second_per_gpu": 9848.57,
      "total_tokens": 8991281
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.466796875,
      "learning_rate": 1.6985113652489374e-05,
      "loss": 1.0391,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 48,
      "tokens_per_second_per_gpu": 10189.26,
      "total_tokens": 9182092
    },
    {
      "epoch": 1.0208333333333333,
      "grad_norm": 0.44921875,
      "learning_rate": 1.6810158587867973e-05,
      "loss": 1.0117,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 49,
      "tokens_per_second_per_gpu": 10109.46,
      "total_tokens": 9375827
    },
    {
      "epoch": 1.0416666666666667,
      "grad_norm": 0.451171875,
      "learning_rate": 1.6631226582407954e-05,
      "loss": 1.0312,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 50,
      "tokens_per_second_per_gpu": 9167.36,
      "total_tokens": 9559737
    },
    {
      "epoch": 1.0625,
      "grad_norm": 0.458984375,
      "learning_rate": 1.6448422127361707e-05,
      "loss": 1.0703,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 51,
      "tokens_per_second_per_gpu": 10531.34,
      "total_tokens": 9754323
    },
    {
      "epoch": 1.0833333333333333,
      "grad_norm": 0.42578125,
      "learning_rate": 1.626185197538314e-05,
      "loss": 0.9688,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 52,
      "tokens_per_second_per_gpu": 10570.62,
      "total_tokens": 9954072
    },
    {
      "epoch": 1.1041666666666667,
      "grad_norm": 0.4609375,
      "learning_rate": 1.6071625078187113e-05,
      "loss": 1.0859,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 53,
      "tokens_per_second_per_gpu": 10132.93,
      "total_tokens": 10142721
    },
    {
      "epoch": 1.125,
      "grad_norm": 0.4609375,
      "learning_rate": 1.5877852522924733e-05,
      "loss": 1.0703,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 54,
      "tokens_per_second_per_gpu": 10263.22,
      "total_tokens": 10327857
    },
    {
      "epoch": 1.1458333333333333,
      "grad_norm": 0.44921875,
      "learning_rate": 1.568064746731156e-05,
      "loss": 1.0176,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 55,
      "tokens_per_second_per_gpu": 10786.39,
      "total_tokens": 10521099
    },
    {
      "epoch": 1.1666666666666667,
      "grad_norm": 0.443359375,
      "learning_rate": 1.5480125073546705e-05,
      "loss": 1.0352,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 56,
      "tokens_per_second_per_gpu": 10023.87,
      "total_tokens": 10715386
    },
    {
      "epoch": 1.1875,
      "grad_norm": 0.45703125,
      "learning_rate": 1.527640244106133e-05,
      "loss": 1.043,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 57,
      "tokens_per_second_per_gpu": 9848.52,
      "total_tokens": 10897513
    },
    {
      "epoch": 1.2083333333333333,
      "grad_norm": 0.482421875,
      "learning_rate": 1.5069598538135905e-05,
      "loss": 1.0352,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 58,
      "tokens_per_second_per_gpu": 10004.24,
      "total_tokens": 11081757
    },
    {
      "epoch": 1.2291666666666667,
      "grad_norm": 0.451171875,
      "learning_rate": 1.485983413242606e-05,
      "loss": 1.0195,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 59,
      "tokens_per_second_per_gpu": 10483.32,
      "total_tokens": 11272324
    },
    {
      "epoch": 1.25,
      "grad_norm": 0.4609375,
      "learning_rate": 1.4647231720437687e-05,
      "loss": 1.0293,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 60,
      "tokens_per_second_per_gpu": 10478.12,
      "total_tokens": 11459770
    },
    {
      "epoch": 1.2708333333333333,
      "grad_norm": 0.451171875,
      "learning_rate": 1.4431915455992416e-05,
      "loss": 1.0039,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 61,
      "tokens_per_second_per_gpu": 11157.45,
      "total_tokens": 11658004
    },
    {
      "epoch": 1.2916666666666667,
      "grad_norm": 0.4296875,
      "learning_rate": 1.4214011077725293e-05,
      "loss": 0.9414,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 62,
      "tokens_per_second_per_gpu": 10464.97,
      "total_tokens": 11859473
    },
    {
      "epoch": 1.3125,
      "grad_norm": 0.45703125,
      "learning_rate": 1.3993645835656955e-05,
      "loss": 0.9844,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 63,
      "tokens_per_second_per_gpu": 10802.11,
      "total_tokens": 12058657
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 0.435546875,
      "learning_rate": 1.3770948416883205e-05,
      "loss": 0.9746,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 64,
      "tokens_per_second_per_gpu": 10169.34,
      "total_tokens": 12247484
    },
    {
      "epoch": 1.3541666666666667,
      "grad_norm": 0.470703125,
      "learning_rate": 1.3546048870425356e-05,
      "loss": 0.9434,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 65,
      "tokens_per_second_per_gpu": 10566.69,
      "total_tokens": 12442194
    },
    {
      "epoch": 1.375,
      "grad_norm": 0.4453125,
      "learning_rate": 1.3319078531285286e-05,
      "loss": 1.002,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 66,
      "tokens_per_second_per_gpu": 9699.95,
      "total_tokens": 12624793
    },
    {
      "epoch": 1.3958333333333333,
      "grad_norm": 0.4375,
      "learning_rate": 1.3090169943749475e-05,
      "loss": 0.9336,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 67,
      "tokens_per_second_per_gpu": 10187.26,
      "total_tokens": 12820441
    },
    {
      "epoch": 1.4166666666666667,
      "grad_norm": 0.458984375,
      "learning_rate": 1.2859456783986892e-05,
      "loss": 0.9629,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 68,
      "tokens_per_second_per_gpu": 10254.48,
      "total_tokens": 13008834
    },
    {
      "epoch": 1.4375,
      "grad_norm": 0.447265625,
      "learning_rate": 1.262707378198587e-05,
      "loss": 0.9648,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 69,
      "tokens_per_second_per_gpu": 9940.27,
      "total_tokens": 13197036
    },
    {
      "epoch": 1.4583333333333333,
      "grad_norm": 0.4609375,
      "learning_rate": 1.2393156642875579e-05,
      "loss": 0.9902,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 70,
      "tokens_per_second_per_gpu": 10499.3,
      "total_tokens": 13390400
    },
    {
      "epoch": 1.4791666666666667,
      "grad_norm": 0.451171875,
      "learning_rate": 1.2157841967678064e-05,
      "loss": 1.0215,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 71,
      "tokens_per_second_per_gpu": 10092.49,
      "total_tokens": 13576215
    },
    {
      "epoch": 1.5,
      "grad_norm": 0.427734375,
      "learning_rate": 1.1921267173537085e-05,
      "loss": 0.9707,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 72,
      "tokens_per_second_per_gpu": 10734.78,
      "total_tokens": 13770987
    },
    {
      "epoch": 1.5208333333333335,
      "grad_norm": 0.451171875,
      "learning_rate": 1.1683570413470384e-05,
      "loss": 1.0039,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 73,
      "tokens_per_second_per_gpu": 10374.13,
      "total_tokens": 13963692
    },
    {
      "epoch": 1.5416666666666665,
      "grad_norm": 0.451171875,
      "learning_rate": 1.1444890495692214e-05,
      "loss": 0.9863,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 74,
      "tokens_per_second_per_gpu": 9925.41,
      "total_tokens": 14145019
    },
    {
      "epoch": 1.5625,
      "grad_norm": 0.4375,
      "learning_rate": 1.1205366802553231e-05,
      "loss": 0.9844,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 75,
      "tokens_per_second_per_gpu": 10193.32,
      "total_tokens": 14336248
    },
    {
      "epoch": 1.5833333333333335,
      "grad_norm": 0.431640625,
      "learning_rate": 1.0965139209145153e-05,
      "loss": 0.957,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 76,
      "tokens_per_second_per_gpu": 10648.45,
      "total_tokens": 14528405
    },
    {
      "epoch": 1.6041666666666665,
      "grad_norm": 0.431640625,
      "learning_rate": 1.0724348001617626e-05,
      "loss": 0.9902,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 77,
      "tokens_per_second_per_gpu": 10212.71,
      "total_tokens": 14719656
    },
    {
      "epoch": 1.625,
      "grad_norm": 0.431640625,
      "learning_rate": 1.0483133795255072e-05,
      "loss": 0.9941,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 78,
      "tokens_per_second_per_gpu": 9784.17,
      "total_tokens": 14902480
    },
    {
      "epoch": 1.6458333333333335,
      "grad_norm": 0.453125,
      "learning_rate": 1.0241637452361323e-05,
      "loss": 0.9883,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 79,
      "tokens_per_second_per_gpu": 10663.55,
      "total_tokens": 15089054
    },
    {
      "epoch": 1.6666666666666665,
      "grad_norm": 0.4375,
      "learning_rate": 1e-05,
      "loss": 1.0078,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 80,
      "tokens_per_second_per_gpu": 10776.49,
      "total_tokens": 15286310
    },
    {
      "epoch": 1.6875,
      "grad_norm": 0.4453125,
      "learning_rate": 9.75836254763868e-06,
      "loss": 1.0234,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 81,
      "tokens_per_second_per_gpu": 10157.52,
      "total_tokens": 15473663
    },
    {
      "epoch": 1.7083333333333335,
      "grad_norm": 0.470703125,
      "learning_rate": 9.516866204744932e-06,
      "loss": 1.0,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 82,
      "tokens_per_second_per_gpu": 10701.67,
      "total_tokens": 15658442
    },
    {
      "epoch": 1.7291666666666665,
      "grad_norm": 0.466796875,
      "learning_rate": 9.275651998382377e-06,
      "loss": 1.0039,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 83,
      "tokens_per_second_per_gpu": 9974.39,
      "total_tokens": 15842525
    },
    {
      "epoch": 1.75,
      "grad_norm": 0.455078125,
      "learning_rate": 9.034860790854848e-06,
      "loss": 0.9863,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 84,
      "tokens_per_second_per_gpu": 10560.42,
      "total_tokens": 16035753
    },
    {
      "epoch": 1.7708333333333335,
      "grad_norm": 0.455078125,
      "learning_rate": 8.79463319744677e-06,
      "loss": 1.0312,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 85,
      "tokens_per_second_per_gpu": 10498.04,
      "total_tokens": 16223979
    },
    {
      "epoch": 1.7916666666666665,
      "grad_norm": 0.42578125,
      "learning_rate": 8.55510950430779e-06,
      "loss": 0.9492,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 86,
      "tokens_per_second_per_gpu": 10922.51,
      "total_tokens": 16424221
    },
    {
      "epoch": 1.8125,
      "grad_norm": 0.431640625,
      "learning_rate": 8.316429586529616e-06,
      "loss": 0.9746,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 87,
      "tokens_per_second_per_gpu": 10848.98,
      "total_tokens": 16624193
    },
    {
      "epoch": 1.8333333333333335,
      "grad_norm": 0.4296875,
      "learning_rate": 8.078732826462917e-06,
      "loss": 0.9355,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 88,
      "tokens_per_second_per_gpu": 9565.65,
      "total_tokens": 16819542
    },
    {
      "epoch": 1.8541666666666665,
      "grad_norm": 0.43359375,
      "learning_rate": 7.84215803232194e-06,
      "loss": 1.0234,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 89,
      "tokens_per_second_per_gpu": 10503.54,
      "total_tokens": 17012659
    },
    {
      "epoch": 1.875,
      "grad_norm": 0.4296875,
      "learning_rate": 7.606843357124426e-06,
      "loss": 1.0215,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 90,
      "tokens_per_second_per_gpu": 10559.42,
      "total_tokens": 17207748
    },
    {
      "epoch": 1.8958333333333335,
      "grad_norm": 0.4375,
      "learning_rate": 7.372926218014131e-06,
      "loss": 0.9668,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 91,
      "tokens_per_second_per_gpu": 10102.48,
      "total_tokens": 17394857
    },
    {
      "epoch": 1.9166666666666665,
      "grad_norm": 0.4453125,
      "learning_rate": 7.140543216013109e-06,
      "loss": 0.9785,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 92,
      "tokens_per_second_per_gpu": 10165.22,
      "total_tokens": 17590988
    },
    {
      "epoch": 1.9375,
      "grad_norm": 0.423828125,
      "learning_rate": 6.909830056250527e-06,
      "loss": 0.9492,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 93,
      "tokens_per_second_per_gpu": 10355.66,
      "total_tokens": 17790368
    },
    {
      "epoch": 1.9583333333333335,
      "grad_norm": 0.431640625,
      "learning_rate": 6.680921468714718e-06,
      "loss": 1.0508,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 94,
      "tokens_per_second_per_gpu": 10134.55,
      "total_tokens": 17981944
    },
    {
      "epoch": 1.9791666666666665,
      "grad_norm": 0.40625,
      "learning_rate": 6.453951129574644e-06,
      "loss": 0.9199,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 95,
      "tokens_per_second_per_gpu": 9778.3,
      "total_tokens": 18173373
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.43359375,
      "learning_rate": 6.229051583116796e-06,
      "loss": 0.9863,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 96,
      "tokens_per_second_per_gpu": 10231.38,
      "total_tokens": 18364184
    },
    {
      "epoch": 2.0208333333333335,
      "grad_norm": 0.546875,
      "learning_rate": 6.006354164343047e-06,
      "loss": 0.959,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 97,
      "tokens_per_second_per_gpu": 10122.23,
      "total_tokens": 18557919
    },
    {
      "epoch": 2.0416666666666665,
      "grad_norm": 0.4296875,
      "learning_rate": 5.785988922274711e-06,
      "loss": 0.9824,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 98,
      "tokens_per_second_per_gpu": 9172.58,
      "total_tokens": 18741829
    },
    {
      "epoch": 2.0625,
      "grad_norm": 0.42578125,
      "learning_rate": 5.5680845440075885e-06,
      "loss": 1.0273,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 99,
      "tokens_per_second_per_gpu": 10557.98,
      "total_tokens": 18936415
    },
    {
      "epoch": 2.0833333333333335,
      "grad_norm": 0.40234375,
      "learning_rate": 5.352768279562315e-06,
      "loss": 0.9277,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 100,
      "tokens_per_second_per_gpu": 10657.64,
      "total_tokens": 19136164
    },
    {
      "epoch": 2.1041666666666665,
      "grad_norm": 0.435546875,
      "learning_rate": 5.14016586757394e-06,
      "loss": 1.0391,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 101,
      "tokens_per_second_per_gpu": 9843.32,
      "total_tokens": 19324813
    },
    {
      "epoch": 2.125,
      "grad_norm": 0.4453125,
      "learning_rate": 4.930401461864099e-06,
      "loss": 1.0273,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 102,
      "tokens_per_second_per_gpu": 10233.37,
      "total_tokens": 19509949
    },
    {
      "epoch": 2.1458333333333335,
      "grad_norm": 0.431640625,
      "learning_rate": 4.7235975589386715e-06,
      "loss": 0.9805,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 103,
      "tokens_per_second_per_gpu": 10815.24,
      "total_tokens": 19703191
    },
    {
      "epoch": 2.1666666666666665,
      "grad_norm": 0.447265625,
      "learning_rate": 4.519874926453303e-06,
      "loss": 1.0,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 104,
      "tokens_per_second_per_gpu": 10097.8,
      "total_tokens": 19897478
    },
    {
      "epoch": 2.1875,
      "grad_norm": 0.4375,
      "learning_rate": 4.319352532688444e-06,
      "loss": 1.0059,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 105,
      "tokens_per_second_per_gpu": 9860.18,
      "total_tokens": 20079605
    },
    {
      "epoch": 2.2083333333333335,
      "grad_norm": 0.439453125,
      "learning_rate": 4.12214747707527e-06,
      "loss": 1.0039,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 106,
      "tokens_per_second_per_gpu": 9991.56,
      "total_tokens": 20263849
    },
    {
      "epoch": 2.2291666666666665,
      "grad_norm": 0.43359375,
      "learning_rate": 3.9283749218128885e-06,
      "loss": 0.9863,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 107,
      "tokens_per_second_per_gpu": 10478.45,
      "total_tokens": 20454416
    },
    {
      "epoch": 2.25,
      "grad_norm": 0.45703125,
      "learning_rate": 3.738148024616863e-06,
      "loss": 0.998,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 108,
      "tokens_per_second_per_gpu": 10475.15,
      "total_tokens": 20641862
    },
    {
      "epoch": 2.2708333333333335,
      "grad_norm": 0.443359375,
      "learning_rate": 3.5515778726382967e-06,
      "loss": 0.9727,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 109,
      "tokens_per_second_per_gpu": 11180.87,
      "total_tokens": 20840096
    },
    {
      "epoch": 2.2916666666666665,
      "grad_norm": 0.41796875,
      "learning_rate": 3.3687734175920505e-06,
      "loss": 0.916,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 110,
      "tokens_per_second_per_gpu": 10457.13,
      "total_tokens": 21041565
    },
    {
      "epoch": 2.3125,
      "grad_norm": 0.42578125,
      "learning_rate": 3.1898414121320277e-06,
      "loss": 0.959,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 111,
      "tokens_per_second_per_gpu": 10793.07,
      "total_tokens": 21240749
    },
    {
      "epoch": 2.3333333333333335,
      "grad_norm": 0.439453125,
      "learning_rate": 3.0148863475106315e-06,
      "loss": 0.9453,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 112,
      "tokens_per_second_per_gpu": 10139.81,
      "total_tokens": 21429576
    },
    {
      "epoch": 2.3541666666666665,
      "grad_norm": 0.42578125,
      "learning_rate": 2.8440103925587904e-06,
      "loss": 0.9199,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 113,
      "tokens_per_second_per_gpu": 10567.3,
      "total_tokens": 21624286
    },
    {
      "epoch": 2.375,
      "grad_norm": 0.4375,
      "learning_rate": 2.6773133340222677e-06,
      "loss": 0.9746,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 114,
      "tokens_per_second_per_gpu": 9746.37,
      "total_tokens": 21806885
    },
    {
      "epoch": 2.3958333333333335,
      "grad_norm": 0.4453125,
      "learning_rate": 2.514892518288988e-06,
      "loss": 0.9121,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 115,
      "tokens_per_second_per_gpu": 10113.82,
      "total_tokens": 22002533
    },
    {
      "epoch": 2.4166666666666665,
      "grad_norm": 0.56640625,
      "learning_rate": 2.3568427945415163e-06,
      "loss": 0.9434,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 116,
      "tokens_per_second_per_gpu": 10289.14,
      "total_tokens": 22190926
    },
    {
      "epoch": 2.4375,
      "grad_norm": 0.5703125,
      "learning_rate": 2.2032564593677773e-06,
      "loss": 0.9434,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 117,
      "tokens_per_second_per_gpu": 10024.32,
      "total_tokens": 22379128
    },
    {
      "epoch": 2.4583333333333335,
      "grad_norm": 0.4375,
      "learning_rate": 2.0542232028624585e-06,
      "loss": 0.9707,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 118,
      "tokens_per_second_per_gpu": 10599.9,
      "total_tokens": 22572492
    },
    {
      "epoch": 2.4791666666666665,
      "grad_norm": 0.4375,
      "learning_rate": 1.9098300562505266e-06,
      "loss": 1.0039,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 119,
      "tokens_per_second_per_gpu": 10064.53,
      "total_tokens": 22758307
    },
    {
      "epoch": 2.5,
      "grad_norm": 0.421875,
      "learning_rate": 1.7701613410634367e-06,
      "loss": 0.9551,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 120,
      "tokens_per_second_per_gpu": 10637.75,
      "total_tokens": 22953079
    },
    {
      "epoch": 2.5208333333333335,
      "grad_norm": 0.43359375,
      "learning_rate": 1.6352986198977327e-06,
      "loss": 0.9844,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 121,
      "tokens_per_second_per_gpu": 10333.95,
      "total_tokens": 23145784
    },
    {
      "epoch": 2.5416666666666665,
      "grad_norm": 0.453125,
      "learning_rate": 1.5053206487847916e-06,
      "loss": 0.9707,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 122,
      "tokens_per_second_per_gpu": 9911.59,
      "total_tokens": 23327111
    },
    {
      "epoch": 2.5625,
      "grad_norm": 0.421875,
      "learning_rate": 1.3803033311995072e-06,
      "loss": 0.9688,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 123,
      "tokens_per_second_per_gpu": 10171.63,
      "total_tokens": 23518340
    },
    {
      "epoch": 2.5833333333333335,
      "grad_norm": 0.41796875,
      "learning_rate": 1.2603196737348211e-06,
      "loss": 0.9434,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 124,
      "tokens_per_second_per_gpu": 10639.7,
      "total_tokens": 23710497
    },
    {
      "epoch": 2.6041666666666665,
      "grad_norm": 0.41796875,
      "learning_rate": 1.1454397434679022e-06,
      "loss": 0.9805,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 125,
      "tokens_per_second_per_gpu": 10197.71,
      "total_tokens": 23901748
    },
    {
      "epoch": 2.625,
      "grad_norm": 0.419921875,
      "learning_rate": 1.0357306270429623e-06,
      "loss": 0.9785,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 126,
      "tokens_per_second_per_gpu": 9804.54,
      "total_tokens": 24084572
    },
    {
      "epoch": 2.6458333333333335,
      "grad_norm": 0.4296875,
      "learning_rate": 9.312563914945461e-07,
      "loss": 0.9766,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 127,
      "tokens_per_second_per_gpu": 10687.65,
      "total_tokens": 24271146
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 0.41796875,
      "learning_rate": 8.320780468341761e-07,
      "loss": 0.9961,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 128,
      "tokens_per_second_per_gpu": 10789.28,
      "total_tokens": 24468402
    },
    {
      "epoch": 2.6875,
      "grad_norm": 0.4296875,
      "learning_rate": 7.382535104222366e-07,
      "loss": 1.0117,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 129,
      "tokens_per_second_per_gpu": 10137.94,
      "total_tokens": 24655755
    },
    {
      "epoch": 2.7083333333333335,
      "grad_norm": 0.4296875,
      "learning_rate": 6.498375731458529e-07,
      "loss": 0.9883,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 130,
      "tokens_per_second_per_gpu": 10640.27,
      "total_tokens": 24840534
    },
    {
      "epoch": 2.7291666666666665,
      "grad_norm": 0.439453125,
      "learning_rate": 5.668818674225684e-07,
      "loss": 0.9941,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 131,
      "tokens_per_second_per_gpu": 10018.08,
      "total_tokens": 25024617
    },
    {
      "epoch": 2.75,
      "grad_norm": 0.427734375,
      "learning_rate": 4.894348370484648e-07,
      "loss": 0.9746,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 132,
      "tokens_per_second_per_gpu": 10544.96,
      "total_tokens": 25217845
    },
    {
      "epoch": 2.7708333333333335,
      "grad_norm": 0.455078125,
      "learning_rate": 4.1754170890833777e-07,
      "loss": 1.0234,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 133,
      "tokens_per_second_per_gpu": 10490.58,
      "total_tokens": 25406071
    },
    {
      "epoch": 2.7916666666666665,
      "grad_norm": 0.419921875,
      "learning_rate": 3.5124446656448654e-07,
      "loss": 0.9414,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 134,
      "tokens_per_second_per_gpu": 10903.67,
      "total_tokens": 25606313
    },
    {
      "epoch": 2.8125,
      "grad_norm": 0.416015625,
      "learning_rate": 2.905818257394799e-07,
      "loss": 0.9688,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 135,
      "tokens_per_second_per_gpu": 10824.22,
      "total_tokens": 25806285
    },
    {
      "epoch": 2.8333333333333335,
      "grad_norm": 0.421875,
      "learning_rate": 2.355892117072789e-07,
      "loss": 0.9277,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 136,
      "tokens_per_second_per_gpu": 9550.25,
      "total_tokens": 26001634
    },
    {
      "epoch": 2.8541666666666665,
      "grad_norm": 0.56640625,
      "learning_rate": 1.8629873860586567e-07,
      "loss": 1.0156,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 137,
      "tokens_per_second_per_gpu": 10503.94,
      "total_tokens": 26194751
    },
    {
      "epoch": 2.875,
      "grad_norm": 0.42578125,
      "learning_rate": 1.4273919068349184e-07,
      "loss": 1.0176,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 138,
      "tokens_per_second_per_gpu": 10613.72,
      "total_tokens": 26389840
    },
    {
      "epoch": 2.8958333333333335,
      "grad_norm": 0.55078125,
      "learning_rate": 1.0493600548948879e-07,
      "loss": 0.959,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 139,
      "tokens_per_second_per_gpu": 10081.76,
      "total_tokens": 26576949
    },
    {
      "epoch": 2.9166666666666665,
      "grad_norm": 0.421875,
      "learning_rate": 7.291125901946027e-08,
      "loss": 0.9727,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 140,
      "tokens_per_second_per_gpu": 10169.04,
      "total_tokens": 26773080
    },
    {
      "epoch": 2.9375,
      "grad_norm": 0.40625,
      "learning_rate": 4.6683652823513725e-08,
      "loss": 0.9434,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 141,
      "tokens_per_second_per_gpu": 10390.82,
      "total_tokens": 26972460
    },
    {
      "epoch": 2.9583333333333335,
      "grad_norm": 0.423828125,
      "learning_rate": 2.6268503085089547e-08,
      "loss": 1.0469,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 142,
      "tokens_per_second_per_gpu": 10107.1,
      "total_tokens": 27164036
    },
    {
      "epoch": 2.9791666666666665,
      "grad_norm": 0.400390625,
      "learning_rate": 1.1677731676733584e-08,
      "loss": 0.916,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 143,
      "tokens_per_second_per_gpu": 9845.86,
      "total_tokens": 27355465
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.421875,
      "learning_rate": 2.9198591980705847e-09,
      "loss": 0.9844,
      "memory/device_reserved (GiB)": 77.12,
      "memory/max_active (GiB)": 69.81,
      "memory/max_allocated (GiB)": 69.81,
      "step": 144,
      "tokens_per_second_per_gpu": 10249.72,
      "total_tokens": 27546276
    }
  ],
  "logging_steps": 1,
  "max_steps": 144,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 48,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 8.229628582947717e+17,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}