Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f49a59670c717e56d6a4311ed243bbd975926e8990d9e46a3f5234b3c409750c
+oid sha256:18bcd1c1737518486f8d3c4c8949b8d131644b4eca28f7962a7aeca7a70338a7
 size 503128704

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:317ce9079bf77ed5e4d3e2644b763fed8157cb653f508847c9507ff162ebe752
+oid sha256:abadb84a828ae1cde4669f147e6c74709632b3d334ead5406ea61de144dab15f
 size 1006351290

checkpoint-10000/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d4f8db97fd2d786dd18e1660a629749563cc5b59d400bd62a512ac18ce773e2a
+oid sha256:95d415f5a2495efee5f0bd162340bac975693e203cab718136e9d616a832b275
 size 14244

checkpoint-10000/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e5a954fd0891b9a63bb5018718359621b31d6887975b43dbbefda70f75ac15a5
+oid sha256:baaddc0dc177f093cb7927eae851400a36fde6f45129bcaf2df443bdde5269f4
 size 1064

checkpoint-10000/trainer_state.json

@@ -1,7 +1,7 @@
 {
-  "best_metric": 3.7542877197265625,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_634/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.7473487854003906,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_high_2000_634/checkpoint-10000",
+  "epoch": 1.0781671159029649,
   "eval_steps": 1000,
   "global_step": 10000,
   "is_hyper_param_search": false,
@@ -9,1498 +9,1498 @@
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.005381552039608223,
-      "grad_norm": 0.947643518447876,
-      "learning_rate": 0.00028199999999999997,
-      "loss": 9.02,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 1.4153761863708496,
+      "learning_rate": 0.0003,
+      "loss": 8.6258,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 2.6564507484436035,
-      "learning_rate": 0.0005819999999999999,
-      "loss": 6.9671,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 4.798243045806885,
+      "learning_rate": 0.0006,
+      "loss": 6.9282,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 2.533383369445801,
-      "learning_rate": 0.0005996961534317422,
-      "loss": 6.5213,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 2.44549560546875,
+      "learning_rate": 0.0005996762007555315,
+      "loss": 6.5183,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 3.2595882415771484,
-      "learning_rate": 0.0005993729124016807,
-      "loss": 6.2948,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.1002956628799438,
+      "learning_rate": 0.000599352401511063,
+      "loss": 6.263,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.9747371673583984,
-      "learning_rate": 0.0005990496713716194,
-      "loss": 6.1227,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 0.991017758846283,
+      "learning_rate": 0.0005990286022665946,
+      "loss": 6.1066,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 1.3486372232437134,
-      "learning_rate": 0.000598726430341558,
-      "loss": 5.989,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 1.7810102701187134,
+      "learning_rate": 0.0005987048030221263,
+      "loss": 5.9974,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 2.030965805053711,
-      "learning_rate": 0.0005984031893114966,
-      "loss": 5.8853,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 1.4339019060134888,
+      "learning_rate": 0.0005983810037776578,
+      "loss": 5.8832,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 1.8007402420043945,
-      "learning_rate": 0.0005980799482814351,
-      "loss": 5.8246,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 1.7068945169448853,
+      "learning_rate": 0.0005980572045331894,
+      "loss": 5.7989,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.0434612035751343,
-      "learning_rate": 0.0005977567072513737,
-      "loss": 5.7759,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 1.2276661396026611,
+      "learning_rate": 0.0005977334052887209,
+      "loss": 5.7294,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 1.30581796169281,
-      "learning_rate": 0.0005974334662213123,
-      "loss": 5.6783,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 1.4345930814743042,
+      "learning_rate": 0.0005974096060442526,
+      "loss": 5.6735,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 1.3575013875961304,
-      "learning_rate": 0.0005971102251912509,
-      "loss": 5.6197,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 1.651483178138733,
+      "learning_rate": 0.0005970858067997841,
+      "loss": 5.5941,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.027186632156372,
-      "learning_rate": 0.0005967869841611895,
-      "loss": 5.5417,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.4423189163208008,
+      "learning_rate": 0.0005967620075553157,
+      "loss": 5.528,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 1.0279064178466797,
-      "learning_rate": 0.000596463743131128,
-      "loss": 5.4727,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 0.9989488124847412,
+      "learning_rate": 0.0005964382083108472,
+      "loss": 5.4406,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 1.1722652912139893,
-      "learning_rate": 0.0005961405021010667,
-      "loss": 5.418,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 1.5697565078735352,
+      "learning_rate": 0.0005961144090663788,
+      "loss": 5.3854,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 0.9741182327270508,
-      "learning_rate": 0.0005958172610710052,
-      "loss": 5.3362,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.3698347806930542,
+      "learning_rate": 0.0005957906098219104,
+      "loss": 5.316,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 1.3713257312774658,
-      "learning_rate": 0.0005954940200409439,
-      "loss": 5.2961,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 0.9174061417579651,
+      "learning_rate": 0.0005954668105774419,
+      "loss": 5.2632,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.626237392425537,
-      "learning_rate": 0.0005951707790108824,
-      "loss": 5.2518,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 0.8781394958496094,
+      "learning_rate": 0.0005951430113329735,
+      "loss": 5.2044,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.4366179704666138,
-      "learning_rate": 0.0005948475379808209,
-      "loss": 5.2072,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 0.9764671325683594,
+      "learning_rate": 0.0005948192120885051,
+      "loss": 5.1792,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 1.0901954174041748,
-      "learning_rate": 0.0005945242969507596,
-      "loss": 5.1598,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 1.7501579523086548,
+      "learning_rate": 0.0005944954128440366,
+      "loss": 5.1443,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.056020975112915,
-      "learning_rate": 0.0005942010559206981,
-      "loss": 5.1238,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 1.8602075576782227,
+      "learning_rate": 0.0005941716135995682,
+      "loss": 5.0898,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.22448784016638668,
-      "eval_loss": 5.047372341156006,
-      "eval_runtime": 186.0066,
-      "eval_samples_per_second": 96.83,
-      "eval_steps_per_second": 6.054,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22601321805841418,
+      "eval_loss": 5.037795066833496,
+      "eval_runtime": 187.186,
+      "eval_samples_per_second": 96.22,
+      "eval_steps_per_second": 6.015,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 1.2061820030212402,
-      "learning_rate": 0.0005938778148906367,
-      "loss": 5.0644,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 0.9920917749404907,
+      "learning_rate": 0.0005938478143550997,
+      "loss": 5.0498,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 0.8459726572036743,
-      "learning_rate": 0.0005935545738605753,
-      "loss": 5.0343,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.1556609869003296,
+      "learning_rate": 0.0005935240151106314,
+      "loss": 5.0218,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 1.117684006690979,
-      "learning_rate": 0.000593231332830514,
-      "loss": 4.9795,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 1.180494785308838,
+      "learning_rate": 0.0005932002158661629,
+      "loss": 4.9949,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 1.4578666687011719,
-      "learning_rate": 0.0005929080918004525,
-      "loss": 4.9842,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 1.130318522453308,
+      "learning_rate": 0.0005928764166216945,
+      "loss": 4.9329,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 0.9818866848945618,
-      "learning_rate": 0.000592584850770391,
-      "loss": 4.9541,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 1.003656029701233,
+      "learning_rate": 0.000592552617377226,
+      "loss": 4.9114,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 0.8982384204864502,
-      "learning_rate": 0.0005922616097403296,
-      "loss": 4.9085,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 1.564238429069519,
+      "learning_rate": 0.0005922288181327577,
+      "loss": 4.8916,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 1.4174669981002808,
-      "learning_rate": 0.0005919383687102682,
-      "loss": 4.8986,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 0.8959649801254272,
+      "learning_rate": 0.0005919050188882893,
+      "loss": 4.8536,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 0.9739001393318176,
-      "learning_rate": 0.0005916151276802069,
-      "loss": 4.873,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 0.8476941585540771,
+      "learning_rate": 0.0005915812196438207,
+      "loss": 4.8471,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 1.062203049659729,
-      "learning_rate": 0.0005912918866501454,
-      "loss": 4.8505,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 1.2242474555969238,
+      "learning_rate": 0.0005912574203993524,
+      "loss": 4.8488,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 1.0389671325683594,
-      "learning_rate": 0.000590968645620084,
-      "loss": 4.8168,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 0.8395281434059143,
+      "learning_rate": 0.0005909336211548839,
+      "loss": 4.8122,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 1.0054130554199219,
-      "learning_rate": 0.0005906454045900226,
-      "loss": 4.8127,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 1.100954294204712,
+      "learning_rate": 0.0005906098219104155,
+      "loss": 4.7627,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 1.2191665172576904,
-      "learning_rate": 0.0005903221635599612,
-      "loss": 4.7919,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 0.8952234387397766,
+      "learning_rate": 0.000590286022665947,
+      "loss": 4.7551,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 0.8363758325576782,
-      "learning_rate": 0.0005899989225298998,
-      "loss": 4.7589,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 0.9563343524932861,
+      "learning_rate": 0.0005899622234214787,
+      "loss": 4.7286,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 1.000256061553955,
-      "learning_rate": 0.0005896756814998383,
-      "loss": 4.7222,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.9019905924797058,
+      "learning_rate": 0.0005896384241770102,
+      "loss": 4.7041,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 1.0218900442123413,
-      "learning_rate": 0.0005893524404697769,
-      "loss": 4.7129,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 1.0329313278198242,
+      "learning_rate": 0.0005893146249325418,
+      "loss": 4.714,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 0.8401052951812744,
-      "learning_rate": 0.0005890291994397155,
-      "loss": 4.6975,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 1.022636890411377,
+      "learning_rate": 0.0005889908256880733,
+      "loss": 4.679,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 1.0717366933822632,
-      "learning_rate": 0.0005887059584096541,
-      "loss": 4.6681,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 1.012805700302124,
+      "learning_rate": 0.0005886670264436049,
+      "loss": 4.6485,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 0.8167716264724731,
-      "learning_rate": 0.0005883827173795926,
-      "loss": 4.6492,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 1.017619252204895,
+      "learning_rate": 0.0005883432271991365,
+      "loss": 4.6367,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 0.7771570682525635,
-      "learning_rate": 0.0005880594763495313,
-      "loss": 4.6232,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 1.0787638425827026,
+      "learning_rate": 0.0005880194279546681,
+      "loss": 4.6371,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 1.1614071130752563,
-      "learning_rate": 0.0005877362353194698,
-      "loss": 4.6101,
+      "epoch": 0.215633423180593,
+      "grad_norm": 1.0098811388015747,
+      "learning_rate": 0.0005876956287101996,
+      "loss": 4.5856,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.2659886798902302,
-      "eval_loss": 4.5428266525268555,
-      "eval_runtime": 184.7289,
-      "eval_samples_per_second": 97.5,
-      "eval_steps_per_second": 6.095,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.26980304816984535,
+      "eval_loss": 4.516435146331787,
+      "eval_runtime": 185.8323,
+      "eval_samples_per_second": 96.921,
+      "eval_steps_per_second": 6.059,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 0.8067775368690491,
-      "learning_rate": 0.0005874129942894084,
-      "loss": 4.5945,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.8449752330780029,
+      "learning_rate": 0.0005873718294657312,
+      "loss": 4.5674,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 1.3052033185958862,
-      "learning_rate": 0.000587089753259347,
-      "loss": 4.576,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 0.6898910999298096,
+      "learning_rate": 0.0005870480302212628,
+      "loss": 4.5584,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 1.0830671787261963,
-      "learning_rate": 0.0005867665122292855,
-      "loss": 4.573,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 0.9888122081756592,
+      "learning_rate": 0.0005867242309767943,
+      "loss": 4.5375,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 1.00840425491333,
-      "learning_rate": 0.0005864432711992242,
-      "loss": 4.5518,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 0.8149335384368896,
+      "learning_rate": 0.0005864004317323259,
+      "loss": 4.4984,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 1.4997429847717285,
-      "learning_rate": 0.0005861200301691628,
-      "loss": 4.5326,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 1.0397859811782837,
+      "learning_rate": 0.0005860766324878575,
+      "loss": 4.5126,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 0.8558619618415833,
-      "learning_rate": 0.0005857967891391014,
-      "loss": 4.4986,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 0.8346261978149414,
+      "learning_rate": 0.000585752833243389,
+      "loss": 4.4873,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.8113582134246826,
-      "learning_rate": 0.0005854735481090399,
-      "loss": 4.5091,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.9166735410690308,
+      "learning_rate": 0.0005854290339989206,
+      "loss": 4.4653,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 0.98548823595047,
-      "learning_rate": 0.0005851503070789784,
-      "loss": 4.4801,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 0.9180364012718201,
+      "learning_rate": 0.0005851052347544521,
+      "loss": 4.4583,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 0.8301020860671997,
-      "learning_rate": 0.0005848270660489171,
-      "loss": 4.4595,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 0.8483642339706421,
+      "learning_rate": 0.0005847814355099838,
+      "loss": 4.4384,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 1.0569989681243896,
-      "learning_rate": 0.0005845038250188556,
-      "loss": 4.4474,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 0.7761411070823669,
+      "learning_rate": 0.0005844576362655154,
+      "loss": 4.4318,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.9343509674072266,
-      "learning_rate": 0.0005841805839887943,
-      "loss": 4.4274,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.8839765787124634,
+      "learning_rate": 0.0005841338370210469,
+      "loss": 4.4097,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 0.8506145477294922,
-      "learning_rate": 0.0005838573429587328,
-      "loss": 4.4131,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 1.1479021310806274,
+      "learning_rate": 0.0005838100377765785,
+      "loss": 4.3848,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.9148919582366943,
-      "learning_rate": 0.0005835341019286715,
-      "loss": 4.3894,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 0.8199343085289001,
+      "learning_rate": 0.0005834862385321101,
+      "loss": 4.3774,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.8359649777412415,
-      "learning_rate": 0.00058321086089861,
-      "loss": 4.4011,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 0.8475073575973511,
+      "learning_rate": 0.0005831624392876417,
+      "loss": 4.3724,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.9812710881233215,
-      "learning_rate": 0.0005828876198685486,
-      "loss": 4.3753,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.9192224144935608,
+      "learning_rate": 0.0005828386400431731,
+      "loss": 4.3824,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 1.0767602920532227,
-      "learning_rate": 0.0005825643788384872,
-      "loss": 4.374,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.7423735857009888,
+      "learning_rate": 0.0005825148407987048,
+      "loss": 4.3731,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 0.8539724349975586,
-      "learning_rate": 0.0005822411378084257,
-      "loss": 4.3591,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.8002654314041138,
+      "learning_rate": 0.0005821910415542363,
+      "loss": 4.3368,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 0.8400486707687378,
-      "learning_rate": 0.0005819178967783644,
-      "loss": 4.3441,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.8412148356437683,
+      "learning_rate": 0.0005818672423097679,
+      "loss": 4.3234,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.9034419059753418,
-      "learning_rate": 0.0005815946557483029,
-      "loss": 4.3381,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.846697986125946,
+      "learning_rate": 0.0005815434430652994,
+      "loss": 4.3543,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.8509873151779175,
-      "learning_rate": 0.0005812714147182415,
-      "loss": 4.327,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.7227766513824463,
+      "learning_rate": 0.0005812196438208311,
+      "loss": 4.318,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.2964122490484995,
-      "eval_loss": 4.251293659210205,
-      "eval_runtime": 184.751,
-      "eval_samples_per_second": 97.488,
-      "eval_steps_per_second": 6.095,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.29839266523469404,
+      "eval_loss": 4.236840724945068,
+      "eval_runtime": 179.5407,
+      "eval_samples_per_second": 100.317,
+      "eval_steps_per_second": 6.272,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.8212876915931702,
-      "learning_rate": 0.0005809481736881801,
-      "loss": 4.3028,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 0.8503046631813049,
+      "learning_rate": 0.0005808958445763626,
+      "loss": 4.309,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.7386678457260132,
-      "learning_rate": 0.0005806249326581187,
-      "loss": 4.3224,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.8151911497116089,
+      "learning_rate": 0.0005805720453318942,
+      "loss": 4.2919,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.8408326506614685,
-      "learning_rate": 0.0005803016916280573,
-      "loss": 4.3054,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.8732330203056335,
+      "learning_rate": 0.0005802482460874257,
+      "loss": 4.2785,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.7891979217529297,
-      "learning_rate": 0.0005799784505979959,
-      "loss": 4.2832,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.7436563372612,
+      "learning_rate": 0.0005799244468429573,
+      "loss": 4.2827,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 1.1157724857330322,
-      "learning_rate": 0.0005796552095679344,
-      "loss": 4.2875,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 1.0856059789657593,
+      "learning_rate": 0.0005796006475984889,
+      "loss": 4.2564,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.7911354899406433,
-      "learning_rate": 0.000579331968537873,
-      "loss": 4.2468,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.6840450167655945,
+      "learning_rate": 0.0005792768483540205,
+      "loss": 4.2669,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.670813798904419,
-      "learning_rate": 0.0005790087275078116,
-      "loss": 4.2655,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.9430235028266907,
+      "learning_rate": 0.000578953049109552,
+      "loss": 4.2636,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.7761114835739136,
-      "learning_rate": 0.0005786854864777502,
-      "loss": 4.2688,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.7581385970115662,
+      "learning_rate": 0.0005786292498650836,
+      "loss": 4.2397,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.8588844537734985,
-      "learning_rate": 0.0005783622454476888,
-      "loss": 4.2547,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.9222409725189209,
+      "learning_rate": 0.0005783054506206152,
+      "loss": 4.2354,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.8100800514221191,
-      "learning_rate": 0.0005780390044176273,
-      "loss": 4.2286,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.6859261989593506,
+      "learning_rate": 0.0005779816513761467,
+      "loss": 4.2277,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.8621886372566223,
-      "learning_rate": 0.000577715763387566,
-      "loss": 4.2537,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.8531029224395752,
+      "learning_rate": 0.0005776578521316782,
+      "loss": 4.2054,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.681401252746582,
-      "learning_rate": 0.0005773925223575045,
-      "loss": 4.2342,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.8664888739585876,
+      "learning_rate": 0.0005773340528872099,
+      "loss": 4.2126,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.8861917853355408,
-      "learning_rate": 0.0005770692813274432,
-      "loss": 4.2298,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 0.7247611880302429,
+      "learning_rate": 0.0005770102536427414,
+      "loss": 4.2271,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.7586415410041809,
-      "learning_rate": 0.0005767460402973817,
-      "loss": 4.1935,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.7640480995178223,
+      "learning_rate": 0.000576686454398273,
+      "loss": 4.2139,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.7726384401321411,
-      "learning_rate": 0.0005764227992673203,
-      "loss": 4.1981,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.7702681422233582,
+      "learning_rate": 0.0005763626551538045,
+      "loss": 4.1992,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.8231189846992493,
-      "learning_rate": 0.0005760995582372589,
-      "loss": 4.1916,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.7853514552116394,
+      "learning_rate": 0.0005760388559093362,
+      "loss": 4.189,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.6635996103286743,
-      "learning_rate": 0.0005757763172071974,
-      "loss": 4.194,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.7235543727874756,
+      "learning_rate": 0.0005757150566648678,
+      "loss": 4.1822,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.6655986905097961,
-      "learning_rate": 0.000575453076177136,
-      "loss": 4.2108,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 0.7994705438613892,
+      "learning_rate": 0.0005753912574203993,
+      "loss": 4.176,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 0.6464502215385437,
-      "learning_rate": 0.0005751298351470746,
-      "loss": 4.1773,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.7909198999404907,
+      "learning_rate": 0.0005750674581759309,
+      "loss": 4.1754,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 1.0790879726409912,
-      "learning_rate": 0.0005748065941170133,
-      "loss": 4.18,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.8087334632873535,
+      "learning_rate": 0.0005747436589314624,
+      "loss": 4.1599,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.3111644862386228,
-      "eval_loss": 4.104041576385498,
-      "eval_runtime": 184.532,
-      "eval_samples_per_second": 97.604,
-      "eval_steps_per_second": 6.102,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.3121017260489105,
+      "eval_loss": 4.09109354019165,
+      "eval_runtime": 179.4252,
+      "eval_samples_per_second": 100.382,
+      "eval_steps_per_second": 6.276,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.7633262872695923,
-      "learning_rate": 0.0005744833530869518,
-      "loss": 4.1895,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.790748655796051,
+      "learning_rate": 0.0005744198596869941,
+      "loss": 4.1715,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.7926012277603149,
-      "learning_rate": 0.0005741601120568903,
-      "loss": 4.1728,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.6250526905059814,
+      "learning_rate": 0.0005740960604425255,
+      "loss": 4.1647,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.6977135539054871,
-      "learning_rate": 0.0005738368710268289,
-      "loss": 4.1502,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.7327976822853088,
+      "learning_rate": 0.0005737722611980572,
+      "loss": 4.1432,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.7949364185333252,
-      "learning_rate": 0.0005735136299967675,
-      "loss": 4.181,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 0.6393489241600037,
+      "learning_rate": 0.0005734484619535887,
+      "loss": 4.1223,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.6541943550109863,
-      "learning_rate": 0.0005731903889667062,
-      "loss": 4.1484,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.6468517780303955,
+      "learning_rate": 0.0005731246627091203,
+      "loss": 4.1544,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.7301810383796692,
-      "learning_rate": 0.0005728671479366447,
-      "loss": 4.1502,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.631144106388092,
+      "learning_rate": 0.0005728008634646518,
+      "loss": 4.1296,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.7047900557518005,
-      "learning_rate": 0.0005725439069065833,
-      "loss": 4.1516,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.7250542640686035,
+      "learning_rate": 0.0005724770642201835,
+      "loss": 4.1325,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.6980259418487549,
-      "learning_rate": 0.0005722206658765219,
-      "loss": 4.1214,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.7701312303543091,
+      "learning_rate": 0.000572153264975715,
+      "loss": 4.128,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 0.7646435499191284,
-      "learning_rate": 0.0005718974248464605,
-      "loss": 4.1348,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.7531642317771912,
+      "learning_rate": 0.0005718294657312466,
+      "loss": 4.1011,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.7655755877494812,
-      "learning_rate": 0.000571574183816399,
-      "loss": 4.1344,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.6209056377410889,
+      "learning_rate": 0.0005715056664867781,
+      "loss": 4.0979,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.7866830825805664,
-      "learning_rate": 0.0005712509427863376,
-      "loss": 4.1116,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.7149604558944702,
+      "learning_rate": 0.0005711818672423097,
+      "loss": 4.0913,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.7541496753692627,
-      "learning_rate": 0.0005709277017562762,
-      "loss": 4.0941,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.621547281742096,
+      "learning_rate": 0.0005708580679978413,
+      "loss": 4.0978,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.7112383842468262,
-      "learning_rate": 0.0005706044607262148,
-      "loss": 4.0955,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.8008401989936829,
+      "learning_rate": 0.0005705342687533729,
+      "loss": 4.0927,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 0.7648486495018005,
-      "learning_rate": 0.0005702812196961534,
-      "loss": 4.1013,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.7120444774627686,
+      "learning_rate": 0.0005702104695089044,
+      "loss": 4.0748,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.6658865213394165,
-      "learning_rate": 0.0005699579786660919,
-      "loss": 4.0833,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.8041061758995056,
+      "learning_rate": 0.000569886670264436,
+      "loss": 4.0861,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.6460030674934387,
-      "learning_rate": 0.0005696347376360306,
-      "loss": 4.0976,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.7002294063568115,
+      "learning_rate": 0.0005695628710199675,
+      "loss": 4.0773,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.7306205630302429,
-      "learning_rate": 0.0005693114966059691,
-      "loss": 4.0811,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.5887237191200256,
+      "learning_rate": 0.0005692390717754991,
+      "loss": 4.0713,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.6842286586761475,
-      "learning_rate": 0.0005689882555759077,
-      "loss": 4.0634,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.6446994543075562,
+      "learning_rate": 0.0005689152725310306,
+      "loss": 4.0624,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.8138746619224548,
-      "learning_rate": 0.0005686650145458463,
-      "loss": 4.0802,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.6368472576141357,
+      "learning_rate": 0.0005685914732865623,
+      "loss": 4.0632,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.6573743224143982,
-      "learning_rate": 0.0005683417735157848,
-      "loss": 4.0788,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.6480772495269775,
+      "learning_rate": 0.0005682676740420939,
+      "loss": 4.0769,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.3209131492920015,
-      "eval_loss": 4.001318454742432,
-      "eval_runtime": 184.7779,
-      "eval_samples_per_second": 97.474,
-      "eval_steps_per_second": 6.094,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.321377857693184,
+      "eval_loss": 3.994414806365967,
+      "eval_runtime": 179.4261,
+      "eval_samples_per_second": 100.381,
+      "eval_steps_per_second": 6.276,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.621402382850647,
-      "learning_rate": 0.0005680185324857235,
-      "loss": 4.0548,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.7073789834976196,
+      "learning_rate": 0.0005679438747976254,
+      "loss": 4.0512,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.8887001872062683,
-      "learning_rate": 0.0005676952914556621,
-      "loss": 4.0738,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.6471377015113831,
+      "learning_rate": 0.000567620075553157,
+      "loss": 4.0507,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.7014524936676025,
-      "learning_rate": 0.0005673720504256007,
-      "loss": 4.0823,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.7392793297767639,
+      "learning_rate": 0.0005672962763086886,
+      "loss": 4.0555,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.7665441036224365,
-      "learning_rate": 0.0005670488093955392,
-      "loss": 4.0523,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.7037752866744995,
+      "learning_rate": 0.0005669724770642202,
+      "loss": 4.0371,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.6104534864425659,
-      "learning_rate": 0.0005667255683654777,
-      "loss": 4.0362,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.6513423323631287,
+      "learning_rate": 0.0005666486778197517,
+      "loss": 4.034,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.6866149306297302,
-      "learning_rate": 0.0005664023273354164,
-      "loss": 4.0581,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.6977173686027527,
+      "learning_rate": 0.0005663248785752833,
+      "loss": 4.0483,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6914085745811462,
-      "learning_rate": 0.0005660790863053549,
-      "loss": 4.0439,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.6718908548355103,
+      "learning_rate": 0.0005660010793308148,
+      "loss": 4.0295,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.604280948638916,
-      "learning_rate": 0.0005657558452752936,
-      "loss": 4.0293,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.6981878280639648,
+      "learning_rate": 0.0005656772800863465,
+      "loss": 4.0463,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.8009333610534668,
-      "learning_rate": 0.0005654326042452321,
-      "loss": 4.0299,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.6827294230461121,
+      "learning_rate": 0.0005653534808418779,
+      "loss": 4.0234,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.7749080061912537,
-      "learning_rate": 0.0005651093632151708,
-      "loss": 4.0229,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.6582399010658264,
+      "learning_rate": 0.0005650296815974096,
+      "loss": 4.0268,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.7085071802139282,
-      "learning_rate": 0.0005647861221851093,
-      "loss": 4.0196,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6177818775177002,
+      "learning_rate": 0.0005647058823529411,
+      "loss": 4.0223,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.6664130687713623,
-      "learning_rate": 0.0005644628811550479,
-      "loss": 4.0094,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.5578612685203552,
+      "learning_rate": 0.0005643820831084727,
+      "loss": 4.0108,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.7595982551574707,
-      "learning_rate": 0.0005641396401249865,
-      "loss": 4.0446,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.6668821573257446,
+      "learning_rate": 0.0005640582838640042,
+      "loss": 4.0237,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.6277487874031067,
-      "learning_rate": 0.000563816399094925,
-      "loss": 4.0226,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.742559552192688,
+      "learning_rate": 0.0005637344846195358,
+      "loss": 4.0175,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.5675387978553772,
-      "learning_rate": 0.0005634931580648637,
-      "loss": 4.0005,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.6238442063331604,
+      "learning_rate": 0.0005634106853750674,
+      "loss": 4.0245,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.6105149388313293,
-      "learning_rate": 0.0005631699170348022,
-      "loss": 4.0048,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.8193546533584595,
+      "learning_rate": 0.000563086886130599,
+      "loss": 3.9917,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.822024941444397,
-      "learning_rate": 0.0005628466760047408,
-      "loss": 4.0139,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.729979395866394,
+      "learning_rate": 0.0005627630868861305,
+      "loss": 3.9921,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.5809662342071533,
-      "learning_rate": 0.0005625234349746794,
-      "loss": 3.9919,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.5888447165489197,
+      "learning_rate": 0.0005624392876416621,
+      "loss": 3.9782,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.6869716048240662,
-      "learning_rate": 0.000562200193944618,
-      "loss": 4.0053,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.705590009689331,
+      "learning_rate": 0.0005621154883971937,
+      "loss": 3.9846,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.7093315720558167,
-      "learning_rate": 0.0005618769529145566,
-      "loss": 3.9809,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.7365871667861938,
+      "learning_rate": 0.0005617916891527253,
+      "loss": 3.9652,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.3273809297449341,
-      "eval_loss": 3.9289309978485107,
-      "eval_runtime": 184.1772,
-      "eval_samples_per_second": 97.792,
-      "eval_steps_per_second": 6.114,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32846506825954697,
+      "eval_loss": 3.9196362495422363,
+      "eval_runtime": 185.9381,
+      "eval_samples_per_second": 96.866,
+      "eval_steps_per_second": 6.056,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.5664177536964417,
-      "learning_rate": 0.0005615537118844952,
-      "loss": 4.0017,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.7241402864456177,
+      "learning_rate": 0.0005614678899082568,
+      "loss": 3.9989,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.6920751929283142,
-      "learning_rate": 0.0005612304708544337,
-      "loss": 3.9825,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.6035869121551514,
+      "learning_rate": 0.0005611440906637884,
+      "loss": 3.9857,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.5844790935516357,
-      "learning_rate": 0.0005609072298243723,
-      "loss": 4.0016,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.8426706194877625,
+      "learning_rate": 0.00056082029141932,
+      "loss": 3.9792,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.6675333976745605,
-      "learning_rate": 0.0005605839887943109,
-      "loss": 3.9938,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.5762391090393066,
+      "learning_rate": 0.0005604964921748515,
+      "loss": 3.9917,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.6434072256088257,
-      "learning_rate": 0.0005602607477642495,
-      "loss": 3.9887,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.5884817242622375,
+      "learning_rate": 0.000560172692930383,
+      "loss": 3.9506,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.660064160823822,
-      "learning_rate": 0.0005599375067341881,
-      "loss": 3.9893,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.6221009492874146,
+      "learning_rate": 0.0005598488936859147,
+      "loss": 3.9624,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.7599339485168457,
-      "learning_rate": 0.0005596142657041266,
-      "loss": 3.969,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.7877587080001831,
+      "learning_rate": 0.0005595250944414463,
+      "loss": 3.9535,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.675517201423645,
-      "learning_rate": 0.0005592910246740653,
-      "loss": 3.9902,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.6189385652542114,
+      "learning_rate": 0.0005592012951969778,
+      "loss": 3.946,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.7125230431556702,
-      "learning_rate": 0.0005589677836440038,
-      "loss": 3.9791,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.7304018139839172,
+      "learning_rate": 0.0005588774959525094,
+      "loss": 3.9736,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.6018815636634827,
-      "learning_rate": 0.0005586445426139425,
-      "loss": 3.9474,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6186996698379517,
+      "learning_rate": 0.000558553696708041,
+      "loss": 3.9576,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.6196553111076355,
-      "learning_rate": 0.000558321301583881,
-      "loss": 3.9771,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.5706896185874939,
+      "learning_rate": 0.0005582298974635726,
+      "loss": 3.9541,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.5549377799034119,
-      "learning_rate": 0.0005579980605538196,
-      "loss": 3.9651,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.5761089324951172,
+      "learning_rate": 0.0005579060982191041,
+      "loss": 3.9519,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.6142178177833557,
-      "learning_rate": 0.0005576748195237582,
-      "loss": 3.9533,
+      "epoch": 0.7169811320754716,
+      "grad_norm": Infinity,
+      "learning_rate": 0.0005575887749595251,
+      "loss": 3.9458,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.6508194208145142,
-      "learning_rate": 0.0005573515784936967,
-      "loss": 3.972,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.6021038889884949,
+      "learning_rate": 0.0005572649757150566,
+      "loss": 3.9413,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6435700058937073,
-      "learning_rate": 0.0005570283374636353,
-      "loss": 3.9371,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.7275946140289307,
+      "learning_rate": 0.0005569411764705882,
+      "loss": 3.9426,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.6877986192703247,
-      "learning_rate": 0.0005567050964335739,
-      "loss": 3.9625,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.5942234992980957,
+      "learning_rate": 0.0005566173772261198,
+      "loss": 3.9527,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.6768865585327148,
-      "learning_rate": 0.0005563818554035126,
-      "loss": 3.9651,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.5786459445953369,
+      "learning_rate": 0.0005562935779816513,
+      "loss": 3.9367,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.5656898021697998,
-      "learning_rate": 0.0005560586143734511,
-      "loss": 3.9244,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.6273922920227051,
+      "learning_rate": 0.0005559697787371828,
+      "loss": 3.9311,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.7026826739311218,
-      "learning_rate": 0.0005557353733433896,
-      "loss": 3.9553,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.6769258379936218,
+      "learning_rate": 0.0005556459794927145,
+      "loss": 3.9179,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.6179747581481934,
-      "learning_rate": 0.0005554121323133283,
-      "loss": 3.9432,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.5907636284828186,
+      "learning_rate": 0.000555322180248246,
+      "loss": 3.9319,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.332839433522958,
-      "eval_loss": 3.8685381412506104,
-      "eval_runtime": 184.8874,
-      "eval_samples_per_second": 97.416,
-      "eval_steps_per_second": 6.09,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.33351166893986617,
+      "eval_loss": 3.861981153488159,
+      "eval_runtime": 179.533,
+      "eval_samples_per_second": 100.321,
+      "eval_steps_per_second": 6.272,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.5971046090126038,
-      "learning_rate": 0.0005550888912832668,
-      "loss": 3.9327,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.5700045824050903,
+      "learning_rate": 0.0005549983810037776,
+      "loss": 3.9355,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.6499930620193481,
-      "learning_rate": 0.0005547656502532055,
-      "loss": 3.9357,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 0.5413492918014526,
+      "learning_rate": 0.0005546745817593091,
+      "loss": 3.9333,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.652708113193512,
-      "learning_rate": 0.000554442409223144,
-      "loss": 3.9197,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.6539698839187622,
+      "learning_rate": 0.0005543507825148408,
+      "loss": 3.9116,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.6613510847091675,
-      "learning_rate": 0.0005541191681930826,
-      "loss": 3.9122,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.6693196892738342,
+      "learning_rate": 0.0005540269832703723,
+      "loss": 3.92,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.6252796053886414,
-      "learning_rate": 0.0005537959271630212,
-      "loss": 3.9254,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.6446160078048706,
+      "learning_rate": 0.0005537031840259039,
+      "loss": 3.9226,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.6944841742515564,
-      "learning_rate": 0.0005534726861329598,
-      "loss": 3.9347,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.7050884962081909,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9085,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.6067506074905396,
-      "learning_rate": 0.0005531494451028983,
-      "loss": 3.9181,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.6198840141296387,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9174,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.671901524066925,
-      "learning_rate": 0.0005528262040728369,
-      "loss": 3.9295,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.612169623374939,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.8806,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.5454155802726746,
-      "learning_rate": 0.0005525029630427755,
-      "loss": 3.8984,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.6216044425964355,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.8935,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.6592690348625183,
-      "learning_rate": 0.0005521797220127141,
-      "loss": 3.9373,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.7052186727523804,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.9062,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.5588394999504089,
-      "learning_rate": 0.0005518564809826527,
-      "loss": 3.9046,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.6003553867340088,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.8995,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.7025091648101807,
-      "learning_rate": 0.0005515332399525912,
-      "loss": 3.9059,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.5882673263549805,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.9185,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.6414318084716797,
-      "learning_rate": 0.0005512099989225299,
-      "loss": 3.9109,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.6166543960571289,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.9029,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.6233903169631958,
-      "learning_rate": 0.0005508867578924685,
-      "loss": 3.9003,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.6360199451446533,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.8742,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.690694272518158,
-      "learning_rate": 0.000550563516862407,
-      "loss": 3.9203,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.7083805203437805,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.8881,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.5982156991958618,
-      "learning_rate": 0.0005502402758323456,
-      "loss": 3.9269,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.580138087272644,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.904,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.5323824882507324,
-      "learning_rate": 0.0005499170348022841,
-      "loss": 3.9035,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.6983124017715454,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8882,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6599534749984741,
-      "learning_rate": 0.0005495937937722228,
-      "loss": 3.8998,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.7539551854133606,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8887,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.5557947158813477,
-      "learning_rate": 0.0005492705527421614,
-      "loss": 3.9154,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.589042603969574,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.8886,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.6077902317047119,
-      "learning_rate": 0.0005489473117121,
-      "loss": 3.8951,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.5650387406349182,
+      "learning_rate": 0.0005488461953588775,
+      "loss": 3.8678,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.3374400140987987,
-      "eval_loss": 3.822192907333374,
-      "eval_runtime": 184.6794,
-      "eval_samples_per_second": 97.526,
-      "eval_steps_per_second": 6.097,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.33783116449493084,
+      "eval_loss": 3.815941572189331,
+      "eval_runtime": 179.2962,
+      "eval_samples_per_second": 100.454,
+      "eval_steps_per_second": 6.28,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.571945309638977,
-      "learning_rate": 0.0005486240706820385,
-      "loss": 3.8844,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.587796151638031,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.8843,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.5621715784072876,
-      "learning_rate": 0.0005483008296519772,
-      "loss": 3.8981,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.7792789936065674,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.8637,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.6691741943359375,
-      "learning_rate": 0.0005479775886219157,
-      "loss": 3.9025,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5374471545219421,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.8739,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.6594086289405823,
-      "learning_rate": 0.0005476543475918542,
-      "loss": 3.8895,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.6031334400177002,
+      "learning_rate": 0.0005475509983810037,
+      "loss": 3.8634,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.56702721118927,
-      "learning_rate": 0.0005473311065617929,
-      "loss": 3.8608,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.63655024766922,
+      "learning_rate": 0.0005472271991365352,
+      "loss": 3.8795,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.5397810935974121,
-      "learning_rate": 0.0005470078655317314,
-      "loss": 3.8865,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.5650659799575806,
+      "learning_rate": 0.0005469033998920669,
+      "loss": 3.8526,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.5843083262443542,
-      "learning_rate": 0.0005466846245016701,
-      "loss": 3.8676,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.5938332676887512,
+      "learning_rate": 0.0005465796006475984,
+      "loss": 3.87,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.7101298570632935,
-      "learning_rate": 0.0005463613834716086,
-      "loss": 3.8729,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.6128816604614258,
+      "learning_rate": 0.00054625580140313,
+      "loss": 3.87,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.6375505924224854,
-      "learning_rate": 0.0005460381424415472,
-      "loss": 3.8818,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.6226370930671692,
+      "learning_rate": 0.0005459320021586615,
+      "loss": 3.8402,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.6644571423530579,
-      "learning_rate": 0.0005457149014114858,
-      "loss": 3.8883,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.5840190052986145,
+      "learning_rate": 0.0005456082029141932,
+      "loss": 3.8508,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.654179036617279,
-      "learning_rate": 0.0005453916603814243,
-      "loss": 3.8855,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.6211468577384949,
+      "learning_rate": 0.0005452844036697248,
+      "loss": 3.8534,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.6557681560516357,
-      "learning_rate": 0.000545068419351363,
-      "loss": 3.8691,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.5937489867210388,
+      "learning_rate": 0.0005449606044252563,
+      "loss": 3.8805,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.5975877642631531,
-      "learning_rate": 0.0005447451783213015,
-      "loss": 3.8716,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.6649389266967773,
+      "learning_rate": 0.0005446368051807879,
+      "loss": 3.8549,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.5875534415245056,
-      "learning_rate": 0.0005444219372912401,
-      "loss": 3.8659,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.5720226764678955,
+      "learning_rate": 0.0005443130059363194,
+      "loss": 3.8637,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.5575305223464966,
-      "learning_rate": 0.0005440986962611787,
-      "loss": 3.8586,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5358853936195374,
+      "learning_rate": 0.0005439892066918511,
+      "loss": 3.8465,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.5902034640312195,
-      "learning_rate": 0.0005437754552311173,
-      "loss": 3.859,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.631186842918396,
+      "learning_rate": 0.0005436654074473825,
+      "loss": 3.8466,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.6572847366333008,
-      "learning_rate": 0.0005434522142010559,
-      "loss": 3.8656,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.5419930815696716,
+      "learning_rate": 0.0005433416082029142,
+      "loss": 3.8474,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.5779463648796082,
-      "learning_rate": 0.0005431289731709945,
-      "loss": 3.8557,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.661422848701477,
+      "learning_rate": 0.0005430178089584457,
+      "loss": 3.8489,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.6097962260246277,
-      "learning_rate": 0.000542805732140933,
-      "loss": 3.8618,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.7400330901145935,
+      "learning_rate": 0.0005426940097139773,
+      "loss": 3.8496,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.6319593191146851,
-      "learning_rate": 0.0005424824911108716,
-      "loss": 3.8465,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.6218024492263794,
+      "learning_rate": 0.0005423766864543982,
+      "loss": 3.8659,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.34105967640345036,
-      "eval_loss": 3.7847001552581787,
-      "eval_runtime": 184.8533,
-      "eval_samples_per_second": 97.434,
-      "eval_steps_per_second": 6.091,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.34161282825531386,
+      "eval_loss": 3.7764477729797363,
+      "eval_runtime": 179.4617,
+      "eval_samples_per_second": 100.361,
+      "eval_steps_per_second": 6.274,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.6114616394042969,
-      "learning_rate": 0.0005421592500808102,
-      "loss": 3.859,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.6361964344978333,
+      "learning_rate": 0.0005420528872099298,
+      "loss": 3.8413,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5971494317054749,
-      "learning_rate": 0.0005418360090507488,
-      "loss": 3.8401,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.6142780184745789,
+      "learning_rate": 0.0005417290879654613,
+      "loss": 3.8326,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.6119955778121948,
-      "learning_rate": 0.0005415127680206874,
-      "loss": 3.8575,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.6248621344566345,
+      "learning_rate": 0.000541405288720993,
+      "loss": 3.8448,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.5487962365150452,
-      "learning_rate": 0.0005411895269906259,
-      "loss": 3.8552,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5891006588935852,
+      "learning_rate": 0.0005410814894765245,
+      "loss": 3.8482,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.631289005279541,
-      "learning_rate": 0.0005408662859605646,
-      "loss": 3.8509,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.5713793635368347,
+      "learning_rate": 0.0005407576902320561,
+      "loss": 3.8387,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.615967333316803,
-      "learning_rate": 0.0005405430449305031,
-      "loss": 3.8556,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5939117074012756,
+      "learning_rate": 0.0005404338909875876,
+      "loss": 3.8174,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.5872229337692261,
-      "learning_rate": 0.0005402198039004416,
-      "loss": 3.7921,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.5924010276794434,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.7836,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.5617460012435913,
-      "learning_rate": 0.0005398965628703803,
-      "loss": 3.7792,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.6352487206459045,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.7956,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.6194376349449158,
-      "learning_rate": 0.0005395733218403189,
-      "loss": 3.7967,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.5576720833778381,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.7835,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.5620608329772949,
-      "learning_rate": 0.0005392500808102575,
-      "loss": 3.7958,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.560811460018158,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.781,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.6231437921524048,
-      "learning_rate": 0.000538926839780196,
-      "loss": 3.8046,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.5907502174377441,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7925,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.6144508719444275,
-      "learning_rate": 0.0005386035987501346,
-      "loss": 3.7608,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.5446668863296509,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.7719,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.8581711649894714,
-      "learning_rate": 0.0005382803577200732,
-      "loss": 3.7768,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.5410051941871643,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.7634,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 0.5918218493461609,
-      "learning_rate": 0.0005379571166900119,
-      "loss": 3.7755,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.587929368019104,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.7742,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.6462705135345459,
-      "learning_rate": 0.0005376338756599504,
-      "loss": 3.7889,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.609221339225769,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.7665,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.7305418848991394,
-      "learning_rate": 0.0005373106346298889,
-      "loss": 3.7776,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.6146703958511353,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.7712,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.6626386642456055,
-      "learning_rate": 0.0005369873935998276,
-      "loss": 3.7625,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.6446210145950317,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.7547,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.5469115376472473,
-      "learning_rate": 0.0005366641525697661,
-      "loss": 3.7704,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.6286680102348328,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7662,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.607982337474823,
-      "learning_rate": 0.0005363409115397048,
-      "loss": 3.7885,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.5983867049217224,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7644,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.5802627205848694,
-      "learning_rate": 0.0005360176705096433,
-      "loss": 3.7776,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.622414767742157,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.7684,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.34371732603939253,
-      "eval_loss": 3.7542877197265625,
-      "eval_runtime": 184.5847,
-      "eval_samples_per_second": 97.576,
-      "eval_steps_per_second": 6.1,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.3455192255309079,
+      "eval_loss": 3.7473487854003906,
+      "eval_runtime": 179.6608,
+      "eval_samples_per_second": 100.25,
+      "eval_steps_per_second": 6.267,
       "step": 10000
     }
   ],
   "logging_steps": 50,
-  "max_steps": 92910,
+  "max_steps": 92750,
   "num_input_tokens_seen": 0,
   "num_train_epochs": 10,
   "save_steps": 10000,
@@ -1516,7 +1516,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 8.3610314735616e+16,
+  "total_flos": 8.36069179392e+16,
   "train_batch_size": 32,
   "trial_name": null,
   "trial_params": null

checkpoint-10000/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cfee9d1ca2636462af1f5dd1f5348dda3ebb5ebff6f794ebb5861a958cddc3d9
+oid sha256:17e61b03de797e84cad5a5a1386a832b5d860e232565c820b1f9056f83f9e849
 size 5304