{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.06325110689437065,
  "eval_steps": 25,
  "global_step": 50,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.001265022137887413,
      "grad_norm": 35.139339447021484,
      "learning_rate": 5e-05,
      "loss": 4.1852,
      "step": 1
    },
    {
      "epoch": 0.001265022137887413,
      "eval_loss": 2.3645200729370117,
      "eval_runtime": 4.0111,
      "eval_samples_per_second": 12.465,
      "eval_steps_per_second": 3.241,
      "step": 1
    },
    {
      "epoch": 0.002530044275774826,
      "grad_norm": 7.04626989364624,
      "learning_rate": 0.0001,
      "loss": 1.8438,
      "step": 2
    },
    {
      "epoch": 0.003795066413662239,
      "grad_norm": 1.121877670288086,
      "learning_rate": 9.990365154573717e-05,
      "loss": 1.9922,
      "step": 3
    },
    {
      "epoch": 0.005060088551549652,
      "grad_norm": 1.3969320058822632,
      "learning_rate": 9.961501876182148e-05,
      "loss": 2.0901,
      "step": 4
    },
    {
      "epoch": 0.006325110689437065,
      "grad_norm": 0.9031108021736145,
      "learning_rate": 9.913533761814537e-05,
      "loss": 2.0232,
      "step": 5
    },
    {
      "epoch": 0.007590132827324478,
      "grad_norm": 0.9523929953575134,
      "learning_rate": 9.846666218300807e-05,
      "loss": 2.0277,
      "step": 6
    },
    {
      "epoch": 0.008855154965211892,
      "grad_norm": 0.9598959684371948,
      "learning_rate": 9.761185582727977e-05,
      "loss": 2.0421,
      "step": 7
    },
    {
      "epoch": 0.010120177103099304,
      "grad_norm": 1.3133834600448608,
      "learning_rate": 9.657457896300791e-05,
      "loss": 2.0316,
      "step": 8
    },
    {
      "epoch": 0.011385199240986717,
      "grad_norm": 10.652119636535645,
      "learning_rate": 9.535927336897098e-05,
      "loss": 1.9813,
      "step": 9
    },
    {
      "epoch": 0.01265022137887413,
      "grad_norm": 2.1253440380096436,
      "learning_rate": 9.397114317029975e-05,
      "loss": 2.0884,
      "step": 10
    },
    {
      "epoch": 0.013915243516761544,
      "grad_norm": 1.9068113565444946,
      "learning_rate": 9.241613255361455e-05,
      "loss": 2.1017,
      "step": 11
    },
    {
      "epoch": 0.015180265654648957,
      "grad_norm": 1.6747380495071411,
      "learning_rate": 9.070090031310558e-05,
      "loss": 2.2796,
      "step": 12
    },
    {
      "epoch": 0.01644528779253637,
      "grad_norm": 29.196680068969727,
      "learning_rate": 8.883279133655399e-05,
      "loss": 7.4892,
      "step": 13
    },
    {
      "epoch": 0.017710309930423784,
      "grad_norm": 2.4710311889648438,
      "learning_rate": 8.681980515339464e-05,
      "loss": 1.8346,
      "step": 14
    },
    {
      "epoch": 0.018975332068311195,
      "grad_norm": 1.579148292541504,
      "learning_rate": 8.467056167950311e-05,
      "loss": 1.9865,
      "step": 15
    },
    {
      "epoch": 0.02024035420619861,
      "grad_norm": 1.1874217987060547,
      "learning_rate": 8.239426430539243e-05,
      "loss": 1.8998,
      "step": 16
    },
    {
      "epoch": 0.021505376344086023,
      "grad_norm": 0.92023766040802,
      "learning_rate": 8.000066048588211e-05,
      "loss": 1.8687,
      "step": 17
    },
    {
      "epoch": 0.022770398481973434,
      "grad_norm": 0.9192359447479248,
      "learning_rate": 7.75e-05,
      "loss": 1.9686,
      "step": 18
    },
    {
      "epoch": 0.02403542061986085,
      "grad_norm": 3.2033963203430176,
      "learning_rate": 7.490299105985507e-05,
      "loss": 2.1434,
      "step": 19
    },
    {
      "epoch": 0.02530044275774826,
      "grad_norm": 1.0458755493164062,
      "learning_rate": 7.222075445642904e-05,
      "loss": 2.0357,
      "step": 20
    },
    {
      "epoch": 0.026565464895635674,
      "grad_norm": 1.1212831735610962,
      "learning_rate": 6.946477593864228e-05,
      "loss": 1.8479,
      "step": 21
    },
    {
      "epoch": 0.027830487033523088,
      "grad_norm": 1.1566752195358276,
      "learning_rate": 6.664685702961344e-05,
      "loss": 2.0032,
      "step": 22
    },
    {
      "epoch": 0.0290955091714105,
      "grad_norm": 1.2244004011154175,
      "learning_rate": 6.377906449072578e-05,
      "loss": 2.0365,
      "step": 23
    },
    {
      "epoch": 0.030360531309297913,
      "grad_norm": 1.3064579963684082,
      "learning_rate": 6.087367864990233e-05,
      "loss": 2.1573,
      "step": 24
    },
    {
      "epoch": 0.031625553447185324,
      "grad_norm": 1.693029522895813,
      "learning_rate": 5.794314081535644e-05,
      "loss": 2.2897,
      "step": 25
    },
    {
      "epoch": 0.031625553447185324,
      "eval_loss": 1.932918667793274,
      "eval_runtime": 4.0675,
      "eval_samples_per_second": 12.293,
      "eval_steps_per_second": 3.196,
      "step": 25
    },
    {
      "epoch": 0.03289057558507274,
      "grad_norm": 37.96943283081055,
      "learning_rate": 5.500000000000001e-05,
      "loss": 5.7969,
      "step": 26
    },
    {
      "epoch": 0.03415559772296015,
      "grad_norm": 1.5687322616577148,
      "learning_rate": 5.205685918464356e-05,
      "loss": 1.7264,
      "step": 27
    },
    {
      "epoch": 0.03542061986084757,
      "grad_norm": 1.1628891229629517,
      "learning_rate": 4.912632135009769e-05,
      "loss": 1.8082,
      "step": 28
    },
    {
      "epoch": 0.036685641998734975,
      "grad_norm": 0.8394140005111694,
      "learning_rate": 4.6220935509274235e-05,
      "loss": 1.9195,
      "step": 29
    },
    {
      "epoch": 0.03795066413662239,
      "grad_norm": 0.9346224665641785,
      "learning_rate": 4.3353142970386564e-05,
      "loss": 1.9093,
      "step": 30
    },
    {
      "epoch": 0.0392156862745098,
      "grad_norm": 0.9901193976402283,
      "learning_rate": 4.053522406135775e-05,
      "loss": 2.0222,
      "step": 31
    },
    {
      "epoch": 0.04048070841239722,
      "grad_norm": 0.9796273112297058,
      "learning_rate": 3.777924554357096e-05,
      "loss": 1.9273,
      "step": 32
    },
    {
      "epoch": 0.04174573055028463,
      "grad_norm": 0.9287394881248474,
      "learning_rate": 3.509700894014496e-05,
      "loss": 1.9015,
      "step": 33
    },
    {
      "epoch": 0.043010752688172046,
      "grad_norm": 0.9738181233406067,
      "learning_rate": 3.250000000000001e-05,
      "loss": 1.7521,
      "step": 34
    },
    {
      "epoch": 0.044275774826059454,
      "grad_norm": 1.0583573579788208,
      "learning_rate": 2.9999339514117912e-05,
      "loss": 2.0511,
      "step": 35
    },
    {
      "epoch": 0.04554079696394687,
      "grad_norm": 1.0245318412780762,
      "learning_rate": 2.760573569460757e-05,
      "loss": 2.1022,
      "step": 36
    },
    {
      "epoch": 0.04680581910183428,
      "grad_norm": 1.2444181442260742,
      "learning_rate": 2.53294383204969e-05,
      "loss": 2.0539,
      "step": 37
    },
    {
      "epoch": 0.0480708412397217,
      "grad_norm": 24.339767456054688,
      "learning_rate": 2.3180194846605367e-05,
      "loss": 3.6826,
      "step": 38
    },
    {
      "epoch": 0.04933586337760911,
      "grad_norm": 0.6691582202911377,
      "learning_rate": 2.1167208663446025e-05,
      "loss": 1.6845,
      "step": 39
    },
    {
      "epoch": 0.05060088551549652,
      "grad_norm": 0.7083836793899536,
      "learning_rate": 1.9299099686894423e-05,
      "loss": 1.776,
      "step": 40
    },
    {
      "epoch": 0.05186590765338393,
      "grad_norm": 0.7264069318771362,
      "learning_rate": 1.758386744638546e-05,
      "loss": 1.8355,
      "step": 41
    },
    {
      "epoch": 0.05313092979127135,
      "grad_norm": 0.7508512139320374,
      "learning_rate": 1.602885682970026e-05,
      "loss": 1.8773,
      "step": 42
    },
    {
      "epoch": 0.05439595192915876,
      "grad_norm": 0.7919434905052185,
      "learning_rate": 1.464072663102903e-05,
      "loss": 1.8648,
      "step": 43
    },
    {
      "epoch": 0.055660974067046176,
      "grad_norm": 0.8442027568817139,
      "learning_rate": 1.3425421036992098e-05,
      "loss": 1.7584,
      "step": 44
    },
    {
      "epoch": 0.056925996204933584,
      "grad_norm": 0.8228534460067749,
      "learning_rate": 1.2388144172720251e-05,
      "loss": 1.9638,
      "step": 45
    },
    {
      "epoch": 0.058191018342821,
      "grad_norm": 0.8046168088912964,
      "learning_rate": 1.1533337816991932e-05,
      "loss": 1.8872,
      "step": 46
    },
    {
      "epoch": 0.05945604048070841,
      "grad_norm": 0.8686741590499878,
      "learning_rate": 1.0864662381854632e-05,
      "loss": 1.89,
      "step": 47
    },
    {
      "epoch": 0.06072106261859583,
      "grad_norm": 0.9693074226379395,
      "learning_rate": 1.0384981238178534e-05,
      "loss": 1.9709,
      "step": 48
    },
    {
      "epoch": 0.06198608475648324,
      "grad_norm": 1.1496561765670776,
      "learning_rate": 1.0096348454262845e-05,
      "loss": 2.0629,
      "step": 49
    },
    {
      "epoch": 0.06325110689437065,
      "grad_norm": 1.4522725343704224,
      "learning_rate": 1e-05,
      "loss": 2.1552,
      "step": 50
    },
    {
      "epoch": 0.06325110689437065,
      "eval_loss": 1.9020295143127441,
      "eval_runtime": 4.0552,
      "eval_samples_per_second": 12.33,
      "eval_steps_per_second": 3.206,
      "step": 50
    }
  ],
  "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": true
      },
      "attributes": {}
    }
  },
  "total_flos": 6.174479344467968e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}