{
  "best_metric": 0.8676381707191467,
  "best_model_checkpoint": "./results_v1/checkpoint-10200",
  "epoch": 25.0,
  "eval_steps": 500,
  "global_step": 10200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.24509803921568626,
      "grad_norm": 10681.4521484375,
      "learning_rate": 0.0004950980392156863,
      "loss": 2.5241,
      "step": 100
    },
    {
      "epoch": 0.49019607843137253,
      "grad_norm": 6339.50927734375,
      "learning_rate": 0.0004901960784313725,
      "loss": 2.0561,
      "step": 200
    },
    {
      "epoch": 0.7352941176470589,
      "grad_norm": 16086.7705078125,
      "learning_rate": 0.0004852941176470588,
      "loss": 1.9379,
      "step": 300
    },
    {
      "epoch": 0.9803921568627451,
      "grad_norm": 10268.5185546875,
      "learning_rate": 0.00048039215686274513,
      "loss": 1.8333,
      "step": 400
    },
    {
      "epoch": 1.0,
      "eval_loss": 1.6950186491012573,
      "eval_runtime": 17.2704,
      "eval_samples_per_second": 463.222,
      "eval_steps_per_second": 3.648,
      "step": 408
    },
    {
      "epoch": 1.2254901960784315,
      "grad_norm": 10859.7197265625,
      "learning_rate": 0.00047549019607843134,
      "loss": 1.7372,
      "step": 500
    },
    {
      "epoch": 1.4705882352941178,
      "grad_norm": 13147.400390625,
      "learning_rate": 0.00047058823529411766,
      "loss": 1.6714,
      "step": 600
    },
    {
      "epoch": 1.715686274509804,
      "grad_norm": 11227.5166015625,
      "learning_rate": 0.0004656862745098039,
      "loss": 1.5845,
      "step": 700
    },
    {
      "epoch": 1.9607843137254903,
      "grad_norm": 10829.8447265625,
      "learning_rate": 0.0004607843137254902,
      "loss": 1.5283,
      "step": 800
    },
    {
      "epoch": 2.0,
      "eval_loss": 1.389205813407898,
      "eval_runtime": 16.6762,
      "eval_samples_per_second": 479.726,
      "eval_steps_per_second": 3.778,
      "step": 816
    },
    {
      "epoch": 2.2058823529411766,
      "grad_norm": 13556.080078125,
      "learning_rate": 0.00045588235294117646,
      "loss": 1.4638,
      "step": 900
    },
    {
      "epoch": 2.450980392156863,
      "grad_norm": 10844.283203125,
      "learning_rate": 0.0004509803921568628,
      "loss": 1.4344,
      "step": 1000
    },
    {
      "epoch": 2.696078431372549,
      "grad_norm": 12495.66796875,
      "learning_rate": 0.00044607843137254904,
      "loss": 1.3873,
      "step": 1100
    },
    {
      "epoch": 2.9411764705882355,
      "grad_norm": 11954.7265625,
      "learning_rate": 0.0004411764705882353,
      "loss": 1.3733,
      "step": 1200
    },
    {
      "epoch": 3.0,
      "eval_loss": 1.241082787513733,
      "eval_runtime": 17.1795,
      "eval_samples_per_second": 465.671,
      "eval_steps_per_second": 3.667,
      "step": 1224
    },
    {
      "epoch": 3.186274509803922,
      "grad_norm": 12640.005859375,
      "learning_rate": 0.0004362745098039216,
      "loss": 1.32,
      "step": 1300
    },
    {
      "epoch": 3.431372549019608,
      "grad_norm": 14839.830078125,
      "learning_rate": 0.0004313725490196079,
      "loss": 1.2993,
      "step": 1400
    },
    {
      "epoch": 3.6764705882352944,
      "grad_norm": 17634.357421875,
      "learning_rate": 0.0004264705882352941,
      "loss": 1.284,
      "step": 1500
    },
    {
      "epoch": 3.9215686274509802,
      "grad_norm": 11596.447265625,
      "learning_rate": 0.00042156862745098043,
      "loss": 1.2622,
      "step": 1600
    },
    {
      "epoch": 4.0,
      "eval_loss": 1.152040958404541,
      "eval_runtime": 16.7185,
      "eval_samples_per_second": 478.513,
      "eval_steps_per_second": 3.768,
      "step": 1632
    },
    {
      "epoch": 4.166666666666667,
      "grad_norm": 10679.4697265625,
      "learning_rate": 0.0004166666666666667,
      "loss": 1.2385,
      "step": 1700
    },
    {
      "epoch": 4.411764705882353,
      "grad_norm": 10487.45703125,
      "learning_rate": 0.0004117647058823529,
      "loss": 1.2131,
      "step": 1800
    },
    {
      "epoch": 4.6568627450980395,
      "grad_norm": 16033.8701171875,
      "learning_rate": 0.0004068627450980392,
      "loss": 1.2052,
      "step": 1900
    },
    {
      "epoch": 4.901960784313726,
      "grad_norm": 13594.8916015625,
      "learning_rate": 0.0004019607843137255,
      "loss": 1.1878,
      "step": 2000
    },
    {
      "epoch": 5.0,
      "eval_loss": 1.0944446325302124,
      "eval_runtime": 16.7053,
      "eval_samples_per_second": 478.89,
      "eval_steps_per_second": 3.771,
      "step": 2040
    },
    {
      "epoch": 5.147058823529412,
      "grad_norm": 12540.892578125,
      "learning_rate": 0.00039705882352941176,
      "loss": 1.1714,
      "step": 2100
    },
    {
      "epoch": 5.392156862745098,
      "grad_norm": 11532.8291015625,
      "learning_rate": 0.000392156862745098,
      "loss": 1.1554,
      "step": 2200
    },
    {
      "epoch": 5.637254901960784,
      "grad_norm": 13462.1943359375,
      "learning_rate": 0.00038725490196078434,
      "loss": 1.1487,
      "step": 2300
    },
    {
      "epoch": 5.882352941176471,
      "grad_norm": 14919.408203125,
      "learning_rate": 0.00038235294117647055,
      "loss": 1.1286,
      "step": 2400
    },
    {
      "epoch": 6.0,
      "eval_loss": 1.0514171123504639,
      "eval_runtime": 17.2713,
      "eval_samples_per_second": 463.195,
      "eval_steps_per_second": 3.648,
      "step": 2448
    },
    {
      "epoch": 6.127450980392156,
      "grad_norm": 12428.1318359375,
      "learning_rate": 0.0003774509803921569,
      "loss": 1.1157,
      "step": 2500
    },
    {
      "epoch": 6.372549019607844,
      "grad_norm": 12143.3505859375,
      "learning_rate": 0.00037254901960784314,
      "loss": 1.1092,
      "step": 2600
    },
    {
      "epoch": 6.617647058823529,
      "grad_norm": 11514.5810546875,
      "learning_rate": 0.00036764705882352946,
      "loss": 1.0906,
      "step": 2700
    },
    {
      "epoch": 6.862745098039216,
      "grad_norm": 18561.646484375,
      "learning_rate": 0.00036274509803921567,
      "loss": 1.0951,
      "step": 2800
    },
    {
      "epoch": 7.0,
      "eval_loss": 1.0217130184173584,
      "eval_runtime": 16.6786,
      "eval_samples_per_second": 479.657,
      "eval_steps_per_second": 3.777,
      "step": 2856
    },
    {
      "epoch": 7.107843137254902,
      "grad_norm": 16182.7216796875,
      "learning_rate": 0.000357843137254902,
      "loss": 1.0806,
      "step": 2900
    },
    {
      "epoch": 7.352941176470588,
      "grad_norm": 15809.748046875,
      "learning_rate": 0.00035294117647058826,
      "loss": 1.0551,
      "step": 3000
    },
    {
      "epoch": 7.598039215686274,
      "grad_norm": 11234.1923828125,
      "learning_rate": 0.0003480392156862745,
      "loss": 1.0648,
      "step": 3100
    },
    {
      "epoch": 7.8431372549019605,
      "grad_norm": 13925.9716796875,
      "learning_rate": 0.0003431372549019608,
      "loss": 1.0551,
      "step": 3200
    },
    {
      "epoch": 8.0,
      "eval_loss": 0.9909818172454834,
      "eval_runtime": 16.7369,
      "eval_samples_per_second": 477.987,
      "eval_steps_per_second": 3.764,
      "step": 3264
    },
    {
      "epoch": 8.088235294117647,
      "grad_norm": 13518.5712890625,
      "learning_rate": 0.0003382352941176471,
      "loss": 1.0471,
      "step": 3300
    },
    {
      "epoch": 8.333333333333334,
      "grad_norm": 12719.046875,
      "learning_rate": 0.0003333333333333333,
      "loss": 1.0271,
      "step": 3400
    },
    {
      "epoch": 8.57843137254902,
      "grad_norm": 13152.646484375,
      "learning_rate": 0.0003284313725490196,
      "loss": 1.0161,
      "step": 3500
    },
    {
      "epoch": 8.823529411764707,
      "grad_norm": 11174.1962890625,
      "learning_rate": 0.0003235294117647059,
      "loss": 1.0259,
      "step": 3600
    },
    {
      "epoch": 9.0,
      "eval_loss": 0.9708221554756165,
      "eval_runtime": 17.2479,
      "eval_samples_per_second": 463.825,
      "eval_steps_per_second": 3.653,
      "step": 3672
    },
    {
      "epoch": 9.068627450980392,
      "grad_norm": 12128.1474609375,
      "learning_rate": 0.0003186274509803921,
      "loss": 1.0239,
      "step": 3700
    },
    {
      "epoch": 9.313725490196079,
      "grad_norm": 10996.591796875,
      "learning_rate": 0.00031372549019607844,
      "loss": 0.9963,
      "step": 3800
    },
    {
      "epoch": 9.558823529411764,
      "grad_norm": 12174.8466796875,
      "learning_rate": 0.0003088235294117647,
      "loss": 1.006,
      "step": 3900
    },
    {
      "epoch": 9.803921568627452,
      "grad_norm": 11552.1796875,
      "learning_rate": 0.00030392156862745097,
      "loss": 1.0001,
      "step": 4000
    },
    {
      "epoch": 10.0,
      "eval_loss": 0.9560179114341736,
      "eval_runtime": 17.2671,
      "eval_samples_per_second": 463.308,
      "eval_steps_per_second": 3.649,
      "step": 4080
    },
    {
      "epoch": 10.049019607843137,
      "grad_norm": 14723.912109375,
      "learning_rate": 0.00029901960784313724,
      "loss": 0.9901,
      "step": 4100
    },
    {
      "epoch": 10.294117647058824,
      "grad_norm": 16030.7060546875,
      "learning_rate": 0.00029411764705882356,
      "loss": 0.9763,
      "step": 4200
    },
    {
      "epoch": 10.53921568627451,
      "grad_norm": 12146.142578125,
      "learning_rate": 0.0002892156862745098,
      "loss": 0.9638,
      "step": 4300
    },
    {
      "epoch": 10.784313725490197,
      "grad_norm": 12161.0078125,
      "learning_rate": 0.0002843137254901961,
      "loss": 0.9891,
      "step": 4400
    },
    {
      "epoch": 11.0,
      "eval_loss": 0.9404354095458984,
      "eval_runtime": 16.7426,
      "eval_samples_per_second": 477.824,
      "eval_steps_per_second": 3.763,
      "step": 4488
    },
    {
      "epoch": 11.029411764705882,
      "grad_norm": 11092.498046875,
      "learning_rate": 0.00027941176470588236,
      "loss": 0.9653,
      "step": 4500
    },
    {
      "epoch": 11.27450980392157,
      "grad_norm": 15073.44140625,
      "learning_rate": 0.0002745098039215687,
      "loss": 0.9608,
      "step": 4600
    },
    {
      "epoch": 11.519607843137255,
      "grad_norm": 12903.4189453125,
      "learning_rate": 0.0002696078431372549,
      "loss": 0.9572,
      "step": 4700
    },
    {
      "epoch": 11.764705882352942,
      "grad_norm": 14061.0234375,
      "learning_rate": 0.0002647058823529412,
      "loss": 0.9523,
      "step": 4800
    },
    {
      "epoch": 12.0,
      "eval_loss": 0.9268321394920349,
      "eval_runtime": 16.7172,
      "eval_samples_per_second": 478.548,
      "eval_steps_per_second": 3.769,
      "step": 4896
    },
    {
      "epoch": 12.009803921568627,
      "grad_norm": 13276.89453125,
      "learning_rate": 0.0002598039215686275,
      "loss": 0.9492,
      "step": 4900
    },
    {
      "epoch": 12.254901960784313,
      "grad_norm": 11237.9267578125,
      "learning_rate": 0.00025490196078431374,
      "loss": 0.9417,
      "step": 5000
    },
    {
      "epoch": 12.5,
      "grad_norm": 11576.23046875,
      "learning_rate": 0.00025,
      "loss": 0.9379,
      "step": 5100
    },
    {
      "epoch": 12.745098039215687,
      "grad_norm": 14009.544921875,
      "learning_rate": 0.00024509803921568627,
      "loss": 0.9349,
      "step": 5200
    },
    {
      "epoch": 12.990196078431373,
      "grad_norm": 12153.9677734375,
      "learning_rate": 0.00024019607843137256,
      "loss": 0.9329,
      "step": 5300
    },
    {
      "epoch": 13.0,
      "eval_loss": 0.9184120893478394,
      "eval_runtime": 17.2276,
      "eval_samples_per_second": 464.372,
      "eval_steps_per_second": 3.657,
      "step": 5304
    },
    {
      "epoch": 13.235294117647058,
      "grad_norm": 12770.306640625,
      "learning_rate": 0.00023529411764705883,
      "loss": 0.9219,
      "step": 5400
    },
    {
      "epoch": 13.480392156862745,
      "grad_norm": 12070.76171875,
      "learning_rate": 0.0002303921568627451,
      "loss": 0.9192,
      "step": 5500
    },
    {
      "epoch": 13.72549019607843,
      "grad_norm": 12918.3779296875,
      "learning_rate": 0.0002254901960784314,
      "loss": 0.925,
      "step": 5600
    },
    {
      "epoch": 13.970588235294118,
      "grad_norm": 11361.103515625,
      "learning_rate": 0.00022058823529411765,
      "loss": 0.9109,
      "step": 5700
    },
    {
      "epoch": 14.0,
      "eval_loss": 0.9040451645851135,
      "eval_runtime": 16.707,
      "eval_samples_per_second": 478.841,
      "eval_steps_per_second": 3.771,
      "step": 5712
    },
    {
      "epoch": 14.215686274509803,
      "grad_norm": 14390.365234375,
      "learning_rate": 0.00021568627450980395,
      "loss": 0.8986,
      "step": 5800
    },
    {
      "epoch": 14.46078431372549,
      "grad_norm": 12093.57421875,
      "learning_rate": 0.00021078431372549021,
      "loss": 0.9064,
      "step": 5900
    },
    {
      "epoch": 14.705882352941176,
      "grad_norm": 9752.62890625,
      "learning_rate": 0.00020588235294117645,
      "loss": 0.9165,
      "step": 6000
    },
    {
      "epoch": 14.950980392156863,
      "grad_norm": 16142.40625,
      "learning_rate": 0.00020098039215686275,
      "loss": 0.9027,
      "step": 6100
    },
    {
      "epoch": 15.0,
      "eval_loss": 0.9005336761474609,
      "eval_runtime": 17.261,
      "eval_samples_per_second": 463.471,
      "eval_steps_per_second": 3.65,
      "step": 6120
    },
    {
      "epoch": 15.196078431372548,
      "grad_norm": 14545.95703125,
      "learning_rate": 0.000196078431372549,
      "loss": 0.9078,
      "step": 6200
    },
    {
      "epoch": 15.441176470588236,
      "grad_norm": 10779.703125,
      "learning_rate": 0.00019117647058823528,
      "loss": 0.8906,
      "step": 6300
    },
    {
      "epoch": 15.686274509803921,
      "grad_norm": 11541.22265625,
      "learning_rate": 0.00018627450980392157,
      "loss": 0.8925,
      "step": 6400
    },
    {
      "epoch": 15.931372549019608,
      "grad_norm": 11552.533203125,
      "learning_rate": 0.00018137254901960784,
      "loss": 0.8872,
      "step": 6500
    },
    {
      "epoch": 16.0,
      "eval_loss": 0.8934208154678345,
      "eval_runtime": 16.6878,
      "eval_samples_per_second": 479.393,
      "eval_steps_per_second": 3.775,
      "step": 6528
    },
    {
      "epoch": 16.176470588235293,
      "grad_norm": 11795.5029296875,
      "learning_rate": 0.00017647058823529413,
      "loss": 0.8854,
      "step": 6600
    },
    {
      "epoch": 16.42156862745098,
      "grad_norm": 11742.291015625,
      "learning_rate": 0.0001715686274509804,
      "loss": 0.8853,
      "step": 6700
    },
    {
      "epoch": 16.666666666666668,
      "grad_norm": 11037.279296875,
      "learning_rate": 0.00016666666666666666,
      "loss": 0.8805,
      "step": 6800
    },
    {
      "epoch": 16.91176470588235,
      "grad_norm": 13446.0478515625,
      "learning_rate": 0.00016176470588235295,
      "loss": 0.8754,
      "step": 6900
    },
    {
      "epoch": 17.0,
      "eval_loss": 0.8874623775482178,
      "eval_runtime": 16.6727,
      "eval_samples_per_second": 479.827,
      "eval_steps_per_second": 3.779,
      "step": 6936
    },
    {
      "epoch": 17.15686274509804,
      "grad_norm": 13627.7470703125,
      "learning_rate": 0.00015686274509803922,
      "loss": 0.8801,
      "step": 7000
    },
    {
      "epoch": 17.401960784313726,
      "grad_norm": 14940.4482421875,
      "learning_rate": 0.00015196078431372549,
      "loss": 0.8703,
      "step": 7100
    },
    {
      "epoch": 17.647058823529413,
      "grad_norm": 17162.34765625,
      "learning_rate": 0.00014705882352941178,
      "loss": 0.8732,
      "step": 7200
    },
    {
      "epoch": 17.892156862745097,
      "grad_norm": 13630.0888671875,
      "learning_rate": 0.00014215686274509804,
      "loss": 0.872,
      "step": 7300
    },
    {
      "epoch": 18.0,
      "eval_loss": 0.8821218013763428,
      "eval_runtime": 17.107,
      "eval_samples_per_second": 467.645,
      "eval_steps_per_second": 3.683,
      "step": 7344
    },
    {
      "epoch": 18.137254901960784,
      "grad_norm": 12382.7177734375,
      "learning_rate": 0.00013725490196078434,
      "loss": 0.8609,
      "step": 7400
    },
    {
      "epoch": 18.38235294117647,
      "grad_norm": 12689.1533203125,
      "learning_rate": 0.0001323529411764706,
      "loss": 0.8643,
      "step": 7500
    },
    {
      "epoch": 18.627450980392158,
      "grad_norm": 13463.681640625,
      "learning_rate": 0.00012745098039215687,
      "loss": 0.8624,
      "step": 7600
    },
    {
      "epoch": 18.872549019607842,
      "grad_norm": 11256.0576171875,
      "learning_rate": 0.00012254901960784314,
      "loss": 0.8648,
      "step": 7700
    },
    {
      "epoch": 19.0,
      "eval_loss": 0.878471314907074,
      "eval_runtime": 16.6776,
      "eval_samples_per_second": 479.684,
      "eval_steps_per_second": 3.778,
      "step": 7752
    },
    {
      "epoch": 19.11764705882353,
      "grad_norm": 13279.96875,
      "learning_rate": 0.00011764705882352942,
      "loss": 0.8581,
      "step": 7800
    },
    {
      "epoch": 19.362745098039216,
      "grad_norm": 12439.5205078125,
      "learning_rate": 0.0001127450980392157,
      "loss": 0.8624,
      "step": 7900
    },
    {
      "epoch": 19.607843137254903,
      "grad_norm": 11728.5849609375,
      "learning_rate": 0.00010784313725490197,
      "loss": 0.8517,
      "step": 8000
    },
    {
      "epoch": 19.852941176470587,
      "grad_norm": 11004.234375,
      "learning_rate": 0.00010294117647058823,
      "loss": 0.8476,
      "step": 8100
    },
    {
      "epoch": 20.0,
      "eval_loss": 0.8746265769004822,
      "eval_runtime": 16.7166,
      "eval_samples_per_second": 478.566,
      "eval_steps_per_second": 3.769,
      "step": 8160
    },
    {
      "epoch": 20.098039215686274,
      "grad_norm": 12775.7421875,
      "learning_rate": 9.80392156862745e-05,
      "loss": 0.8636,
      "step": 8200
    },
    {
      "epoch": 20.34313725490196,
      "grad_norm": 11157.79296875,
      "learning_rate": 9.313725490196079e-05,
      "loss": 0.8482,
      "step": 8300
    },
    {
      "epoch": 20.58823529411765,
      "grad_norm": 12324.103515625,
      "learning_rate": 8.823529411764706e-05,
      "loss": 0.844,
      "step": 8400
    },
    {
      "epoch": 20.833333333333332,
      "grad_norm": 12305.396484375,
      "learning_rate": 8.333333333333333e-05,
      "loss": 0.847,
      "step": 8500
    },
    {
      "epoch": 21.0,
      "eval_loss": 0.8715333342552185,
      "eval_runtime": 16.7289,
      "eval_samples_per_second": 478.213,
      "eval_steps_per_second": 3.766,
      "step": 8568
    },
    {
      "epoch": 21.07843137254902,
      "grad_norm": 14880.2978515625,
      "learning_rate": 7.843137254901961e-05,
      "loss": 0.8532,
      "step": 8600
    },
    {
      "epoch": 21.323529411764707,
      "grad_norm": 11197.8076171875,
      "learning_rate": 7.352941176470589e-05,
      "loss": 0.8484,
      "step": 8700
    },
    {
      "epoch": 21.568627450980394,
      "grad_norm": 10673.302734375,
      "learning_rate": 6.862745098039217e-05,
      "loss": 0.8395,
      "step": 8800
    },
    {
      "epoch": 21.813725490196077,
      "grad_norm": 10516.341796875,
      "learning_rate": 6.372549019607843e-05,
      "loss": 0.8472,
      "step": 8900
    },
    {
      "epoch": 22.0,
      "eval_loss": 0.8717733025550842,
      "eval_runtime": 16.6953,
      "eval_samples_per_second": 479.176,
      "eval_steps_per_second": 3.774,
      "step": 8976
    },
    {
      "epoch": 22.058823529411764,
      "grad_norm": 11651.111328125,
      "learning_rate": 5.882352941176471e-05,
      "loss": 0.8404,
      "step": 9000
    },
    {
      "epoch": 22.30392156862745,
      "grad_norm": 13144.53515625,
      "learning_rate": 5.392156862745099e-05,
      "loss": 0.8362,
      "step": 9100
    },
    {
      "epoch": 22.54901960784314,
      "grad_norm": 11174.5419921875,
      "learning_rate": 4.901960784313725e-05,
      "loss": 0.8405,
      "step": 9200
    },
    {
      "epoch": 22.794117647058822,
      "grad_norm": 10513.1904296875,
      "learning_rate": 4.411764705882353e-05,
      "loss": 0.8417,
      "step": 9300
    },
    {
      "epoch": 23.0,
      "eval_loss": 0.8683518171310425,
      "eval_runtime": 17.2108,
      "eval_samples_per_second": 464.825,
      "eval_steps_per_second": 3.66,
      "step": 9384
    },
    {
      "epoch": 23.03921568627451,
      "grad_norm": 13963.318359375,
      "learning_rate": 3.9215686274509805e-05,
      "loss": 0.8438,
      "step": 9400
    },
    {
      "epoch": 23.284313725490197,
      "grad_norm": 15601.267578125,
      "learning_rate": 3.4313725490196084e-05,
      "loss": 0.8422,
      "step": 9500
    },
    {
      "epoch": 23.529411764705884,
      "grad_norm": 12826.8515625,
      "learning_rate": 2.9411764705882354e-05,
      "loss": 0.8304,
      "step": 9600
    },
    {
      "epoch": 23.774509803921568,
      "grad_norm": 13277.5234375,
      "learning_rate": 2.4509803921568626e-05,
      "loss": 0.8357,
      "step": 9700
    },
    {
      "epoch": 24.0,
      "eval_loss": 0.8688600659370422,
      "eval_runtime": 16.7261,
      "eval_samples_per_second": 478.293,
      "eval_steps_per_second": 3.767,
      "step": 9792
    },
    {
      "epoch": 24.019607843137255,
      "grad_norm": 11718.4677734375,
      "learning_rate": 1.9607843137254903e-05,
      "loss": 0.8361,
      "step": 9800
    },
    {
      "epoch": 24.264705882352942,
      "grad_norm": 13389.1337890625,
      "learning_rate": 1.4705882352941177e-05,
      "loss": 0.8371,
      "step": 9900
    },
    {
      "epoch": 24.509803921568626,
      "grad_norm": 10126.4189453125,
      "learning_rate": 9.803921568627451e-06,
      "loss": 0.8234,
      "step": 10000
    },
    {
      "epoch": 24.754901960784313,
      "grad_norm": 15717.1884765625,
      "learning_rate": 4.901960784313726e-06,
      "loss": 0.8405,
      "step": 10100
    },
    {
      "epoch": 25.0,
      "grad_norm": 27583.41015625,
      "learning_rate": 0.0,
      "loss": 0.8359,
      "step": 10200
    },
    {
      "epoch": 25.0,
      "eval_loss": 0.8676381707191467,
      "eval_runtime": 16.6913,
      "eval_samples_per_second": 479.291,
      "eval_steps_per_second": 3.774,
      "step": 10200
    }
  ],
  "logging_steps": 100,
  "max_steps": 10200,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 25,
  "save_steps": 500,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 2,
        "early_stopping_threshold": 0.0
      },
      "attributes": {
        "early_stopping_patience_counter": 0
      }
    },
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 4.40829394550784e+16,
  "train_batch_size": 256,
  "trial_name": null,
  "trial_params": null
}