{
  "best_metric": 2.1908211708068848,
  "best_model_checkpoint": "output/mistral_squad/checkpoint-1000",
  "epoch": 0.4910385465259023,
  "eval_steps": 1000,
  "global_step": 2000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 2.5e-05,
      "loss": 2.1039,
      "step": 25
    },
    {
      "epoch": 0.01,
      "learning_rate": 5e-05,
      "loss": 1.9227,
      "step": 50
    },
    {
      "epoch": 0.02,
      "learning_rate": 7.500000000000001e-05,
      "loss": 1.8559,
      "step": 75
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0001,
      "loss": 1.8136,
      "step": 100
    },
    {
      "epoch": 0.03,
      "learning_rate": 9.937075257991442e-05,
      "loss": 1.795,
      "step": 125
    },
    {
      "epoch": 0.04,
      "learning_rate": 9.874150515982885e-05,
      "loss": 1.7844,
      "step": 150
    },
    {
      "epoch": 0.04,
      "learning_rate": 9.811225773974327e-05,
      "loss": 1.7486,
      "step": 175
    },
    {
      "epoch": 0.05,
      "learning_rate": 9.74830103196577e-05,
      "loss": 1.719,
      "step": 200
    },
    {
      "epoch": 0.06,
      "learning_rate": 9.685376289957212e-05,
      "loss": 1.7017,
      "step": 225
    },
    {
      "epoch": 0.06,
      "learning_rate": 9.622451547948655e-05,
      "loss": 1.6495,
      "step": 250
    },
    {
      "epoch": 0.07,
      "learning_rate": 9.559526805940097e-05,
      "loss": 1.5982,
      "step": 275
    },
    {
      "epoch": 0.07,
      "learning_rate": 9.496602063931538e-05,
      "loss": 1.5912,
      "step": 300
    },
    {
      "epoch": 0.08,
      "learning_rate": 9.43367732192298e-05,
      "loss": 1.5603,
      "step": 325
    },
    {
      "epoch": 0.09,
      "learning_rate": 9.370752579914423e-05,
      "loss": 1.5453,
      "step": 350
    },
    {
      "epoch": 0.09,
      "learning_rate": 9.307827837905865e-05,
      "loss": 1.4757,
      "step": 375
    },
    {
      "epoch": 0.1,
      "learning_rate": 9.244903095897306e-05,
      "loss": 1.4421,
      "step": 400
    },
    {
      "epoch": 0.1,
      "learning_rate": 9.18197835388875e-05,
      "loss": 1.3738,
      "step": 425
    },
    {
      "epoch": 0.11,
      "learning_rate": 9.119053611880191e-05,
      "loss": 1.3607,
      "step": 450
    },
    {
      "epoch": 0.12,
      "learning_rate": 9.056128869871634e-05,
      "loss": 1.354,
      "step": 475
    },
    {
      "epoch": 0.12,
      "learning_rate": 8.993204127863076e-05,
      "loss": 1.3505,
      "step": 500
    },
    {
      "epoch": 0.13,
      "learning_rate": 8.930279385854519e-05,
      "loss": 1.3025,
      "step": 525
    },
    {
      "epoch": 0.14,
      "learning_rate": 8.867354643845961e-05,
      "loss": 1.2577,
      "step": 550
    },
    {
      "epoch": 0.14,
      "learning_rate": 8.804429901837403e-05,
      "loss": 1.3072,
      "step": 575
    },
    {
      "epoch": 0.15,
      "learning_rate": 8.741505159828846e-05,
      "loss": 1.2621,
      "step": 600
    },
    {
      "epoch": 0.15,
      "learning_rate": 8.678580417820287e-05,
      "loss": 1.213,
      "step": 625
    },
    {
      "epoch": 0.16,
      "learning_rate": 8.615655675811729e-05,
      "loss": 1.1523,
      "step": 650
    },
    {
      "epoch": 0.17,
      "learning_rate": 8.552730933803171e-05,
      "loss": 1.1324,
      "step": 675
    },
    {
      "epoch": 0.17,
      "learning_rate": 8.489806191794614e-05,
      "loss": 1.1531,
      "step": 700
    },
    {
      "epoch": 0.18,
      "learning_rate": 8.426881449786056e-05,
      "loss": 1.0632,
      "step": 725
    },
    {
      "epoch": 0.18,
      "learning_rate": 8.363956707777499e-05,
      "loss": 1.1238,
      "step": 750
    },
    {
      "epoch": 0.19,
      "learning_rate": 8.30103196576894e-05,
      "loss": 1.1197,
      "step": 775
    },
    {
      "epoch": 0.2,
      "learning_rate": 8.238107223760383e-05,
      "loss": 1.04,
      "step": 800
    },
    {
      "epoch": 0.2,
      "learning_rate": 8.175182481751825e-05,
      "loss": 1.1003,
      "step": 825
    },
    {
      "epoch": 0.21,
      "learning_rate": 8.112257739743267e-05,
      "loss": 1.0161,
      "step": 850
    },
    {
      "epoch": 0.21,
      "learning_rate": 8.04933299773471e-05,
      "loss": 0.972,
      "step": 875
    },
    {
      "epoch": 0.22,
      "learning_rate": 7.986408255726152e-05,
      "loss": 0.9713,
      "step": 900
    },
    {
      "epoch": 0.23,
      "learning_rate": 7.923483513717595e-05,
      "loss": 0.9788,
      "step": 925
    },
    {
      "epoch": 0.23,
      "learning_rate": 7.860558771709036e-05,
      "loss": 0.9189,
      "step": 950
    },
    {
      "epoch": 0.24,
      "learning_rate": 7.79763402970048e-05,
      "loss": 0.9001,
      "step": 975
    },
    {
      "epoch": 0.25,
      "learning_rate": 7.734709287691921e-05,
      "loss": 0.8809,
      "step": 1000
    },
    {
      "epoch": 0.25,
      "eval_loss": 2.1908211708068848,
      "eval_runtime": 596.3037,
      "eval_samples_per_second": 19.911,
      "eval_steps_per_second": 0.624,
      "step": 1000
    },
    {
      "epoch": 0.25,
      "learning_rate": 7.671784545683363e-05,
      "loss": 0.8418,
      "step": 1025
    },
    {
      "epoch": 0.26,
      "learning_rate": 7.608859803674805e-05,
      "loss": 0.8633,
      "step": 1050
    },
    {
      "epoch": 0.26,
      "learning_rate": 7.545935061666248e-05,
      "loss": 0.8767,
      "step": 1075
    },
    {
      "epoch": 0.27,
      "learning_rate": 7.48301031965769e-05,
      "loss": 0.8018,
      "step": 1100
    },
    {
      "epoch": 0.28,
      "learning_rate": 7.420085577649131e-05,
      "loss": 0.817,
      "step": 1125
    },
    {
      "epoch": 0.28,
      "learning_rate": 7.357160835640574e-05,
      "loss": 0.7738,
      "step": 1150
    },
    {
      "epoch": 0.29,
      "learning_rate": 7.294236093632016e-05,
      "loss": 0.7264,
      "step": 1175
    },
    {
      "epoch": 0.29,
      "learning_rate": 7.231311351623459e-05,
      "loss": 0.8269,
      "step": 1200
    },
    {
      "epoch": 0.3,
      "learning_rate": 7.168386609614901e-05,
      "loss": 0.7176,
      "step": 1225
    },
    {
      "epoch": 0.31,
      "learning_rate": 7.105461867606344e-05,
      "loss": 0.7344,
      "step": 1250
    },
    {
      "epoch": 0.31,
      "learning_rate": 7.042537125597786e-05,
      "loss": 0.7132,
      "step": 1275
    },
    {
      "epoch": 0.32,
      "learning_rate": 6.979612383589229e-05,
      "loss": 0.6838,
      "step": 1300
    },
    {
      "epoch": 0.33,
      "learning_rate": 6.91668764158067e-05,
      "loss": 0.7343,
      "step": 1325
    },
    {
      "epoch": 0.33,
      "learning_rate": 6.853762899572112e-05,
      "loss": 0.6807,
      "step": 1350
    },
    {
      "epoch": 0.34,
      "learning_rate": 6.790838157563554e-05,
      "loss": 0.6712,
      "step": 1375
    },
    {
      "epoch": 0.34,
      "learning_rate": 6.727913415554995e-05,
      "loss": 0.6808,
      "step": 1400
    },
    {
      "epoch": 0.35,
      "learning_rate": 6.664988673546439e-05,
      "loss": 0.6305,
      "step": 1425
    },
    {
      "epoch": 0.36,
      "learning_rate": 6.60206393153788e-05,
      "loss": 0.645,
      "step": 1450
    },
    {
      "epoch": 0.36,
      "learning_rate": 6.539139189529323e-05,
      "loss": 0.6229,
      "step": 1475
    },
    {
      "epoch": 0.37,
      "learning_rate": 6.476214447520765e-05,
      "loss": 0.5976,
      "step": 1500
    },
    {
      "epoch": 0.37,
      "learning_rate": 6.413289705512208e-05,
      "loss": 0.6214,
      "step": 1525
    },
    {
      "epoch": 0.38,
      "learning_rate": 6.35036496350365e-05,
      "loss": 0.5981,
      "step": 1550
    },
    {
      "epoch": 0.39,
      "learning_rate": 6.287440221495093e-05,
      "loss": 0.564,
      "step": 1575
    },
    {
      "epoch": 0.39,
      "learning_rate": 6.224515479486535e-05,
      "loss": 0.5757,
      "step": 1600
    },
    {
      "epoch": 0.4,
      "learning_rate": 6.161590737477976e-05,
      "loss": 0.5611,
      "step": 1625
    },
    {
      "epoch": 0.41,
      "learning_rate": 6.0986659954694194e-05,
      "loss": 0.5462,
      "step": 1650
    },
    {
      "epoch": 0.41,
      "learning_rate": 6.035741253460862e-05,
      "loss": 0.5172,
      "step": 1675
    },
    {
      "epoch": 0.42,
      "learning_rate": 5.9728165114523035e-05,
      "loss": 0.5087,
      "step": 1700
    },
    {
      "epoch": 0.42,
      "learning_rate": 5.909891769443746e-05,
      "loss": 0.5148,
      "step": 1725
    },
    {
      "epoch": 0.43,
      "learning_rate": 5.8469670274351876e-05,
      "loss": 0.5247,
      "step": 1750
    },
    {
      "epoch": 0.44,
      "learning_rate": 5.784042285426629e-05,
      "loss": 0.5356,
      "step": 1775
    },
    {
      "epoch": 0.44,
      "learning_rate": 5.721117543418072e-05,
      "loss": 0.4837,
      "step": 1800
    },
    {
      "epoch": 0.45,
      "learning_rate": 5.658192801409514e-05,
      "loss": 0.533,
      "step": 1825
    },
    {
      "epoch": 0.45,
      "learning_rate": 5.5952680594009565e-05,
      "loss": 0.485,
      "step": 1850
    },
    {
      "epoch": 0.46,
      "learning_rate": 5.532343317392399e-05,
      "loss": 0.4728,
      "step": 1875
    },
    {
      "epoch": 0.47,
      "learning_rate": 5.469418575383841e-05,
      "loss": 0.4873,
      "step": 1900
    },
    {
      "epoch": 0.47,
      "learning_rate": 5.406493833375284e-05,
      "loss": 0.4693,
      "step": 1925
    },
    {
      "epoch": 0.48,
      "learning_rate": 5.343569091366726e-05,
      "loss": 0.4951,
      "step": 1950
    },
    {
      "epoch": 0.48,
      "learning_rate": 5.280644349358168e-05,
      "loss": 0.4445,
      "step": 1975
    },
    {
      "epoch": 0.49,
      "learning_rate": 5.21771960734961e-05,
      "loss": 0.4423,
      "step": 2000
    },
    {
      "epoch": 0.49,
      "eval_loss": 2.3274576663970947,
      "eval_runtime": 596.4629,
      "eval_samples_per_second": 19.906,
      "eval_steps_per_second": 0.624,
      "step": 2000
    }
  ],
  "logging_steps": 25,
  "max_steps": 4073,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 1000,
  "total_flos": 7.64977027743744e+17,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}