{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.991533396048918,
  "eval_steps": 500,
  "global_step": 795,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.04,
      "grad_norm": 2.453125,
      "learning_rate": 2.5e-06,
      "loss": 1.6646,
      "step": 10
    },
    {
      "epoch": 0.08,
      "grad_norm": 2.71875,
      "learning_rate": 5e-06,
      "loss": 1.633,
      "step": 20
    },
    {
      "epoch": 0.11,
      "grad_norm": 1.453125,
      "learning_rate": 7.500000000000001e-06,
      "loss": 1.5999,
      "step": 30
    },
    {
      "epoch": 0.15,
      "grad_norm": 1.921875,
      "learning_rate": 1e-05,
      "loss": 1.6187,
      "step": 40
    },
    {
      "epoch": 0.19,
      "grad_norm": 2.625,
      "learning_rate": 1.25e-05,
      "loss": 1.5918,
      "step": 50
    },
    {
      "epoch": 0.23,
      "grad_norm": 0.59765625,
      "learning_rate": 1.5000000000000002e-05,
      "loss": 1.5936,
      "step": 60
    },
    {
      "epoch": 0.26,
      "grad_norm": 2.015625,
      "learning_rate": 1.7500000000000002e-05,
      "loss": 1.5263,
      "step": 70
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.92578125,
      "learning_rate": 2e-05,
      "loss": 1.4784,
      "step": 80
    },
    {
      "epoch": 0.34,
      "grad_norm": 0.474609375,
      "learning_rate": 1.999034865600726e-05,
      "loss": 1.3839,
      "step": 90
    },
    {
      "epoch": 0.38,
      "grad_norm": 1.2109375,
      "learning_rate": 1.9961413253717214e-05,
      "loss": 1.3698,
      "step": 100
    },
    {
      "epoch": 0.41,
      "grad_norm": 1.0390625,
      "learning_rate": 1.9913249646234072e-05,
      "loss": 1.2916,
      "step": 110
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.35546875,
      "learning_rate": 1.9845950802266584e-05,
      "loss": 1.2607,
      "step": 120
    },
    {
      "epoch": 0.49,
      "grad_norm": 0.375,
      "learning_rate": 1.9759646626673445e-05,
      "loss": 1.2405,
      "step": 130
    },
    {
      "epoch": 0.53,
      "grad_norm": 0.8828125,
      "learning_rate": 1.9654503709711984e-05,
      "loss": 1.2615,
      "step": 140
    },
    {
      "epoch": 0.56,
      "grad_norm": 1.2421875,
      "learning_rate": 1.9530725005474195e-05,
      "loss": 1.1433,
      "step": 150
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.404296875,
      "learning_rate": 1.93885494401308e-05,
      "loss": 1.1599,
      "step": 160
    },
    {
      "epoch": 0.64,
      "grad_norm": 0.91796875,
      "learning_rate": 1.9228251450739495e-05,
      "loss": 1.1325,
      "step": 170
    },
    {
      "epoch": 0.68,
      "grad_norm": 0.55859375,
      "learning_rate": 1.905014045550767e-05,
      "loss": 1.0797,
      "step": 180
    },
    {
      "epoch": 0.71,
      "grad_norm": 0.375,
      "learning_rate": 1.8854560256532098e-05,
      "loss": 1.041,
      "step": 190
    },
    {
      "epoch": 0.75,
      "grad_norm": 1.2265625,
      "learning_rate": 1.8641888376168483e-05,
      "loss": 1.0286,
      "step": 200
    },
    {
      "epoch": 0.79,
      "grad_norm": 0.287109375,
      "learning_rate": 1.8412535328311813e-05,
      "loss": 0.9831,
      "step": 210
    },
    {
      "epoch": 0.83,
      "grad_norm": 0.28515625,
      "learning_rate": 1.816694382599422e-05,
      "loss": 0.9927,
      "step": 220
    },
    {
      "epoch": 0.87,
      "grad_norm": 3.25,
      "learning_rate": 1.7905587926829815e-05,
      "loss": 1.0075,
      "step": 230
    },
    {
      "epoch": 0.9,
      "grad_norm": 0.255859375,
      "learning_rate": 1.762897211795607e-05,
      "loss": 1.0232,
      "step": 240
    },
    {
      "epoch": 0.94,
      "grad_norm": 0.6484375,
      "learning_rate": 1.733763034223804e-05,
      "loss": 0.9476,
      "step": 250
    },
    {
      "epoch": 0.98,
      "grad_norm": 0.67578125,
      "learning_rate": 1.7032124967615112e-05,
      "loss": 0.9338,
      "step": 260
    },
    {
      "epoch": 1.02,
      "grad_norm": 0.41796875,
      "learning_rate": 1.6713045701579705e-05,
      "loss": 0.9243,
      "step": 270
    },
    {
      "epoch": 1.05,
      "grad_norm": 0.232421875,
      "learning_rate": 1.638100845288331e-05,
      "loss": 0.9084,
      "step": 280
    },
    {
      "epoch": 1.09,
      "grad_norm": 0.349609375,
      "learning_rate": 1.6036654142667043e-05,
      "loss": 0.896,
      "step": 290
    },
    {
      "epoch": 1.13,
      "grad_norm": 0.25390625,
      "learning_rate": 1.568064746731156e-05,
      "loss": 0.9311,
      "step": 300
    },
    {
      "epoch": 1.17,
      "grad_norm": 0.236328125,
      "learning_rate": 1.5313675615394373e-05,
      "loss": 0.9182,
      "step": 310
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.326171875,
      "learning_rate": 1.4936446941231186e-05,
      "loss": 0.8967,
      "step": 320
    },
    {
      "epoch": 1.24,
      "grad_norm": 0.2451171875,
      "learning_rate": 1.4549689597561652e-05,
      "loss": 0.8965,
      "step": 330
    },
    {
      "epoch": 1.28,
      "grad_norm": 0.435546875,
      "learning_rate": 1.4154150130018867e-05,
      "loss": 0.9105,
      "step": 340
    },
    {
      "epoch": 1.32,
      "grad_norm": 0.3359375,
      "learning_rate": 1.375059203609562e-05,
      "loss": 0.9089,
      "step": 350
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.2578125,
      "learning_rate": 1.3339794291389015e-05,
      "loss": 0.9066,
      "step": 360
    },
    {
      "epoch": 1.39,
      "grad_norm": 0.271484375,
      "learning_rate": 1.2922549845968174e-05,
      "loss": 0.8878,
      "step": 370
    },
    {
      "epoch": 1.43,
      "grad_norm": 0.30078125,
      "learning_rate": 1.2499664093767458e-05,
      "loss": 0.8909,
      "step": 380
    },
    {
      "epoch": 1.47,
      "grad_norm": 2.8125,
      "learning_rate": 1.2071953317959692e-05,
      "loss": 0.8659,
      "step": 390
    },
    {
      "epoch": 1.51,
      "grad_norm": 0.3125,
      "learning_rate": 1.1640243115310219e-05,
      "loss": 0.8685,
      "step": 400
    },
    {
      "epoch": 1.54,
      "grad_norm": 0.216796875,
      "learning_rate": 1.1205366802553231e-05,
      "loss": 0.8959,
      "step": 410
    },
    {
      "epoch": 1.58,
      "grad_norm": 0.310546875,
      "learning_rate": 1.076816380786647e-05,
      "loss": 0.9412,
      "step": 420
    },
    {
      "epoch": 1.62,
      "grad_norm": 0.99609375,
      "learning_rate": 1.0329478050549208e-05,
      "loss": 0.8929,
      "step": 430
    },
    {
      "epoch": 1.66,
      "grad_norm": 0.3359375,
      "learning_rate": 9.890156312031165e-06,
      "loss": 0.9102,
      "step": 440
    },
    {
      "epoch": 1.69,
      "grad_norm": 0.37109375,
      "learning_rate": 9.451046601356725e-06,
      "loss": 0.8645,
      "step": 450
    },
    {
      "epoch": 1.73,
      "grad_norm": 0.259765625,
      "learning_rate": 9.012996518299547e-06,
      "loss": 0.8554,
      "step": 460
    },
    {
      "epoch": 1.77,
      "grad_norm": 0.314453125,
      "learning_rate": 8.576851617267151e-06,
      "loss": 0.9034,
      "step": 470
    },
    {
      "epoch": 1.81,
      "grad_norm": 0.2001953125,
      "learning_rate": 8.143453775153646e-06,
      "loss": 0.8696,
      "step": 480
    },
    {
      "epoch": 1.84,
      "grad_norm": 0.671875,
      "learning_rate": 7.713639566291028e-06,
      "loss": 0.8801,
      "step": 490
    },
    {
      "epoch": 1.88,
      "grad_norm": 0.291015625,
      "learning_rate": 7.2882386476358304e-06,
      "loss": 0.8872,
      "step": 500
    },
    {
      "epoch": 1.92,
      "grad_norm": 0.37109375,
      "learning_rate": 6.868072157308213e-06,
      "loss": 0.8586,
      "step": 510
    },
    {
      "epoch": 1.96,
      "grad_norm": 0.7109375,
      "learning_rate": 6.453951129574644e-06,
      "loss": 0.8727,
      "step": 520
    },
    {
      "epoch": 1.99,
      "grad_norm": 1.15625,
      "learning_rate": 6.046674929333787e-06,
      "loss": 0.8833,
      "step": 530
    },
    {
      "epoch": 2.03,
      "grad_norm": 0.2431640625,
      "learning_rate": 5.647029709127355e-06,
      "loss": 0.9192,
      "step": 540
    },
    {
      "epoch": 2.07,
      "grad_norm": 0.5859375,
      "learning_rate": 5.2557868916543996e-06,
      "loss": 0.9092,
      "step": 550
    },
    {
      "epoch": 2.11,
      "grad_norm": 0.392578125,
      "learning_rate": 4.873701680718146e-06,
      "loss": 0.8563,
      "step": 560
    },
    {
      "epoch": 2.14,
      "grad_norm": 0.28515625,
      "learning_rate": 4.501511603479653e-06,
      "loss": 0.8795,
      "step": 570
    },
    {
      "epoch": 2.18,
      "grad_norm": 0.283203125,
      "learning_rate": 4.13993508683214e-06,
      "loss": 0.8673,
      "step": 580
    },
    {
      "epoch": 2.22,
      "grad_norm": 0.283203125,
      "learning_rate": 3.7896700706439826e-06,
      "loss": 0.8649,
      "step": 590
    },
    {
      "epoch": 2.26,
      "grad_norm": 0.40625,
      "learning_rate": 3.4513926605471504e-06,
      "loss": 0.8592,
      "step": 600
    },
    {
      "epoch": 2.3,
      "grad_norm": 0.2490234375,
      "learning_rate": 3.125755822871607e-06,
      "loss": 0.8603,
      "step": 610
    },
    {
      "epoch": 2.33,
      "grad_norm": 0.23828125,
      "learning_rate": 2.813388124244778e-06,
      "loss": 0.8614,
      "step": 620
    },
    {
      "epoch": 2.37,
      "grad_norm": 0.251953125,
      "learning_rate": 2.514892518288988e-06,
      "loss": 0.8799,
      "step": 630
    },
    {
      "epoch": 2.41,
      "grad_norm": 0.27734375,
      "learning_rate": 2.230845181758928e-06,
      "loss": 0.8356,
      "step": 640
    },
    {
      "epoch": 2.45,
      "grad_norm": 0.453125,
      "learning_rate": 1.961794402365611e-06,
      "loss": 0.8888,
      "step": 650
    },
    {
      "epoch": 2.48,
      "grad_norm": 0.2470703125,
      "learning_rate": 1.7082595204337183e-06,
      "loss": 0.8796,
      "step": 660
    },
    {
      "epoch": 2.52,
      "grad_norm": 0.482421875,
      "learning_rate": 1.4707299264351914e-06,
      "loss": 0.8694,
      "step": 670
    },
    {
      "epoch": 2.56,
      "grad_norm": 0.259765625,
      "learning_rate": 1.2496641163340562e-06,
      "loss": 0.8944,
      "step": 680
    },
    {
      "epoch": 2.6,
      "grad_norm": 0.51953125,
      "learning_rate": 1.0454888065659775e-06,
      "loss": 0.8564,
      "step": 690
    },
    {
      "epoch": 2.63,
      "grad_norm": 0.2265625,
      "learning_rate": 8.585981103608343e-07,
      "loss": 0.8361,
      "step": 700
    },
    {
      "epoch": 2.67,
      "grad_norm": 0.23828125,
      "learning_rate": 6.8935277699825e-07,
      "loss": 0.869,
      "step": 710
    },
    {
      "epoch": 2.71,
      "grad_norm": 0.2294921875,
      "learning_rate": 5.380794954645141e-07,
      "loss": 0.8573,
      "step": 720
    },
    {
      "epoch": 2.75,
      "grad_norm": 0.35546875,
      "learning_rate": 4.0507026385502747e-07,
      "loss": 0.8439,
      "step": 730
    },
    {
      "epoch": 2.78,
      "grad_norm": 0.216796875,
      "learning_rate": 2.905818257394799e-07,
      "loss": 0.9043,
      "step": 740
    },
    {
      "epoch": 2.82,
      "grad_norm": 0.2177734375,
      "learning_rate": 1.9483517457776436e-07,
      "loss": 0.8677,
      "step": 750
    },
    {
      "epoch": 2.86,
      "grad_norm": 0.3203125,
      "learning_rate": 1.1801512714318286e-07,
      "loss": 0.8762,
      "step": 760
    },
    {
      "epoch": 2.9,
      "grad_norm": 0.390625,
      "learning_rate": 6.026996677640062e-08,
      "loss": 0.8791,
      "step": 770
    },
    {
      "epoch": 2.94,
      "grad_norm": 0.322265625,
      "learning_rate": 2.171115715874139e-08,
      "loss": 0.91,
      "step": 780
    },
    {
      "epoch": 2.97,
      "grad_norm": 0.212890625,
      "learning_rate": 2.4131271573191172e-09,
      "loss": 0.8937,
      "step": 790
    }
  ],
  "logging_steps": 10,
  "max_steps": 795,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "total_flos": 1.5541346535840154e+17,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}