{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 5.888888888888889,
  "eval_steps": 500,
  "global_step": 42,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.25396825396825395,
      "grad_norm": 22.339479446411133,
      "learning_rate": 9.523809523809524e-05,
      "loss": 72.1042,
      "step": 2
    },
    {
      "epoch": 0.5079365079365079,
      "grad_norm": 23.86441421508789,
      "learning_rate": 9.047619047619048e-05,
      "loss": 70.3059,
      "step": 4
    },
    {
      "epoch": 0.7619047619047619,
      "grad_norm": 27.62677001953125,
      "learning_rate": 8.571428571428571e-05,
      "loss": 66.5863,
      "step": 6
    },
    {
      "epoch": 1.126984126984127,
      "grad_norm": 57.27507781982422,
      "learning_rate": 8.095238095238096e-05,
      "loss": 82.8912,
      "step": 8
    },
    {
      "epoch": 1.380952380952381,
      "grad_norm": 38.90316390991211,
      "learning_rate": 7.619047619047618e-05,
      "loss": 57.8037,
      "step": 10
    },
    {
      "epoch": 1.6349206349206349,
      "grad_norm": 28.1763916015625,
      "learning_rate": 7.142857142857143e-05,
      "loss": 52.8797,
      "step": 12
    },
    {
      "epoch": 1.8888888888888888,
      "grad_norm": 30.55934715270996,
      "learning_rate": 6.666666666666667e-05,
      "loss": 52.8374,
      "step": 14
    },
    {
      "epoch": 2.253968253968254,
      "grad_norm": 26.450092315673828,
      "learning_rate": 6.19047619047619e-05,
      "loss": 69.0564,
      "step": 16
    },
    {
      "epoch": 2.507936507936508,
      "grad_norm": 34.066776275634766,
      "learning_rate": 5.714285714285714e-05,
      "loss": 46.8763,
      "step": 18
    },
    {
      "epoch": 2.761904761904762,
      "grad_norm": 24.209619522094727,
      "learning_rate": 5.2380952380952384e-05,
      "loss": 46.5825,
      "step": 20
    },
    {
      "epoch": 3.126984126984127,
      "grad_norm": 35.07560729980469,
      "learning_rate": 4.761904761904762e-05,
      "loss": 62.4426,
      "step": 22
    },
    {
      "epoch": 3.380952380952381,
      "grad_norm": 14.254531860351562,
      "learning_rate": 4.2857142857142856e-05,
      "loss": 43.6682,
      "step": 24
    },
    {
      "epoch": 3.634920634920635,
      "grad_norm": 10.83181095123291,
      "learning_rate": 3.809523809523809e-05,
      "loss": 44.4259,
      "step": 26
    },
    {
      "epoch": 3.888888888888889,
      "grad_norm": 15.05122184753418,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 43.1349,
      "step": 28
    },
    {
      "epoch": 4.253968253968254,
      "grad_norm": 10.117740631103516,
      "learning_rate": 2.857142857142857e-05,
      "loss": 58.7522,
      "step": 30
    },
    {
      "epoch": 4.507936507936508,
      "grad_norm": 12.624483108520508,
      "learning_rate": 2.380952380952381e-05,
      "loss": 41.6464,
      "step": 32
    },
    {
      "epoch": 4.761904761904762,
      "grad_norm": 8.7578125,
      "learning_rate": 1.9047619047619046e-05,
      "loss": 41.2899,
      "step": 34
    },
    {
      "epoch": 5.1269841269841265,
      "grad_norm": 14.106217384338379,
      "learning_rate": 1.4285714285714285e-05,
      "loss": 58.7284,
      "step": 36
    },
    {
      "epoch": 5.380952380952381,
      "grad_norm": 11.060308456420898,
      "learning_rate": 9.523809523809523e-06,
      "loss": 39.9459,
      "step": 38
    },
    {
      "epoch": 5.634920634920634,
      "grad_norm": 11.781099319458008,
      "learning_rate": 4.7619047619047615e-06,
      "loss": 43.8549,
      "step": 40
    },
    {
      "epoch": 5.888888888888889,
      "grad_norm": 8.562162399291992,
      "learning_rate": 0.0,
      "loss": 42.2842,
      "step": 42
    },
    {
      "epoch": 5.888888888888889,
      "step": 42,
      "total_flos": 674107298249544.0,
      "train_loss": 54.19509578886486,
      "train_runtime": 1597.8044,
      "train_samples_per_second": 0.473,
      "train_steps_per_second": 0.026
    }
  ],
  "logging_steps": 2,
  "max_steps": 42,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 6,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 674107298249544.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}