{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 4.071246819338422,
  "eval_steps": 500,
  "global_step": 400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.05089058524173028,
      "grad_norm": 2.2103662490844727,
      "learning_rate": 0.0001999771640780308,
      "loss": 3.0938,
      "num_input_tokens_seen": 190176,
      "step": 5
    },
    {
      "epoch": 0.10178117048346055,
      "grad_norm": 0.8035047054290771,
      "learning_rate": 0.0001998844110196681,
      "loss": 1.471,
      "num_input_tokens_seen": 378656,
      "step": 10
    },
    {
      "epoch": 0.15267175572519084,
      "grad_norm": 0.44624239206314087,
      "learning_rate": 0.00019972037971811802,
      "loss": 1.2452,
      "num_input_tokens_seen": 570208,
      "step": 15
    },
    {
      "epoch": 0.2035623409669211,
      "grad_norm": 0.41621240973472595,
      "learning_rate": 0.00019948518722731206,
      "loss": 1.1957,
      "num_input_tokens_seen": 763040,
      "step": 20
    },
    {
      "epoch": 0.2544529262086514,
      "grad_norm": 0.3875686824321747,
      "learning_rate": 0.0001991790013823246,
      "loss": 1.1319,
      "num_input_tokens_seen": 959200,
      "step": 25
    },
    {
      "epoch": 0.3053435114503817,
      "grad_norm": 0.4382984936237335,
      "learning_rate": 0.00019880204067960472,
      "loss": 1.1209,
      "num_input_tokens_seen": 1153728,
      "step": 30
    },
    {
      "epoch": 0.356234096692112,
      "grad_norm": 0.45683977007865906,
      "learning_rate": 0.00019835457412105528,
      "loss": 1.123,
      "num_input_tokens_seen": 1351616,
      "step": 35
    },
    {
      "epoch": 0.4071246819338422,
      "grad_norm": 0.4628298580646515,
      "learning_rate": 0.00019783692102207155,
      "loss": 1.0532,
      "num_input_tokens_seen": 1537696,
      "step": 40
    },
    {
      "epoch": 0.4580152671755725,
      "grad_norm": 0.5620412826538086,
      "learning_rate": 0.00019724945078367513,
      "loss": 1.0805,
      "num_input_tokens_seen": 1730144,
      "step": 45
    },
    {
      "epoch": 0.5089058524173028,
      "grad_norm": 0.4734291732311249,
      "learning_rate": 0.00019659258262890683,
      "loss": 1.0572,
      "num_input_tokens_seen": 1916608,
      "step": 50
    },
    {
      "epoch": 0.5597964376590331,
      "grad_norm": 0.46405333280563354,
      "learning_rate": 0.00019586678530366606,
      "loss": 1.0069,
      "num_input_tokens_seen": 2103712,
      "step": 55
    },
    {
      "epoch": 0.6106870229007634,
      "grad_norm": 0.4856356084346771,
      "learning_rate": 0.00019507257674221027,
      "loss": 1.0293,
      "num_input_tokens_seen": 2289888,
      "step": 60
    },
    {
      "epoch": 0.6615776081424937,
      "grad_norm": 0.4761466085910797,
      "learning_rate": 0.00019421052369755334,
      "loss": 0.9994,
      "num_input_tokens_seen": 2477888,
      "step": 65
    },
    {
      "epoch": 0.712468193384224,
      "grad_norm": 0.5239170789718628,
      "learning_rate": 0.0001932812413370265,
      "loss": 1.0159,
      "num_input_tokens_seen": 2662848,
      "step": 70
    },
    {
      "epoch": 0.7633587786259542,
      "grad_norm": 0.4161457419395447,
      "learning_rate": 0.00019228539280329038,
      "loss": 1.0233,
      "num_input_tokens_seen": 2861024,
      "step": 75
    },
    {
      "epoch": 0.8142493638676844,
      "grad_norm": 0.5579718351364136,
      "learning_rate": 0.00019122368874111172,
      "loss": 1.0109,
      "num_input_tokens_seen": 3043712,
      "step": 80
    },
    {
      "epoch": 0.8651399491094147,
      "grad_norm": 0.4595694839954376,
      "learning_rate": 0.0001900968867902419,
      "loss": 1.0132,
      "num_input_tokens_seen": 3231840,
      "step": 85
    },
    {
      "epoch": 0.916030534351145,
      "grad_norm": 0.49271535873413086,
      "learning_rate": 0.00018890579104475995,
      "loss": 1.0079,
      "num_input_tokens_seen": 3429632,
      "step": 90
    },
    {
      "epoch": 0.9669211195928753,
      "grad_norm": 0.5259862542152405,
      "learning_rate": 0.00018765125147926476,
      "loss": 0.9515,
      "num_input_tokens_seen": 3620000,
      "step": 95
    },
    {
      "epoch": 1.0178117048346056,
      "grad_norm": 0.444477379322052,
      "learning_rate": 0.00018633416334232753,
      "loss": 0.93,
      "num_input_tokens_seen": 3809824,
      "step": 100
    },
    {
      "epoch": 1.0687022900763359,
      "grad_norm": 0.42680156230926514,
      "learning_rate": 0.0001849554665176354,
      "loss": 0.8925,
      "num_input_tokens_seen": 4005472,
      "step": 105
    },
    {
      "epoch": 1.1195928753180662,
      "grad_norm": 0.6088462471961975,
      "learning_rate": 0.00018351614485328388,
      "loss": 0.8715,
      "num_input_tokens_seen": 4192480,
      "step": 110
    },
    {
      "epoch": 1.1704834605597965,
      "grad_norm": 0.5572230815887451,
      "learning_rate": 0.0001820172254596956,
      "loss": 0.8569,
      "num_input_tokens_seen": 4379296,
      "step": 115
    },
    {
      "epoch": 1.2213740458015268,
      "grad_norm": 0.5509177446365356,
      "learning_rate": 0.00018045977797666684,
      "loss": 0.8269,
      "num_input_tokens_seen": 4568352,
      "step": 120
    },
    {
      "epoch": 1.272264631043257,
      "grad_norm": 0.67293781042099,
      "learning_rate": 0.00017884491381006478,
      "loss": 0.84,
      "num_input_tokens_seen": 4752928,
      "step": 125
    },
    {
      "epoch": 1.3231552162849873,
      "grad_norm": 0.5845524668693542,
      "learning_rate": 0.00017717378533872017,
      "loss": 0.8243,
      "num_input_tokens_seen": 4939008,
      "step": 130
    },
    {
      "epoch": 1.3740458015267176,
      "grad_norm": 0.5623295307159424,
      "learning_rate": 0.00017544758509208146,
      "loss": 0.8624,
      "num_input_tokens_seen": 5133664,
      "step": 135
    },
    {
      "epoch": 1.424936386768448,
      "grad_norm": 0.5870885848999023,
      "learning_rate": 0.00017366754489921694,
      "loss": 0.8543,
      "num_input_tokens_seen": 5332000,
      "step": 140
    },
    {
      "epoch": 1.4758269720101782,
      "grad_norm": 0.5874502062797546,
      "learning_rate": 0.00017183493500977278,
      "loss": 0.842,
      "num_input_tokens_seen": 5520640,
      "step": 145
    },
    {
      "epoch": 1.5267175572519083,
      "grad_norm": 0.5786451101303101,
      "learning_rate": 0.0001699510631875134,
      "loss": 0.8268,
      "num_input_tokens_seen": 5710528,
      "step": 150
    },
    {
      "epoch": 1.5776081424936388,
      "grad_norm": 0.5929097533226013,
      "learning_rate": 0.00016801727377709194,
      "loss": 0.8238,
      "num_input_tokens_seen": 5907264,
      "step": 155
    },
    {
      "epoch": 1.6284987277353689,
      "grad_norm": 0.570965051651001,
      "learning_rate": 0.00016603494674471593,
      "loss": 0.8324,
      "num_input_tokens_seen": 6095776,
      "step": 160
    },
    {
      "epoch": 1.6793893129770994,
      "grad_norm": 0.6001121401786804,
      "learning_rate": 0.0001640054966933935,
      "loss": 0.8431,
      "num_input_tokens_seen": 6281280,
      "step": 165
    },
    {
      "epoch": 1.7302798982188294,
      "grad_norm": 0.5865895748138428,
      "learning_rate": 0.00016193037185346224,
      "loss": 0.8054,
      "num_input_tokens_seen": 6470368,
      "step": 170
    },
    {
      "epoch": 1.78117048346056,
      "grad_norm": 0.6706877946853638,
      "learning_rate": 0.00015981105304912162,
      "loss": 0.8255,
      "num_input_tokens_seen": 6658944,
      "step": 175
    },
    {
      "epoch": 1.83206106870229,
      "grad_norm": 0.6549966931343079,
      "learning_rate": 0.0001576490526417059,
      "loss": 0.8334,
      "num_input_tokens_seen": 6842880,
      "step": 180
    },
    {
      "epoch": 1.8829516539440203,
      "grad_norm": 0.6813002228736877,
      "learning_rate": 0.0001554459134504523,
      "loss": 0.8354,
      "num_input_tokens_seen": 7034432,
      "step": 185
    },
    {
      "epoch": 1.9338422391857506,
      "grad_norm": 0.6717746257781982,
      "learning_rate": 0.00015320320765153367,
      "loss": 0.8294,
      "num_input_tokens_seen": 7228672,
      "step": 190
    },
    {
      "epoch": 1.984732824427481,
      "grad_norm": 0.6509274840354919,
      "learning_rate": 0.00015092253565614233,
      "loss": 0.8561,
      "num_input_tokens_seen": 7414144,
      "step": 195
    },
    {
      "epoch": 2.035623409669211,
      "grad_norm": 0.6088241338729858,
      "learning_rate": 0.00014860552496842494,
      "loss": 0.6968,
      "num_input_tokens_seen": 7603104,
      "step": 200
    },
    {
      "epoch": 2.0865139949109412,
      "grad_norm": 0.7045843601226807,
      "learning_rate": 0.00014625382902408356,
      "loss": 0.6587,
      "num_input_tokens_seen": 7794016,
      "step": 205
    },
    {
      "epoch": 2.1374045801526718,
      "grad_norm": 0.6975011825561523,
      "learning_rate": 0.00014386912601047213,
      "loss": 0.6276,
      "num_input_tokens_seen": 7987168,
      "step": 210
    },
    {
      "epoch": 2.188295165394402,
      "grad_norm": 0.7472736835479736,
      "learning_rate": 0.00014145311766902957,
      "loss": 0.6451,
      "num_input_tokens_seen": 8180224,
      "step": 215
    },
    {
      "epoch": 2.2391857506361323,
      "grad_norm": 0.7247308492660522,
      "learning_rate": 0.00013900752808090468,
      "loss": 0.6776,
      "num_input_tokens_seen": 8371136,
      "step": 220
    },
    {
      "epoch": 2.2900763358778624,
      "grad_norm": 0.8300191164016724,
      "learning_rate": 0.00013653410243663952,
      "loss": 0.6307,
      "num_input_tokens_seen": 8552064,
      "step": 225
    },
    {
      "epoch": 2.340966921119593,
      "grad_norm": 0.791684627532959,
      "learning_rate": 0.00013403460579078833,
      "loss": 0.6623,
      "num_input_tokens_seen": 8737120,
      "step": 230
    },
    {
      "epoch": 2.391857506361323,
      "grad_norm": 0.7664393782615662,
      "learning_rate": 0.0001315108218023621,
      "loss": 0.6754,
      "num_input_tokens_seen": 8928608,
      "step": 235
    },
    {
      "epoch": 2.4427480916030535,
      "grad_norm": 0.7355414032936096,
      "learning_rate": 0.0001289645514619963,
      "loss": 0.6518,
      "num_input_tokens_seen": 9120032,
      "step": 240
    },
    {
      "epoch": 2.4936386768447836,
      "grad_norm": 0.7242420315742493,
      "learning_rate": 0.00012639761180675098,
      "loss": 0.6331,
      "num_input_tokens_seen": 9307840,
      "step": 245
    },
    {
      "epoch": 2.544529262086514,
      "grad_norm": 0.7869409322738647,
      "learning_rate": 0.00012381183462345982,
      "loss": 0.6588,
      "num_input_tokens_seen": 9499104,
      "step": 250
    },
    {
      "epoch": 2.595419847328244,
      "grad_norm": 0.7066891193389893,
      "learning_rate": 0.0001212090651415537,
      "loss": 0.6657,
      "num_input_tokens_seen": 9694752,
      "step": 255
    },
    {
      "epoch": 2.6463104325699747,
      "grad_norm": 0.8193861842155457,
      "learning_rate": 0.00011859116071629149,
      "loss": 0.6707,
      "num_input_tokens_seen": 9889824,
      "step": 260
    },
    {
      "epoch": 2.6972010178117047,
      "grad_norm": 0.7292617559432983,
      "learning_rate": 0.00011595998950333793,
      "loss": 0.6624,
      "num_input_tokens_seen": 10080896,
      "step": 265
    },
    {
      "epoch": 2.7480916030534353,
      "grad_norm": 0.7750299572944641,
      "learning_rate": 0.00011331742912563413,
      "loss": 0.6671,
      "num_input_tokens_seen": 10278144,
      "step": 270
    },
    {
      "epoch": 2.7989821882951653,
      "grad_norm": 0.7738965153694153,
      "learning_rate": 0.00011066536533351202,
      "loss": 0.6654,
      "num_input_tokens_seen": 10466528,
      "step": 275
    },
    {
      "epoch": 2.849872773536896,
      "grad_norm": 0.8403007984161377,
      "learning_rate": 0.00010800569065900933,
      "loss": 0.6681,
      "num_input_tokens_seen": 10649024,
      "step": 280
    },
    {
      "epoch": 2.900763358778626,
      "grad_norm": 0.8651427626609802,
      "learning_rate": 0.0001053403030653449,
      "loss": 0.6718,
      "num_input_tokens_seen": 10834816,
      "step": 285
    },
    {
      "epoch": 2.9516539440203564,
      "grad_norm": 0.8150460720062256,
      "learning_rate": 0.00010267110459251823,
      "loss": 0.637,
      "num_input_tokens_seen": 11027616,
      "step": 290
    },
    {
      "epoch": 3.0025445292620865,
      "grad_norm": 0.7902624011039734,
      "learning_rate": 0.0001,
      "loss": 0.6567,
      "num_input_tokens_seen": 11217664,
      "step": 295
    },
    {
      "epoch": 3.053435114503817,
      "grad_norm": 0.7837531566619873,
      "learning_rate": 9.73288954074818e-05,
      "loss": 0.5179,
      "num_input_tokens_seen": 11406624,
      "step": 300
    },
    {
      "epoch": 3.104325699745547,
      "grad_norm": 0.893497109413147,
      "learning_rate": 9.46596969346551e-05,
      "loss": 0.5329,
      "num_input_tokens_seen": 11596096,
      "step": 305
    },
    {
      "epoch": 3.1552162849872776,
      "grad_norm": 0.8641018867492676,
      "learning_rate": 9.199430934099068e-05,
      "loss": 0.5039,
      "num_input_tokens_seen": 11788992,
      "step": 310
    },
    {
      "epoch": 3.2061068702290076,
      "grad_norm": 0.9465248584747314,
      "learning_rate": 8.933463466648798e-05,
      "loss": 0.5085,
      "num_input_tokens_seen": 11986976,
      "step": 315
    },
    {
      "epoch": 3.2569974554707377,
      "grad_norm": 0.8461422920227051,
      "learning_rate": 8.66825708743659e-05,
      "loss": 0.5065,
      "num_input_tokens_seen": 12179136,
      "step": 320
    },
    {
      "epoch": 3.3078880407124682,
      "grad_norm": 0.9939844608306885,
      "learning_rate": 8.404001049666211e-05,
      "loss": 0.5059,
      "num_input_tokens_seen": 12368608,
      "step": 325
    },
    {
      "epoch": 3.3587786259541983,
      "grad_norm": 0.8322144150733948,
      "learning_rate": 8.140883928370855e-05,
      "loss": 0.5089,
      "num_input_tokens_seen": 12556256,
      "step": 330
    },
    {
      "epoch": 3.409669211195929,
      "grad_norm": 0.8849288821220398,
      "learning_rate": 7.879093485844635e-05,
      "loss": 0.4853,
      "num_input_tokens_seen": 12746784,
      "step": 335
    },
    {
      "epoch": 3.460559796437659,
      "grad_norm": 0.898762047290802,
      "learning_rate": 7.618816537654018e-05,
      "loss": 0.5119,
      "num_input_tokens_seen": 12944096,
      "step": 340
    },
    {
      "epoch": 3.5114503816793894,
      "grad_norm": 0.9055672883987427,
      "learning_rate": 7.360238819324903e-05,
      "loss": 0.5024,
      "num_input_tokens_seen": 13139904,
      "step": 345
    },
    {
      "epoch": 3.5623409669211195,
      "grad_norm": 0.8258459568023682,
      "learning_rate": 7.10354485380037e-05,
      "loss": 0.5074,
      "num_input_tokens_seen": 13328480,
      "step": 350
    },
    {
      "epoch": 3.61323155216285,
      "grad_norm": 0.8779503703117371,
      "learning_rate": 6.848917819763793e-05,
      "loss": 0.5078,
      "num_input_tokens_seen": 13518112,
      "step": 355
    },
    {
      "epoch": 3.66412213740458,
      "grad_norm": 0.8894705176353455,
      "learning_rate": 6.596539420921171e-05,
      "loss": 0.5211,
      "num_input_tokens_seen": 13705632,
      "step": 360
    },
    {
      "epoch": 3.7150127226463106,
      "grad_norm": 0.8531593680381775,
      "learning_rate": 6.34658975633605e-05,
      "loss": 0.5042,
      "num_input_tokens_seen": 13896896,
      "step": 365
    },
    {
      "epoch": 3.7659033078880406,
      "grad_norm": 0.859188973903656,
      "learning_rate": 6.0992471919095315e-05,
      "loss": 0.5038,
      "num_input_tokens_seen": 14081888,
      "step": 370
    },
    {
      "epoch": 3.816793893129771,
      "grad_norm": 0.9424790740013123,
      "learning_rate": 5.854688233097045e-05,
      "loss": 0.5046,
      "num_input_tokens_seen": 14270560,
      "step": 375
    },
    {
      "epoch": 3.867684478371501,
      "grad_norm": 0.90933758020401,
      "learning_rate": 5.613087398952792e-05,
      "loss": 0.5196,
      "num_input_tokens_seen": 14461120,
      "step": 380
    },
    {
      "epoch": 3.9185750636132317,
      "grad_norm": 0.9108320474624634,
      "learning_rate": 5.37461709759165e-05,
      "loss": 0.4898,
      "num_input_tokens_seen": 14648096,
      "step": 385
    },
    {
      "epoch": 3.969465648854962,
      "grad_norm": 0.8353007435798645,
      "learning_rate": 5.139447503157513e-05,
      "loss": 0.5283,
      "num_input_tokens_seen": 14839840,
      "step": 390
    },
    {
      "epoch": 4.020356234096692,
      "grad_norm": 0.8553094863891602,
      "learning_rate": 4.9077464343857694e-05,
      "loss": 0.4485,
      "num_input_tokens_seen": 15020448,
      "step": 395
    },
    {
      "epoch": 4.071246819338422,
      "grad_norm": 0.9495893120765686,
      "learning_rate": 4.6796792348466356e-05,
      "loss": 0.4041,
      "num_input_tokens_seen": 15210368,
      "step": 400
    }
  ],
  "logging_steps": 5,
  "max_steps": 588,
  "num_input_tokens_seen": 15210368,
  "num_train_epochs": 6,
  "save_steps": 100,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 6.868299045194957e+17,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}