checkpoint-200/trainer_state.json

{
  "best_metric": 0.6959671974182129,
  "best_model_checkpoint": "../../models/lora/llama-2-13b-multitask_cot_reflection/checkpoint-200",
  "epoch": 12.5,
  "global_step": 200,
  "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.1339,
      "step": 10
    },
    {
      "epoch": 1.25,
      "learning_rate": 5.9999999999999995e-05,
      "loss": 2.0943,
      "step": 20
    },
    {
      "epoch": 1.88,
      "learning_rate": 8.999999999999999e-05,
      "loss": 1.8574,
      "step": 30
    },
    {
      "epoch": 2.5,
      "learning_rate": 0.00011999999999999999,
      "loss": 1.4125,
      "step": 40
    },
    {
      "epoch": 3.12,
      "learning_rate": 0.00015,
      "loss": 1.0679,
      "step": 50
    },
    {
      "epoch": 3.75,
      "learning_rate": 0.00017999999999999998,
      "loss": 0.9275,
      "step": 60
    },
    {
      "epoch": 4.38,
      "learning_rate": 0.00020999999999999998,
      "loss": 0.8729,
      "step": 70
    },
    {
      "epoch": 5.0,
      "learning_rate": 0.00023999999999999998,
      "loss": 0.8581,
      "step": 80
    },
    {
      "epoch": 5.62,
      "learning_rate": 0.00027,
      "loss": 0.8374,
      "step": 90
    },
    {
      "epoch": 6.25,
      "learning_rate": 0.0003,
      "loss": 0.7997,
      "step": 100
    },
    {
      "epoch": 6.88,
      "learning_rate": 0.0002921052631578947,
      "loss": 0.7573,
      "step": 110
    },
    {
      "epoch": 7.5,
      "learning_rate": 0.0002842105263157894,
      "loss": 0.7445,
      "step": 120
    },
    {
      "epoch": 8.12,
      "learning_rate": 0.0002763157894736842,
      "loss": 0.7352,
      "step": 130
    },
    {
      "epoch": 8.75,
      "learning_rate": 0.0002684210526315789,
      "loss": 0.7108,
      "step": 140
    },
    {
      "epoch": 9.38,
      "learning_rate": 0.0002605263157894737,
      "loss": 0.7131,
      "step": 150
    },
    {
      "epoch": 10.0,
      "learning_rate": 0.00025263157894736836,
      "loss": 0.7051,
      "step": 160
    },
    {
      "epoch": 10.62,
      "learning_rate": 0.00024473684210526314,
      "loss": 0.708,
      "step": 170
    },
    {
      "epoch": 11.25,
      "learning_rate": 0.00023684210526315788,
      "loss": 0.6888,
      "step": 180
    },
    {
      "epoch": 11.88,
      "learning_rate": 0.00022894736842105263,
      "loss": 0.6695,
      "step": 190
    },
    {
      "epoch": 12.5,
      "learning_rate": 0.00022105263157894733,
      "loss": 0.6579,
      "step": 200
    },
    {
      "epoch": 12.5,
      "eval_loss": 0.6959671974182129,
      "eval_runtime": 1.6927,
      "eval_samples_per_second": 12.406,
      "eval_steps_per_second": 1.772,
      "step": 200
    }
  ],
  "max_steps": 480,
  "num_train_epochs": 30,
  "total_flos": 5.103476994686976e+17,
  "trial_name": null,
  "trial_params": null
}