{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.362281822051937,
  "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.017028522775649212,
      "grad_norm": 3.096078395843506,
      "learning_rate": 0.0001988623435722412,
      "loss": 2.9252,
      "mean_token_accuracy": 0.5491172857582569,
      "num_tokens": 4766.0,
      "step": 5
    },
    {
      "epoch": 0.034057045551298425,
      "grad_norm": 0.695277214050293,
      "learning_rate": 0.00019772468714448238,
      "loss": 0.924,
      "mean_token_accuracy": 0.8361630752682686,
      "num_tokens": 9556.0,
      "step": 10
    },
    {
      "epoch": 0.05108556832694764,
      "grad_norm": 0.541026771068573,
      "learning_rate": 0.00019658703071672356,
      "loss": 0.5477,
      "mean_token_accuracy": 0.8945453137159347,
      "num_tokens": 14116.0,
      "step": 15
    },
    {
      "epoch": 0.06811409110259685,
      "grad_norm": 0.5480035543441772,
      "learning_rate": 0.00019544937428896475,
      "loss": 0.5796,
      "mean_token_accuracy": 0.8836526513099671,
      "num_tokens": 18879.0,
      "step": 20
    },
    {
      "epoch": 0.08514261387824607,
      "grad_norm": 0.5408394932746887,
      "learning_rate": 0.00019431171786120593,
      "loss": 0.6289,
      "mean_token_accuracy": 0.8782178282737731,
      "num_tokens": 23699.0,
      "step": 25
    },
    {
      "epoch": 0.10217113665389528,
      "grad_norm": 0.583755612373352,
      "learning_rate": 0.00019317406143344712,
      "loss": 0.5507,
      "mean_token_accuracy": 0.8905435591936112,
      "num_tokens": 28425.0,
      "step": 30
    },
    {
      "epoch": 0.11919965942954448,
      "grad_norm": 0.5654754638671875,
      "learning_rate": 0.0001920364050056883,
      "loss": 0.5163,
      "mean_token_accuracy": 0.8963450387120246,
      "num_tokens": 33060.0,
      "step": 35
    },
    {
      "epoch": 0.1362281822051937,
      "grad_norm": 0.5462912321090698,
      "learning_rate": 0.00019089874857792946,
      "loss": 0.6819,
      "mean_token_accuracy": 0.8708900690078736,
      "num_tokens": 38017.0,
      "step": 40
    },
    {
      "epoch": 0.1532567049808429,
      "grad_norm": 0.45884689688682556,
      "learning_rate": 0.00018976109215017067,
      "loss": 0.5452,
      "mean_token_accuracy": 0.8896441653370857,
      "num_tokens": 42760.0,
      "step": 45
    },
    {
      "epoch": 0.17028522775649213,
      "grad_norm": 0.5886663198471069,
      "learning_rate": 0.00018862343572241183,
      "loss": 0.5337,
      "mean_token_accuracy": 0.8954117730259895,
      "num_tokens": 47455.0,
      "step": 50
    },
    {
      "epoch": 0.18731375053214133,
      "grad_norm": 0.5249215364456177,
      "learning_rate": 0.00018748577929465302,
      "loss": 0.4948,
      "mean_token_accuracy": 0.9001417383551598,
      "num_tokens": 52186.0,
      "step": 55
    },
    {
      "epoch": 0.20434227330779056,
      "grad_norm": 0.40858832001686096,
      "learning_rate": 0.0001863481228668942,
      "loss": 0.4616,
      "mean_token_accuracy": 0.9116851478815079,
      "num_tokens": 56765.0,
      "step": 60
    },
    {
      "epoch": 0.22137079608343976,
      "grad_norm": 0.4287504553794861,
      "learning_rate": 0.0001852104664391354,
      "loss": 0.5232,
      "mean_token_accuracy": 0.895601749420166,
      "num_tokens": 61436.0,
      "step": 65
    },
    {
      "epoch": 0.23839931885908897,
      "grad_norm": 0.5480343103408813,
      "learning_rate": 0.00018407281001137657,
      "loss": 0.5757,
      "mean_token_accuracy": 0.8878508538007737,
      "num_tokens": 66190.0,
      "step": 70
    },
    {
      "epoch": 0.2554278416347382,
      "grad_norm": 0.4539305567741394,
      "learning_rate": 0.00018293515358361776,
      "loss": 0.5136,
      "mean_token_accuracy": 0.8946248903870583,
      "num_tokens": 70888.0,
      "step": 75
    },
    {
      "epoch": 0.2724563644103874,
      "grad_norm": 0.5497062802314758,
      "learning_rate": 0.00018179749715585894,
      "loss": 0.4996,
      "mean_token_accuracy": 0.9008801370859146,
      "num_tokens": 75588.0,
      "step": 80
    },
    {
      "epoch": 0.2894848871860366,
      "grad_norm": 0.5979815721511841,
      "learning_rate": 0.00018065984072810013,
      "loss": 0.5454,
      "mean_token_accuracy": 0.8904410973191261,
      "num_tokens": 80394.0,
      "step": 85
    },
    {
      "epoch": 0.3065134099616858,
      "grad_norm": 0.44527795910835266,
      "learning_rate": 0.0001795221843003413,
      "loss": 0.4675,
      "mean_token_accuracy": 0.9040657833218575,
      "num_tokens": 85088.0,
      "step": 90
    },
    {
      "epoch": 0.32354193273733506,
      "grad_norm": 0.5105463862419128,
      "learning_rate": 0.0001783845278725825,
      "loss": 0.5391,
      "mean_token_accuracy": 0.8978576123714447,
      "num_tokens": 89799.0,
      "step": 95
    },
    {
      "epoch": 0.34057045551298426,
      "grad_norm": 0.5285545587539673,
      "learning_rate": 0.00017724687144482368,
      "loss": 0.5426,
      "mean_token_accuracy": 0.890034094452858,
      "num_tokens": 94573.0,
      "step": 100
    },
    {
      "epoch": 0.35759897828863346,
      "grad_norm": 0.5576184988021851,
      "learning_rate": 0.00017610921501706487,
      "loss": 0.5258,
      "mean_token_accuracy": 0.8902270376682282,
      "num_tokens": 99357.0,
      "step": 105
    },
    {
      "epoch": 0.37462750106428266,
      "grad_norm": 0.5248317718505859,
      "learning_rate": 0.00017497155858930602,
      "loss": 0.5373,
      "mean_token_accuracy": 0.8933822825551033,
      "num_tokens": 104092.0,
      "step": 110
    },
    {
      "epoch": 0.39165602383993187,
      "grad_norm": 0.45350518822669983,
      "learning_rate": 0.00017383390216154724,
      "loss": 0.5542,
      "mean_token_accuracy": 0.8876873329281807,
      "num_tokens": 108902.0,
      "step": 115
    },
    {
      "epoch": 0.4086845466155811,
      "grad_norm": 0.5155291557312012,
      "learning_rate": 0.0001726962457337884,
      "loss": 0.5019,
      "mean_token_accuracy": 0.9029773235321045,
      "num_tokens": 113571.0,
      "step": 120
    },
    {
      "epoch": 0.4257130693912303,
      "grad_norm": 0.5053763389587402,
      "learning_rate": 0.0001715585893060296,
      "loss": 0.5595,
      "mean_token_accuracy": 0.8906429842114448,
      "num_tokens": 118402.0,
      "step": 125
    },
    {
      "epoch": 0.4427415921668795,
      "grad_norm": 0.5493877530097961,
      "learning_rate": 0.00017042093287827076,
      "loss": 0.4821,
      "mean_token_accuracy": 0.9028852418065071,
      "num_tokens": 123147.0,
      "step": 130
    },
    {
      "epoch": 0.45977011494252873,
      "grad_norm": 0.426513135433197,
      "learning_rate": 0.00016928327645051198,
      "loss": 0.5311,
      "mean_token_accuracy": 0.8920483842492104,
      "num_tokens": 127878.0,
      "step": 135
    },
    {
      "epoch": 0.47679863771817793,
      "grad_norm": 0.5025385022163391,
      "learning_rate": 0.00016814562002275313,
      "loss": 0.4699,
      "mean_token_accuracy": 0.9039205580949783,
      "num_tokens": 132591.0,
      "step": 140
    },
    {
      "epoch": 0.49382716049382713,
      "grad_norm": 0.4677104651927948,
      "learning_rate": 0.00016700796359499432,
      "loss": 0.4876,
      "mean_token_accuracy": 0.9038951709866524,
      "num_tokens": 137221.0,
      "step": 145
    },
    {
      "epoch": 0.5108556832694764,
      "grad_norm": 0.5534231066703796,
      "learning_rate": 0.0001658703071672355,
      "loss": 0.4969,
      "mean_token_accuracy": 0.897729343175888,
      "num_tokens": 141981.0,
      "step": 150
    },
    {
      "epoch": 0.5278842060451255,
      "grad_norm": 0.4998553395271301,
      "learning_rate": 0.0001647326507394767,
      "loss": 0.5476,
      "mean_token_accuracy": 0.8921540513634681,
      "num_tokens": 146807.0,
      "step": 155
    },
    {
      "epoch": 0.5449127288207748,
      "grad_norm": 0.545978307723999,
      "learning_rate": 0.00016359499431171787,
      "loss": 0.4543,
      "mean_token_accuracy": 0.9018902003765106,
      "num_tokens": 151544.0,
      "step": 160
    },
    {
      "epoch": 0.561941251596424,
      "grad_norm": 0.5749042630195618,
      "learning_rate": 0.00016245733788395906,
      "loss": 0.5064,
      "mean_token_accuracy": 0.8976296812295914,
      "num_tokens": 156280.0,
      "step": 165
    },
    {
      "epoch": 0.5789697743720732,
      "grad_norm": 0.5217350125312805,
      "learning_rate": 0.00016131968145620024,
      "loss": 0.4741,
      "mean_token_accuracy": 0.9045222416520119,
      "num_tokens": 161011.0,
      "step": 170
    },
    {
      "epoch": 0.5959982971477225,
      "grad_norm": 0.5088939666748047,
      "learning_rate": 0.00016018202502844143,
      "loss": 0.4655,
      "mean_token_accuracy": 0.9078934848308563,
      "num_tokens": 165670.0,
      "step": 175
    },
    {
      "epoch": 0.6130268199233716,
      "grad_norm": 0.5888277292251587,
      "learning_rate": 0.0001590443686006826,
      "loss": 0.4692,
      "mean_token_accuracy": 0.9044368535280227,
      "num_tokens": 170377.0,
      "step": 180
    },
    {
      "epoch": 0.6300553426990209,
      "grad_norm": 0.6187074780464172,
      "learning_rate": 0.0001579067121729238,
      "loss": 0.5105,
      "mean_token_accuracy": 0.8983473464846611,
      "num_tokens": 175090.0,
      "step": 185
    },
    {
      "epoch": 0.6470838654746701,
      "grad_norm": 0.5764688849449158,
      "learning_rate": 0.00015676905574516496,
      "loss": 0.5022,
      "mean_token_accuracy": 0.8986694201827049,
      "num_tokens": 179933.0,
      "step": 190
    },
    {
      "epoch": 0.6641123882503193,
      "grad_norm": 0.5491649508476257,
      "learning_rate": 0.00015563139931740617,
      "loss": 0.5028,
      "mean_token_accuracy": 0.8957466080784797,
      "num_tokens": 184767.0,
      "step": 195
    },
    {
      "epoch": 0.6811409110259685,
      "grad_norm": 0.4836632311344147,
      "learning_rate": 0.00015449374288964733,
      "loss": 0.5325,
      "mean_token_accuracy": 0.8894904315471649,
      "num_tokens": 189624.0,
      "step": 200
    },
    {
      "epoch": 0.6981694338016177,
      "grad_norm": 0.4630073308944702,
      "learning_rate": 0.00015335608646188854,
      "loss": 0.4529,
      "mean_token_accuracy": 0.9051140874624253,
      "num_tokens": 194357.0,
      "step": 205
    },
    {
      "epoch": 0.7151979565772669,
      "grad_norm": 0.49463990330696106,
      "learning_rate": 0.0001522184300341297,
      "loss": 0.4165,
      "mean_token_accuracy": 0.9074924409389495,
      "num_tokens": 199038.0,
      "step": 210
    },
    {
      "epoch": 0.7322264793529162,
      "grad_norm": 0.5512949824333191,
      "learning_rate": 0.00015108077360637088,
      "loss": 0.4798,
      "mean_token_accuracy": 0.9015959203243256,
      "num_tokens": 203805.0,
      "step": 215
    },
    {
      "epoch": 0.7492550021285653,
      "grad_norm": 0.58586585521698,
      "learning_rate": 0.00014994311717861207,
      "loss": 0.5301,
      "mean_token_accuracy": 0.8944625854492188,
      "num_tokens": 208579.0,
      "step": 220
    },
    {
      "epoch": 0.7662835249042146,
      "grad_norm": 0.4895232319831848,
      "learning_rate": 0.00014880546075085325,
      "loss": 0.4694,
      "mean_token_accuracy": 0.9098370909690857,
      "num_tokens": 213237.0,
      "step": 225
    },
    {
      "epoch": 0.7833120476798637,
      "grad_norm": 0.5064729452133179,
      "learning_rate": 0.00014766780432309444,
      "loss": 0.5047,
      "mean_token_accuracy": 0.8984142780303955,
      "num_tokens": 217964.0,
      "step": 230
    },
    {
      "epoch": 0.800340570455513,
      "grad_norm": 0.5260823965072632,
      "learning_rate": 0.00014653014789533562,
      "loss": 0.5159,
      "mean_token_accuracy": 0.8943046078085899,
      "num_tokens": 222821.0,
      "step": 235
    },
    {
      "epoch": 0.8173690932311622,
      "grad_norm": 0.5452587604522705,
      "learning_rate": 0.0001453924914675768,
      "loss": 0.4438,
      "mean_token_accuracy": 0.905141781270504,
      "num_tokens": 227453.0,
      "step": 240
    },
    {
      "epoch": 0.8343976160068114,
      "grad_norm": 0.5265963077545166,
      "learning_rate": 0.000144254835039818,
      "loss": 0.472,
      "mean_token_accuracy": 0.9038746282458305,
      "num_tokens": 232153.0,
      "step": 245
    },
    {
      "epoch": 0.8514261387824607,
      "grad_norm": 0.655539333820343,
      "learning_rate": 0.00014311717861205915,
      "loss": 0.542,
      "mean_token_accuracy": 0.8909124419093132,
      "num_tokens": 236925.0,
      "step": 250
    },
    {
      "epoch": 0.8684546615581098,
      "grad_norm": 0.4830772578716278,
      "learning_rate": 0.00014197952218430036,
      "loss": 0.4432,
      "mean_token_accuracy": 0.9110842600464821,
      "num_tokens": 241546.0,
      "step": 255
    },
    {
      "epoch": 0.885483184333759,
      "grad_norm": 0.5120635032653809,
      "learning_rate": 0.00014084186575654152,
      "loss": 0.4758,
      "mean_token_accuracy": 0.904394218325615,
      "num_tokens": 246278.0,
      "step": 260
    },
    {
      "epoch": 0.9025117071094083,
      "grad_norm": 0.49142977595329285,
      "learning_rate": 0.00013970420932878273,
      "loss": 0.481,
      "mean_token_accuracy": 0.8970154285430908,
      "num_tokens": 251040.0,
      "step": 265
    },
    {
      "epoch": 0.9195402298850575,
      "grad_norm": 0.5381618738174438,
      "learning_rate": 0.0001385665529010239,
      "loss": 0.4988,
      "mean_token_accuracy": 0.8992804601788521,
      "num_tokens": 255781.0,
      "step": 270
    },
    {
      "epoch": 0.9365687526607067,
      "grad_norm": 0.5239886045455933,
      "learning_rate": 0.0001374288964732651,
      "loss": 0.4957,
      "mean_token_accuracy": 0.9006548926234246,
      "num_tokens": 260512.0,
      "step": 275
    },
    {
      "epoch": 0.9535972754363559,
      "grad_norm": 0.5325700640678406,
      "learning_rate": 0.00013629124004550626,
      "loss": 0.4876,
      "mean_token_accuracy": 0.8976434960961341,
      "num_tokens": 265234.0,
      "step": 280
    },
    {
      "epoch": 0.9706257982120051,
      "grad_norm": 0.52351975440979,
      "learning_rate": 0.00013515358361774744,
      "loss": 0.4408,
      "mean_token_accuracy": 0.9042558416724205,
      "num_tokens": 269939.0,
      "step": 285
    },
    {
      "epoch": 0.9876543209876543,
      "grad_norm": 0.5632317066192627,
      "learning_rate": 0.00013401592718998863,
      "loss": 0.4267,
      "mean_token_accuracy": 0.9127328768372536,
      "num_tokens": 274580.0,
      "step": 290
    },
    {
      "epoch": 1.0046828437633035,
      "grad_norm": 0.43729496002197266,
      "learning_rate": 0.00013287827076222981,
      "loss": 0.5103,
      "mean_token_accuracy": 0.9010089665651322,
      "num_tokens": 279321.0,
      "step": 295
    },
    {
      "epoch": 1.0217113665389528,
      "grad_norm": 0.45043033361434937,
      "learning_rate": 0.000131740614334471,
      "loss": 0.376,
      "mean_token_accuracy": 0.9193378701806069,
      "num_tokens": 284033.0,
      "step": 300
    },
    {
      "epoch": 1.038739889314602,
      "grad_norm": 0.6295264363288879,
      "learning_rate": 0.00013060295790671218,
      "loss": 0.3561,
      "mean_token_accuracy": 0.9209762245416642,
      "num_tokens": 288815.0,
      "step": 305
    },
    {
      "epoch": 1.055768412090251,
      "grad_norm": 0.5746341347694397,
      "learning_rate": 0.00012946530147895337,
      "loss": 0.361,
      "mean_token_accuracy": 0.9210935369133949,
      "num_tokens": 293454.0,
      "step": 310
    },
    {
      "epoch": 1.0727969348659003,
      "grad_norm": 0.5386441349983215,
      "learning_rate": 0.00012832764505119455,
      "loss": 0.3631,
      "mean_token_accuracy": 0.923116135597229,
      "num_tokens": 298136.0,
      "step": 315
    },
    {
      "epoch": 1.0898254576415496,
      "grad_norm": 0.637798547744751,
      "learning_rate": 0.0001271899886234357,
      "loss": 0.375,
      "mean_token_accuracy": 0.9151263251900673,
      "num_tokens": 302962.0,
      "step": 320
    },
    {
      "epoch": 1.1068539804171988,
      "grad_norm": 0.6014487147331238,
      "learning_rate": 0.00012605233219567692,
      "loss": 0.3584,
      "mean_token_accuracy": 0.9245758667588234,
      "num_tokens": 307661.0,
      "step": 325
    },
    {
      "epoch": 1.123882503192848,
      "grad_norm": 0.6004022359848022,
      "learning_rate": 0.00012491467576791808,
      "loss": 0.3569,
      "mean_token_accuracy": 0.921052211523056,
      "num_tokens": 312330.0,
      "step": 330
    },
    {
      "epoch": 1.1409110259684971,
      "grad_norm": 0.8666739463806152,
      "learning_rate": 0.0001237770193401593,
      "loss": 0.3778,
      "mean_token_accuracy": 0.9176680579781532,
      "num_tokens": 317107.0,
      "step": 335
    },
    {
      "epoch": 1.1579395487441464,
      "grad_norm": 0.5575281381607056,
      "learning_rate": 0.00012263936291240045,
      "loss": 0.3765,
      "mean_token_accuracy": 0.9149453848600387,
      "num_tokens": 321895.0,
      "step": 340
    },
    {
      "epoch": 1.1749680715197957,
      "grad_norm": 0.5432224273681641,
      "learning_rate": 0.00012150170648464165,
      "loss": 0.3546,
      "mean_token_accuracy": 0.9211765006184578,
      "num_tokens": 326592.0,
      "step": 345
    },
    {
      "epoch": 1.191996594295445,
      "grad_norm": 0.5993474721908569,
      "learning_rate": 0.00012036405005688282,
      "loss": 0.3721,
      "mean_token_accuracy": 0.9195932745933533,
      "num_tokens": 331391.0,
      "step": 350
    },
    {
      "epoch": 1.2090251170710942,
      "grad_norm": 0.6452960968017578,
      "learning_rate": 0.000119226393629124,
      "loss": 0.4292,
      "mean_token_accuracy": 0.9088605523109436,
      "num_tokens": 336238.0,
      "step": 355
    },
    {
      "epoch": 1.2260536398467432,
      "grad_norm": 0.5383236408233643,
      "learning_rate": 0.00011808873720136519,
      "loss": 0.3192,
      "mean_token_accuracy": 0.9233780831098557,
      "num_tokens": 340817.0,
      "step": 360
    },
    {
      "epoch": 1.2430821626223925,
      "grad_norm": 0.6566250324249268,
      "learning_rate": 0.00011695108077360638,
      "loss": 0.3617,
      "mean_token_accuracy": 0.916201414167881,
      "num_tokens": 345509.0,
      "step": 365
    },
    {
      "epoch": 1.2601106853980417,
      "grad_norm": 0.6461586952209473,
      "learning_rate": 0.00011581342434584756,
      "loss": 0.3971,
      "mean_token_accuracy": 0.9110044553875923,
      "num_tokens": 350303.0,
      "step": 370
    },
    {
      "epoch": 1.277139208173691,
      "grad_norm": 0.700552225112915,
      "learning_rate": 0.00011467576791808873,
      "loss": 0.3439,
      "mean_token_accuracy": 0.9284276351332664,
      "num_tokens": 354873.0,
      "step": 375
    },
    {
      "epoch": 1.29416773094934,
      "grad_norm": 0.6307278871536255,
      "learning_rate": 0.00011353811149032993,
      "loss": 0.3619,
      "mean_token_accuracy": 0.9204769432544708,
      "num_tokens": 359564.0,
      "step": 380
    },
    {
      "epoch": 1.3111962537249893,
      "grad_norm": 0.5559372901916504,
      "learning_rate": 0.0001124004550625711,
      "loss": 0.337,
      "mean_token_accuracy": 0.9221480071544648,
      "num_tokens": 364285.0,
      "step": 385
    },
    {
      "epoch": 1.3282247765006385,
      "grad_norm": 0.7321580648422241,
      "learning_rate": 0.00011126279863481229,
      "loss": 0.3864,
      "mean_token_accuracy": 0.917863991856575,
      "num_tokens": 369021.0,
      "step": 390
    },
    {
      "epoch": 1.3452532992762878,
      "grad_norm": 0.6482062339782715,
      "learning_rate": 0.00011012514220705347,
      "loss": 0.342,
      "mean_token_accuracy": 0.9234487116336823,
      "num_tokens": 373711.0,
      "step": 395
    },
    {
      "epoch": 1.362281822051937,
      "grad_norm": 0.7216457724571228,
      "learning_rate": 0.00010898748577929466,
      "loss": 0.3494,
      "mean_token_accuracy": 0.9197908401489258,
      "num_tokens": 378403.0,
      "step": 400
    }
  ],
  "logging_steps": 5,
  "max_steps": 879,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 200,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 1.6144918131772416e+16,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}