{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 1197,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.05,
      "grad_norm": 0.96484375,
      "learning_rate": 1.1111111111111113e-05,
      "loss": 0.4919,
      "step": 20
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.68359375,
      "learning_rate": 1.9999414238190722e-05,
      "loss": 0.3018,
      "step": 40
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.59765625,
      "learning_rate": 1.9978919779420427e-05,
      "loss": 0.2632,
      "step": 60
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.57421875,
      "learning_rate": 1.9929205817437334e-05,
      "loss": 0.2528,
      "step": 80
    },
    {
      "epoch": 0.25,
      "grad_norm": 0.5625,
      "learning_rate": 1.9850417920830427e-05,
      "loss": 0.2423,
      "step": 100
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.58203125,
      "learning_rate": 1.9742786790241548e-05,
      "loss": 0.2366,
      "step": 120
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.53515625,
      "learning_rate": 1.9606627582845587e-05,
      "loss": 0.2325,
      "step": 140
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.54296875,
      "learning_rate": 1.944233898953152e-05,
      "loss": 0.2294,
      "step": 160
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.57421875,
      "learning_rate": 1.9250402067486523e-05,
      "loss": 0.2258,
      "step": 180
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.5625,
      "learning_rate": 1.9031378831601425e-05,
      "loss": 0.2233,
      "step": 200
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.5625,
      "learning_rate": 1.878591060882203e-05,
      "loss": 0.2191,
      "step": 220
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.56640625,
      "learning_rate": 1.851471616026505e-05,
      "loss": 0.2178,
      "step": 240
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.57421875,
      "learning_rate": 1.8218589576597203e-05,
      "loss": 0.2141,
      "step": 260
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.5625,
      "learning_rate": 1.789839795284022e-05,
      "loss": 0.2169,
      "step": 280
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.546875,
      "learning_rate": 1.7555078849410044e-05,
      "loss": 0.2144,
      "step": 300
    },
    {
      "epoch": 0.8,
      "grad_norm": 0.81640625,
      "learning_rate": 1.7189637546824752e-05,
      "loss": 0.2083,
      "step": 320
    },
    {
      "epoch": 0.85,
      "grad_norm": 0.57421875,
      "learning_rate": 1.680314410211967e-05,
      "loss": 0.2087,
      "step": 340
    },
    {
      "epoch": 0.9,
      "grad_norm": 0.56640625,
      "learning_rate": 1.6396730215588913e-05,
      "loss": 0.2089,
      "step": 360
    },
    {
      "epoch": 0.95,
      "grad_norm": 0.5390625,
      "learning_rate": 1.5971585917027864e-05,
      "loss": 0.2073,
      "step": 380
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.57421875,
      "learning_rate": 1.552895608117971e-05,
      "loss": 0.2058,
      "step": 400
    },
    {
      "epoch": 1.05,
      "grad_norm": 0.5703125,
      "learning_rate": 1.5070136782589236e-05,
      "loss": 0.1761,
      "step": 420
    },
    {
      "epoch": 1.1,
      "grad_norm": 0.58984375,
      "learning_rate": 1.4596471500537334e-05,
      "loss": 0.1746,
      "step": 440
    },
    {
      "epoch": 1.15,
      "grad_norm": 0.56640625,
      "learning_rate": 1.4109347185168577e-05,
      "loss": 0.175,
      "step": 460
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.6015625,
      "learning_rate": 1.3610190196330777e-05,
      "loss": 0.1722,
      "step": 480
    },
    {
      "epoch": 1.25,
      "grad_norm": 0.62109375,
      "learning_rate": 1.3100462127018108e-05,
      "loss": 0.1749,
      "step": 500
    },
    {
      "epoch": 1.3,
      "grad_norm": 0.58984375,
      "learning_rate": 1.2581655523647234e-05,
      "loss": 0.1742,
      "step": 520
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.61328125,
      "learning_rate": 1.2055289515698008e-05,
      "loss": 0.1729,
      "step": 540
    },
    {
      "epoch": 1.4,
      "grad_norm": 0.5625,
      "learning_rate": 1.1522905367515705e-05,
      "loss": 0.1733,
      "step": 560
    },
    {
      "epoch": 1.45,
      "grad_norm": 0.77734375,
      "learning_rate": 1.0986061965299646e-05,
      "loss": 0.1711,
      "step": 580
    },
    {
      "epoch": 1.5,
      "grad_norm": 0.57421875,
      "learning_rate": 1.0446331252492864e-05,
      "loss": 0.1741,
      "step": 600
    },
    {
      "epoch": 1.55,
      "grad_norm": 0.5625,
      "learning_rate": 9.905293626938564e-06,
      "loss": 0.1712,
      "step": 620
    },
    {
      "epoch": 1.6,
      "grad_norm": 0.60546875,
      "learning_rate": 9.364533313281056e-06,
      "loss": 0.1699,
      "step": 640
    },
    {
      "epoch": 1.65,
      "grad_norm": 0.609375,
      "learning_rate": 8.825633724161336e-06,
      "loss": 0.1714,
      "step": 660
    },
    {
      "epoch": 1.7,
      "grad_norm": 0.57421875,
      "learning_rate": 8.290172823790337e-06,
      "loss": 0.1689,
      "step": 680
    },
    {
      "epoch": 1.75,
      "grad_norm": 0.6015625,
      "learning_rate": 7.759718507475822e-06,
      "loss": 0.1703,
      "step": 700
    },
    {
      "epoch": 1.8,
      "grad_norm": 0.578125,
      "learning_rate": 7.235824010632284e-06,
      "loss": 0.1708,
      "step": 720
    },
    {
      "epoch": 1.85,
      "grad_norm": 0.59765625,
      "learning_rate": 6.720023360716818e-06,
      "loss": 0.171,
      "step": 740
    },
    {
      "epoch": 1.9,
      "grad_norm": 0.60546875,
      "learning_rate": 6.2138268854082094e-06,
      "loss": 0.1667,
      "step": 760
    },
    {
      "epoch": 1.95,
      "grad_norm": 0.6015625,
      "learning_rate": 5.7187167901819665e-06,
      "loss": 0.1709,
      "step": 780
    },
    {
      "epoch": 2.01,
      "grad_norm": 0.56640625,
      "learning_rate": 5.236142818230518e-06,
      "loss": 0.1676,
      "step": 800
    },
    {
      "epoch": 2.06,
      "grad_norm": 0.59765625,
      "learning_rate": 4.767518005437073e-06,
      "loss": 0.1536,
      "step": 820
    },
    {
      "epoch": 2.11,
      "grad_norm": 0.609375,
      "learning_rate": 4.314214542832889e-06,
      "loss": 0.1526,
      "step": 840
    },
    {
      "epoch": 2.16,
      "grad_norm": 0.59375,
      "learning_rate": 3.877559758653358e-06,
      "loss": 0.1559,
      "step": 860
    },
    {
      "epoch": 2.21,
      "grad_norm": 0.63671875,
      "learning_rate": 3.4588322317578395e-06,
      "loss": 0.1529,
      "step": 880
    },
    {
      "epoch": 2.26,
      "grad_norm": 0.5703125,
      "learning_rate": 3.0592580477936606e-06,
      "loss": 0.1515,
      "step": 900
    },
    {
      "epoch": 2.31,
      "grad_norm": 0.61328125,
      "learning_rate": 2.6800072090667085e-06,
      "loss": 0.1516,
      "step": 920
    },
    {
      "epoch": 2.36,
      "grad_norm": 0.61328125,
      "learning_rate": 2.3221902086308943e-06,
      "loss": 0.1515,
      "step": 940
    },
    {
      "epoch": 2.41,
      "grad_norm": 0.6015625,
      "learning_rate": 1.986854778628067e-06,
      "loss": 0.1533,
      "step": 960
    },
    {
      "epoch": 2.46,
      "grad_norm": 0.64453125,
      "learning_rate": 1.67498282239956e-06,
      "loss": 0.1528,
      "step": 980
    },
    {
      "epoch": 2.51,
      "grad_norm": 0.59375,
      "learning_rate": 1.3874875393525722e-06,
      "loss": 0.1508,
      "step": 1000
    },
    {
      "epoch": 2.56,
      "grad_norm": 0.61328125,
      "learning_rate": 1.1252107510000843e-06,
      "loss": 0.1519,
      "step": 1020
    },
    {
      "epoch": 2.61,
      "grad_norm": 0.625,
      "learning_rate": 8.889204360040471e-07,
      "loss": 0.1502,
      "step": 1040
    },
    {
      "epoch": 2.66,
      "grad_norm": 0.62890625,
      "learning_rate": 6.793084814394945e-07,
      "loss": 0.1526,
      "step": 1060
    },
    {
      "epoch": 2.71,
      "grad_norm": 0.62109375,
      "learning_rate": 4.969886568641757e-07,
      "loss": 0.1513,
      "step": 1080
    },
    {
      "epoch": 2.76,
      "grad_norm": 0.61328125,
      "learning_rate": 3.424948171258824e-07,
      "loss": 0.1525,
      "step": 1100
    },
    {
      "epoch": 2.81,
      "grad_norm": 0.62890625,
      "learning_rate": 2.16279339169847e-07,
      "loss": 0.1549,
      "step": 1120
    },
    {
      "epoch": 2.86,
      "grad_norm": 0.6171875,
      "learning_rate": 1.1871179742348416e-07,
      "loss": 0.1494,
      "step": 1140
    },
    {
      "epoch": 2.91,
      "grad_norm": 0.61328125,
      "learning_rate": 5.007788163703842e-08,
      "loss": 0.1536,
      "step": 1160
    },
    {
      "epoch": 2.96,
      "grad_norm": 0.65234375,
      "learning_rate": 1.0578560348905031e-08,
      "loss": 0.1532,
      "step": 1180
    },
    {
      "epoch": 3.0,
      "step": 1197,
      "total_flos": 1.7005983975572767e+18,
      "train_loss": 0.18855523903766272,
      "train_runtime": 5513.3521,
      "train_samples_per_second": 31.264,
      "train_steps_per_second": 0.217
    }
  ],
  "logging_steps": 20,
  "max_steps": 1197,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 10000,
  "total_flos": 1.7005983975572767e+18,
  "train_batch_size": 36,
  "trial_name": null,
  "trial_params": null
}