{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.8072382361844539,
  "eval_steps": 500,
  "global_step": 6000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 4.9997567688496474e-05,
      "loss": 2.9132,
      "step": 100
    },
    {
      "epoch": 0.03,
      "learning_rate": 4.9990172715142793e-05,
      "loss": 0.0721,
      "step": 200
    },
    {
      "epoch": 0.04,
      "learning_rate": 4.997781629993153e-05,
      "loss": 0.0757,
      "step": 300
    },
    {
      "epoch": 0.05,
      "learning_rate": 4.9960500896052476e-05,
      "loss": 0.0599,
      "step": 400
    },
    {
      "epoch": 0.07,
      "learning_rate": 4.9938477167054135e-05,
      "loss": 0.0353,
      "step": 500
    },
    {
      "epoch": 0.08,
      "learning_rate": 4.9911304569050045e-05,
      "loss": 0.0256,
      "step": 600
    },
    {
      "epoch": 0.09,
      "learning_rate": 4.987918618733232e-05,
      "loss": 0.0173,
      "step": 700
    },
    {
      "epoch": 0.11,
      "learning_rate": 4.98421283985469e-05,
      "loss": 0.0276,
      "step": 800
    },
    {
      "epoch": 0.12,
      "learning_rate": 4.9800138559988466e-05,
      "loss": 0.0225,
      "step": 900
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.97532250081397e-05,
      "loss": 0.026,
      "step": 1000
    },
    {
      "epoch": 0.15,
      "learning_rate": 4.970139705701628e-05,
      "loss": 0.0271,
      "step": 1100
    },
    {
      "epoch": 0.16,
      "learning_rate": 4.9644664996317616e-05,
      "loss": 0.0231,
      "step": 1200
    },
    {
      "epoch": 0.17,
      "learning_rate": 4.958304008938407e-05,
      "loss": 0.0196,
      "step": 1300
    },
    {
      "epoch": 0.19,
      "learning_rate": 4.951653457096072e-05,
      "loss": 0.0124,
      "step": 1400
    },
    {
      "epoch": 0.2,
      "learning_rate": 4.944516164476834e-05,
      "loss": 0.0159,
      "step": 1500
    },
    {
      "epoch": 0.22,
      "learning_rate": 4.9368935480882034e-05,
      "loss": 0.0144,
      "step": 1600
    },
    {
      "epoch": 0.23,
      "learning_rate": 4.9287871212917866e-05,
      "loss": 0.0182,
      "step": 1700
    },
    {
      "epoch": 0.24,
      "learning_rate": 4.9201984935028426e-05,
      "loss": 0.018,
      "step": 1800
    },
    {
      "epoch": 0.26,
      "learning_rate": 4.911129369870746e-05,
      "loss": 0.0156,
      "step": 1900
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.90158155094046e-05,
      "loss": 0.0146,
      "step": 2000
    },
    {
      "epoch": 0.28,
      "learning_rate": 4.8915569322950615e-05,
      "loss": 0.0196,
      "step": 2100
    },
    {
      "epoch": 0.3,
      "learning_rate": 4.8810575041794e-05,
      "loss": 0.02,
      "step": 2200
    },
    {
      "epoch": 0.31,
      "learning_rate": 4.8700853511049656e-05,
      "loss": 0.0143,
      "step": 2300
    },
    {
      "epoch": 0.32,
      "learning_rate": 4.858642651436035e-05,
      "loss": 0.0111,
      "step": 2400
    },
    {
      "epoch": 0.34,
      "learning_rate": 4.846731676957191e-05,
      "loss": 0.0179,
      "step": 2500
    },
    {
      "epoch": 0.35,
      "learning_rate": 4.834354792422293e-05,
      "loss": 0.0133,
      "step": 2600
    },
    {
      "epoch": 0.36,
      "learning_rate": 4.821514455084985e-05,
      "loss": 0.0112,
      "step": 2700
    },
    {
      "epoch": 0.38,
      "learning_rate": 4.8082132142108465e-05,
      "loss": 0.0131,
      "step": 2800
    },
    {
      "epoch": 0.39,
      "learning_rate": 4.794453710571272e-05,
      "loss": 0.0134,
      "step": 2900
    },
    {
      "epoch": 0.4,
      "learning_rate": 4.780238675919182e-05,
      "loss": 0.0084,
      "step": 3000
    },
    {
      "epoch": 0.42,
      "learning_rate": 4.765570932446672e-05,
      "loss": 0.0069,
      "step": 3100
    },
    {
      "epoch": 0.43,
      "learning_rate": 4.75045339222471e-05,
      "loss": 0.0186,
      "step": 3200
    },
    {
      "epoch": 0.44,
      "learning_rate": 4.734889056624983e-05,
      "loss": 0.0138,
      "step": 3300
    },
    {
      "epoch": 0.46,
      "learning_rate": 4.718881015724017e-05,
      "loss": 0.012,
      "step": 3400
    },
    {
      "epoch": 0.47,
      "learning_rate": 4.702432447689692e-05,
      "loss": 0.0121,
      "step": 3500
    },
    {
      "epoch": 0.48,
      "learning_rate": 4.6855466181502544e-05,
      "loss": 0.0104,
      "step": 3600
    },
    {
      "epoch": 0.5,
      "learning_rate": 4.66822687954598e-05,
      "loss": 0.014,
      "step": 3700
    },
    {
      "epoch": 0.51,
      "learning_rate": 4.65047667046359e-05,
      "loss": 0.0085,
      "step": 3800
    },
    {
      "epoch": 0.52,
      "learning_rate": 4.632299514953571e-05,
      "loss": 0.0111,
      "step": 3900
    },
    {
      "epoch": 0.54,
      "learning_rate": 4.613699021830524e-05,
      "loss": 0.0122,
      "step": 4000
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.59467888395669e-05,
      "loss": 0.0082,
      "step": 4100
    },
    {
      "epoch": 0.57,
      "learning_rate": 4.575242877508777e-05,
      "loss": 0.0087,
      "step": 4200
    },
    {
      "epoch": 0.58,
      "learning_rate": 4.5553948612282607e-05,
      "loss": 0.0115,
      "step": 4300
    },
    {
      "epoch": 0.59,
      "learning_rate": 4.5351387756552846e-05,
      "loss": 0.0069,
      "step": 4400
    },
    {
      "epoch": 0.61,
      "learning_rate": 4.51447864234632e-05,
      "loss": 0.0099,
      "step": 4500
    },
    {
      "epoch": 0.62,
      "learning_rate": 4.4934185630757484e-05,
      "loss": 0.0087,
      "step": 4600
    },
    {
      "epoch": 0.63,
      "learning_rate": 4.4719627190215064e-05,
      "loss": 0.0096,
      "step": 4700
    },
    {
      "epoch": 0.65,
      "learning_rate": 4.450115369934976e-05,
      "loss": 0.0122,
      "step": 4800
    },
    {
      "epoch": 0.66,
      "learning_rate": 4.427880853295274e-05,
      "loss": 0.0076,
      "step": 4900
    },
    {
      "epoch": 0.67,
      "learning_rate": 4.4052635834481025e-05,
      "loss": 0.0071,
      "step": 5000
    },
    {
      "epoch": 0.69,
      "learning_rate": 4.3822680507293455e-05,
      "loss": 0.0105,
      "step": 5100
    },
    {
      "epoch": 0.7,
      "learning_rate": 4.358898820573581e-05,
      "loss": 0.0108,
      "step": 5200
    },
    {
      "epoch": 0.71,
      "learning_rate": 4.3351605326076724e-05,
      "loss": 0.0063,
      "step": 5300
    },
    {
      "epoch": 0.73,
      "learning_rate": 4.3110578997296416e-05,
      "loss": 0.0147,
      "step": 5400
    },
    {
      "epoch": 0.74,
      "learning_rate": 4.286595707172986e-05,
      "loss": 0.0135,
      "step": 5500
    },
    {
      "epoch": 0.75,
      "learning_rate": 4.261778811556646e-05,
      "loss": 0.0105,
      "step": 5600
    },
    {
      "epoch": 0.77,
      "learning_rate": 4.236612139920786e-05,
      "loss": 0.0131,
      "step": 5700
    },
    {
      "epoch": 0.78,
      "learning_rate": 4.2111006887486035e-05,
      "loss": 0.0085,
      "step": 5800
    },
    {
      "epoch": 0.79,
      "learning_rate": 4.185249522974346e-05,
      "loss": 0.0052,
      "step": 5900
    },
    {
      "epoch": 0.81,
      "learning_rate": 4.159063774977748e-05,
      "loss": 0.008,
      "step": 6000
    }
  ],
  "logging_steps": 100,
  "max_steps": 22296,
  "num_train_epochs": 3,
  "save_steps": 3000,
  "total_flos": 2.424525286093947e+18,
  "trial_name": null,
  "trial_params": null
}