{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.7395629183152757,
  "eval_steps": 500,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 4.999925019651876e-05,
      "loss": 3.2124,
      "step": 100
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9997000831051485e-05,
      "loss": 0.2834,
      "step": 200
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.999325203852471e-05,
      "loss": 0.2178,
      "step": 300
    },
    {
      "epoch": 0.03,
      "learning_rate": 4.998800404380707e-05,
      "loss": 0.1782,
      "step": 400
    },
    {
      "epoch": 0.04,
      "learning_rate": 4.9981257161695735e-05,
      "loss": 0.1501,
      "step": 500
    },
    {
      "epoch": 0.04,
      "learning_rate": 4.997301179689756e-05,
      "loss": 0.1376,
      "step": 600
    },
    {
      "epoch": 0.05,
      "learning_rate": 4.996326844400481e-05,
      "loss": 0.1161,
      "step": 700
    },
    {
      "epoch": 0.06,
      "learning_rate": 4.9952027687465465e-05,
      "loss": 0.1163,
      "step": 800
    },
    {
      "epoch": 0.07,
      "learning_rate": 4.9939290201548214e-05,
      "loss": 0.1157,
      "step": 900
    },
    {
      "epoch": 0.07,
      "learning_rate": 4.992505675030195e-05,
      "loss": 0.1048,
      "step": 1000
    },
    {
      "epoch": 0.08,
      "learning_rate": 4.9909328187509964e-05,
      "loss": 0.1073,
      "step": 1100
    },
    {
      "epoch": 0.09,
      "learning_rate": 4.989210545663877e-05,
      "loss": 0.1047,
      "step": 1200
    },
    {
      "epoch": 0.1,
      "learning_rate": 4.987338959078144e-05,
      "loss": 0.1066,
      "step": 1300
    },
    {
      "epoch": 0.1,
      "learning_rate": 4.9853181712595686e-05,
      "loss": 0.1011,
      "step": 1400
    },
    {
      "epoch": 0.11,
      "learning_rate": 4.98314830342365e-05,
      "loss": 0.0947,
      "step": 1500
    },
    {
      "epoch": 0.12,
      "learning_rate": 4.9808294857283454e-05,
      "loss": 0.0916,
      "step": 1600
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.97836185726626e-05,
      "loss": 0.0933,
      "step": 1700
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.9757455660563085e-05,
      "loss": 0.089,
      "step": 1800
    },
    {
      "epoch": 0.14,
      "learning_rate": 4.9729807690348297e-05,
      "loss": 0.0872,
      "step": 1900
    },
    {
      "epoch": 0.15,
      "learning_rate": 4.9700676320461794e-05,
      "loss": 0.0898,
      "step": 2000
    },
    {
      "epoch": 0.16,
      "learning_rate": 4.9670376756771875e-05,
      "loss": 0.0894,
      "step": 2100
    },
    {
      "epoch": 0.16,
      "learning_rate": 4.963829870746861e-05,
      "loss": 0.0859,
      "step": 2200
    },
    {
      "epoch": 0.17,
      "learning_rate": 4.960474274759395e-05,
      "loss": 0.0826,
      "step": 2300
    },
    {
      "epoch": 0.18,
      "learning_rate": 4.956971088997792e-05,
      "loss": 0.0816,
      "step": 2400
    },
    {
      "epoch": 0.18,
      "learning_rate": 4.953320523598123e-05,
      "loss": 0.074,
      "step": 2500
    },
    {
      "epoch": 0.19,
      "learning_rate": 4.9495227975369186e-05,
      "loss": 0.0896,
      "step": 2600
    },
    {
      "epoch": 0.2,
      "learning_rate": 4.945578138618039e-05,
      "loss": 0.0768,
      "step": 2700
    },
    {
      "epoch": 0.21,
      "learning_rate": 4.941486783459001e-05,
      "loss": 0.0779,
      "step": 2800
    },
    {
      "epoch": 0.21,
      "learning_rate": 4.937248977476793e-05,
      "loss": 0.0814,
      "step": 2900
    },
    {
      "epoch": 0.22,
      "learning_rate": 4.9328649748731495e-05,
      "loss": 0.0727,
      "step": 3000
    },
    {
      "epoch": 0.23,
      "learning_rate": 4.928335038619302e-05,
      "loss": 0.0806,
      "step": 3100
    },
    {
      "epoch": 0.24,
      "learning_rate": 4.9236594404402104e-05,
      "loss": 0.0709,
      "step": 3200
    },
    {
      "epoch": 0.24,
      "learning_rate": 4.9188384607982565e-05,
      "loss": 0.0746,
      "step": 3300
    },
    {
      "epoch": 0.25,
      "learning_rate": 4.913872388876427e-05,
      "loss": 0.0805,
      "step": 3400
    },
    {
      "epoch": 0.26,
      "learning_rate": 4.908761522560962e-05,
      "loss": 0.0726,
      "step": 3500
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.903506168423491e-05,
      "loss": 0.0715,
      "step": 3600
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.898106641702639e-05,
      "loss": 0.0751,
      "step": 3700
    },
    {
      "epoch": 0.28,
      "learning_rate": 4.892563266285121e-05,
      "loss": 0.0818,
      "step": 3800
    },
    {
      "epoch": 0.29,
      "learning_rate": 4.886876374686313e-05,
      "loss": 0.0651,
      "step": 3900
    },
    {
      "epoch": 0.3,
      "learning_rate": 4.881046308030303e-05,
      "loss": 0.0806,
      "step": 4000
    },
    {
      "epoch": 0.3,
      "learning_rate": 4.87513385076558e-05,
      "loss": 0.0803,
      "step": 4100
    },
    {
      "epoch": 0.31,
      "learning_rate": 4.869019914568559e-05,
      "loss": 0.0717,
      "step": 4200
    },
    {
      "epoch": 0.32,
      "learning_rate": 4.8627638744212125e-05,
      "loss": 0.0678,
      "step": 4300
    },
    {
      "epoch": 0.33,
      "learning_rate": 4.8563661055875955e-05,
      "loss": 0.0681,
      "step": 4400
    },
    {
      "epoch": 0.33,
      "learning_rate": 4.849826991833256e-05,
      "loss": 0.0668,
      "step": 4500
    },
    {
      "epoch": 0.34,
      "learning_rate": 4.8431469254022146e-05,
      "loss": 0.0717,
      "step": 4600
    },
    {
      "epoch": 0.35,
      "learning_rate": 4.8363263069934364e-05,
      "loss": 0.0681,
      "step": 4700
    },
    {
      "epoch": 0.35,
      "learning_rate": 4.829365545736794e-05,
      "loss": 0.0703,
      "step": 4800
    },
    {
      "epoch": 0.36,
      "learning_rate": 4.8222650591685316e-05,
      "loss": 0.0625,
      "step": 4900
    },
    {
      "epoch": 0.37,
      "learning_rate": 4.815025273206212e-05,
      "loss": 0.066,
      "step": 5000
    },
    {
      "epoch": 0.38,
      "learning_rate": 4.807646622123172e-05,
      "loss": 0.0707,
      "step": 5100
    },
    {
      "epoch": 0.38,
      "learning_rate": 4.800129548522474e-05,
      "loss": 0.061,
      "step": 5200
    },
    {
      "epoch": 0.39,
      "learning_rate": 4.7924745033103533e-05,
      "loss": 0.0645,
      "step": 5300
    },
    {
      "epoch": 0.4,
      "learning_rate": 4.784681945669176e-05,
      "loss": 0.0669,
      "step": 5400
    },
    {
      "epoch": 0.41,
      "learning_rate": 4.776752343029888e-05,
      "loss": 0.059,
      "step": 5500
    },
    {
      "epoch": 0.41,
      "learning_rate": 4.768686171043982e-05,
      "loss": 0.0641,
      "step": 5600
    },
    {
      "epoch": 0.42,
      "learning_rate": 4.760483913554966e-05,
      "loss": 0.0619,
      "step": 5700
    },
    {
      "epoch": 0.43,
      "learning_rate": 4.7522301106315205e-05,
      "loss": 0.0616,
      "step": 5800
    },
    {
      "epoch": 0.44,
      "learning_rate": 4.7437585147204955e-05,
      "loss": 0.0625,
      "step": 5900
    },
    {
      "epoch": 0.44,
      "learning_rate": 4.7351523285738384e-05,
      "loss": 0.0594,
      "step": 6000
    },
    {
      "epoch": 0.45,
      "learning_rate": 4.726412068427418e-05,
      "loss": 0.0616,
      "step": 6100
    },
    {
      "epoch": 0.46,
      "learning_rate": 4.717538258559431e-05,
      "loss": 0.0657,
      "step": 6200
    },
    {
      "epoch": 0.47,
      "learning_rate": 4.7085314312589614e-05,
      "loss": 0.0631,
      "step": 6300
    },
    {
      "epoch": 0.47,
      "learning_rate": 4.699392126794045e-05,
      "loss": 0.0503,
      "step": 6400
    },
    {
      "epoch": 0.48,
      "learning_rate": 4.6901208933792675e-05,
      "loss": 0.0652,
      "step": 6500
    },
    {
      "epoch": 0.49,
      "learning_rate": 4.680718287142875e-05,
      "loss": 0.0571,
      "step": 6600
    },
    {
      "epoch": 0.5,
      "learning_rate": 4.671184872093419e-05,
      "loss": 0.054,
      "step": 6700
    },
    {
      "epoch": 0.5,
      "learning_rate": 4.661521220085923e-05,
      "loss": 0.0628,
      "step": 6800
    },
    {
      "epoch": 0.51,
      "learning_rate": 4.65172791078758e-05,
      "loss": 0.0554,
      "step": 6900
    },
    {
      "epoch": 0.52,
      "learning_rate": 4.641805531642983e-05,
      "loss": 0.059,
      "step": 7000
    },
    {
      "epoch": 0.53,
      "learning_rate": 4.631754677838885e-05,
      "loss": 0.0557,
      "step": 7100
    },
    {
      "epoch": 0.53,
      "learning_rate": 4.621575952268501e-05,
      "loss": 0.0579,
      "step": 7200
    },
    {
      "epoch": 0.54,
      "learning_rate": 4.6112699654953394e-05,
      "loss": 0.0594,
      "step": 7300
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.600837335716581e-05,
      "loss": 0.0652,
      "step": 7400
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.590278688725998e-05,
      "loss": 0.0509,
      "step": 7500
    },
    {
      "epoch": 0.56,
      "learning_rate": 4.579594657876408e-05,
      "loss": 0.0658,
      "step": 7600
    },
    {
      "epoch": 0.57,
      "learning_rate": 4.568785884041696e-05,
      "loss": 0.054,
      "step": 7700
    },
    {
      "epoch": 0.58,
      "learning_rate": 4.557853015578361e-05,
      "loss": 0.0543,
      "step": 7800
    },
    {
      "epoch": 0.58,
      "learning_rate": 4.5467967082866306e-05,
      "loss": 0.0552,
      "step": 7900
    },
    {
      "epoch": 0.59,
      "learning_rate": 4.535730021743883e-05,
      "loss": 0.0615,
      "step": 8000
    },
    {
      "epoch": 0.6,
      "learning_rate": 4.52443005148096e-05,
      "loss": 0.0567,
      "step": 8100
    },
    {
      "epoch": 0.61,
      "learning_rate": 4.5130086472420265e-05,
      "loss": 0.0519,
      "step": 8200
    },
    {
      "epoch": 0.61,
      "learning_rate": 4.5014664941317766e-05,
      "loss": 0.0573,
      "step": 8300
    },
    {
      "epoch": 0.62,
      "learning_rate": 4.489804284497936e-05,
      "loss": 0.053,
      "step": 8400
    },
    {
      "epoch": 0.63,
      "learning_rate": 4.4780227178897366e-05,
      "loss": 0.0658,
      "step": 8500
    },
    {
      "epoch": 0.64,
      "learning_rate": 4.46612250101595e-05,
      "loss": 0.0574,
      "step": 8600
    },
    {
      "epoch": 0.64,
      "learning_rate": 4.4541043477025e-05,
      "loss": 0.0608,
      "step": 8700
    },
    {
      "epoch": 0.65,
      "learning_rate": 4.441968978849641e-05,
      "loss": 0.0471,
      "step": 8800
    },
    {
      "epoch": 0.66,
      "learning_rate": 4.4297171223887194e-05,
      "loss": 0.0598,
      "step": 8900
    },
    {
      "epoch": 0.67,
      "learning_rate": 4.4173495132385035e-05,
      "loss": 0.0549,
      "step": 9000
    },
    {
      "epoch": 0.67,
      "learning_rate": 4.404866893261106e-05,
      "loss": 0.0529,
      "step": 9100
    },
    {
      "epoch": 0.68,
      "learning_rate": 4.39227001121748e-05,
      "loss": 0.0529,
      "step": 9200
    },
    {
      "epoch": 0.69,
      "learning_rate": 4.3795596227225066e-05,
      "loss": 0.0568,
      "step": 9300
    },
    {
      "epoch": 0.7,
      "learning_rate": 4.366736490199668e-05,
      "loss": 0.0602,
      "step": 9400
    },
    {
      "epoch": 0.7,
      "learning_rate": 4.353801382835318e-05,
      "loss": 0.053,
      "step": 9500
    },
    {
      "epoch": 0.71,
      "learning_rate": 4.340755076532538e-05,
      "loss": 0.0498,
      "step": 9600
    },
    {
      "epoch": 0.72,
      "learning_rate": 4.3275983538645995e-05,
      "loss": 0.0509,
      "step": 9700
    },
    {
      "epoch": 0.72,
      "learning_rate": 4.314332004028019e-05,
      "loss": 0.0552,
      "step": 9800
    },
    {
      "epoch": 0.73,
      "learning_rate": 4.3009568227952194e-05,
      "loss": 0.0565,
      "step": 9900
    },
    {
      "epoch": 0.74,
      "learning_rate": 4.287473612466796e-05,
      "loss": 0.0573,
      "step": 10000
    }
  ],
  "logging_steps": 100,
  "max_steps": 40563,
  "num_train_epochs": 3,
  "save_steps": 5000,
  "total_flos": 2.2312434217439724e+18,
  "trial_name": null,
  "trial_params": null
}