{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.100840336134454,
  "eval_steps": 100,
  "global_step": 500,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.10504201680672269,
      "grad_norm": 8.704133033752441,
      "learning_rate": 4.166666666666667e-05,
      "loss": 3.5481,
      "step": 25
    },
    {
      "epoch": 0.21008403361344538,
      "grad_norm": 10.129899978637695,
      "learning_rate": 4.9947570655942796e-05,
      "loss": 3.0062,
      "step": 50
    },
    {
      "epoch": 0.31512605042016806,
      "grad_norm": 6.111865997314453,
      "learning_rate": 4.9734953280908904e-05,
      "loss": 2.9095,
      "step": 75
    },
    {
      "epoch": 0.42016806722689076,
      "grad_norm": 4.366275787353516,
      "learning_rate": 4.936026311617316e-05,
      "loss": 2.9062,
      "step": 100
    },
    {
      "epoch": 0.42016806722689076,
      "eval_loss": 2.872298240661621,
      "eval_runtime": 35.9441,
      "eval_samples_per_second": 13.02,
      "eval_steps_per_second": 1.641,
      "step": 100
    },
    {
      "epoch": 0.5252100840336135,
      "grad_norm": 25.31424903869629,
      "learning_rate": 4.882595527372152e-05,
      "loss": 2.874,
      "step": 125
    },
    {
      "epoch": 0.6302521008403361,
      "grad_norm": 53.229515075683594,
      "learning_rate": 4.813553074106761e-05,
      "loss": 2.887,
      "step": 150
    },
    {
      "epoch": 0.7352941176470589,
      "grad_norm": 7.568065166473389,
      "learning_rate": 4.7293513441455364e-05,
      "loss": 2.8903,
      "step": 175
    },
    {
      "epoch": 0.8403361344537815,
      "grad_norm": 6.394404888153076,
      "learning_rate": 4.630542059139924e-05,
      "loss": 2.7455,
      "step": 200
    },
    {
      "epoch": 0.8403361344537815,
      "eval_loss": 2.743553638458252,
      "eval_runtime": 35.9282,
      "eval_samples_per_second": 13.026,
      "eval_steps_per_second": 1.642,
      "step": 200
    },
    {
      "epoch": 0.9453781512605042,
      "grad_norm": 3.6301393508911133,
      "learning_rate": 4.517772654979023e-05,
      "loss": 2.7609,
      "step": 225
    },
    {
      "epoch": 1.050420168067227,
      "grad_norm": 21.900226593017578,
      "learning_rate": 4.391782039544238e-05,
      "loss": 2.7411,
      "step": 250
    },
    {
      "epoch": 1.1554621848739495,
      "grad_norm": 9.00839900970459,
      "learning_rate": 4.253395751104748e-05,
      "loss": 2.3212,
      "step": 275
    },
    {
      "epoch": 1.2605042016806722,
      "grad_norm": 3.9047327041625977,
      "learning_rate": 4.10352054907785e-05,
      "loss": 2.0506,
      "step": 300
    },
    {
      "epoch": 1.2605042016806722,
      "eval_loss": 2.8556525707244873,
      "eval_runtime": 35.8984,
      "eval_samples_per_second": 13.037,
      "eval_steps_per_second": 1.644,
      "step": 300
    },
    {
      "epoch": 1.365546218487395,
      "grad_norm": 3.455016613006592,
      "learning_rate": 3.943138472597549e-05,
      "loss": 1.826,
      "step": 325
    },
    {
      "epoch": 1.4705882352941178,
      "grad_norm": 10.288714408874512,
      "learning_rate": 3.773300405821908e-05,
      "loss": 1.8779,
      "step": 350
    },
    {
      "epoch": 1.5756302521008403,
      "grad_norm": 2.957385540008545,
      "learning_rate": 3.595119192141706e-05,
      "loss": 1.85,
      "step": 375
    },
    {
      "epoch": 1.680672268907563,
      "grad_norm": 3.40120267868042,
      "learning_rate": 3.409762342408719e-05,
      "loss": 1.835,
      "step": 400
    },
    {
      "epoch": 1.680672268907563,
      "eval_loss": 2.7827320098876953,
      "eval_runtime": 36.0584,
      "eval_samples_per_second": 12.979,
      "eval_steps_per_second": 1.636,
      "step": 400
    },
    {
      "epoch": 1.7857142857142856,
      "grad_norm": 3.446842670440674,
      "learning_rate": 3.218444384962071e-05,
      "loss": 1.8477,
      "step": 425
    },
    {
      "epoch": 1.8907563025210083,
      "grad_norm": 3.1697998046875,
      "learning_rate": 3.0224189075781884e-05,
      "loss": 1.7934,
      "step": 450
    },
    {
      "epoch": 1.995798319327731,
      "grad_norm": 2.8750455379486084,
      "learning_rate": 2.8229703434885163e-05,
      "loss": 1.7526,
      "step": 475
    },
    {
      "epoch": 2.100840336134454,
      "grad_norm": 2.977962017059326,
      "learning_rate": 2.621405555286121e-05,
      "loss": 0.8246,
      "step": 500
    },
    {
      "epoch": 2.100840336134454,
      "eval_loss": 3.001242160797119,
      "eval_runtime": 35.8998,
      "eval_samples_per_second": 13.036,
      "eval_steps_per_second": 1.643,
      "step": 500
    }
  ],
  "logging_steps": 25,
  "max_steps": 1000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 250,
  "total_flos": 4.61102588624896e+16,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}