{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.078412153883852,
  "eval_steps": 5,
  "global_step": 20,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0039206076941926,
      "grad_norm": 1.108206868171692,
      "learning_rate": 1e-05,
      "loss": 0.9312,
      "step": 1
    },
    {
      "epoch": 0.0039206076941926,
      "eval_loss": 1.099448561668396,
      "eval_runtime": 64.3877,
      "eval_samples_per_second": 6.678,
      "eval_steps_per_second": 3.339,
      "step": 1
    },
    {
      "epoch": 0.0078412153883852,
      "grad_norm": 1.2041150331497192,
      "learning_rate": 2e-05,
      "loss": 1.0575,
      "step": 2
    },
    {
      "epoch": 0.0117618230825778,
      "grad_norm": 1.2153946161270142,
      "learning_rate": 3e-05,
      "loss": 1.0726,
      "step": 3
    },
    {
      "epoch": 0.0156824307767704,
      "grad_norm": 1.1603235006332397,
      "learning_rate": 4e-05,
      "loss": 1.0766,
      "step": 4
    },
    {
      "epoch": 0.019603038470963,
      "grad_norm": 1.2363840341567993,
      "learning_rate": 5e-05,
      "loss": 1.0818,
      "step": 5
    },
    {
      "epoch": 0.019603038470963,
      "eval_loss": 0.9847230315208435,
      "eval_runtime": 6.9547,
      "eval_samples_per_second": 61.829,
      "eval_steps_per_second": 30.914,
      "step": 5
    },
    {
      "epoch": 0.0235236461651556,
      "grad_norm": 1.3062018156051636,
      "learning_rate": 6e-05,
      "loss": 1.1497,
      "step": 6
    },
    {
      "epoch": 0.0274442538593482,
      "grad_norm": 0.8951370120048523,
      "learning_rate": 7e-05,
      "loss": 0.9674,
      "step": 7
    },
    {
      "epoch": 0.0313648615535408,
      "grad_norm": 0.9190327525138855,
      "learning_rate": 8e-05,
      "loss": 0.8878,
      "step": 8
    },
    {
      "epoch": 0.035285469247733396,
      "grad_norm": 0.7922409772872925,
      "learning_rate": 9e-05,
      "loss": 0.7617,
      "step": 9
    },
    {
      "epoch": 0.039206076941926,
      "grad_norm": 0.9028505086898804,
      "learning_rate": 0.0001,
      "loss": 0.8743,
      "step": 10
    },
    {
      "epoch": 0.039206076941926,
      "eval_loss": 0.7476094365119934,
      "eval_runtime": 6.9611,
      "eval_samples_per_second": 61.772,
      "eval_steps_per_second": 30.886,
      "step": 10
    },
    {
      "epoch": 0.0431266846361186,
      "grad_norm": 0.7901967167854309,
      "learning_rate": 9.755282581475769e-05,
      "loss": 0.7681,
      "step": 11
    },
    {
      "epoch": 0.0470472923303112,
      "grad_norm": 0.7195383310317993,
      "learning_rate": 9.045084971874738e-05,
      "loss": 0.7462,
      "step": 12
    },
    {
      "epoch": 0.0509679000245038,
      "grad_norm": 0.7095159292221069,
      "learning_rate": 7.938926261462366e-05,
      "loss": 0.6836,
      "step": 13
    },
    {
      "epoch": 0.0548885077186964,
      "grad_norm": 0.5935850143432617,
      "learning_rate": 6.545084971874738e-05,
      "loss": 0.5753,
      "step": 14
    },
    {
      "epoch": 0.058809115412888996,
      "grad_norm": 0.5951647758483887,
      "learning_rate": 5e-05,
      "loss": 0.5539,
      "step": 15
    },
    {
      "epoch": 0.058809115412888996,
      "eval_loss": 0.5578025579452515,
      "eval_runtime": 6.9612,
      "eval_samples_per_second": 61.771,
      "eval_steps_per_second": 30.886,
      "step": 15
    },
    {
      "epoch": 0.0627297231070816,
      "grad_norm": 0.633608877658844,
      "learning_rate": 3.4549150281252636e-05,
      "loss": 0.5572,
      "step": 16
    },
    {
      "epoch": 0.0666503308012742,
      "grad_norm": 0.6298178434371948,
      "learning_rate": 2.061073738537635e-05,
      "loss": 0.5276,
      "step": 17
    },
    {
      "epoch": 0.07057093849546679,
      "grad_norm": 0.6569718718528748,
      "learning_rate": 9.549150281252633e-06,
      "loss": 0.5219,
      "step": 18
    },
    {
      "epoch": 0.07449154618965939,
      "grad_norm": 0.6659846305847168,
      "learning_rate": 2.4471741852423237e-06,
      "loss": 0.5346,
      "step": 19
    },
    {
      "epoch": 0.078412153883852,
      "grad_norm": 0.695641815662384,
      "learning_rate": 0.0,
      "loss": 0.4932,
      "step": 20
    },
    {
      "epoch": 0.078412153883852,
      "eval_loss": 0.5049919486045837,
      "eval_runtime": 6.9646,
      "eval_samples_per_second": 61.741,
      "eval_steps_per_second": 30.871,
      "step": 20
    }
  ],
  "logging_steps": 1,
  "max_steps": 20,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 5,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 5501993254649856.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}