{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0,
  "eval_steps": 500,
  "global_step": 58,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03508771929824561,
      "grad_norm": 2.0672700448576804,
      "learning_rate": 0.0,
      "loss": 0.866,
      "step": 1
    },
    {
      "epoch": 0.07017543859649122,
      "grad_norm": 2.2902360202645,
      "learning_rate": 7.142857142857143e-07,
      "loss": 1.043,
      "step": 2
    },
    {
      "epoch": 0.10526315789473684,
      "grad_norm": 2.064587564315747,
      "learning_rate": 1.4285714285714286e-06,
      "loss": 0.9479,
      "step": 3
    },
    {
      "epoch": 0.14035087719298245,
      "grad_norm": 2.1456251662755625,
      "learning_rate": 2.1428571428571427e-06,
      "loss": 1.0053,
      "step": 4
    },
    {
      "epoch": 0.17543859649122806,
      "grad_norm": 2.118290312395124,
      "learning_rate": 2.8571428571428573e-06,
      "loss": 1.0228,
      "step": 5
    },
    {
      "epoch": 0.21052631578947367,
      "grad_norm": 1.9727110398188104,
      "learning_rate": 3.5714285714285718e-06,
      "loss": 0.9994,
      "step": 6
    },
    {
      "epoch": 0.24561403508771928,
      "grad_norm": 1.811780274017235,
      "learning_rate": 4.2857142857142855e-06,
      "loss": 0.9832,
      "step": 7
    },
    {
      "epoch": 0.2807017543859649,
      "grad_norm": 1.60280335465616,
      "learning_rate": 5e-06,
      "loss": 1.0057,
      "step": 8
    },
    {
      "epoch": 0.3157894736842105,
      "grad_norm": 1.4946232465242921,
      "learning_rate": 4.999302593030069e-06,
      "loss": 1.0198,
      "step": 9
    },
    {
      "epoch": 0.3508771929824561,
      "grad_norm": 1.219848341043154,
      "learning_rate": 4.99721076122146e-06,
      "loss": 0.9945,
      "step": 10
    },
    {
      "epoch": 0.38596491228070173,
      "grad_norm": 1.0507879904789832,
      "learning_rate": 4.99372567166064e-06,
      "loss": 0.8896,
      "step": 11
    },
    {
      "epoch": 0.42105263157894735,
      "grad_norm": 1.2565080469257828,
      "learning_rate": 4.98884926876821e-06,
      "loss": 0.9066,
      "step": 12
    },
    {
      "epoch": 0.45614035087719296,
      "grad_norm": 1.3416635061508715,
      "learning_rate": 4.982584273214061e-06,
      "loss": 0.973,
      "step": 13
    },
    {
      "epoch": 0.49122807017543857,
      "grad_norm": 1.1635476799198454,
      "learning_rate": 4.974934180399447e-06,
      "loss": 0.8802,
      "step": 14
    },
    {
      "epoch": 0.5263157894736842,
      "grad_norm": 1.2273198837996009,
      "learning_rate": 4.965903258506806e-06,
      "loss": 0.9058,
      "step": 15
    },
    {
      "epoch": 0.5614035087719298,
      "grad_norm": 0.9397916351800223,
      "learning_rate": 4.955496546118439e-06,
      "loss": 0.851,
      "step": 16
    },
    {
      "epoch": 0.5964912280701754,
      "grad_norm": 0.8970172914695361,
      "learning_rate": 4.943719849405347e-06,
      "loss": 0.959,
      "step": 17
    },
    {
      "epoch": 0.631578947368421,
      "grad_norm": 0.8557735235888474,
      "learning_rate": 4.930579738887827e-06,
      "loss": 0.8591,
      "step": 18
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 0.8228693160594547,
      "learning_rate": 4.9160835457696075e-06,
      "loss": 0.9227,
      "step": 19
    },
    {
      "epoch": 0.7017543859649122,
      "grad_norm": 0.9746558682735013,
      "learning_rate": 4.900239357847582e-06,
      "loss": 0.9185,
      "step": 20
    },
    {
      "epoch": 0.7368421052631579,
      "grad_norm": 0.783405238430667,
      "learning_rate": 4.883056014999423e-06,
      "loss": 0.8853,
      "step": 21
    },
    {
      "epoch": 0.7719298245614035,
      "grad_norm": 1.1560714233050355,
      "learning_rate": 4.864543104251587e-06,
      "loss": 0.837,
      "step": 22
    },
    {
      "epoch": 0.8070175438596491,
      "grad_norm": 0.6779891999963303,
      "learning_rate": 4.844710954430464e-06,
      "loss": 0.9006,
      "step": 23
    },
    {
      "epoch": 0.8421052631578947,
      "grad_norm": 0.6140368603132732,
      "learning_rate": 4.823570630399665e-06,
      "loss": 0.8713,
      "step": 24
    },
    {
      "epoch": 0.8771929824561403,
      "grad_norm": 0.5959623784138675,
      "learning_rate": 4.8011339268866505e-06,
      "loss": 0.8573,
      "step": 25
    },
    {
      "epoch": 0.9122807017543859,
      "grad_norm": 0.6645642684704514,
      "learning_rate": 4.777413361902152e-06,
      "loss": 0.9244,
      "step": 26
    },
    {
      "epoch": 0.9473684210526315,
      "grad_norm": 0.5749254662708289,
      "learning_rate": 4.752422169756048e-06,
      "loss": 0.8214,
      "step": 27
    },
    {
      "epoch": 0.9824561403508771,
      "grad_norm": 0.49471199829315,
      "learning_rate": 4.726174293673612e-06,
      "loss": 0.8501,
      "step": 28
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.49471199829315,
      "learning_rate": 4.698684378016223e-06,
      "loss": 0.7761,
      "step": 29
    },
    {
      "epoch": 1.0350877192982457,
      "grad_norm": 0.8408225702174914,
      "learning_rate": 4.669967760110908e-06,
      "loss": 0.8702,
      "step": 30
    },
    {
      "epoch": 1.0701754385964912,
      "grad_norm": 0.5019249070368396,
      "learning_rate": 4.6400404616932505e-06,
      "loss": 0.8376,
      "step": 31
    },
    {
      "epoch": 1.1052631578947367,
      "grad_norm": 0.4303555090359822,
      "learning_rate": 4.608919179968457e-06,
      "loss": 0.8834,
      "step": 32
    },
    {
      "epoch": 1.1403508771929824,
      "grad_norm": 0.446333451933402,
      "learning_rate": 4.576621278295558e-06,
      "loss": 0.8631,
      "step": 33
    },
    {
      "epoch": 1.1754385964912282,
      "grad_norm": 0.4783577116779664,
      "learning_rate": 4.543164776499945e-06,
      "loss": 0.8283,
      "step": 34
    },
    {
      "epoch": 1.2105263157894737,
      "grad_norm": 0.4079076293014929,
      "learning_rate": 4.508568340819654e-06,
      "loss": 0.8868,
      "step": 35
    },
    {
      "epoch": 1.2456140350877192,
      "grad_norm": 0.40536005006600634,
      "learning_rate": 4.472851273490985e-06,
      "loss": 0.8674,
      "step": 36
    },
    {
      "epoch": 1.280701754385965,
      "grad_norm": 0.4152500781803562,
      "learning_rate": 4.436033501979299e-06,
      "loss": 0.859,
      "step": 37
    },
    {
      "epoch": 1.3157894736842106,
      "grad_norm": 0.4347010646426558,
      "learning_rate": 4.398135567860972e-06,
      "loss": 0.8254,
      "step": 38
    },
    {
      "epoch": 1.3508771929824561,
      "grad_norm": 0.3788287405374111,
      "learning_rate": 4.359178615362725e-06,
      "loss": 0.7829,
      "step": 39
    },
    {
      "epoch": 1.3859649122807016,
      "grad_norm": 0.4670693473167291,
      "learning_rate": 4.319184379564716e-06,
      "loss": 0.8918,
      "step": 40
    },
    {
      "epoch": 1.4210526315789473,
      "grad_norm": 0.47835984972537143,
      "learning_rate": 4.278175174273989e-06,
      "loss": 0.8179,
      "step": 41
    },
    {
      "epoch": 1.456140350877193,
      "grad_norm": 0.4499444593333873,
      "learning_rate": 4.236173879575022e-06,
      "loss": 0.7871,
      "step": 42
    },
    {
      "epoch": 1.4912280701754386,
      "grad_norm": 0.3878622142611574,
      "learning_rate": 4.1932039290643534e-06,
      "loss": 0.7789,
      "step": 43
    },
    {
      "epoch": 1.526315789473684,
      "grad_norm": 0.3316720314372251,
      "learning_rate": 4.149289296776369e-06,
      "loss": 0.91,
      "step": 44
    },
    {
      "epoch": 1.5614035087719298,
      "grad_norm": 0.36941265676632506,
      "learning_rate": 4.104454483807579e-06,
      "loss": 0.8607,
      "step": 45
    },
    {
      "epoch": 1.5964912280701755,
      "grad_norm": 0.3620626145776103,
      "learning_rate": 4.058724504646834e-06,
      "loss": 0.7937,
      "step": 46
    },
    {
      "epoch": 1.631578947368421,
      "grad_norm": 0.3338291240702826,
      "learning_rate": 4.012124873219094e-06,
      "loss": 0.7815,
      "step": 47
    },
    {
      "epoch": 1.6666666666666665,
      "grad_norm": 0.472336322290524,
      "learning_rate": 3.964681588650562e-06,
      "loss": 0.8394,
      "step": 48
    },
    {
      "epoch": 1.7017543859649122,
      "grad_norm": 0.3420578655212163,
      "learning_rate": 3.916421120763106e-06,
      "loss": 0.8502,
      "step": 49
    },
    {
      "epoch": 1.736842105263158,
      "grad_norm": 0.38000589787590366,
      "learning_rate": 3.8673703953060685e-06,
      "loss": 0.8347,
      "step": 50
    },
    {
      "epoch": 1.7719298245614035,
      "grad_norm": 0.3654714309302888,
      "learning_rate": 3.817556778933697e-06,
      "loss": 0.8581,
      "step": 51
    },
    {
      "epoch": 1.807017543859649,
      "grad_norm": 0.3918540158095545,
      "learning_rate": 3.7670080639366e-06,
      "loss": 0.8769,
      "step": 52
    },
    {
      "epoch": 1.8421052631578947,
      "grad_norm": 0.35070825923075866,
      "learning_rate": 3.7157524527357036e-06,
      "loss": 0.7163,
      "step": 53
    },
    {
      "epoch": 1.8771929824561404,
      "grad_norm": 0.35412184971484173,
      "learning_rate": 3.663818542147409e-06,
      "loss": 0.8756,
      "step": 54
    },
    {
      "epoch": 1.912280701754386,
      "grad_norm": 0.3580769918085387,
      "learning_rate": 3.61123530742869e-06,
      "loss": 0.8221,
      "step": 55
    },
    {
      "epoch": 1.9473684210526314,
      "grad_norm": 0.33901277879515357,
      "learning_rate": 3.5580320861110627e-06,
      "loss": 0.8703,
      "step": 56
    },
    {
      "epoch": 1.9824561403508771,
      "grad_norm": 0.3463865269661244,
      "learning_rate": 3.5042385616324243e-06,
      "loss": 0.8721,
      "step": 57
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.5122368904802833,
      "learning_rate": 3.4498847467759e-06,
      "loss": 0.8607,
      "step": 58
    }
  ],
  "logging_steps": 1,
  "max_steps": 140,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 25934832402432.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}