Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8305ddb7dcf6cfe5a35bb48a2f25351223e7b9f898277680d5d6fa2d3db2244f
+oid sha256:53750843de32f8db3e77747274fbf050884944ef260742015078d10740e9e435
 size 503128704

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14d6af0b69f631e4761ab098bfa9fa3774cebaea8a68bbe42d54414319db9409
+oid sha256:2789df46a3e32c34a8c21b6bfdf22dae483b2ed113da178edd743064e0580477
 size 1006351290

checkpoint-10000/rng_state.pth

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

checkpoint-10000/scheduler.pt

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

checkpoint-10000/trainer_state.json

@@ -1,7 +1,7 @@
 {
-  "best_metric": 3.7440133094787598,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_495/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.746513843536377,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_high_500_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": 1.6117655038833618,
-      "learning_rate": 0.00027,
-      "loss": 8.7843,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 2.169931173324585,
+      "learning_rate": 0.0003,
+      "loss": 8.5202,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 1.1765116453170776,
-      "learning_rate": 0.00057,
-      "loss": 6.9711,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 4.110340118408203,
+      "learning_rate": 0.0006,
+      "loss": 6.9257,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 1.6633868217468262,
-      "learning_rate": 0.0005997090830729447,
-      "loss": 6.5373,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 1.283010482788086,
+      "learning_rate": 0.0005996762007555315,
+      "loss": 6.5197,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 1.8836766481399536,
-      "learning_rate": 0.0005993858420428833,
-      "loss": 6.2429,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.4407877922058105,
+      "learning_rate": 0.000599352401511063,
+      "loss": 6.2691,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.5037431716918945,
-      "learning_rate": 0.0005990626010128218,
-      "loss": 6.0966,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 0.850818395614624,
+      "learning_rate": 0.0005990286022665946,
+      "loss": 6.0888,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 2.2753920555114746,
-      "learning_rate": 0.0005987393599827604,
-      "loss": 5.953,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 1.4988986253738403,
+      "learning_rate": 0.0005987048030221263,
+      "loss": 5.9867,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 3.0884439945220947,
-      "learning_rate": 0.000598416118952699,
-      "loss": 5.869,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 1.1998897790908813,
+      "learning_rate": 0.0005983810037776578,
+      "loss": 5.8798,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 1.4899078607559204,
-      "learning_rate": 0.0005980928779226376,
-      "loss": 5.8032,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 1.2407352924346924,
+      "learning_rate": 0.0005980572045331894,
+      "loss": 5.8211,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.125209093093872,
-      "learning_rate": 0.0005977696368925762,
-      "loss": 5.7355,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 0.9971290826797485,
+      "learning_rate": 0.0005977334052887209,
+      "loss": 5.7361,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 1.960442066192627,
-      "learning_rate": 0.0005974463958625148,
-      "loss": 5.6583,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 1.255806803703308,
+      "learning_rate": 0.0005974096060442526,
+      "loss": 5.683,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 1.044176697731018,
-      "learning_rate": 0.0005971231548324534,
-      "loss": 5.5711,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 1.75487220287323,
+      "learning_rate": 0.0005970858067997841,
+      "loss": 5.5637,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.6373082399368286,
-      "learning_rate": 0.0005967999138023919,
-      "loss": 5.5099,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.2305444478988647,
+      "learning_rate": 0.0005967620075553157,
+      "loss": 5.5257,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 1.8287220001220703,
-      "learning_rate": 0.0005964766727723304,
-      "loss": 5.4242,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 1.346718668937683,
+      "learning_rate": 0.0005964382083108472,
+      "loss": 5.4252,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 1.3936927318572998,
-      "learning_rate": 0.0005961534317422691,
-      "loss": 5.3571,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 1.0916610956192017,
+      "learning_rate": 0.0005961144090663788,
+      "loss": 5.3969,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 0.9875015020370483,
-      "learning_rate": 0.0005958301907122077,
-      "loss": 5.3186,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.0708554983139038,
+      "learning_rate": 0.0005957906098219104,
+      "loss": 5.3257,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 1.0579822063446045,
-      "learning_rate": 0.0005955069496821463,
-      "loss": 5.2725,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 1.4724178314208984,
+      "learning_rate": 0.0005954668105774419,
+      "loss": 5.2774,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.3142889738082886,
-      "learning_rate": 0.0005951837086520848,
-      "loss": 5.2145,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 1.0166338682174683,
+      "learning_rate": 0.0005951430113329735,
+      "loss": 5.2416,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.207129716873169,
-      "learning_rate": 0.0005948604676220235,
-      "loss": 5.1809,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 1.2486152648925781,
+      "learning_rate": 0.0005948192120885051,
+      "loss": 5.155,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 1.3603378534317017,
-      "learning_rate": 0.000594537226591962,
-      "loss": 5.1166,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 0.8233184814453125,
+      "learning_rate": 0.0005944954128440366,
+      "loss": 5.1494,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.0100822448730469,
-      "learning_rate": 0.0005942139855619007,
-      "loss": 5.0872,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 1.1141846179962158,
+      "learning_rate": 0.0005941716135995682,
+      "loss": 5.0928,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.22757434275053265,
-      "eval_loss": 5.0165934562683105,
-      "eval_runtime": 186.2978,
-      "eval_samples_per_second": 96.679,
-      "eval_steps_per_second": 6.044,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22779479945990821,
+      "eval_loss": 5.019223690032959,
+      "eval_runtime": 184.6889,
+      "eval_samples_per_second": 97.521,
+      "eval_steps_per_second": 6.097,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 0.9639580249786377,
-      "learning_rate": 0.0005938907445318392,
-      "loss": 5.0357,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 1.1920228004455566,
+      "learning_rate": 0.0005938478143550997,
+      "loss": 5.0528,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 0.8208778500556946,
-      "learning_rate": 0.0005935675035017777,
-      "loss": 5.0074,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.3988215923309326,
+      "learning_rate": 0.0005935240151106314,
+      "loss": 5.0162,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 0.9588396549224854,
-      "learning_rate": 0.0005932442624717164,
-      "loss": 4.9928,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 1.1493585109710693,
+      "learning_rate": 0.0005932002158661629,
+      "loss": 4.9947,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 0.8537307977676392,
-      "learning_rate": 0.0005929210214416549,
-      "loss": 4.9625,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 0.8649298548698425,
+      "learning_rate": 0.0005928764166216945,
+      "loss": 4.9752,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 1.1293931007385254,
-      "learning_rate": 0.0005925977804115936,
-      "loss": 4.9296,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 1.2792754173278809,
+      "learning_rate": 0.000592552617377226,
+      "loss": 4.9485,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 1.039612054824829,
-      "learning_rate": 0.0005922745393815321,
-      "loss": 4.904,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 0.939761757850647,
+      "learning_rate": 0.0005922288181327577,
+      "loss": 4.8944,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 0.986505925655365,
-      "learning_rate": 0.0005919512983514707,
-      "loss": 4.8739,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 1.2719601392745972,
+      "learning_rate": 0.0005919050188882893,
+      "loss": 4.8786,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 0.9047093391418457,
-      "learning_rate": 0.0005916280573214093,
-      "loss": 4.8174,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 0.7670081853866577,
+      "learning_rate": 0.0005915812196438207,
+      "loss": 4.8498,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 0.770999014377594,
-      "learning_rate": 0.0005913048162913478,
-      "loss": 4.8181,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 1.2284456491470337,
+      "learning_rate": 0.0005912574203993524,
+      "loss": 4.8299,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 0.9082644581794739,
-      "learning_rate": 0.0005909815752612864,
-      "loss": 4.8036,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 0.8652415871620178,
+      "learning_rate": 0.0005909336211548839,
+      "loss": 4.8202,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 0.8358648419380188,
-      "learning_rate": 0.000590658334231225,
-      "loss": 4.7798,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 0.8690969944000244,
+      "learning_rate": 0.0005906098219104155,
+      "loss": 4.7751,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 1.0487140417099,
-      "learning_rate": 0.0005903350932011637,
-      "loss": 4.7364,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 1.03486967086792,
+      "learning_rate": 0.000590286022665947,
+      "loss": 4.7912,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 1.1815866231918335,
-      "learning_rate": 0.0005900118521711022,
-      "loss": 4.7392,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 0.8726552724838257,
+      "learning_rate": 0.0005899622234214787,
+      "loss": 4.7651,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 0.9304591417312622,
-      "learning_rate": 0.0005896886111410408,
-      "loss": 4.7118,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.8985825181007385,
+      "learning_rate": 0.0005896384241770102,
+      "loss": 4.6925,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 1.0020699501037598,
-      "learning_rate": 0.0005893653701109793,
-      "loss": 4.6978,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 1.0985790491104126,
+      "learning_rate": 0.0005893146249325418,
+      "loss": 4.7067,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 0.8822704553604126,
-      "learning_rate": 0.000589042129080918,
-      "loss": 4.6769,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 1.0200293064117432,
+      "learning_rate": 0.0005889908256880733,
+      "loss": 4.6942,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 0.944671630859375,
-      "learning_rate": 0.0005887188880508566,
-      "loss": 4.6617,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 1.0469223260879517,
+      "learning_rate": 0.0005886670264436049,
+      "loss": 4.6484,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 1.0103189945220947,
-      "learning_rate": 0.0005883956470207951,
-      "loss": 4.6347,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 0.8023126721382141,
+      "learning_rate": 0.0005883432271991365,
+      "loss": 4.6433,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 0.9612853527069092,
-      "learning_rate": 0.0005880724059907337,
-      "loss": 4.5904,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 1.0328996181488037,
+      "learning_rate": 0.0005880194279546681,
+      "loss": 4.6104,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 0.9868743419647217,
-      "learning_rate": 0.0005877491649606723,
-      "loss": 4.58,
+      "epoch": 0.215633423180593,
+      "grad_norm": 1.1157277822494507,
+      "learning_rate": 0.0005876956287101996,
+      "loss": 4.586,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.27043877621644785,
-      "eval_loss": 4.506998538970947,
-      "eval_runtime": 184.7948,
-      "eval_samples_per_second": 97.465,
-      "eval_steps_per_second": 6.093,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.26792661279728924,
+      "eval_loss": 4.539734363555908,
+      "eval_runtime": 183.597,
+      "eval_samples_per_second": 98.101,
+      "eval_steps_per_second": 6.133,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 0.8025125861167908,
-      "learning_rate": 0.0005874259239306109,
-      "loss": 4.5503,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.802883505821228,
+      "learning_rate": 0.0005873718294657312,
+      "loss": 4.6011,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 1.0225203037261963,
-      "learning_rate": 0.0005871026829005494,
-      "loss": 4.5634,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 0.9819656014442444,
+      "learning_rate": 0.0005870480302212628,
+      "loss": 4.5627,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 0.9046829342842102,
-      "learning_rate": 0.0005867794418704881,
-      "loss": 4.5213,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 1.0422033071517944,
+      "learning_rate": 0.0005867242309767943,
+      "loss": 4.5141,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 0.7576098442077637,
-      "learning_rate": 0.0005864562008404266,
-      "loss": 4.5007,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 0.9134729504585266,
+      "learning_rate": 0.0005864004317323259,
+      "loss": 4.5497,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 0.9998608231544495,
-      "learning_rate": 0.0005861329598103653,
-      "loss": 4.4993,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 0.7439409494400024,
+      "learning_rate": 0.0005860766324878575,
+      "loss": 4.486,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 1.168628215789795,
-      "learning_rate": 0.0005858097187803038,
-      "loss": 4.4896,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 1.0813612937927246,
+      "learning_rate": 0.000585752833243389,
+      "loss": 4.4643,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.8030685782432556,
-      "learning_rate": 0.0005854864777502423,
-      "loss": 4.4601,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.8331607580184937,
+      "learning_rate": 0.0005854290339989206,
+      "loss": 4.4584,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 1.0477194786071777,
-      "learning_rate": 0.000585163236720181,
-      "loss": 4.4508,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 0.8642703294754028,
+      "learning_rate": 0.0005851052347544521,
+      "loss": 4.4498,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 0.86094731092453,
-      "learning_rate": 0.0005848399956901195,
-      "loss": 4.428,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 0.829148530960083,
+      "learning_rate": 0.0005847814355099838,
+      "loss": 4.453,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 0.9053926467895508,
-      "learning_rate": 0.0005845167546600582,
-      "loss": 4.4063,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 0.9349790215492249,
+      "learning_rate": 0.0005844576362655154,
+      "loss": 4.4526,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.7576990127563477,
-      "learning_rate": 0.0005841935136299967,
-      "loss": 4.4131,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.8753833770751953,
+      "learning_rate": 0.0005841338370210469,
+      "loss": 4.412,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 0.7213003635406494,
-      "learning_rate": 0.0005838702725999353,
-      "loss": 4.3988,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 0.7982025146484375,
+      "learning_rate": 0.0005838100377765785,
+      "loss": 4.4007,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.9967924356460571,
-      "learning_rate": 0.0005835470315698739,
-      "loss": 4.3945,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 1.1045323610305786,
+      "learning_rate": 0.0005834862385321101,
+      "loss": 4.3978,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.7419942617416382,
-      "learning_rate": 0.0005832237905398125,
-      "loss": 4.3636,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 0.873582124710083,
+      "learning_rate": 0.0005831624392876417,
+      "loss": 4.3911,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.7668341994285583,
-      "learning_rate": 0.0005829005495097511,
-      "loss": 4.3658,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.9419072866439819,
+      "learning_rate": 0.0005828386400431731,
+      "loss": 4.3527,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 0.7742299437522888,
-      "learning_rate": 0.0005825773084796896,
-      "loss": 4.334,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.6602005958557129,
+      "learning_rate": 0.0005825148407987048,
+      "loss": 4.3873,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 0.9014106392860413,
-      "learning_rate": 0.0005822540674496282,
-      "loss": 4.327,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.7998479008674622,
+      "learning_rate": 0.0005821910415542363,
+      "loss": 4.3582,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 0.6771566867828369,
-      "learning_rate": 0.0005819308264195668,
-      "loss": 4.3486,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.7541155219078064,
+      "learning_rate": 0.0005818672423097679,
+      "loss": 4.337,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.712129533290863,
-      "learning_rate": 0.0005816075853895054,
-      "loss": 4.3111,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.8266378045082092,
+      "learning_rate": 0.0005815434430652994,
+      "loss": 4.3355,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.6686953902244568,
-      "learning_rate": 0.000581284344359444,
-      "loss": 4.3079,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.7615757584571838,
+      "learning_rate": 0.0005812196438208311,
+      "loss": 4.3331,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.2981868766651734,
-      "eval_loss": 4.241795063018799,
-      "eval_runtime": 184.6997,
-      "eval_samples_per_second": 97.515,
-      "eval_steps_per_second": 6.096,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.2979562065843433,
+      "eval_loss": 4.236091613769531,
+      "eval_runtime": 183.7161,
+      "eval_samples_per_second": 98.037,
+      "eval_steps_per_second": 6.129,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.6886458992958069,
-      "learning_rate": 0.0005809611033293825,
-      "loss": 4.3161,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 0.9040746688842773,
+      "learning_rate": 0.0005808958445763626,
+      "loss": 4.3105,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.7012606859207153,
-      "learning_rate": 0.0005806378622993211,
-      "loss": 4.2908,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.6944011449813843,
+      "learning_rate": 0.0005805720453318942,
+      "loss": 4.3025,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.6854538321495056,
-      "learning_rate": 0.0005803146212692597,
-      "loss": 4.2775,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.6618968844413757,
+      "learning_rate": 0.0005802482460874257,
+      "loss": 4.2711,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.8339408040046692,
-      "learning_rate": 0.0005799913802391983,
-      "loss": 4.27,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.7913966178894043,
+      "learning_rate": 0.0005799244468429573,
+      "loss": 4.279,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 0.8958104848861694,
-      "learning_rate": 0.0005796681392091369,
-      "loss": 4.2535,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 0.7068710923194885,
+      "learning_rate": 0.0005796006475984889,
+      "loss": 4.2522,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.671660304069519,
-      "learning_rate": 0.0005793448981790755,
-      "loss": 4.2516,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.7139419317245483,
+      "learning_rate": 0.0005792768483540205,
+      "loss": 4.2757,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.7408727407455444,
-      "learning_rate": 0.0005790216571490141,
-      "loss": 4.2658,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.7112064957618713,
+      "learning_rate": 0.000578953049109552,
+      "loss": 4.255,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.6920570731163025,
-      "learning_rate": 0.0005786984161189527,
-      "loss": 4.2325,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.824454665184021,
+      "learning_rate": 0.0005786292498650836,
+      "loss": 4.2453,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.8909915685653687,
-      "learning_rate": 0.0005783751750888912,
-      "loss": 4.2163,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.651171863079071,
+      "learning_rate": 0.0005783054506206152,
+      "loss": 4.2315,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.8377681970596313,
-      "learning_rate": 0.0005780519340588297,
-      "loss": 4.2289,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.6538099646568298,
+      "learning_rate": 0.0005779816513761467,
+      "loss": 4.2093,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.6133842468261719,
-      "learning_rate": 0.0005777286930287684,
-      "loss": 4.216,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.6883544325828552,
+      "learning_rate": 0.0005776578521316782,
+      "loss": 4.2153,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.5939428806304932,
-      "learning_rate": 0.000577405451998707,
-      "loss": 4.2227,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.8138344287872314,
+      "learning_rate": 0.0005773340528872099,
+      "loss": 4.2188,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.6408166885375977,
-      "learning_rate": 0.0005770822109686456,
-      "loss": 4.2049,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 0.9981099963188171,
+      "learning_rate": 0.0005770102536427414,
+      "loss": 4.2288,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.819424033164978,
-      "learning_rate": 0.0005767589699385841,
-      "loss": 4.2005,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.7214846014976501,
+      "learning_rate": 0.000576686454398273,
+      "loss": 4.1953,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.6583958864212036,
-      "learning_rate": 0.0005764357289085228,
-      "loss": 4.1926,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.7932049036026001,
+      "learning_rate": 0.0005763626551538045,
+      "loss": 4.1891,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.6472761631011963,
-      "learning_rate": 0.0005761124878784613,
-      "loss": 4.1791,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.7664895057678223,
+      "learning_rate": 0.0005760388559093362,
+      "loss": 4.214,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.6922594308853149,
-      "learning_rate": 0.0005757892468484,
-      "loss": 4.1849,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.8450409770011902,
+      "learning_rate": 0.0005757150566648678,
+      "loss": 4.1912,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.6930834054946899,
-      "learning_rate": 0.0005754660058183385,
-      "loss": 4.1757,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 0.9285402297973633,
+      "learning_rate": 0.0005753912574203993,
+      "loss": 4.1599,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 0.6437042355537415,
-      "learning_rate": 0.000575142764788277,
-      "loss": 4.1714,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.6313654184341431,
+      "learning_rate": 0.0005750674581759309,
+      "loss": 4.1601,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 0.7084268927574158,
-      "learning_rate": 0.0005748195237582157,
-      "loss": 4.1579,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.6990293264389038,
+      "learning_rate": 0.0005747436589314624,
+      "loss": 4.1641,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.31272235134410686,
-      "eval_loss": 4.0872697830200195,
-      "eval_runtime": 184.7092,
-      "eval_samples_per_second": 97.51,
-      "eval_steps_per_second": 6.096,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.31198546745894906,
+      "eval_loss": 4.094001293182373,
+      "eval_runtime": 183.5145,
+      "eval_samples_per_second": 98.145,
+      "eval_steps_per_second": 6.136,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.6244553327560425,
-      "learning_rate": 0.0005744962827281542,
-      "loss": 4.1348,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.6404671669006348,
+      "learning_rate": 0.0005744198596869941,
+      "loss": 4.1621,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.7042115330696106,
-      "learning_rate": 0.0005741730416980928,
-      "loss": 4.1397,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.6013748049736023,
+      "learning_rate": 0.0005740960604425255,
+      "loss": 4.1719,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.7811130881309509,
-      "learning_rate": 0.0005738498006680314,
-      "loss": 4.1573,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.7267091274261475,
+      "learning_rate": 0.0005737722611980572,
+      "loss": 4.1522,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.6470036506652832,
-      "learning_rate": 0.00057352655963797,
-      "loss": 4.1357,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 0.6014766097068787,
+      "learning_rate": 0.0005734484619535887,
+      "loss": 4.1425,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.6716099977493286,
-      "learning_rate": 0.0005732033186079086,
-      "loss": 4.1327,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.5657269358634949,
+      "learning_rate": 0.0005731246627091203,
+      "loss": 4.1337,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.7920870184898376,
-      "learning_rate": 0.0005728800775778471,
-      "loss": 4.1263,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.6378490328788757,
+      "learning_rate": 0.0005728008634646518,
+      "loss": 4.126,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.7230188846588135,
-      "learning_rate": 0.0005725568365477857,
-      "loss": 4.1104,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.739479660987854,
+      "learning_rate": 0.0005724770642201835,
+      "loss": 4.1349,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.7376595735549927,
-      "learning_rate": 0.0005722335955177243,
-      "loss": 4.1103,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.7157710790634155,
+      "learning_rate": 0.000572153264975715,
+      "loss": 4.1215,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 1.1194459199905396,
-      "learning_rate": 0.000571910354487663,
-      "loss": 4.1222,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.7867320775985718,
+      "learning_rate": 0.0005718294657312466,
+      "loss": 4.1252,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.6437053680419922,
-      "learning_rate": 0.0005715871134576015,
-      "loss": 4.1131,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.622015118598938,
+      "learning_rate": 0.0005715056664867781,
+      "loss": 4.108,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.7767078876495361,
-      "learning_rate": 0.0005712638724275401,
-      "loss": 4.1017,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.6655036807060242,
+      "learning_rate": 0.0005711818672423097,
+      "loss": 4.0987,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.6522454023361206,
-      "learning_rate": 0.0005709406313974786,
-      "loss": 4.0979,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.6496259570121765,
+      "learning_rate": 0.0005708580679978413,
+      "loss": 4.0999,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.6566357612609863,
-      "learning_rate": 0.0005706173903674173,
-      "loss": 4.0735,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.6310088038444519,
+      "learning_rate": 0.0005705342687533729,
+      "loss": 4.0796,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 0.5966187715530396,
-      "learning_rate": 0.0005702941493373559,
-      "loss": 4.0895,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.8397471308708191,
+      "learning_rate": 0.0005702104695089044,
+      "loss": 4.0885,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.7359771132469177,
-      "learning_rate": 0.0005699709083072944,
-      "loss": 4.0868,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.6343526840209961,
+      "learning_rate": 0.000569886670264436,
+      "loss": 4.0699,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.7156068086624146,
-      "learning_rate": 0.000569647667277233,
-      "loss": 4.0897,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.6864327192306519,
+      "learning_rate": 0.0005695628710199675,
+      "loss": 4.0858,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.6234465837478638,
-      "learning_rate": 0.0005693244262471716,
-      "loss": 4.0549,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.6211134195327759,
+      "learning_rate": 0.0005692390717754991,
+      "loss": 4.0786,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.7085710167884827,
-      "learning_rate": 0.0005690011852171102,
-      "loss": 4.0802,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.6251089572906494,
+      "learning_rate": 0.0005689152725310306,
+      "loss": 4.0887,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.5667790174484253,
-      "learning_rate": 0.0005686779441870487,
-      "loss": 4.0628,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.691362738609314,
+      "learning_rate": 0.0005685914732865623,
+      "loss": 4.061,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.6275297999382019,
-      "learning_rate": 0.0005683547031569874,
-      "loss": 4.0589,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.6830545663833618,
+      "learning_rate": 0.0005682676740420939,
+      "loss": 4.0484,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.3213402637940002,
-      "eval_loss": 3.990799903869629,
-      "eval_runtime": 184.6417,
-      "eval_samples_per_second": 97.546,
-      "eval_steps_per_second": 6.098,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.3213867672299848,
+      "eval_loss": 3.994671106338501,
+      "eval_runtime": 183.5628,
+      "eval_samples_per_second": 98.119,
+      "eval_steps_per_second": 6.134,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.6828199625015259,
-      "learning_rate": 0.0005680314621269259,
-      "loss": 4.056,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.5907337069511414,
+      "learning_rate": 0.0005679438747976254,
+      "loss": 4.045,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.6843883991241455,
-      "learning_rate": 0.0005677082210968646,
-      "loss": 4.0418,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.8197788000106812,
+      "learning_rate": 0.000567620075553157,
+      "loss": 4.0488,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.5970234274864197,
-      "learning_rate": 0.0005673849800668031,
-      "loss": 4.0339,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.6527996063232422,
+      "learning_rate": 0.0005672962763086886,
+      "loss": 4.0536,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.6066529154777527,
-      "learning_rate": 0.0005670617390367416,
-      "loss": 4.0554,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.6831157803535461,
+      "learning_rate": 0.0005669724770642202,
+      "loss": 4.0314,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.7181238532066345,
-      "learning_rate": 0.0005667384980066803,
-      "loss": 4.0289,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.6056663990020752,
+      "learning_rate": 0.0005666486778197517,
+      "loss": 4.0345,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.6789528131484985,
-      "learning_rate": 0.0005664152569766188,
-      "loss": 4.0306,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.70479416847229,
+      "learning_rate": 0.0005663248785752833,
+      "loss": 4.0151,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6828908920288086,
-      "learning_rate": 0.0005660920159465575,
-      "loss": 4.0383,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.6906822919845581,
+      "learning_rate": 0.0005660010793308148,
+      "loss": 4.0303,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.5343970656394958,
-      "learning_rate": 0.000565768774916496,
-      "loss": 4.0472,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.6968085765838623,
+      "learning_rate": 0.0005656772800863465,
+      "loss": 4.0215,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.6703925728797913,
-      "learning_rate": 0.0005654455338864346,
-      "loss": 4.0139,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.8021159768104553,
+      "learning_rate": 0.0005653534808418779,
+      "loss": 4.0326,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.5589005351066589,
-      "learning_rate": 0.0005651222928563732,
-      "loss": 4.0171,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.6045252084732056,
+      "learning_rate": 0.0005650296815974096,
+      "loss": 4.033,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.6419985890388489,
-      "learning_rate": 0.0005647990518263118,
-      "loss": 3.994,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6504480838775635,
+      "learning_rate": 0.0005647058823529411,
+      "loss": 4.0368,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.5866629481315613,
-      "learning_rate": 0.0005644758107962504,
-      "loss": 4.0104,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.6711751222610474,
+      "learning_rate": 0.0005643820831084727,
+      "loss": 4.0154,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.9215855002403259,
-      "learning_rate": 0.0005641525697661889,
-      "loss": 4.0096,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.5429975390434265,
+      "learning_rate": 0.0005640582838640042,
+      "loss": 4.004,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.5779905319213867,
-      "learning_rate": 0.0005638293287361275,
-      "loss": 3.9754,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.6812418103218079,
+      "learning_rate": 0.0005637344846195358,
+      "loss": 4.0092,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.6425080299377441,
-      "learning_rate": 0.0005635060877060661,
-      "loss": 3.9917,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.67169588804245,
+      "learning_rate": 0.0005634106853750674,
+      "loss": 4.0189,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.6887379884719849,
-      "learning_rate": 0.0005631828466760047,
-      "loss": 3.9941,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.67845219373703,
+      "learning_rate": 0.000563086886130599,
+      "loss": 4.0127,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.6834486722946167,
-      "learning_rate": 0.0005628596056459433,
-      "loss": 3.9953,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.6689065098762512,
+      "learning_rate": 0.0005627630868861305,
+      "loss": 4.0195,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.7504608631134033,
-      "learning_rate": 0.0005625363646158818,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.6098775267601013,
+      "learning_rate": 0.0005624392876416621,
       "loss": 3.9892,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.6510689854621887,
-      "learning_rate": 0.0005622131235858205,
-      "loss": 3.9722,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.5924386382102966,
+      "learning_rate": 0.0005621154883971937,
+      "loss": 3.9839,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.5909599661827087,
-      "learning_rate": 0.000561889882555759,
-      "loss": 3.9673,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.5603781342506409,
+      "learning_rate": 0.0005617916891527253,
+      "loss": 3.9955,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.32834229049631664,
-      "eval_loss": 3.9165666103363037,
-      "eval_runtime": 184.5514,
-      "eval_samples_per_second": 97.593,
-      "eval_steps_per_second": 6.101,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32812498472068763,
+      "eval_loss": 3.916694164276123,
+      "eval_runtime": 183.5467,
+      "eval_samples_per_second": 98.128,
+      "eval_steps_per_second": 6.135,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.6396195292472839,
-      "learning_rate": 0.0005615666415256976,
-      "loss": 3.997,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.7233008742332458,
+      "learning_rate": 0.0005614678899082568,
+      "loss": 3.9679,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.5541490316390991,
-      "learning_rate": 0.0005612434004956361,
-      "loss": 3.981,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.7258248329162598,
+      "learning_rate": 0.0005611440906637884,
+      "loss": 3.9993,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.6580690145492554,
-      "learning_rate": 0.0005609201594655748,
-      "loss": 3.9881,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.6090084910392761,
+      "learning_rate": 0.00056082029141932,
+      "loss": 3.9792,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.6880823373794556,
-      "learning_rate": 0.0005605969184355134,
-      "loss": 3.9573,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.667877197265625,
+      "learning_rate": 0.0005605029681597409,
+      "loss": 3.9585,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.6356266140937805,
-      "learning_rate": 0.000560273677405452,
-      "loss": 3.9975,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.6215923428535461,
+      "learning_rate": 0.0005601791689152725,
+      "loss": 3.9753,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.5227646827697754,
-      "learning_rate": 0.0005599504363753905,
-      "loss": 3.9533,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.6659998893737793,
+      "learning_rate": 0.000559855369670804,
+      "loss": 3.9925,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.6362395882606506,
-      "learning_rate": 0.0005596271953453291,
-      "loss": 3.9563,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.6712725758552551,
+      "learning_rate": 0.0005595315704263357,
+      "loss": 3.9557,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.5801821947097778,
-      "learning_rate": 0.0005593039543152677,
-      "loss": 3.9428,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.5836709141731262,
+      "learning_rate": 0.0005592077711818672,
+      "loss": 3.9608,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.6077560782432556,
-      "learning_rate": 0.0005589807132852063,
-      "loss": 3.9554,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.6350057721138,
+      "learning_rate": 0.0005588839719373988,
+      "loss": 3.9449,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.5769425630569458,
-      "learning_rate": 0.0005586574722551449,
-      "loss": 3.9447,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6150210499763489,
+      "learning_rate": 0.0005585601726929303,
+      "loss": 3.9582,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.6278015971183777,
-      "learning_rate": 0.0005583342312250834,
-      "loss": 3.9678,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.598513126373291,
+      "learning_rate": 0.0005582363734484619,
+      "loss": 3.9816,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.5199827551841736,
-      "learning_rate": 0.0005580109901950221,
-      "loss": 3.9618,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.6591358184814453,
+      "learning_rate": 0.0005579125742039935,
+      "loss": 3.937,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.659668505191803,
-      "learning_rate": 0.0005576877491649606,
-      "loss": 3.9206,
+      "epoch": 0.7169811320754716,
+      "grad_norm": 0.5885124206542969,
+      "learning_rate": 0.0005575887749595251,
+      "loss": 3.9678,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.6791085004806519,
-      "learning_rate": 0.0005573645081348993,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.6395810842514038,
+      "learning_rate": 0.0005572649757150566,
       "loss": 3.9427,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6757652759552002,
-      "learning_rate": 0.0005570412671048378,
-      "loss": 3.9381,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.5958187580108643,
+      "learning_rate": 0.0005569411764705882,
+      "loss": 3.9486,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.5604714155197144,
-      "learning_rate": 0.0005567180260747763,
-      "loss": 3.9385,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.5831095576286316,
+      "learning_rate": 0.0005566173772261198,
+      "loss": 3.9383,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.6604406237602234,
-      "learning_rate": 0.000556394785044715,
-      "loss": 3.9272,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.6585524082183838,
+      "learning_rate": 0.0005562935779816513,
+      "loss": 3.9173,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.564445972442627,
-      "learning_rate": 0.0005560715440146535,
-      "loss": 3.9496,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.705803394317627,
+      "learning_rate": 0.0005559697787371828,
+      "loss": 3.945,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.5737416744232178,
-      "learning_rate": 0.0005557483029845921,
-      "loss": 3.9387,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.634233295917511,
+      "learning_rate": 0.0005556459794927145,
+      "loss": 3.9407,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.5588909983634949,
-      "learning_rate": 0.0005554250619545307,
-      "loss": 3.9074,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.6854012608528137,
+      "learning_rate": 0.000555322180248246,
+      "loss": 3.9273,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.33360815270424543,
-      "eval_loss": 3.859344720840454,
-      "eval_runtime": 184.9069,
-      "eval_samples_per_second": 97.406,
-      "eval_steps_per_second": 6.09,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.3336400966532629,
+      "eval_loss": 3.8616080284118652,
+      "eval_runtime": 183.6309,
+      "eval_samples_per_second": 98.083,
+      "eval_steps_per_second": 6.132,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.6285735964775085,
-      "learning_rate": 0.0005551018209244694,
-      "loss": 3.9333,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.6441482901573181,
+      "learning_rate": 0.0005549983810037776,
+      "loss": 3.9186,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.5597401261329651,
-      "learning_rate": 0.0005547785798944079,
-      "loss": 3.9209,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 0.6202889084815979,
+      "learning_rate": 0.0005546745817593091,
+      "loss": 3.9302,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.578063428401947,
-      "learning_rate": 0.0005544553388643464,
-      "loss": 3.9312,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.518191933631897,
+      "learning_rate": 0.0005543507825148408,
+      "loss": 3.9288,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.5833026170730591,
-      "learning_rate": 0.000554132097834285,
-      "loss": 3.9328,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.6395387053489685,
+      "learning_rate": 0.0005540269832703723,
+      "loss": 3.9317,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.518353283405304,
-      "learning_rate": 0.0005538088568042236,
-      "loss": 3.9029,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.578109860420227,
+      "learning_rate": 0.0005537031840259039,
+      "loss": 3.887,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.6059916019439697,
-      "learning_rate": 0.0005534856157741623,
-      "loss": 3.9199,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.7136623859405518,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9282,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.5463325381278992,
-      "learning_rate": 0.0005531623747441008,
-      "loss": 3.9054,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.615642786026001,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9137,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.6463024616241455,
-      "learning_rate": 0.0005528391337140394,
-      "loss": 3.9191,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.6420246362686157,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.9119,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.6024007797241211,
-      "learning_rate": 0.0005525158926839779,
-      "loss": 3.9284,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.5536246299743652,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.9007,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.5554441213607788,
-      "learning_rate": 0.0005521926516539166,
-      "loss": 3.9015,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.6325304508209229,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.8882,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.5887565612792969,
-      "learning_rate": 0.0005518694106238552,
-      "loss": 3.9024,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.6307028532028198,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.9159,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.575228214263916,
-      "learning_rate": 0.0005515461695937937,
-      "loss": 3.8981,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.5926609039306641,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.9175,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.6179103255271912,
-      "learning_rate": 0.0005512229285637323,
-      "loss": 3.9081,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.5906722545623779,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.8928,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.6597625017166138,
-      "learning_rate": 0.0005508996875336709,
-      "loss": 3.8958,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.5734378099441528,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.8949,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.5965509414672852,
-      "learning_rate": 0.0005505764465036095,
-      "loss": 3.8879,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.56657475233078,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.878,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.512361466884613,
-      "learning_rate": 0.000550253205473548,
-      "loss": 3.8809,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.6782810091972351,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.9073,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.6104485988616943,
-      "learning_rate": 0.0005499299644434867,
-      "loss": 3.8852,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.6757487654685974,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8846,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6456380486488342,
-      "learning_rate": 0.0005496067234134252,
-      "loss": 3.8783,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.5899352431297302,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8869,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.5954079627990723,
-      "learning_rate": 0.0005492834823833639,
-      "loss": 3.889,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.5763821005821228,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.8827,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.6212036609649658,
-      "learning_rate": 0.0005489602413533024,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.6570541858673096,
+      "learning_rate": 0.0005488461953588775,
       "loss": 3.8712,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.33805890094779,
-      "eval_loss": 3.8169474601745605,
-      "eval_runtime": 184.4499,
-      "eval_samples_per_second": 97.647,
-      "eval_steps_per_second": 6.105,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.3377064309797198,
+      "eval_loss": 3.816032886505127,
+      "eval_runtime": 183.5243,
+      "eval_samples_per_second": 98.14,
+      "eval_steps_per_second": 6.135,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.5757076144218445,
-      "learning_rate": 0.0005486370003232409,
-      "loss": 3.8725,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.5602306723594666,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.8805,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.6085401177406311,
-      "learning_rate": 0.0005483137592931796,
-      "loss": 3.8905,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.6562096476554871,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.8669,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.5095037817955017,
-      "learning_rate": 0.0005479905182631181,
-      "loss": 3.8799,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5427864789962769,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.8766,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.5414556860923767,
-      "learning_rate": 0.0005476672772330568,
-      "loss": 3.8747,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.5421739816665649,
+      "learning_rate": 0.0005475574743658931,
+      "loss": 3.8668,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.5936631560325623,
-      "learning_rate": 0.0005473440362029953,
-      "loss": 3.863,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.6107007265090942,
+      "learning_rate": 0.0005472336751214246,
+      "loss": 3.8544,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.5667993426322937,
-      "learning_rate": 0.0005470207951729339,
-      "loss": 3.8607,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.6804000735282898,
+      "learning_rate": 0.0005469098758769562,
+      "loss": 3.8797,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.5451300144195557,
-      "learning_rate": 0.0005466975541428725,
-      "loss": 3.8857,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.6543049216270447,
+      "learning_rate": 0.0005465860766324878,
+      "loss": 3.861,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.5745592713356018,
-      "learning_rate": 0.0005463743131128111,
-      "loss": 3.8554,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.7077866196632385,
+      "learning_rate": 0.0005462622773880194,
+      "loss": 3.874,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.5426753759384155,
-      "learning_rate": 0.0005460510720827497,
-      "loss": 3.8752,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.5970845818519592,
+      "learning_rate": 0.000545938478143551,
+      "loss": 3.8836,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.508082389831543,
-      "learning_rate": 0.0005457278310526882,
-      "loss": 3.8584,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.5863073468208313,
+      "learning_rate": 0.0005456146788990825,
+      "loss": 3.8752,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.532146155834198,
-      "learning_rate": 0.0005454045900226268,
-      "loss": 3.8476,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.5704134702682495,
+      "learning_rate": 0.000545290879654614,
+      "loss": 3.8553,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.5778146386146545,
-      "learning_rate": 0.0005450813489925654,
-      "loss": 3.8612,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.5353178381919861,
+      "learning_rate": 0.0005449670804101457,
+      "loss": 3.8502,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.5984019637107849,
-      "learning_rate": 0.000544758107962504,
-      "loss": 3.8478,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.5742438435554504,
+      "learning_rate": 0.0005446432811656773,
+      "loss": 3.8445,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.609585165977478,
-      "learning_rate": 0.0005444348669324426,
-      "loss": 3.8541,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.6159672141075134,
+      "learning_rate": 0.0005443194819212088,
+      "loss": 3.85,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.5877447724342346,
-      "learning_rate": 0.0005441116259023811,
-      "loss": 3.8298,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5223457217216492,
+      "learning_rate": 0.0005439956826767404,
+      "loss": 3.8475,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.6004450917243958,
-      "learning_rate": 0.0005437883848723198,
-      "loss": 3.8393,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.5645538568496704,
+      "learning_rate": 0.000543671883432272,
+      "loss": 3.8401,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.5448794960975647,
-      "learning_rate": 0.0005434651438422583,
-      "loss": 3.8515,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.5175207853317261,
+      "learning_rate": 0.0005433480841878035,
+      "loss": 3.8302,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.5065131783485413,
-      "learning_rate": 0.0005431419028121969,
-      "loss": 3.8478,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.5572501420974731,
+      "learning_rate": 0.000543024284943335,
+      "loss": 3.8494,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.6335159540176392,
-      "learning_rate": 0.0005428186617821354,
-      "loss": 3.8411,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.6185976266860962,
+      "learning_rate": 0.0005427004856988667,
+      "loss": 3.8528,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.5555748343467712,
-      "learning_rate": 0.0005424954207520741,
-      "loss": 3.8516,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.5853222608566284,
+      "learning_rate": 0.0005423766864543982,
+      "loss": 3.8428,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.34156469502601206,
-      "eval_loss": 3.775557279586792,
-      "eval_runtime": 184.197,
-      "eval_samples_per_second": 97.781,
-      "eval_steps_per_second": 6.113,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.34173299834923665,
+      "eval_loss": 3.77936053276062,
+      "eval_runtime": 183.7681,
+      "eval_samples_per_second": 98.009,
+      "eval_steps_per_second": 6.127,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.6176546812057495,
-      "learning_rate": 0.0005421721797220127,
-      "loss": 3.8444,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.5816961526870728,
+      "learning_rate": 0.0005420528872099298,
+      "loss": 3.8353,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5457547307014465,
-      "learning_rate": 0.0005418489386919513,
-      "loss": 3.8305,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.5886768102645874,
+      "learning_rate": 0.0005417290879654613,
+      "loss": 3.8505,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.5831869840621948,
-      "learning_rate": 0.0005415256976618898,
-      "loss": 3.8196,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.5620021224021912,
+      "learning_rate": 0.000541405288720993,
+      "loss": 3.8231,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.5737512111663818,
-      "learning_rate": 0.0005412024566318284,
-      "loss": 3.8124,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5953831672668457,
+      "learning_rate": 0.0005410814894765245,
+      "loss": 3.8198,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.5983269810676575,
-      "learning_rate": 0.000540879215601767,
-      "loss": 3.8368,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.6049351096153259,
+      "learning_rate": 0.0005407576902320561,
+      "loss": 3.8327,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.5368288159370422,
-      "learning_rate": 0.0005405559745717056,
-      "loss": 3.8048,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5849935412406921,
+      "learning_rate": 0.0005404338909875876,
+      "loss": 3.8049,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.5906519889831543,
-      "learning_rate": 0.0005402327335416442,
-      "loss": 3.7763,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.6319761276245117,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.772,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.6060296893119812,
-      "learning_rate": 0.0005399094925115827,
-      "loss": 3.7644,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.599040687084198,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.7662,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.5626549124717712,
-      "learning_rate": 0.0005395862514815214,
-      "loss": 3.7658,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.6134844422340393,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.7678,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.521769642829895,
-      "learning_rate": 0.0005392630104514599,
-      "loss": 3.7667,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.6510705351829529,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.7646,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.5677669048309326,
-      "learning_rate": 0.0005389397694213986,
-      "loss": 3.7803,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.5537397861480713,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7639,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.5613917708396912,
-      "learning_rate": 0.0005386165283913371,
-      "loss": 3.7582,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.6508437395095825,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.753,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.5857552289962769,
-      "learning_rate": 0.0005382932873612756,
-      "loss": 3.7806,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.6333539485931396,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.7944,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 0.5657854080200195,
-      "learning_rate": 0.0005379700463312143,
-      "loss": 3.7687,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.6207062005996704,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.7775,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.580058217048645,
-      "learning_rate": 0.0005376468053011528,
-      "loss": 3.7603,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.5319339632987976,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.779,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.5268446207046509,
-      "learning_rate": 0.0005373235642710914,
-      "loss": 3.7591,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.586555540561676,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.7858,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.5513815879821777,
-      "learning_rate": 0.00053700032324103,
-      "loss": 3.7748,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.5690563321113586,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.7811,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.5946319699287415,
-      "learning_rate": 0.0005366770822109687,
-      "loss": 3.7816,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.5293869972229004,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7734,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.5772368311882019,
-      "learning_rate": 0.0005363538411809072,
-      "loss": 3.754,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.5877575278282166,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7691,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.6033820509910583,
-      "learning_rate": 0.0005360306001508457,
-      "loss": 3.7639,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.6360712051391602,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.759,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.345006709859087,
-      "eval_loss": 3.7440133094787598,
-      "eval_runtime": 184.605,
-      "eval_samples_per_second": 97.565,
-      "eval_steps_per_second": 6.1,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.34497118036477165,
+      "eval_loss": 3.746513843536377,
+      "eval_runtime": 183.499,
+      "eval_samples_per_second": 98.153,
+      "eval_steps_per_second": 6.136,
       "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.3611098611712e+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:a7e9ad18c4133cca362e8ca729cda30cbeb2c0db779f81e5371625847996c992
+oid sha256:c424126a3e03dc0994f6fc899ff4f1ff7015775f3423397f5c1c16e9ebf50480
 size 5304