{
  "best_metric": 0.7148585915565491,
  "best_model_checkpoint": "../../models/lora/llama-2-7b-multitask_cot_reflection/checkpoint-200",
  "epoch": 25.0,
  "global_step": 400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.62,
      "learning_rate": 2.9999999999999997e-05,
      "loss": 2.4102,
      "step": 10
    },
    {
      "epoch": 1.25,
      "learning_rate": 5.9999999999999995e-05,
      "loss": 2.3324,
      "step": 20
    },
    {
      "epoch": 1.88,
      "learning_rate": 8.999999999999999e-05,
      "loss": 1.9864,
      "step": 30
    },
    {
      "epoch": 2.5,
      "learning_rate": 0.00011999999999999999,
      "loss": 1.6398,
      "step": 40
    },
    {
      "epoch": 3.12,
      "learning_rate": 0.00015,
      "loss": 1.2703,
      "step": 50
    },
    {
      "epoch": 3.75,
      "learning_rate": 0.00017999999999999998,
      "loss": 1.0435,
      "step": 60
    },
    {
      "epoch": 4.38,
      "learning_rate": 0.00020999999999999998,
      "loss": 0.9847,
      "step": 70
    },
    {
      "epoch": 5.0,
      "learning_rate": 0.00023999999999999998,
      "loss": 0.956,
      "step": 80
    },
    {
      "epoch": 5.62,
      "learning_rate": 0.00027,
      "loss": 0.9321,
      "step": 90
    },
    {
      "epoch": 6.25,
      "learning_rate": 0.0003,
      "loss": 0.8775,
      "step": 100
    },
    {
      "epoch": 6.88,
      "learning_rate": 0.0002921052631578947,
      "loss": 0.818,
      "step": 110
    },
    {
      "epoch": 7.5,
      "learning_rate": 0.0002842105263157894,
      "loss": 0.8128,
      "step": 120
    },
    {
      "epoch": 8.12,
      "learning_rate": 0.0002763157894736842,
      "loss": 0.78,
      "step": 130
    },
    {
      "epoch": 8.75,
      "learning_rate": 0.0002684210526315789,
      "loss": 0.7734,
      "step": 140
    },
    {
      "epoch": 9.38,
      "learning_rate": 0.0002605263157894737,
      "loss": 0.7761,
      "step": 150
    },
    {
      "epoch": 10.0,
      "learning_rate": 0.00025263157894736836,
      "loss": 0.7668,
      "step": 160
    },
    {
      "epoch": 10.62,
      "learning_rate": 0.00024473684210526314,
      "loss": 0.7732,
      "step": 170
    },
    {
      "epoch": 11.25,
      "learning_rate": 0.00023684210526315788,
      "loss": 0.7536,
      "step": 180
    },
    {
      "epoch": 11.88,
      "learning_rate": 0.00022894736842105263,
      "loss": 0.7406,
      "step": 190
    },
    {
      "epoch": 12.5,
      "learning_rate": 0.00022105263157894733,
      "loss": 0.7288,
      "step": 200
    },
    {
      "epoch": 12.5,
      "eval_loss": 0.7148585915565491,
      "eval_runtime": 1.0356,
      "eval_samples_per_second": 20.277,
      "eval_steps_per_second": 2.897,
      "step": 200
    },
    {
      "epoch": 13.12,
      "learning_rate": 0.00021315789473684208,
      "loss": 0.7373,
      "step": 210
    },
    {
      "epoch": 13.75,
      "learning_rate": 0.00020526315789473683,
      "loss": 0.6982,
      "step": 220
    },
    {
      "epoch": 14.38,
      "learning_rate": 0.00019736842105263157,
      "loss": 0.7172,
      "step": 230
    },
    {
      "epoch": 15.0,
      "learning_rate": 0.0001894736842105263,
      "loss": 0.7116,
      "step": 240
    },
    {
      "epoch": 15.62,
      "learning_rate": 0.00018157894736842105,
      "loss": 0.7176,
      "step": 250
    },
    {
      "epoch": 16.25,
      "learning_rate": 0.0001736842105263158,
      "loss": 0.6915,
      "step": 260
    },
    {
      "epoch": 16.88,
      "learning_rate": 0.00016578947368421052,
      "loss": 0.6805,
      "step": 270
    },
    {
      "epoch": 17.5,
      "learning_rate": 0.00015789473684210524,
      "loss": 0.6726,
      "step": 280
    },
    {
      "epoch": 18.12,
      "learning_rate": 0.00015,
      "loss": 0.6736,
      "step": 290
    },
    {
      "epoch": 18.75,
      "learning_rate": 0.0001421052631578947,
      "loss": 0.6431,
      "step": 300
    },
    {
      "epoch": 19.38,
      "learning_rate": 0.00013421052631578946,
      "loss": 0.6579,
      "step": 310
    },
    {
      "epoch": 20.0,
      "learning_rate": 0.00012631578947368418,
      "loss": 0.6454,
      "step": 320
    },
    {
      "epoch": 20.62,
      "learning_rate": 0.00011842105263157894,
      "loss": 0.6592,
      "step": 330
    },
    {
      "epoch": 21.25,
      "learning_rate": 0.00011052631578947366,
      "loss": 0.6382,
      "step": 340
    },
    {
      "epoch": 21.88,
      "learning_rate": 0.00010263157894736841,
      "loss": 0.6219,
      "step": 350
    },
    {
      "epoch": 22.5,
      "learning_rate": 9.473684210526315e-05,
      "loss": 0.6203,
      "step": 360
    },
    {
      "epoch": 23.12,
      "learning_rate": 8.68421052631579e-05,
      "loss": 0.6126,
      "step": 370
    },
    {
      "epoch": 23.75,
      "learning_rate": 7.894736842105262e-05,
      "loss": 0.5904,
      "step": 380
    },
    {
      "epoch": 24.38,
      "learning_rate": 7.105263157894735e-05,
      "loss": 0.6079,
      "step": 390
    },
    {
      "epoch": 25.0,
      "learning_rate": 6.315789473684209e-05,
      "loss": 0.5885,
      "step": 400
    },
    {
      "epoch": 25.0,
      "eval_loss": 0.7490605711936951,
      "eval_runtime": 1.0337,
      "eval_samples_per_second": 20.315,
      "eval_steps_per_second": 2.902,
      "step": 400
    }
  ],
  "max_steps": 480,
  "num_train_epochs": 30,
  "total_flos": 5.331925003272192e+17,
  "trial_name": null,
  "trial_params": null
}