{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.5024812976816532,
  "eval_steps": 11,
  "global_step": 53,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.009480779201540626,
      "grad_norm": 1.421875,
      "learning_rate": 2e-05,
      "loss": 0.4503,
      "step": 1
    },
    {
      "epoch": 0.009480779201540626,
      "eval_loss": 0.4263582229614258,
      "eval_runtime": 33.6286,
      "eval_samples_per_second": 21.143,
      "eval_steps_per_second": 21.143,
      "step": 1
    },
    {
      "epoch": 0.018961558403081252,
      "grad_norm": 1.359375,
      "learning_rate": 4e-05,
      "loss": 0.412,
      "step": 2
    },
    {
      "epoch": 0.02844233760462188,
      "grad_norm": 1.1875,
      "learning_rate": 6e-05,
      "loss": 0.4214,
      "step": 3
    },
    {
      "epoch": 0.037923116806162505,
      "grad_norm": 0.80859375,
      "learning_rate": 8e-05,
      "loss": 0.2924,
      "step": 4
    },
    {
      "epoch": 0.04740389600770313,
      "grad_norm": 0.51953125,
      "learning_rate": 0.0001,
      "loss": 0.1896,
      "step": 5
    },
    {
      "epoch": 0.05688467520924376,
      "grad_norm": 0.4453125,
      "learning_rate": 0.00012,
      "loss": 0.1531,
      "step": 6
    },
    {
      "epoch": 0.06636545441078438,
      "grad_norm": 0.27734375,
      "learning_rate": 0.00014,
      "loss": 0.1181,
      "step": 7
    },
    {
      "epoch": 0.07584623361232501,
      "grad_norm": 0.263671875,
      "learning_rate": 0.00016,
      "loss": 0.1143,
      "step": 8
    },
    {
      "epoch": 0.08532701281386564,
      "grad_norm": 0.2197265625,
      "learning_rate": 0.00018,
      "loss": 0.0952,
      "step": 9
    },
    {
      "epoch": 0.09480779201540626,
      "grad_norm": 0.1708984375,
      "learning_rate": 0.0002,
      "loss": 0.0767,
      "step": 10
    },
    {
      "epoch": 0.10428857121694689,
      "grad_norm": 0.1484375,
      "learning_rate": 0.00019999469523400122,
      "loss": 0.0836,
      "step": 11
    },
    {
      "epoch": 0.10428857121694689,
      "eval_loss": 0.07918477058410645,
      "eval_runtime": 33.4231,
      "eval_samples_per_second": 21.273,
      "eval_steps_per_second": 21.273,
      "step": 11
    },
    {
      "epoch": 0.11376935041848751,
      "grad_norm": 0.154296875,
      "learning_rate": 0.00019997878149881574,
      "loss": 0.0757,
      "step": 12
    },
    {
      "epoch": 0.12325012962002814,
      "grad_norm": 0.1513671875,
      "learning_rate": 0.0001999522604828164,
      "loss": 0.0767,
      "step": 13
    },
    {
      "epoch": 0.13273090882156877,
      "grad_norm": 0.1416015625,
      "learning_rate": 0.00019991513499975882,
      "loss": 0.0809,
      "step": 14
    },
    {
      "epoch": 0.1422116880231094,
      "grad_norm": 0.09765625,
      "learning_rate": 0.00019986740898848306,
      "loss": 0.0634,
      "step": 15
    },
    {
      "epoch": 0.15169246722465002,
      "grad_norm": 0.099609375,
      "learning_rate": 0.00019980908751249555,
      "loss": 0.0674,
      "step": 16
    },
    {
      "epoch": 0.16117324642619066,
      "grad_norm": 0.119140625,
      "learning_rate": 0.00019974017675943192,
      "loss": 0.0667,
      "step": 17
    },
    {
      "epoch": 0.17065402562773127,
      "grad_norm": 0.09619140625,
      "learning_rate": 0.0001996606840404006,
      "loss": 0.0632,
      "step": 18
    },
    {
      "epoch": 0.1801348048292719,
      "grad_norm": 0.09130859375,
      "learning_rate": 0.00019957061778920701,
      "loss": 0.0488,
      "step": 19
    },
    {
      "epoch": 0.18961558403081252,
      "grad_norm": 0.0947265625,
      "learning_rate": 0.0001994699875614589,
      "loss": 0.0627,
      "step": 20
    },
    {
      "epoch": 0.19909636323235316,
      "grad_norm": 0.08203125,
      "learning_rate": 0.00019935880403355253,
      "loss": 0.0528,
      "step": 21
    },
    {
      "epoch": 0.20857714243389378,
      "grad_norm": 0.1123046875,
      "learning_rate": 0.00019923707900153982,
      "loss": 0.0532,
      "step": 22
    },
    {
      "epoch": 0.20857714243389378,
      "eval_loss": 0.056647635996341705,
      "eval_runtime": 33.3595,
      "eval_samples_per_second": 21.313,
      "eval_steps_per_second": 21.313,
      "step": 22
    },
    {
      "epoch": 0.21805792163543442,
      "grad_norm": 0.10693359375,
      "learning_rate": 0.00019910482537987702,
      "loss": 0.0583,
      "step": 23
    },
    {
      "epoch": 0.22753870083697503,
      "grad_norm": 0.0791015625,
      "learning_rate": 0.0001989620572000544,
      "loss": 0.0554,
      "step": 24
    },
    {
      "epoch": 0.23701948003851567,
      "grad_norm": 0.123046875,
      "learning_rate": 0.00019880878960910772,
      "loss": 0.0688,
      "step": 25
    },
    {
      "epoch": 0.24650025924005628,
      "grad_norm": 0.10498046875,
      "learning_rate": 0.00019864503886801106,
      "loss": 0.0655,
      "step": 26
    },
    {
      "epoch": 0.2559810384415969,
      "grad_norm": 0.07421875,
      "learning_rate": 0.00019847082234995171,
      "loss": 0.0471,
      "step": 27
    },
    {
      "epoch": 0.26546181764313753,
      "grad_norm": 0.080078125,
      "learning_rate": 0.00019828615853848688,
      "loss": 0.0518,
      "step": 28
    },
    {
      "epoch": 0.27494259684467814,
      "grad_norm": 0.07421875,
      "learning_rate": 0.00019809106702558277,
      "loss": 0.0481,
      "step": 29
    },
    {
      "epoch": 0.2844233760462188,
      "grad_norm": 0.0703125,
      "learning_rate": 0.0001978855685095358,
      "loss": 0.0464,
      "step": 30
    },
    {
      "epoch": 0.2939041552477594,
      "grad_norm": 0.0849609375,
      "learning_rate": 0.00019766968479277683,
      "loss": 0.0566,
      "step": 31
    },
    {
      "epoch": 0.30338493444930004,
      "grad_norm": 0.0859375,
      "learning_rate": 0.00019744343877955788,
      "loss": 0.0517,
      "step": 32
    },
    {
      "epoch": 0.3128657136508407,
      "grad_norm": 0.07177734375,
      "learning_rate": 0.00019720685447352209,
      "loss": 0.0511,
      "step": 33
    },
    {
      "epoch": 0.3128657136508407,
      "eval_loss": 0.04964344948530197,
      "eval_runtime": 33.3741,
      "eval_samples_per_second": 21.304,
      "eval_steps_per_second": 21.304,
      "step": 33
    },
    {
      "epoch": 0.3223464928523813,
      "grad_norm": 0.076171875,
      "learning_rate": 0.0001969599569751571,
      "loss": 0.045,
      "step": 34
    },
    {
      "epoch": 0.33182727205392193,
      "grad_norm": 0.07568359375,
      "learning_rate": 0.00019670277247913205,
      "loss": 0.0543,
      "step": 35
    },
    {
      "epoch": 0.34130805125546254,
      "grad_norm": 0.0791015625,
      "learning_rate": 0.0001964353282715183,
      "loss": 0.0444,
      "step": 36
    },
    {
      "epoch": 0.3507888304570032,
      "grad_norm": 0.0771484375,
      "learning_rate": 0.00019615765272689461,
      "loss": 0.0506,
      "step": 37
    },
    {
      "epoch": 0.3602696096585438,
      "grad_norm": 0.080078125,
      "learning_rate": 0.00019586977530533677,
      "loss": 0.0558,
      "step": 38
    },
    {
      "epoch": 0.36975038886008443,
      "grad_norm": 0.078125,
      "learning_rate": 0.00019557172654929196,
      "loss": 0.0507,
      "step": 39
    },
    {
      "epoch": 0.37923116806162505,
      "grad_norm": 0.06396484375,
      "learning_rate": 0.00019526353808033825,
      "loss": 0.0452,
      "step": 40
    },
    {
      "epoch": 0.3887119472631657,
      "grad_norm": 0.07275390625,
      "learning_rate": 0.00019494524259582992,
      "loss": 0.0481,
      "step": 41
    },
    {
      "epoch": 0.3981927264647063,
      "grad_norm": 0.06591796875,
      "learning_rate": 0.00019461687386542826,
      "loss": 0.0464,
      "step": 42
    },
    {
      "epoch": 0.40767350566624694,
      "grad_norm": 0.06591796875,
      "learning_rate": 0.00019427846672751873,
      "loss": 0.0431,
      "step": 43
    },
    {
      "epoch": 0.41715428486778755,
      "grad_norm": 0.06982421875,
      "learning_rate": 0.00019393005708551498,
      "loss": 0.0511,
      "step": 44
    },
    {
      "epoch": 0.41715428486778755,
      "eval_loss": 0.04574437811970711,
      "eval_runtime": 33.3514,
      "eval_samples_per_second": 21.318,
      "eval_steps_per_second": 21.318,
      "step": 44
    },
    {
      "epoch": 0.4266350640693282,
      "grad_norm": 0.06787109375,
      "learning_rate": 0.00019357168190404936,
      "loss": 0.0443,
      "step": 45
    },
    {
      "epoch": 0.43611584327086883,
      "grad_norm": 0.0751953125,
      "learning_rate": 0.00019320337920505153,
      "loss": 0.0545,
      "step": 46
    },
    {
      "epoch": 0.44559662247240944,
      "grad_norm": 0.07666015625,
      "learning_rate": 0.00019282518806371414,
      "loss": 0.0542,
      "step": 47
    },
    {
      "epoch": 0.45507740167395005,
      "grad_norm": 0.07666015625,
      "learning_rate": 0.0001924371486043473,
      "loss": 0.0589,
      "step": 48
    },
    {
      "epoch": 0.4645581808754907,
      "grad_norm": 0.059326171875,
      "learning_rate": 0.0001920393019961217,
      "loss": 0.0444,
      "step": 49
    },
    {
      "epoch": 0.47403896007703133,
      "grad_norm": 0.0732421875,
      "learning_rate": 0.0001916316904487005,
      "loss": 0.0485,
      "step": 50
    },
    {
      "epoch": 0.48351973927857195,
      "grad_norm": 0.060546875,
      "learning_rate": 0.00019121435720776122,
      "loss": 0.0408,
      "step": 51
    },
    {
      "epoch": 0.49300051848011256,
      "grad_norm": 0.068359375,
      "learning_rate": 0.0001907873465504076,
      "loss": 0.0466,
      "step": 52
    },
    {
      "epoch": 0.5024812976816532,
      "grad_norm": 0.07763671875,
      "learning_rate": 0.00019035070378047204,
      "loss": 0.0426,
      "step": 53
    }
  ],
  "logging_steps": 1,
  "max_steps": 315,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 53,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 9.521933077020672e+16,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}