{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.05485463521667581,
  "eval_steps": 5,
  "global_step": 25,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0021941854086670325,
      "grad_norm": NaN,
      "learning_rate": 0.0,
      "loss": 9.7459,
      "step": 1
    },
    {
      "epoch": 0.0021941854086670325,
      "eval_loss": 9.789505004882812,
      "eval_runtime": 23.1683,
      "eval_samples_per_second": 8.287,
      "eval_steps_per_second": 4.144,
      "step": 1
    },
    {
      "epoch": 0.004388370817334065,
      "grad_norm": 17.736719131469727,
      "learning_rate": 1e-05,
      "loss": 10.0567,
      "step": 2
    },
    {
      "epoch": 0.006582556226001097,
      "grad_norm": 18.50017738342285,
      "learning_rate": 2e-05,
      "loss": 9.2557,
      "step": 3
    },
    {
      "epoch": 0.00877674163466813,
      "grad_norm": 17.195247650146484,
      "learning_rate": 3e-05,
      "loss": 8.6156,
      "step": 4
    },
    {
      "epoch": 0.010970927043335162,
      "grad_norm": 18.478811264038086,
      "learning_rate": 4e-05,
      "loss": 8.6089,
      "step": 5
    },
    {
      "epoch": 0.010970927043335162,
      "eval_loss": 8.13350772857666,
      "eval_runtime": 22.7364,
      "eval_samples_per_second": 8.445,
      "eval_steps_per_second": 4.222,
      "step": 5
    },
    {
      "epoch": 0.013165112452002194,
      "grad_norm": 14.635212898254395,
      "learning_rate": 5e-05,
      "loss": 8.6232,
      "step": 6
    },
    {
      "epoch": 0.015359297860669226,
      "grad_norm": 13.62023639678955,
      "learning_rate": 6e-05,
      "loss": 6.725,
      "step": 7
    },
    {
      "epoch": 0.01755348326933626,
      "grad_norm": 14.408036231994629,
      "learning_rate": 7e-05,
      "loss": 5.464,
      "step": 8
    },
    {
      "epoch": 0.019747668678003292,
      "grad_norm": 17.330280303955078,
      "learning_rate": 8e-05,
      "loss": 4.2439,
      "step": 9
    },
    {
      "epoch": 0.021941854086670324,
      "grad_norm": 15.395853996276855,
      "learning_rate": 9e-05,
      "loss": 2.9754,
      "step": 10
    },
    {
      "epoch": 0.021941854086670324,
      "eval_loss": 1.9544070959091187,
      "eval_runtime": 22.7951,
      "eval_samples_per_second": 8.423,
      "eval_steps_per_second": 4.211,
      "step": 10
    },
    {
      "epoch": 0.024136039495337356,
      "grad_norm": 14.372428894042969,
      "learning_rate": 0.0001,
      "loss": 2.1836,
      "step": 11
    },
    {
      "epoch": 0.026330224904004388,
      "grad_norm": 10.696700096130371,
      "learning_rate": 9.98458666866564e-05,
      "loss": 0.8312,
      "step": 12
    },
    {
      "epoch": 0.02852441031267142,
      "grad_norm": 6.342767238616943,
      "learning_rate": 9.938441702975689e-05,
      "loss": 0.6199,
      "step": 13
    },
    {
      "epoch": 0.030718595721338452,
      "grad_norm": 8.464292526245117,
      "learning_rate": 9.861849601988383e-05,
      "loss": 0.5169,
      "step": 14
    },
    {
      "epoch": 0.03291278113000549,
      "grad_norm": 7.956654071807861,
      "learning_rate": 9.755282581475769e-05,
      "loss": 0.765,
      "step": 15
    },
    {
      "epoch": 0.03291278113000549,
      "eval_loss": 0.5048862099647522,
      "eval_runtime": 22.8072,
      "eval_samples_per_second": 8.418,
      "eval_steps_per_second": 4.209,
      "step": 15
    },
    {
      "epoch": 0.03510696653867252,
      "grad_norm": 10.383864402770996,
      "learning_rate": 9.619397662556435e-05,
      "loss": 0.7324,
      "step": 16
    },
    {
      "epoch": 0.03730115194733955,
      "grad_norm": 4.3315110206604,
      "learning_rate": 9.45503262094184e-05,
      "loss": 0.4558,
      "step": 17
    },
    {
      "epoch": 0.039495337356006584,
      "grad_norm": 19.855714797973633,
      "learning_rate": 9.263200821770461e-05,
      "loss": 1.2013,
      "step": 18
    },
    {
      "epoch": 0.041689522764673616,
      "grad_norm": 8.671701431274414,
      "learning_rate": 9.045084971874738e-05,
      "loss": 0.8788,
      "step": 19
    },
    {
      "epoch": 0.04388370817334065,
      "grad_norm": 8.814596176147461,
      "learning_rate": 8.802029828000156e-05,
      "loss": 0.9201,
      "step": 20
    },
    {
      "epoch": 0.04388370817334065,
      "eval_loss": 0.7541892528533936,
      "eval_runtime": 22.8305,
      "eval_samples_per_second": 8.41,
      "eval_steps_per_second": 4.205,
      "step": 20
    },
    {
      "epoch": 0.04607789358200768,
      "grad_norm": 1.8071426153182983,
      "learning_rate": 8.535533905932738e-05,
      "loss": 0.4432,
      "step": 21
    },
    {
      "epoch": 0.04827207899067471,
      "grad_norm": 4.000314235687256,
      "learning_rate": 8.247240241650918e-05,
      "loss": 0.6024,
      "step": 22
    },
    {
      "epoch": 0.050466264399341744,
      "grad_norm": 4.672601222991943,
      "learning_rate": 7.938926261462366e-05,
      "loss": 0.494,
      "step": 23
    },
    {
      "epoch": 0.052660449808008776,
      "grad_norm": 2.561952590942383,
      "learning_rate": 7.612492823579745e-05,
      "loss": 0.4829,
      "step": 24
    },
    {
      "epoch": 0.05485463521667581,
      "grad_norm": 5.277298927307129,
      "learning_rate": 7.269952498697734e-05,
      "loss": 0.5405,
      "step": 25
    },
    {
      "epoch": 0.05485463521667581,
      "eval_loss": 0.5149335861206055,
      "eval_runtime": 22.8748,
      "eval_samples_per_second": 8.394,
      "eval_steps_per_second": 4.197,
      "step": 25
    }
  ],
  "logging_steps": 1,
  "max_steps": 50,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 25,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 9273591044505600.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}