Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f9ecb7d2cdd0667c4486020af5ba7e4fa897d83b09ba91b38fb038a95b63412
+oid sha256:03ac1bf824faf271bae58dce673b9d7952ef68a6eccdb4f5c5dd65b8f9aa84a7
 size 503128704

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82cb4fcc8c34b3cd40b17526742c583c9bd86a97648c7965ff9715953b8f8c71
+oid sha256:72ab1143702e11fdaaf19b5fff3b4c60a793fc84d845bcd5583d8263cc76e0cb
 size 1006351290

checkpoint-10000/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f1a0d474d254d7072642df0d6e147a602036c487497f36ec913c1166677568b0
+oid sha256:1abdd4c7c045dcb1f8a6404bddc83e4806c8d3e3173eb1017f9b082b3c161e85
 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.7458271980285645,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_495/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.753229856491089,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_high_2000_495/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": 2.1159660816192627,
-      "learning_rate": 0.00028199999999999997,
-      "loss": 8.9316,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 1.4431322813034058,
+      "learning_rate": 0.000294,
+      "loss": 8.5391,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 1.9895623922348022,
-      "learning_rate": 0.0005819999999999999,
-      "loss": 6.9096,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 3.01206111907959,
+      "learning_rate": 0.0005939999999999999,
+      "loss": 6.9498,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 1.8703136444091797,
-      "learning_rate": 0.0005996961534317422,
-      "loss": 6.4463,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 2.0440971851348877,
+      "learning_rate": 0.0005996826767404209,
+      "loss": 6.5454,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 1.2748037576675415,
-      "learning_rate": 0.0005993729124016807,
-      "loss": 6.2017,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.441593885421753,
+      "learning_rate": 0.0005993588774959525,
+      "loss": 6.29,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.3479403257369995,
-      "learning_rate": 0.0005990496713716194,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 1.1196218729019165,
+      "learning_rate": 0.000599035078251484,
       "loss": 6.1031,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 1.0388602018356323,
-      "learning_rate": 0.000598726430341558,
-      "loss": 5.9726,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 1.7114529609680176,
+      "learning_rate": 0.0005987112790070156,
+      "loss": 5.9896,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 1.1332666873931885,
-      "learning_rate": 0.0005984031893114966,
-      "loss": 5.886,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 1.0614324808120728,
+      "learning_rate": 0.0005983874797625472,
+      "loss": 5.8898,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 1.398230791091919,
-      "learning_rate": 0.0005980799482814351,
-      "loss": 5.8019,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 0.9476490616798401,
+      "learning_rate": 0.0005980636805180788,
+      "loss": 5.8227,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.5357556343078613,
-      "learning_rate": 0.0005977567072513737,
-      "loss": 5.7437,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 1.198143482208252,
+      "learning_rate": 0.0005977398812736103,
+      "loss": 5.7384,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 1.175200343132019,
-      "learning_rate": 0.0005974334662213123,
-      "loss": 5.6616,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 1.5752428770065308,
+      "learning_rate": 0.0005974160820291419,
+      "loss": 5.6852,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 1.386881709098816,
-      "learning_rate": 0.0005971102251912509,
-      "loss": 5.5895,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 2.0579349994659424,
+      "learning_rate": 0.0005970922827846734,
+      "loss": 5.5695,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.3578399419784546,
-      "learning_rate": 0.0005967869841611895,
-      "loss": 5.4912,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.3935744762420654,
+      "learning_rate": 0.0005967684835402051,
+      "loss": 5.535,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 1.2151751518249512,
-      "learning_rate": 0.000596463743131128,
-      "loss": 5.4275,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 1.303139090538025,
+      "learning_rate": 0.0005964446842957366,
+      "loss": 5.4307,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 1.3390365839004517,
-      "learning_rate": 0.0005961405021010667,
-      "loss": 5.3731,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 0.969357967376709,
+      "learning_rate": 0.0005961208850512682,
+      "loss": 5.4075,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 1.103965401649475,
-      "learning_rate": 0.0005958172610710052,
-      "loss": 5.3186,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.0970923900604248,
+      "learning_rate": 0.0005957970858067997,
+      "loss": 5.3339,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 1.1866347789764404,
-      "learning_rate": 0.0005954940200409439,
-      "loss": 5.2687,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 1.444526195526123,
+      "learning_rate": 0.0005954732865623314,
+      "loss": 5.2734,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.0097291469573975,
-      "learning_rate": 0.0005951707790108824,
-      "loss": 5.2072,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 1.2398624420166016,
+      "learning_rate": 0.0005951494873178628,
+      "loss": 5.2435,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.3843412399291992,
-      "learning_rate": 0.0005948475379808209,
-      "loss": 5.1549,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 1.0231698751449585,
+      "learning_rate": 0.0005948256880733944,
+      "loss": 5.1615,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 1.1859126091003418,
-      "learning_rate": 0.0005945242969507596,
-      "loss": 5.0977,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 0.8052211403846741,
+      "learning_rate": 0.000594501888828926,
+      "loss": 5.1565,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.1825942993164062,
-      "learning_rate": 0.0005942010559206981,
-      "loss": 5.0644,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 0.9758635759353638,
+      "learning_rate": 0.0005941780895844576,
+      "loss": 5.0973,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.22798678911267642,
-      "eval_loss": 5.015804290771484,
-      "eval_runtime": 186.987,
-      "eval_samples_per_second": 96.322,
-      "eval_steps_per_second": 6.022,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22682387725439804,
+      "eval_loss": 5.0240583419799805,
+      "eval_runtime": 187.7541,
+      "eval_samples_per_second": 95.929,
+      "eval_steps_per_second": 5.997,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 1.036710262298584,
-      "learning_rate": 0.0005938778148906367,
-      "loss": 5.045,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 1.1241644620895386,
+      "learning_rate": 0.0005938542903399891,
+      "loss": 5.0594,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 1.1089870929718018,
-      "learning_rate": 0.0005935545738605753,
-      "loss": 5.0081,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.3577865362167358,
+      "learning_rate": 0.0005935304910955207,
+      "loss": 5.0184,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 0.9135867357254028,
-      "learning_rate": 0.000593231332830514,
-      "loss": 4.9925,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 1.2023968696594238,
+      "learning_rate": 0.0005932066918510523,
+      "loss": 4.9997,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 0.8327316641807556,
-      "learning_rate": 0.0005929080918004525,
-      "loss": 4.9437,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 1.0888073444366455,
+      "learning_rate": 0.0005928828926065839,
+      "loss": 4.9799,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 1.0137102603912354,
-      "learning_rate": 0.000592584850770391,
-      "loss": 4.9288,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 1.1558417081832886,
+      "learning_rate": 0.0005925590933621154,
+      "loss": 4.9525,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 1.620171070098877,
-      "learning_rate": 0.0005922616097403296,
-      "loss": 4.8813,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 0.8390737771987915,
+      "learning_rate": 0.000592235294117647,
+      "loss": 4.8991,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 1.1357035636901855,
-      "learning_rate": 0.0005919383687102682,
-      "loss": 4.8713,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 1.2147760391235352,
+      "learning_rate": 0.0005919114948731785,
+      "loss": 4.8807,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 0.7975968718528748,
-      "learning_rate": 0.0005916151276802069,
-      "loss": 4.835,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 0.8272018432617188,
+      "learning_rate": 0.0005915876956287102,
+      "loss": 4.8533,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 0.9254979491233826,
-      "learning_rate": 0.0005912918866501454,
-      "loss": 4.806,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 0.9127699732780457,
+      "learning_rate": 0.0005912638963842416,
+      "loss": 4.8361,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 0.8283946514129639,
-      "learning_rate": 0.000590968645620084,
-      "loss": 4.8075,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 1.053229808807373,
+      "learning_rate": 0.0005909400971397733,
+      "loss": 4.822,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 1.1234453916549683,
-      "learning_rate": 0.0005906454045900226,
-      "loss": 4.7791,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 0.7915006279945374,
+      "learning_rate": 0.0005906162978953049,
+      "loss": 4.7808,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 0.9318310618400574,
-      "learning_rate": 0.0005903221635599612,
-      "loss": 4.7354,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 1.0942384004592896,
+      "learning_rate": 0.0005902924986508364,
+      "loss": 4.7932,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 0.8706691861152649,
-      "learning_rate": 0.0005899989225298998,
-      "loss": 4.7278,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 1.114819884300232,
+      "learning_rate": 0.000589968699406368,
+      "loss": 4.7716,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 1.0983690023422241,
-      "learning_rate": 0.0005896756814998383,
-      "loss": 4.7116,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.8059048652648926,
+      "learning_rate": 0.0005896449001618996,
+      "loss": 4.6935,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 1.1487178802490234,
-      "learning_rate": 0.0005893524404697769,
-      "loss": 4.7026,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 1.0432419776916504,
+      "learning_rate": 0.0005893211009174312,
+      "loss": 4.7057,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 0.8954223394393921,
-      "learning_rate": 0.0005890291994397155,
-      "loss": 4.6491,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 1.0190030336380005,
+      "learning_rate": 0.0005889973016729627,
+      "loss": 4.6937,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 1.1864057779312134,
-      "learning_rate": 0.0005887059584096541,
-      "loss": 4.6526,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 0.9340100884437561,
+      "learning_rate": 0.0005886735024284943,
+      "loss": 4.6527,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 0.945884108543396,
-      "learning_rate": 0.0005883827173795926,
-      "loss": 4.6132,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 1.0046193599700928,
+      "learning_rate": 0.0005883497031840258,
+      "loss": 4.65,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 1.2601373195648193,
-      "learning_rate": 0.0005880594763495313,
-      "loss": 4.6044,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 1.0945899486541748,
+      "learning_rate": 0.0005880259039395575,
+      "loss": 4.6138,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 0.9447725415229797,
-      "learning_rate": 0.0005877362353194698,
-      "loss": 4.5877,
+      "epoch": 0.215633423180593,
+      "grad_norm": 1.0984909534454346,
+      "learning_rate": 0.000587702104695089,
+      "loss": 4.5905,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.27040933128385014,
-      "eval_loss": 4.508606910705566,
-      "eval_runtime": 184.8398,
-      "eval_samples_per_second": 97.441,
-      "eval_steps_per_second": 6.092,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.26831504687122604,
+      "eval_loss": 4.5331034660339355,
+      "eval_runtime": 188.7106,
+      "eval_samples_per_second": 95.442,
+      "eval_steps_per_second": 5.967,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 0.7851864695549011,
-      "learning_rate": 0.0005874129942894084,
-      "loss": 4.5643,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.9245219230651855,
+      "learning_rate": 0.0005873783054506206,
+      "loss": 4.6042,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 0.862130880355835,
-      "learning_rate": 0.000587089753259347,
-      "loss": 4.5481,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 1.0915874242782593,
+      "learning_rate": 0.0005870545062061521,
+      "loss": 4.5669,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 0.8470994830131531,
-      "learning_rate": 0.0005867665122292855,
-      "loss": 4.5219,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 1.1114705801010132,
+      "learning_rate": 0.0005867307069616837,
+      "loss": 4.513,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 0.8713746070861816,
-      "learning_rate": 0.0005864432711992242,
-      "loss": 4.511,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 1.0507372617721558,
+      "learning_rate": 0.0005864069077172152,
+      "loss": 4.555,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 0.8879591226577759,
-      "learning_rate": 0.0005861200301691628,
-      "loss": 4.5126,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 0.7661042809486389,
+      "learning_rate": 0.0005860831084727468,
+      "loss": 4.4936,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 0.9361649751663208,
-      "learning_rate": 0.0005857967891391014,
-      "loss": 4.4803,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 0.9031612873077393,
+      "learning_rate": 0.0005857593092282784,
+      "loss": 4.4732,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.8480615019798279,
-      "learning_rate": 0.0005854735481090399,
-      "loss": 4.4446,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.8872079849243164,
+      "learning_rate": 0.00058543550998381,
+      "loss": 4.468,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 1.0228937864303589,
-      "learning_rate": 0.0005851503070789784,
-      "loss": 4.4508,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 1.007972002029419,
+      "learning_rate": 0.0005851117107393415,
+      "loss": 4.4588,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 0.9693055152893066,
-      "learning_rate": 0.0005848270660489171,
-      "loss": 4.4283,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 0.9107605218887329,
+      "learning_rate": 0.0005847879114948731,
+      "loss": 4.4611,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 0.7247486114501953,
-      "learning_rate": 0.0005845038250188556,
-      "loss": 4.4015,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 0.8212679624557495,
+      "learning_rate": 0.0005844641122504048,
+      "loss": 4.4549,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.7183290123939514,
-      "learning_rate": 0.0005841805839887943,
-      "loss": 4.3918,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.8247270584106445,
+      "learning_rate": 0.0005841403130059363,
+      "loss": 4.4192,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 0.900417149066925,
-      "learning_rate": 0.0005838573429587328,
-      "loss": 4.3701,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 0.8410664796829224,
+      "learning_rate": 0.0005838165137614679,
+      "loss": 4.4058,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.8292120099067688,
-      "learning_rate": 0.0005835341019286715,
-      "loss": 4.3891,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 0.8341009616851807,
+      "learning_rate": 0.0005834927145169994,
+      "loss": 4.4019,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.798037052154541,
-      "learning_rate": 0.00058321086089861,
-      "loss": 4.36,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 0.7168475985527039,
+      "learning_rate": 0.000583168915272531,
+      "loss": 4.3961,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.8176091313362122,
-      "learning_rate": 0.0005828876198685486,
-      "loss": 4.3491,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.9485689997673035,
+      "learning_rate": 0.0005828451160280626,
+      "loss": 4.3549,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 0.8014965057373047,
-      "learning_rate": 0.0005825643788384872,
-      "loss": 4.3642,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.7494977116584778,
+      "learning_rate": 0.0005825213167835941,
+      "loss": 4.394,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 1.2503150701522827,
-      "learning_rate": 0.0005822411378084257,
-      "loss": 4.3251,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.7997616529464722,
+      "learning_rate": 0.0005821975175391257,
+      "loss": 4.3639,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 0.7830126285552979,
-      "learning_rate": 0.0005819178967783644,
-      "loss": 4.3251,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.7571331262588501,
+      "learning_rate": 0.0005818737182946573,
+      "loss": 4.3431,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.9035398364067078,
-      "learning_rate": 0.0005815946557483029,
-      "loss": 4.3313,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.7700881958007812,
+      "learning_rate": 0.0005815499190501888,
+      "loss": 4.3449,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.6857808828353882,
-      "learning_rate": 0.0005812714147182415,
-      "loss": 4.2989,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.670320451259613,
+      "learning_rate": 0.0005812261198057204,
+      "loss": 4.3425,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.2990024252411089,
-      "eval_loss": 4.2293620109558105,
-      "eval_runtime": 185.031,
-      "eval_samples_per_second": 97.34,
-      "eval_steps_per_second": 6.085,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.2970671000033574,
+      "eval_loss": 4.244440078735352,
+      "eval_runtime": 181.2734,
+      "eval_samples_per_second": 99.358,
+      "eval_steps_per_second": 6.212,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.7547754049301147,
-      "learning_rate": 0.0005809481736881801,
-      "loss": 4.296,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 0.7345431447029114,
+      "learning_rate": 0.0005809023205612519,
+      "loss": 4.3171,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.7557756304740906,
-      "learning_rate": 0.0005806249326581187,
-      "loss": 4.2866,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.7953465580940247,
+      "learning_rate": 0.0005805785213167836,
+      "loss": 4.3147,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.7948616147041321,
-      "learning_rate": 0.0005803016916280573,
-      "loss": 4.2736,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.7929427027702332,
+      "learning_rate": 0.0005802547220723151,
+      "loss": 4.2801,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.864937424659729,
-      "learning_rate": 0.0005799784505979959,
-      "loss": 4.2902,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.8242636322975159,
+      "learning_rate": 0.0005799309228278467,
+      "loss": 4.2861,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 0.8035250902175903,
-      "learning_rate": 0.0005796552095679344,
-      "loss": 4.2524,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 0.9226231575012207,
+      "learning_rate": 0.0005796071235833782,
+      "loss": 4.2607,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.7827973365783691,
-      "learning_rate": 0.000579331968537873,
-      "loss": 4.2573,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.8058574199676514,
+      "learning_rate": 0.0005792833243389099,
+      "loss": 4.2833,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.7145817875862122,
-      "learning_rate": 0.0005790087275078116,
-      "loss": 4.233,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.5928198099136353,
+      "learning_rate": 0.0005789595250944414,
+      "loss": 4.262,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.7260717749595642,
-      "learning_rate": 0.0005786854864777502,
-      "loss": 4.2351,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.7675554752349854,
+      "learning_rate": 0.000578635725849973,
+      "loss": 4.2542,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.846215009689331,
-      "learning_rate": 0.0005783622454476888,
-      "loss": 4.236,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.611778974533081,
+      "learning_rate": 0.0005783119266055045,
+      "loss": 4.2414,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.6381611227989197,
-      "learning_rate": 0.0005780390044176273,
-      "loss": 4.2186,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.6161898970603943,
+      "learning_rate": 0.0005779881273610361,
+      "loss": 4.2189,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.7176727652549744,
-      "learning_rate": 0.000577715763387566,
-      "loss": 4.2159,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.739580512046814,
+      "learning_rate": 0.0005776643281165676,
+      "loss": 4.2229,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.7199215888977051,
-      "learning_rate": 0.0005773925223575045,
-      "loss": 4.2187,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.8803327679634094,
+      "learning_rate": 0.0005773405288720992,
+      "loss": 4.2268,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.7120049595832825,
-      "learning_rate": 0.0005770692813274432,
-      "loss": 4.1858,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 1.1461224555969238,
+      "learning_rate": 0.0005770167296276308,
+      "loss": 4.2344,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.682905912399292,
-      "learning_rate": 0.0005767460402973817,
-      "loss": 4.2067,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.9922895431518555,
+      "learning_rate": 0.0005766929303831624,
+      "loss": 4.2043,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.8106050491333008,
-      "learning_rate": 0.0005764227992673203,
-      "loss": 4.1901,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.7285529375076294,
+      "learning_rate": 0.000576369131138694,
+      "loss": 4.2012,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.6139677166938782,
-      "learning_rate": 0.0005760995582372589,
-      "loss": 4.1922,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.809013307094574,
+      "learning_rate": 0.0005760453318942255,
+      "loss": 4.2178,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.6996195912361145,
-      "learning_rate": 0.0005757763172071974,
-      "loss": 4.1884,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.8063985109329224,
+      "learning_rate": 0.000575721532649757,
+      "loss": 4.2018,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.6972913146018982,
-      "learning_rate": 0.000575453076177136,
-      "loss": 4.1469,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 0.847189724445343,
+      "learning_rate": 0.0005753977334052887,
+      "loss": 4.1669,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 0.7805142998695374,
-      "learning_rate": 0.0005751298351470746,
-      "loss": 4.1681,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.6911249756813049,
+      "learning_rate": 0.0005750739341608203,
+      "loss": 4.1661,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 0.6750819683074951,
-      "learning_rate": 0.0005748065941170133,
-      "loss": 4.1677,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.7047509551048279,
+      "learning_rate": 0.0005747501349163518,
+      "loss": 4.1711,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.3131229545414788,
-      "eval_loss": 4.085065841674805,
-      "eval_runtime": 184.7939,
-      "eval_samples_per_second": 97.465,
-      "eval_steps_per_second": 6.093,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.31155715777518433,
+      "eval_loss": 4.099443435668945,
+      "eval_runtime": 181.2897,
+      "eval_samples_per_second": 99.349,
+      "eval_steps_per_second": 6.211,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.743445098400116,
-      "learning_rate": 0.0005744833530869518,
-      "loss": 4.1605,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.6963608264923096,
+      "learning_rate": 0.0005744263356718834,
+      "loss": 4.1712,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.6816270351409912,
-      "learning_rate": 0.0005741601120568903,
-      "loss": 4.1494,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.5528955459594727,
+      "learning_rate": 0.000574102536427415,
+      "loss": 4.1704,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.6031078696250916,
-      "learning_rate": 0.0005738368710268289,
-      "loss": 4.1373,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.6239518523216248,
+      "learning_rate": 0.0005737787371829465,
+      "loss": 4.1571,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.869440495967865,
-      "learning_rate": 0.0005735136299967675,
-      "loss": 4.1274,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 0.6668544411659241,
+      "learning_rate": 0.0005734549379384781,
+      "loss": 4.1498,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.6965010762214661,
-      "learning_rate": 0.0005731903889667062,
-      "loss": 4.1363,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.6054083704948425,
+      "learning_rate": 0.0005731311386940097,
+      "loss": 4.1415,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.6586082577705383,
-      "learning_rate": 0.0005728671479366447,
-      "loss": 4.1073,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.6274952292442322,
+      "learning_rate": 0.0005728073394495412,
+      "loss": 4.1351,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.6008425951004028,
-      "learning_rate": 0.0005725439069065833,
-      "loss": 4.1131,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.6964167952537537,
+      "learning_rate": 0.0005724835402050728,
+      "loss": 4.1408,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.6233063340187073,
-      "learning_rate": 0.0005722206658765219,
-      "loss": 4.1266,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.7939971685409546,
+      "learning_rate": 0.0005721597409606043,
+      "loss": 4.1304,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 0.7806559205055237,
-      "learning_rate": 0.0005718974248464605,
-      "loss": 4.1052,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.7325452566146851,
+      "learning_rate": 0.000571835941716136,
+      "loss": 4.1334,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.7604005336761475,
-      "learning_rate": 0.000571574183816399,
-      "loss": 4.0944,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.6969997882843018,
+      "learning_rate": 0.0005715121424716675,
+      "loss": 4.1142,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.8495274782180786,
-      "learning_rate": 0.0005712509427863376,
-      "loss": 4.1176,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.6884226202964783,
+      "learning_rate": 0.0005711883432271991,
+      "loss": 4.1028,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.6198326349258423,
-      "learning_rate": 0.0005709277017562762,
-      "loss": 4.1016,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.6321718096733093,
+      "learning_rate": 0.0005708645439827306,
+      "loss": 4.1078,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.6641437411308289,
-      "learning_rate": 0.0005706044607262148,
-      "loss": 4.0835,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.701016366481781,
+      "learning_rate": 0.0005705407447382623,
+      "loss": 4.0883,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 0.7691634297370911,
-      "learning_rate": 0.0005702812196961534,
-      "loss": 4.0815,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.7334260940551758,
+      "learning_rate": 0.0005702169454937938,
+      "loss": 4.0958,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.7156651020050049,
-      "learning_rate": 0.0005699579786660919,
-      "loss": 4.0818,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.5910300612449646,
+      "learning_rate": 0.0005698931462493253,
+      "loss": 4.075,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.6472979784011841,
-      "learning_rate": 0.0005696347376360306,
-      "loss": 4.067,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.6013472080230713,
+      "learning_rate": 0.0005695693470048569,
+      "loss": 4.0922,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.724949300289154,
-      "learning_rate": 0.0005693114966059691,
-      "loss": 4.0615,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.6121587753295898,
+      "learning_rate": 0.0005692455477603885,
+      "loss": 4.0841,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.7559602856636047,
-      "learning_rate": 0.0005689882555759077,
-      "loss": 4.0676,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.613993227481842,
+      "learning_rate": 0.00056892174851592,
+      "loss": 4.0931,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.6394193768501282,
-      "learning_rate": 0.0005686650145458463,
-      "loss": 4.0461,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.6811482310295105,
+      "learning_rate": 0.0005685979492714516,
+      "loss": 4.0664,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.7001257538795471,
-      "learning_rate": 0.0005683417735157848,
-      "loss": 4.0729,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.6156988739967346,
+      "learning_rate": 0.0005682741500269833,
+      "loss": 4.0584,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.320991922635667,
-      "eval_loss": 3.990946054458618,
-      "eval_runtime": 184.8283,
-      "eval_samples_per_second": 97.447,
-      "eval_steps_per_second": 6.092,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.3207276788125022,
+      "eval_loss": 3.9989216327667236,
+      "eval_runtime": 181.0844,
+      "eval_samples_per_second": 99.462,
+      "eval_steps_per_second": 6.218,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.7765357494354248,
-      "learning_rate": 0.0005680185324857235,
-      "loss": 4.0425,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.5171638131141663,
+      "learning_rate": 0.0005679503507825148,
+      "loss": 4.0485,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.6510383486747742,
-      "learning_rate": 0.0005676952914556621,
-      "loss": 4.0239,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.8664311170578003,
+      "learning_rate": 0.0005676265515380464,
+      "loss": 4.0568,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.6882658004760742,
-      "learning_rate": 0.0005673720504256007,
-      "loss": 4.0242,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.5895658731460571,
+      "learning_rate": 0.0005673027522935779,
+      "loss": 4.0612,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.7188264727592468,
-      "learning_rate": 0.0005670488093955392,
-      "loss": 4.0442,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.7551826238632202,
+      "learning_rate": 0.0005669789530491095,
+      "loss": 4.0396,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.6599631905555725,
-      "learning_rate": 0.0005667255683654777,
-      "loss": 4.0497,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.6080760359764099,
+      "learning_rate": 0.0005666551538046411,
+      "loss": 4.0427,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.5674473643302917,
-      "learning_rate": 0.0005664023273354164,
-      "loss": 4.0271,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.6198018193244934,
+      "learning_rate": 0.0005663313545601727,
+      "loss": 4.0237,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6670402884483337,
-      "learning_rate": 0.0005660790863053549,
-      "loss": 4.0284,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.6986076235771179,
+      "learning_rate": 0.0005660075553157042,
+      "loss": 4.037,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.7389664649963379,
-      "learning_rate": 0.0005657558452752936,
-      "loss": 4.0449,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.6296870708465576,
+      "learning_rate": 0.0005656837560712358,
+      "loss": 4.0294,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.6841723322868347,
-      "learning_rate": 0.0005654326042452321,
-      "loss": 4.051,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.7363589406013489,
+      "learning_rate": 0.0005653599568267674,
+      "loss": 4.0366,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.6119431853294373,
-      "learning_rate": 0.0005651093632151708,
-      "loss": 4.0158,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.5726944804191589,
+      "learning_rate": 0.0005650361575822989,
+      "loss": 4.038,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.8850950598716736,
-      "learning_rate": 0.0005647861221851093,
-      "loss": 4.0183,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6348768472671509,
+      "learning_rate": 0.0005647123583378305,
+      "loss": 4.0435,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.6376111507415771,
-      "learning_rate": 0.0005644628811550479,
-      "loss": 4.0076,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.6801629662513733,
+      "learning_rate": 0.0005643885590933621,
+      "loss": 4.0221,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.5115376114845276,
-      "learning_rate": 0.0005641396401249865,
-      "loss": 4.0006,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.5451263785362244,
+      "learning_rate": 0.0005640647598488936,
+      "loss": 4.0095,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.6021561026573181,
-      "learning_rate": 0.000563816399094925,
-      "loss": 4.0056,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.6517798900604248,
+      "learning_rate": 0.0005637409606044252,
+      "loss": 4.0171,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.5795146226882935,
-      "learning_rate": 0.0005634931580648637,
-      "loss": 4.0159,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.5931031107902527,
+      "learning_rate": 0.0005634171613599567,
+      "loss": 4.0271,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.6918138861656189,
-      "learning_rate": 0.0005631699170348022,
-      "loss": 3.991,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.6558272838592529,
+      "learning_rate": 0.0005630933621154884,
+      "loss": 4.0187,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.6135892271995544,
-      "learning_rate": 0.0005628466760047408,
-      "loss": 4.0127,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.6062823534011841,
+      "learning_rate": 0.0005627695628710199,
+      "loss": 4.0266,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.6184191703796387,
-      "learning_rate": 0.0005625234349746794,
-      "loss": 3.9862,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.6019988656044006,
+      "learning_rate": 0.0005624457636265515,
+      "loss": 3.996,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.6259024739265442,
-      "learning_rate": 0.000562200193944618,
-      "loss": 3.9922,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.6201125383377075,
+      "learning_rate": 0.000562121964382083,
+      "loss": 3.9898,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.5602654814720154,
-      "learning_rate": 0.0005618769529145566,
-      "loss": 3.995,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.6050779819488525,
+      "learning_rate": 0.0005617981651376146,
+      "loss": 4.0004,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.32913100180896193,
-      "eval_loss": 3.9143569469451904,
-      "eval_runtime": 184.5106,
-      "eval_samples_per_second": 97.615,
-      "eval_steps_per_second": 6.103,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32684505370223305,
+      "eval_loss": 3.9261674880981445,
+      "eval_runtime": 185.9413,
+      "eval_samples_per_second": 96.864,
+      "eval_steps_per_second": 6.056,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.7381386160850525,
-      "learning_rate": 0.0005615537118844952,
-      "loss": 3.9749,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.6865056753158569,
+      "learning_rate": 0.0005614743658931463,
+      "loss": 3.9731,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.6130694150924683,
-      "learning_rate": 0.0005612304708544337,
-      "loss": 3.9791,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.7325416803359985,
+      "learning_rate": 0.0005611505666486777,
+      "loss": 4.0069,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.6282258629798889,
-      "learning_rate": 0.0005609072298243723,
-      "loss": 3.9766,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.5644854307174683,
+      "learning_rate": 0.0005608267674042094,
+      "loss": 3.9862,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.7320833802223206,
-      "learning_rate": 0.0005605839887943109,
-      "loss": 3.9744,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.6596018075942993,
+      "learning_rate": 0.0005605029681597409,
+      "loss": 3.9661,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.7067676782608032,
-      "learning_rate": 0.0005602607477642495,
-      "loss": 3.9755,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.6351631879806519,
+      "learning_rate": 0.0005601791689152725,
+      "loss": 3.9823,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.6241488456726074,
-      "learning_rate": 0.0005599375067341881,
-      "loss": 3.9639,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.7349021434783936,
+      "learning_rate": 0.000559855369670804,
+      "loss": 4.0002,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.6031643152236938,
-      "learning_rate": 0.0005596142657041266,
-      "loss": 3.9535,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.6522649526596069,
+      "learning_rate": 0.0005595315704263357,
+      "loss": 3.9623,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.5485600829124451,
-      "learning_rate": 0.0005592910246740653,
-      "loss": 3.9616,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.5307708978652954,
+      "learning_rate": 0.0005592077711818672,
+      "loss": 3.9669,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.6337101459503174,
-      "learning_rate": 0.0005589677836440038,
-      "loss": 3.9509,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.5935479998588562,
+      "learning_rate": 0.0005588839719373988,
+      "loss": 3.9545,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.546737790107727,
-      "learning_rate": 0.0005586445426139425,
-      "loss": 3.9416,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6537756323814392,
+      "learning_rate": 0.0005585601726929303,
+      "loss": 3.9651,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.6616541147232056,
-      "learning_rate": 0.000558321301583881,
-      "loss": 3.9618,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.5890859961509705,
+      "learning_rate": 0.0005582363734484619,
+      "loss": 3.9888,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.5739938020706177,
-      "learning_rate": 0.0005579980605538196,
-      "loss": 3.949,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.6212690472602844,
+      "learning_rate": 0.0005579125742039935,
+      "loss": 3.9448,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.5481446981430054,
-      "learning_rate": 0.0005576748195237582,
-      "loss": 3.961,
+      "epoch": 0.7169811320754716,
+      "grad_norm": 0.6124947667121887,
+      "learning_rate": 0.0005575887749595251,
+      "loss": 3.9736,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.6824391484260559,
-      "learning_rate": 0.0005573515784936967,
-      "loss": 3.9376,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.5879746675491333,
+      "learning_rate": 0.0005572649757150566,
+      "loss": 3.9494,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6663408279418945,
-      "learning_rate": 0.0005570283374636353,
-      "loss": 3.9525,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.5893728733062744,
+      "learning_rate": 0.0005569411764705882,
+      "loss": 3.9548,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.5435226559638977,
-      "learning_rate": 0.0005567050964335739,
-      "loss": 3.9249,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.626844584941864,
+      "learning_rate": 0.0005566173772261198,
+      "loss": 3.9442,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.6284464001655579,
-      "learning_rate": 0.0005563818554035126,
-      "loss": 3.9307,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.6960020065307617,
+      "learning_rate": 0.0005562935779816513,
+      "loss": 3.9258,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.5826367139816284,
-      "learning_rate": 0.0005560586143734511,
-      "loss": 3.9251,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.6511778235435486,
+      "learning_rate": 0.0005559697787371828,
+      "loss": 3.9527,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.5755831003189087,
-      "learning_rate": 0.0005557353733433896,
-      "loss": 3.9271,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.6318475604057312,
+      "learning_rate": 0.0005556459794927145,
+      "loss": 3.9468,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.6272289156913757,
-      "learning_rate": 0.0005554121323133283,
-      "loss": 3.9517,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.7971281409263611,
+      "learning_rate": 0.000555322180248246,
+      "loss": 3.9359,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.33305206722441094,
-      "eval_loss": 3.8637194633483887,
-      "eval_runtime": 184.9315,
-      "eval_samples_per_second": 97.393,
-      "eval_steps_per_second": 6.089,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.3327894531945633,
+      "eval_loss": 3.869997501373291,
+      "eval_runtime": 181.2103,
+      "eval_samples_per_second": 99.393,
+      "eval_steps_per_second": 6.214,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.6227930188179016,
-      "learning_rate": 0.0005550888912832668,
-      "loss": 3.916,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.6572127342224121,
+      "learning_rate": 0.0005549983810037776,
+      "loss": 3.9277,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.6726216077804565,
-      "learning_rate": 0.0005547656502532055,
-      "loss": 3.9217,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 0.5682153105735779,
+      "learning_rate": 0.0005546745817593091,
+      "loss": 3.9367,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.5420123338699341,
-      "learning_rate": 0.000554442409223144,
-      "loss": 3.9238,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.5704119801521301,
+      "learning_rate": 0.0005543507825148408,
+      "loss": 3.9349,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.5879136919975281,
-      "learning_rate": 0.0005541191681930826,
-      "loss": 3.9191,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.6218463182449341,
+      "learning_rate": 0.0005540269832703723,
+      "loss": 3.9385,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.5138105750083923,
-      "learning_rate": 0.0005537959271630212,
-      "loss": 3.9145,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.5849258303642273,
+      "learning_rate": 0.0005537031840259039,
+      "loss": 3.8932,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.5751107931137085,
-      "learning_rate": 0.0005534726861329598,
-      "loss": 3.9067,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.5835541486740112,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9344,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.6342102289199829,
-      "learning_rate": 0.0005531494451028983,
-      "loss": 3.9038,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.5649335980415344,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9192,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.5363113284111023,
-      "learning_rate": 0.0005528262040728369,
-      "loss": 3.9121,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.6361489295959473,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.9193,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.6412177681922913,
-      "learning_rate": 0.0005525029630427755,
-      "loss": 3.894,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.5219013094902039,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.9079,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.6429057121276855,
-      "learning_rate": 0.0005521797220127141,
-      "loss": 3.904,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.6409909129142761,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.8963,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.6362873315811157,
-      "learning_rate": 0.0005518564809826527,
-      "loss": 3.9328,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.6128087043762207,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.9236,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.6349565386772156,
-      "learning_rate": 0.0005515332399525912,
-      "loss": 3.8807,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.6087100505828857,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.9237,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.6769669651985168,
-      "learning_rate": 0.0005512099989225299,
-      "loss": 3.8932,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.6042186617851257,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.8995,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.601761519908905,
-      "learning_rate": 0.0005508867578924685,
-      "loss": 3.897,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.566831111907959,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.905,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.6315547823905945,
-      "learning_rate": 0.000550563516862407,
-      "loss": 3.8814,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.544748842716217,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.8854,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.5754820108413696,
-      "learning_rate": 0.0005502402758323456,
-      "loss": 3.9101,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.5742127299308777,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.9136,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.627307653427124,
-      "learning_rate": 0.0005499170348022841,
-      "loss": 3.9035,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.6579509973526001,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8913,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6498885154724121,
-      "learning_rate": 0.0005495937937722228,
-      "loss": 3.9012,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.5239347219467163,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8935,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.5503144264221191,
-      "learning_rate": 0.0005492705527421614,
-      "loss": 3.8942,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.6472986340522766,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.8914,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.6200529336929321,
-      "learning_rate": 0.0005489473117121,
-      "loss": 3.8717,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.5899927616119385,
+      "learning_rate": 0.0005488461953588775,
+      "loss": 3.8808,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.3380203291726159,
-      "eval_loss": 3.8143210411071777,
-      "eval_runtime": 184.8318,
-      "eval_samples_per_second": 97.445,
-      "eval_steps_per_second": 6.092,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.33687349794173405,
+      "eval_loss": 3.8257834911346436,
+      "eval_runtime": 181.2863,
+      "eval_samples_per_second": 99.351,
+      "eval_steps_per_second": 6.211,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.7157406210899353,
-      "learning_rate": 0.0005486240706820385,
-      "loss": 3.8746,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.5935076475143433,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.8878,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.643976628780365,
-      "learning_rate": 0.0005483008296519772,
-      "loss": 3.8859,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.6126903891563416,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.873,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.5374308824539185,
-      "learning_rate": 0.0005479775886219157,
-      "loss": 3.8586,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5333702564239502,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.8833,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.6169512867927551,
-      "learning_rate": 0.0005476543475918542,
-      "loss": 3.8734,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.542793869972229,
+      "learning_rate": 0.0005475509983810037,
+      "loss": 3.874,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.6148831248283386,
-      "learning_rate": 0.0005473311065617929,
-      "loss": 3.8731,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.5891146659851074,
+      "learning_rate": 0.0005472271991365352,
+      "loss": 3.8604,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.5680239796638489,
-      "learning_rate": 0.0005470078655317314,
-      "loss": 3.8674,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.6112461090087891,
+      "learning_rate": 0.0005469033998920669,
+      "loss": 3.8849,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.5679119825363159,
-      "learning_rate": 0.0005466846245016701,
-      "loss": 3.8556,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.5972184538841248,
+      "learning_rate": 0.0005465796006475984,
+      "loss": 3.8678,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.5395457744598389,
-      "learning_rate": 0.0005463613834716086,
-      "loss": 3.8536,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.685249388217926,
+      "learning_rate": 0.00054625580140313,
+      "loss": 3.8825,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.5603843331336975,
-      "learning_rate": 0.0005460381424415472,
-      "loss": 3.8579,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.6276729702949524,
+      "learning_rate": 0.0005459320021586615,
+      "loss": 3.8911,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.5550517439842224,
-      "learning_rate": 0.0005457149014114858,
-      "loss": 3.858,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.5620297193527222,
+      "learning_rate": 0.0005456082029141932,
+      "loss": 3.8825,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.6112827062606812,
-      "learning_rate": 0.0005453916603814243,
-      "loss": 3.8563,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.5803284049034119,
+      "learning_rate": 0.0005452844036697248,
+      "loss": 3.8637,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.6317242383956909,
-      "learning_rate": 0.000545068419351363,
-      "loss": 3.854,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.5740498900413513,
+      "learning_rate": 0.0005449606044252563,
+      "loss": 3.8559,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.574309766292572,
-      "learning_rate": 0.0005447451783213015,
-      "loss": 3.8455,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.5694971680641174,
+      "learning_rate": 0.0005446368051807879,
+      "loss": 3.8485,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.5467745065689087,
-      "learning_rate": 0.0005444219372912401,
-      "loss": 3.858,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.555069625377655,
+      "learning_rate": 0.0005443130059363194,
+      "loss": 3.857,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.5530813932418823,
-      "learning_rate": 0.0005440986962611787,
-      "loss": 3.8518,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5356409549713135,
+      "learning_rate": 0.0005439892066918511,
+      "loss": 3.8539,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.7071423530578613,
-      "learning_rate": 0.0005437754552311173,
-      "loss": 3.8473,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.5425722002983093,
+      "learning_rate": 0.0005436654074473825,
+      "loss": 3.8465,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.5127449631690979,
-      "learning_rate": 0.0005434522142010559,
-      "loss": 3.8471,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.5436581373214722,
+      "learning_rate": 0.0005433416082029142,
+      "loss": 3.8369,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.5855455994606018,
-      "learning_rate": 0.0005431289731709945,
-      "loss": 3.8589,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.5613462328910828,
+      "learning_rate": 0.0005430178089584457,
+      "loss": 3.8548,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.6145156025886536,
-      "learning_rate": 0.000542805732140933,
-      "loss": 3.8295,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.5960737466812134,
+      "learning_rate": 0.0005426940097139773,
+      "loss": 3.8608,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.5550306439399719,
-      "learning_rate": 0.0005424824911108716,
-      "loss": 3.8378,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.5757009387016296,
+      "learning_rate": 0.0005423702104695088,
+      "loss": 3.8477,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.3415591537287335,
-      "eval_loss": 3.7806262969970703,
-      "eval_runtime": 184.435,
-      "eval_samples_per_second": 97.655,
-      "eval_steps_per_second": 6.105,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.340650489627941,
+      "eval_loss": 3.7848594188690186,
+      "eval_runtime": 181.5098,
+      "eval_samples_per_second": 99.229,
+      "eval_steps_per_second": 6.204,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.5801270008087158,
-      "learning_rate": 0.0005421592500808102,
-      "loss": 3.8502,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.5870667695999146,
+      "learning_rate": 0.0005420464112250404,
+      "loss": 3.8409,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5607963800430298,
-      "learning_rate": 0.0005418360090507488,
-      "loss": 3.825,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.6312296986579895,
+      "learning_rate": 0.000541722611980572,
+      "loss": 3.8558,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.5534636974334717,
-      "learning_rate": 0.0005415127680206874,
-      "loss": 3.8361,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.5529806613922119,
+      "learning_rate": 0.0005413988127361036,
+      "loss": 3.8301,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.692423939704895,
-      "learning_rate": 0.0005411895269906259,
-      "loss": 3.8318,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5621755123138428,
+      "learning_rate": 0.0005410750134916351,
+      "loss": 3.8279,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.5722038745880127,
-      "learning_rate": 0.0005408662859605646,
-      "loss": 3.8376,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.564895510673523,
+      "learning_rate": 0.0005407512142471667,
+      "loss": 3.8412,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.6226180195808411,
-      "learning_rate": 0.0005405430449305031,
-      "loss": 3.8294,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5690886974334717,
+      "learning_rate": 0.0005404274150026983,
+      "loss": 3.8157,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.6096346974372864,
-      "learning_rate": 0.0005402198039004416,
-      "loss": 3.7878,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.544973611831665,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.7794,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.5371617078781128,
-      "learning_rate": 0.0005398965628703803,
-      "loss": 3.7627,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.61162269115448,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.774,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.6130861639976501,
-      "learning_rate": 0.0005395733218403189,
-      "loss": 3.769,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.6406456828117371,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.773,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.5652937889099121,
-      "learning_rate": 0.0005392500808102575,
-      "loss": 3.7714,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.6477643847465515,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.7741,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.5810158252716064,
-      "learning_rate": 0.000538926839780196,
-      "loss": 3.7566,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.5913816094398499,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7717,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.6366006731987,
-      "learning_rate": 0.0005386035987501346,
-      "loss": 3.7647,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.6406410932540894,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.76,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.5298298597335815,
-      "learning_rate": 0.0005382803577200732,
-      "loss": 3.7488,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.6126238107681274,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.801,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 1.2535181045532227,
-      "learning_rate": 0.0005379571166900119,
-      "loss": 3.7539,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.5817605257034302,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.7861,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.601189374923706,
-      "learning_rate": 0.0005376338756599504,
-      "loss": 3.7654,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.5051379203796387,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.7861,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.6602137684822083,
-      "learning_rate": 0.0005373106346298889,
-      "loss": 3.7686,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.5685185194015503,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.7939,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.6176168918609619,
-      "learning_rate": 0.0005369873935998276,
-      "loss": 3.7589,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.561697244644165,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.788,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.6260287165641785,
-      "learning_rate": 0.0005366641525697661,
-      "loss": 3.7655,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.5285218954086304,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7792,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.5720372796058655,
-      "learning_rate": 0.0005363409115397048,
-      "loss": 3.7559,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.6429543495178223,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7742,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.5716636180877686,
-      "learning_rate": 0.0005360176705096433,
-      "loss": 3.7573,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.6265733242034912,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.7666,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.3451164492757796,
-      "eval_loss": 3.7458271980285645,
-      "eval_runtime": 185.1344,
-      "eval_samples_per_second": 97.286,
-      "eval_steps_per_second": 6.082,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.34411564752612045,
+      "eval_loss": 3.753229856491089,
+      "eval_runtime": 181.6252,
+      "eval_samples_per_second": 99.166,
+      "eval_steps_per_second": 6.2,
       "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:afce0a72ad125f75905f2f0d22e2b9b901626403e511a920d55b73209c30e029
+oid sha256:b9e13e50542c0805c5428fafff83c01e0b5b324f63cb6bdf0fab92b8f96e924c
 size 5304