{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.9815668202764978,
  "eval_steps": 500,
  "global_step": 144,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.013824884792626729,
      "grad_norm": 34.963134765625,
      "learning_rate": 5.0000000000000004e-08,
      "loss": 2.5476,
      "step": 1
    },
    {
      "epoch": 0.027649769585253458,
      "grad_norm": 35.32600021362305,
      "learning_rate": 1.0000000000000001e-07,
      "loss": 2.6058,
      "step": 2
    },
    {
      "epoch": 0.041474654377880185,
      "grad_norm": 34.955448150634766,
      "learning_rate": 1.5000000000000002e-07,
      "loss": 2.5871,
      "step": 3
    },
    {
      "epoch": 0.055299539170506916,
      "grad_norm": 35.09806442260742,
      "learning_rate": 2.0000000000000002e-07,
      "loss": 2.5912,
      "step": 4
    },
    {
      "epoch": 0.06912442396313365,
      "grad_norm": 34.88739776611328,
      "learning_rate": 2.5000000000000004e-07,
      "loss": 2.592,
      "step": 5
    },
    {
      "epoch": 0.08294930875576037,
      "grad_norm": 34.84288024902344,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 2.5609,
      "step": 6
    },
    {
      "epoch": 0.0967741935483871,
      "grad_norm": 35.0090217590332,
      "learning_rate": 3.5000000000000004e-07,
      "loss": 2.5651,
      "step": 7
    },
    {
      "epoch": 0.11059907834101383,
      "grad_norm": 35.03983688354492,
      "learning_rate": 4.0000000000000003e-07,
      "loss": 2.5437,
      "step": 8
    },
    {
      "epoch": 0.12442396313364056,
      "grad_norm": 34.802833557128906,
      "learning_rate": 4.5000000000000003e-07,
      "loss": 2.5448,
      "step": 9
    },
    {
      "epoch": 0.1382488479262673,
      "grad_norm": 34.5220947265625,
      "learning_rate": 5.000000000000001e-07,
      "loss": 2.504,
      "step": 10
    },
    {
      "epoch": 0.15207373271889402,
      "grad_norm": 34.401580810546875,
      "learning_rate": 5.5e-07,
      "loss": 2.4814,
      "step": 11
    },
    {
      "epoch": 0.16589861751152074,
      "grad_norm": 33.76997375488281,
      "learning_rate": 6.000000000000001e-07,
      "loss": 2.4282,
      "step": 12
    },
    {
      "epoch": 0.17972350230414746,
      "grad_norm": 33.53415298461914,
      "learning_rate": 6.5e-07,
      "loss": 2.4216,
      "step": 13
    },
    {
      "epoch": 0.1935483870967742,
      "grad_norm": 32.401580810546875,
      "learning_rate": 7.000000000000001e-07,
      "loss": 2.3362,
      "step": 14
    },
    {
      "epoch": 0.2073732718894009,
      "grad_norm": 33.636661529541016,
      "learning_rate": 7.5e-07,
      "loss": 2.2978,
      "step": 15
    },
    {
      "epoch": 0.22119815668202766,
      "grad_norm": 31.3782901763916,
      "learning_rate": 8.000000000000001e-07,
      "loss": 2.1358,
      "step": 16
    },
    {
      "epoch": 0.2350230414746544,
      "grad_norm": 30.72391700744629,
      "learning_rate": 8.500000000000001e-07,
      "loss": 2.0652,
      "step": 17
    },
    {
      "epoch": 0.2488479262672811,
      "grad_norm": 30.817584991455078,
      "learning_rate": 9.000000000000001e-07,
      "loss": 2.0115,
      "step": 18
    },
    {
      "epoch": 0.2626728110599078,
      "grad_norm": 29.683996200561523,
      "learning_rate": 9.500000000000001e-07,
      "loss": 1.8668,
      "step": 19
    },
    {
      "epoch": 0.2764976958525346,
      "grad_norm": 29.506683349609375,
      "learning_rate": 1.0000000000000002e-06,
      "loss": 1.7796,
      "step": 20
    },
    {
      "epoch": 0.2903225806451613,
      "grad_norm": 27.55340003967285,
      "learning_rate": 1.0500000000000001e-06,
      "loss": 1.5656,
      "step": 21
    },
    {
      "epoch": 0.30414746543778803,
      "grad_norm": 27.78036880493164,
      "learning_rate": 1.1e-06,
      "loss": 1.5112,
      "step": 22
    },
    {
      "epoch": 0.31797235023041476,
      "grad_norm": 26.36115264892578,
      "learning_rate": 1.1500000000000002e-06,
      "loss": 1.3283,
      "step": 23
    },
    {
      "epoch": 0.3317972350230415,
      "grad_norm": 25.388761520385742,
      "learning_rate": 1.2000000000000002e-06,
      "loss": 1.137,
      "step": 24
    },
    {
      "epoch": 0.3456221198156682,
      "grad_norm": 25.21432876586914,
      "learning_rate": 1.25e-06,
      "loss": 0.9867,
      "step": 25
    },
    {
      "epoch": 0.35944700460829493,
      "grad_norm": 24.924489974975586,
      "learning_rate": 1.3e-06,
      "loss": 0.7122,
      "step": 26
    },
    {
      "epoch": 0.37327188940092165,
      "grad_norm": 21.881420135498047,
      "learning_rate": 1.3500000000000002e-06,
      "loss": 0.4952,
      "step": 27
    },
    {
      "epoch": 0.3870967741935484,
      "grad_norm": 17.67154884338379,
      "learning_rate": 1.4000000000000001e-06,
      "loss": 0.3602,
      "step": 28
    },
    {
      "epoch": 0.4009216589861751,
      "grad_norm": 11.489490509033203,
      "learning_rate": 1.45e-06,
      "loss": 0.2432,
      "step": 29
    },
    {
      "epoch": 0.4147465437788018,
      "grad_norm": 7.622438907623291,
      "learning_rate": 1.5e-06,
      "loss": 0.189,
      "step": 30
    },
    {
      "epoch": 0.42857142857142855,
      "grad_norm": 4.340638637542725,
      "learning_rate": 1.5500000000000002e-06,
      "loss": 0.1302,
      "step": 31
    },
    {
      "epoch": 0.4423963133640553,
      "grad_norm": 3.079514980316162,
      "learning_rate": 1.6000000000000001e-06,
      "loss": 0.1075,
      "step": 32
    },
    {
      "epoch": 0.45622119815668205,
      "grad_norm": 2.355943441390991,
      "learning_rate": 1.6500000000000003e-06,
      "loss": 0.0998,
      "step": 33
    },
    {
      "epoch": 0.4700460829493088,
      "grad_norm": 1.9480725526809692,
      "learning_rate": 1.7000000000000002e-06,
      "loss": 0.0926,
      "step": 34
    },
    {
      "epoch": 0.4838709677419355,
      "grad_norm": 1.8598166704177856,
      "learning_rate": 1.75e-06,
      "loss": 0.0733,
      "step": 35
    },
    {
      "epoch": 0.4976958525345622,
      "grad_norm": 0.9892730712890625,
      "learning_rate": 1.8000000000000001e-06,
      "loss": 0.0664,
      "step": 36
    },
    {
      "epoch": 0.511520737327189,
      "grad_norm": 0.8992418050765991,
      "learning_rate": 1.85e-06,
      "loss": 0.0709,
      "step": 37
    },
    {
      "epoch": 0.5253456221198156,
      "grad_norm": 0.7340101599693298,
      "learning_rate": 1.9000000000000002e-06,
      "loss": 0.0535,
      "step": 38
    },
    {
      "epoch": 0.5391705069124424,
      "grad_norm": 0.7032178044319153,
      "learning_rate": 1.9500000000000004e-06,
      "loss": 0.0573,
      "step": 39
    },
    {
      "epoch": 0.5529953917050692,
      "grad_norm": 0.6449429392814636,
      "learning_rate": 2.0000000000000003e-06,
      "loss": 0.0576,
      "step": 40
    },
    {
      "epoch": 0.5668202764976958,
      "grad_norm": 0.6358592510223389,
      "learning_rate": 2.05e-06,
      "loss": 0.0502,
      "step": 41
    },
    {
      "epoch": 0.5806451612903226,
      "grad_norm": 0.572036623954773,
      "learning_rate": 2.1000000000000002e-06,
      "loss": 0.0556,
      "step": 42
    },
    {
      "epoch": 0.5944700460829493,
      "grad_norm": 0.6538863778114319,
      "learning_rate": 2.15e-06,
      "loss": 0.0556,
      "step": 43
    },
    {
      "epoch": 0.6082949308755761,
      "grad_norm": 0.3532159626483917,
      "learning_rate": 2.2e-06,
      "loss": 0.0452,
      "step": 44
    },
    {
      "epoch": 0.6221198156682027,
      "grad_norm": 0.4853012263774872,
      "learning_rate": 2.25e-06,
      "loss": 0.0471,
      "step": 45
    },
    {
      "epoch": 0.6359447004608295,
      "grad_norm": 0.4761648178100586,
      "learning_rate": 2.3000000000000004e-06,
      "loss": 0.0469,
      "step": 46
    },
    {
      "epoch": 0.6497695852534562,
      "grad_norm": 0.6094638109207153,
      "learning_rate": 2.35e-06,
      "loss": 0.047,
      "step": 47
    },
    {
      "epoch": 0.663594470046083,
      "grad_norm": 0.5211306214332581,
      "learning_rate": 2.4000000000000003e-06,
      "loss": 0.0402,
      "step": 48
    },
    {
      "epoch": 0.6774193548387096,
      "grad_norm": 0.2997778356075287,
      "learning_rate": 2.4500000000000003e-06,
      "loss": 0.0425,
      "step": 49
    },
    {
      "epoch": 0.6912442396313364,
      "grad_norm": 0.37834689021110535,
      "learning_rate": 2.5e-06,
      "loss": 0.0446,
      "step": 50
    },
    {
      "epoch": 0.7050691244239631,
      "grad_norm": 0.31011995673179626,
      "learning_rate": 2.55e-06,
      "loss": 0.0406,
      "step": 51
    },
    {
      "epoch": 0.7188940092165899,
      "grad_norm": 0.3113131523132324,
      "learning_rate": 2.6e-06,
      "loss": 0.0368,
      "step": 52
    },
    {
      "epoch": 0.7327188940092166,
      "grad_norm": 0.5685846209526062,
      "learning_rate": 2.6500000000000005e-06,
      "loss": 0.0389,
      "step": 53
    },
    {
      "epoch": 0.7465437788018433,
      "grad_norm": 0.29334983229637146,
      "learning_rate": 2.7000000000000004e-06,
      "loss": 0.0423,
      "step": 54
    },
    {
      "epoch": 0.7603686635944701,
      "grad_norm": 0.5776861906051636,
      "learning_rate": 2.7500000000000004e-06,
      "loss": 0.0399,
      "step": 55
    },
    {
      "epoch": 0.7741935483870968,
      "grad_norm": 0.35423165559768677,
      "learning_rate": 2.8000000000000003e-06,
      "loss": 0.0357,
      "step": 56
    },
    {
      "epoch": 0.7880184331797235,
      "grad_norm": 0.37902742624282837,
      "learning_rate": 2.85e-06,
      "loss": 0.0407,
      "step": 57
    },
    {
      "epoch": 0.8018433179723502,
      "grad_norm": 0.26948878169059753,
      "learning_rate": 2.9e-06,
      "loss": 0.0351,
      "step": 58
    },
    {
      "epoch": 0.815668202764977,
      "grad_norm": 0.35688117146492004,
      "learning_rate": 2.95e-06,
      "loss": 0.0377,
      "step": 59
    },
    {
      "epoch": 0.8294930875576036,
      "grad_norm": 0.5287911891937256,
      "learning_rate": 3e-06,
      "loss": 0.0377,
      "step": 60
    },
    {
      "epoch": 0.8433179723502304,
      "grad_norm": 0.2950785756111145,
      "learning_rate": 3.05e-06,
      "loss": 0.0361,
      "step": 61
    },
    {
      "epoch": 0.8571428571428571,
      "grad_norm": 0.2789723575115204,
      "learning_rate": 3.1000000000000004e-06,
      "loss": 0.032,
      "step": 62
    },
    {
      "epoch": 0.8709677419354839,
      "grad_norm": 0.2802198529243469,
      "learning_rate": 3.1500000000000003e-06,
      "loss": 0.0394,
      "step": 63
    },
    {
      "epoch": 0.8847926267281107,
      "grad_norm": 0.286981463432312,
      "learning_rate": 3.2000000000000003e-06,
      "loss": 0.033,
      "step": 64
    },
    {
      "epoch": 0.8986175115207373,
      "grad_norm": 0.37392762303352356,
      "learning_rate": 3.2500000000000002e-06,
      "loss": 0.0335,
      "step": 65
    },
    {
      "epoch": 0.9124423963133641,
      "grad_norm": 0.25025588274002075,
      "learning_rate": 3.3000000000000006e-06,
      "loss": 0.0311,
      "step": 66
    },
    {
      "epoch": 0.9262672811059908,
      "grad_norm": 0.4292861521244049,
      "learning_rate": 3.3500000000000005e-06,
      "loss": 0.0362,
      "step": 67
    },
    {
      "epoch": 0.9400921658986175,
      "grad_norm": 0.4717651307582855,
      "learning_rate": 3.4000000000000005e-06,
      "loss": 0.0303,
      "step": 68
    },
    {
      "epoch": 0.9539170506912442,
      "grad_norm": 0.49291253089904785,
      "learning_rate": 3.45e-06,
      "loss": 0.0352,
      "step": 69
    },
    {
      "epoch": 0.967741935483871,
      "grad_norm": 0.3729935586452484,
      "learning_rate": 3.5e-06,
      "loss": 0.0297,
      "step": 70
    },
    {
      "epoch": 0.9815668202764977,
      "grad_norm": 0.27150583267211914,
      "learning_rate": 3.5500000000000003e-06,
      "loss": 0.0326,
      "step": 71
    },
    {
      "epoch": 0.9953917050691244,
      "grad_norm": 0.34516096115112305,
      "learning_rate": 3.6000000000000003e-06,
      "loss": 0.0336,
      "step": 72
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.34516096115112305,
      "learning_rate": 3.65e-06,
      "loss": 0.0274,
      "step": 73
    },
    {
      "epoch": 1.0138248847926268,
      "grad_norm": 0.6282734870910645,
      "learning_rate": 3.7e-06,
      "loss": 0.0289,
      "step": 74
    },
    {
      "epoch": 1.0276497695852536,
      "grad_norm": 0.2935558557510376,
      "learning_rate": 3.7500000000000005e-06,
      "loss": 0.0308,
      "step": 75
    },
    {
      "epoch": 1.0414746543778801,
      "grad_norm": 0.3166769742965698,
      "learning_rate": 3.8000000000000005e-06,
      "loss": 0.0277,
      "step": 76
    },
    {
      "epoch": 1.055299539170507,
      "grad_norm": 0.38190239667892456,
      "learning_rate": 3.85e-06,
      "loss": 0.0338,
      "step": 77
    },
    {
      "epoch": 1.0691244239631337,
      "grad_norm": 0.2779421806335449,
      "learning_rate": 3.900000000000001e-06,
      "loss": 0.03,
      "step": 78
    },
    {
      "epoch": 1.0829493087557605,
      "grad_norm": 0.4055996537208557,
      "learning_rate": 3.95e-06,
      "loss": 0.0295,
      "step": 79
    },
    {
      "epoch": 1.096774193548387,
      "grad_norm": 0.2987312972545624,
      "learning_rate": 4.000000000000001e-06,
      "loss": 0.028,
      "step": 80
    },
    {
      "epoch": 1.1105990783410138,
      "grad_norm": 0.2674776017665863,
      "learning_rate": 4.05e-06,
      "loss": 0.0243,
      "step": 81
    },
    {
      "epoch": 1.1244239631336406,
      "grad_norm": 0.29042816162109375,
      "learning_rate": 4.1e-06,
      "loss": 0.0318,
      "step": 82
    },
    {
      "epoch": 1.1382488479262673,
      "grad_norm": 0.2904883027076721,
      "learning_rate": 4.15e-06,
      "loss": 0.0257,
      "step": 83
    },
    {
      "epoch": 1.1520737327188941,
      "grad_norm": 0.30603015422821045,
      "learning_rate": 4.2000000000000004e-06,
      "loss": 0.0284,
      "step": 84
    },
    {
      "epoch": 1.1658986175115207,
      "grad_norm": 0.23131045699119568,
      "learning_rate": 4.25e-06,
      "loss": 0.0285,
      "step": 85
    },
    {
      "epoch": 1.1797235023041475,
      "grad_norm": 0.26788002252578735,
      "learning_rate": 4.3e-06,
      "loss": 0.0269,
      "step": 86
    },
    {
      "epoch": 1.1935483870967742,
      "grad_norm": 0.2639651894569397,
      "learning_rate": 4.350000000000001e-06,
      "loss": 0.0289,
      "step": 87
    },
    {
      "epoch": 1.2073732718894008,
      "grad_norm": 0.25068584084510803,
      "learning_rate": 4.4e-06,
      "loss": 0.0275,
      "step": 88
    },
    {
      "epoch": 1.2211981566820276,
      "grad_norm": 0.25494542717933655,
      "learning_rate": 4.450000000000001e-06,
      "loss": 0.0275,
      "step": 89
    },
    {
      "epoch": 1.2350230414746544,
      "grad_norm": 0.31125035881996155,
      "learning_rate": 4.5e-06,
      "loss": 0.0251,
      "step": 90
    },
    {
      "epoch": 1.2488479262672811,
      "grad_norm": 0.2691773474216461,
      "learning_rate": 4.5500000000000005e-06,
      "loss": 0.0267,
      "step": 91
    },
    {
      "epoch": 1.262672811059908,
      "grad_norm": 0.20079147815704346,
      "learning_rate": 4.600000000000001e-06,
      "loss": 0.0263,
      "step": 92
    },
    {
      "epoch": 1.2764976958525347,
      "grad_norm": 0.28027331829071045,
      "learning_rate": 4.65e-06,
      "loss": 0.0227,
      "step": 93
    },
    {
      "epoch": 1.2903225806451613,
      "grad_norm": 0.40053099393844604,
      "learning_rate": 4.7e-06,
      "loss": 0.0246,
      "step": 94
    },
    {
      "epoch": 1.304147465437788,
      "grad_norm": 0.33066362142562866,
      "learning_rate": 4.75e-06,
      "loss": 0.0221,
      "step": 95
    },
    {
      "epoch": 1.3179723502304148,
      "grad_norm": 0.2531339228153229,
      "learning_rate": 4.800000000000001e-06,
      "loss": 0.0216,
      "step": 96
    },
    {
      "epoch": 1.3317972350230414,
      "grad_norm": 0.37544378638267517,
      "learning_rate": 4.85e-06,
      "loss": 0.0247,
      "step": 97
    },
    {
      "epoch": 1.3456221198156681,
      "grad_norm": 0.34273672103881836,
      "learning_rate": 4.9000000000000005e-06,
      "loss": 0.0217,
      "step": 98
    },
    {
      "epoch": 1.359447004608295,
      "grad_norm": 0.2338661253452301,
      "learning_rate": 4.95e-06,
      "loss": 0.0237,
      "step": 99
    },
    {
      "epoch": 1.3732718894009217,
      "grad_norm": 0.30151981115341187,
      "learning_rate": 5e-06,
      "loss": 0.0248,
      "step": 100
    },
    {
      "epoch": 1.3870967741935485,
      "grad_norm": 0.3205336630344391,
      "learning_rate": 4.999888074163108e-06,
      "loss": 0.0232,
      "step": 101
    },
    {
      "epoch": 1.400921658986175,
      "grad_norm": 0.2705315351486206,
      "learning_rate": 4.999552306674345e-06,
      "loss": 0.0245,
      "step": 102
    },
    {
      "epoch": 1.4147465437788018,
      "grad_norm": 0.2564137578010559,
      "learning_rate": 4.998992727598557e-06,
      "loss": 0.0274,
      "step": 103
    },
    {
      "epoch": 1.4285714285714286,
      "grad_norm": 0.1967611312866211,
      "learning_rate": 4.998209387040829e-06,
      "loss": 0.0173,
      "step": 104
    },
    {
      "epoch": 1.4423963133640554,
      "grad_norm": 0.2568240761756897,
      "learning_rate": 4.9972023551419995e-06,
      "loss": 0.0223,
      "step": 105
    },
    {
      "epoch": 1.456221198156682,
      "grad_norm": 0.2236352413892746,
      "learning_rate": 4.995971722072379e-06,
      "loss": 0.0202,
      "step": 106
    },
    {
      "epoch": 1.4700460829493087,
      "grad_norm": 0.3389627933502197,
      "learning_rate": 4.9945175980236745e-06,
      "loss": 0.0214,
      "step": 107
    },
    {
      "epoch": 1.4838709677419355,
      "grad_norm": 0.31428012251853943,
      "learning_rate": 4.992840113199131e-06,
      "loss": 0.0188,
      "step": 108
    },
    {
      "epoch": 1.4976958525345623,
      "grad_norm": 0.41508516669273376,
      "learning_rate": 4.990939417801859e-06,
      "loss": 0.0213,
      "step": 109
    },
    {
      "epoch": 1.511520737327189,
      "grad_norm": 0.19615545868873596,
      "learning_rate": 4.988815682021398e-06,
      "loss": 0.0191,
      "step": 110
    },
    {
      "epoch": 1.5253456221198156,
      "grad_norm": 0.2059931755065918,
      "learning_rate": 4.986469096018472e-06,
      "loss": 0.0208,
      "step": 111
    },
    {
      "epoch": 1.5391705069124424,
      "grad_norm": 0.26946336030960083,
      "learning_rate": 4.983899869907963e-06,
      "loss": 0.0192,
      "step": 112
    },
    {
      "epoch": 1.5529953917050692,
      "grad_norm": 0.3227538466453552,
      "learning_rate": 4.981108233740096e-06,
      "loss": 0.0169,
      "step": 113
    },
    {
      "epoch": 1.5668202764976957,
      "grad_norm": 0.2811918258666992,
      "learning_rate": 4.978094437479843e-06,
      "loss": 0.0151,
      "step": 114
    },
    {
      "epoch": 1.5806451612903225,
      "grad_norm": 0.32980477809906006,
      "learning_rate": 4.97485875098454e-06,
      "loss": 0.0182,
      "step": 115
    },
    {
      "epoch": 1.5944700460829493,
      "grad_norm": 0.2759259045124054,
      "learning_rate": 4.971401463979722e-06,
      "loss": 0.0192,
      "step": 116
    },
    {
      "epoch": 1.608294930875576,
      "grad_norm": 0.2572178840637207,
      "learning_rate": 4.967722886033181e-06,
      "loss": 0.0198,
      "step": 117
    },
    {
      "epoch": 1.6221198156682028,
      "grad_norm": 0.3238658905029297,
      "learning_rate": 4.963823346527249e-06,
      "loss": 0.0186,
      "step": 118
    },
    {
      "epoch": 1.6359447004608296,
      "grad_norm": 0.3834918737411499,
      "learning_rate": 4.959703194629304e-06,
      "loss": 0.0188,
      "step": 119
    },
    {
      "epoch": 1.6497695852534562,
      "grad_norm": 0.23881244659423828,
      "learning_rate": 4.955362799260507e-06,
      "loss": 0.0182,
      "step": 120
    },
    {
      "epoch": 1.663594470046083,
      "grad_norm": 0.1885918825864792,
      "learning_rate": 4.950802549062764e-06,
      "loss": 0.0183,
      "step": 121
    },
    {
      "epoch": 1.6774193548387095,
      "grad_norm": 0.34959614276885986,
      "learning_rate": 4.946022852363932e-06,
      "loss": 0.0173,
      "step": 122
    },
    {
      "epoch": 1.6912442396313363,
      "grad_norm": 0.22990310192108154,
      "learning_rate": 4.9410241371412525e-06,
      "loss": 0.0135,
      "step": 123
    },
    {
      "epoch": 1.705069124423963,
      "grad_norm": 0.2790350615978241,
      "learning_rate": 4.935806850983034e-06,
      "loss": 0.0159,
      "step": 124
    },
    {
      "epoch": 1.7188940092165899,
      "grad_norm": 0.3218020796775818,
      "learning_rate": 4.9303714610485705e-06,
      "loss": 0.0176,
      "step": 125
    },
    {
      "epoch": 1.7327188940092166,
      "grad_norm": 0.2294609695672989,
      "learning_rate": 4.924718454026318e-06,
      "loss": 0.0149,
      "step": 126
    },
    {
      "epoch": 1.7465437788018434,
      "grad_norm": 0.3427927494049072,
      "learning_rate": 4.918848336090309e-06,
      "loss": 0.0165,
      "step": 127
    },
    {
      "epoch": 1.7603686635944702,
      "grad_norm": 0.22731825709342957,
      "learning_rate": 4.912761632854834e-06,
      "loss": 0.0145,
      "step": 128
    },
    {
      "epoch": 1.7741935483870968,
      "grad_norm": 0.35364386439323425,
      "learning_rate": 4.906458889327375e-06,
      "loss": 0.0161,
      "step": 129
    },
    {
      "epoch": 1.7880184331797235,
      "grad_norm": 0.29476454854011536,
      "learning_rate": 4.899940669859807e-06,
      "loss": 0.0154,
      "step": 130
    },
    {
      "epoch": 1.80184331797235,
      "grad_norm": 0.28667864203453064,
      "learning_rate": 4.893207558097867e-06,
      "loss": 0.0143,
      "step": 131
    },
    {
      "epoch": 1.8156682027649769,
      "grad_norm": 0.2731999158859253,
      "learning_rate": 4.8862601569288885e-06,
      "loss": 0.0141,
      "step": 132
    },
    {
      "epoch": 1.8294930875576036,
      "grad_norm": 0.2670470178127289,
      "learning_rate": 4.879099088427824e-06,
      "loss": 0.0131,
      "step": 133
    },
    {
      "epoch": 1.8433179723502304,
      "grad_norm": 0.23313525319099426,
      "learning_rate": 4.871724993801541e-06,
      "loss": 0.012,
      "step": 134
    },
    {
      "epoch": 1.8571428571428572,
      "grad_norm": 0.2192607820034027,
      "learning_rate": 4.864138533331411e-06,
      "loss": 0.0125,
      "step": 135
    },
    {
      "epoch": 1.870967741935484,
      "grad_norm": 0.26603585481643677,
      "learning_rate": 4.8563403863141825e-06,
      "loss": 0.0121,
      "step": 136
    },
    {
      "epoch": 1.8847926267281108,
      "grad_norm": 0.32500001788139343,
      "learning_rate": 4.84833125100116e-06,
      "loss": 0.0116,
      "step": 137
    },
    {
      "epoch": 1.8986175115207373,
      "grad_norm": 0.24893291294574738,
      "learning_rate": 4.840111844535682e-06,
      "loss": 0.0119,
      "step": 138
    },
    {
      "epoch": 1.912442396313364,
      "grad_norm": 0.17670764029026031,
      "learning_rate": 4.8316829028889076e-06,
      "loss": 0.0096,
      "step": 139
    },
    {
      "epoch": 1.9262672811059907,
      "grad_norm": 0.16747575998306274,
      "learning_rate": 4.823045180793914e-06,
      "loss": 0.0113,
      "step": 140
    },
    {
      "epoch": 1.9400921658986174,
      "grad_norm": 0.19587458670139313,
      "learning_rate": 4.8141994516781196e-06,
      "loss": 0.0111,
      "step": 141
    },
    {
      "epoch": 1.9539170506912442,
      "grad_norm": 0.237543985247612,
      "learning_rate": 4.805146507594034e-06,
      "loss": 0.0088,
      "step": 142
    },
    {
      "epoch": 1.967741935483871,
      "grad_norm": 0.22710399329662323,
      "learning_rate": 4.7958871591483305e-06,
      "loss": 0.0085,
      "step": 143
    },
    {
      "epoch": 1.9815668202764978,
      "grad_norm": 0.2946629822254181,
      "learning_rate": 4.786422235429269e-06,
      "loss": 0.0122,
      "step": 144
    }
  ],
  "logging_steps": 1,
  "max_steps": 432,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 6,
  "save_steps": 72,
  "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.220896013978436e+18,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}