Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bd8db750e953aace95f8a4dffaa6d2d88d4f6e7884c50e06721151e4ad5c7c36
+oid sha256:5c74a5cd0edd7f22e792c239f9ee2fc37997d24312aa4e6e425fba8fa6d318cb
 size 503128704

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0155faac0b124f0bd5a396ebf688ad0c78bf6435c38c71e298b1457a2e8300c4
+oid sha256:ac89e009d5f5adc2e6012e87bdf60e5cc1849e94356b2ad2b9fa43c94d8f13f5
 size 1006351290

checkpoint-10000/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c3d3adf43228da0c54191da5477ce45109713e09d997154dddfab883dc23db3b
+oid sha256:793dbdfe1dd819d2a2df7a014dacf916da5d2986ca0f4bc141c769a4b74f4323
 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.750166893005371,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_8397/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.7437524795532227,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_high_1000_8397/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.6844520568847656,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 2.05204176902771,
       "learning_rate": 0.0003,
-      "loss": 8.6253,
+      "loss": 8.6345,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 2.7981810569763184,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 3.2708382606506348,
       "learning_rate": 0.0006,
-      "loss": 6.9074,
+      "loss": 6.9055,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 2.5690064430236816,
-      "learning_rate": 0.0005996767589699385,
-      "loss": 6.4568,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 1.4124608039855957,
+      "learning_rate": 0.0005996762007555315,
+      "loss": 6.4518,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 1.3536978960037231,
-      "learning_rate": 0.0005993535179398771,
-      "loss": 6.228,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.6837894916534424,
+      "learning_rate": 0.000599352401511063,
+      "loss": 6.2297,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.5520764589309692,
-      "learning_rate": 0.0005990302769098158,
-      "loss": 6.0714,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 1.635530948638916,
+      "learning_rate": 0.0005990286022665946,
+      "loss": 6.0754,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 1.1894077062606812,
-      "learning_rate": 0.0005987070358797543,
-      "loss": 5.9155,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 2.2319276332855225,
+      "learning_rate": 0.0005987048030221263,
+      "loss": 5.9754,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 1.9455897808074951,
-      "learning_rate": 0.0005983837948496929,
-      "loss": 5.8681,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 0.9215444922447205,
+      "learning_rate": 0.0005983810037776578,
+      "loss": 5.8714,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 1.525094747543335,
-      "learning_rate": 0.0005980605538196314,
-      "loss": 5.8018,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 1.0773382186889648,
+      "learning_rate": 0.0005980572045331894,
+      "loss": 5.8189,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.710013747215271,
-      "learning_rate": 0.0005977373127895701,
-      "loss": 5.7214,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 1.5438169240951538,
+      "learning_rate": 0.0005977334052887209,
+      "loss": 5.7462,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 2.247425079345703,
-      "learning_rate": 0.0005974140717595086,
-      "loss": 5.6246,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 1.2330684661865234,
+      "learning_rate": 0.0005974096060442526,
+      "loss": 5.6427,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 1.3971604108810425,
-      "learning_rate": 0.0005970908307294472,
-      "loss": 5.5482,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 1.0057889223098755,
+      "learning_rate": 0.0005970858067997841,
+      "loss": 5.561,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.3233646154403687,
-      "learning_rate": 0.0005967675896993858,
-      "loss": 5.4968,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.1920849084854126,
+      "learning_rate": 0.0005967620075553157,
+      "loss": 5.485,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 2.039377212524414,
-      "learning_rate": 0.0005964443486693243,
-      "loss": 5.4117,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 1.1211256980895996,
+      "learning_rate": 0.0005964382083108472,
+      "loss": 5.423,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 1.130034327507019,
-      "learning_rate": 0.000596121107639263,
-      "loss": 5.3297,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 1.492863416671753,
+      "learning_rate": 0.0005961144090663788,
+      "loss": 5.373,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 1.8250740766525269,
-      "learning_rate": 0.0005957978666092015,
-      "loss": 5.2953,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.3866064548492432,
+      "learning_rate": 0.0005957906098219104,
+      "loss": 5.3021,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 0.9127852320671082,
-      "learning_rate": 0.0005954746255791401,
-      "loss": 5.2404,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 1.4208452701568604,
+      "learning_rate": 0.0005954668105774419,
+      "loss": 5.2585,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.2714794874191284,
-      "learning_rate": 0.0005951513845490787,
-      "loss": 5.209,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 1.7031372785568237,
+      "learning_rate": 0.0005951430113329735,
+      "loss": 5.2191,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.1769967079162598,
-      "learning_rate": 0.0005948281435190174,
-      "loss": 5.1444,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 1.3381309509277344,
+      "learning_rate": 0.0005948192120885051,
+      "loss": 5.1628,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 1.3081032037734985,
-      "learning_rate": 0.0005945049024889559,
-      "loss": 5.127,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 0.9775306582450867,
+      "learning_rate": 0.0005944954128440366,
+      "loss": 5.1378,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.2607852220535278,
-      "learning_rate": 0.0005941816614588944,
-      "loss": 5.081,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 0.9237972497940063,
+      "learning_rate": 0.0005941716135995682,
+      "loss": 5.0938,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.22925976634631087,
-      "eval_loss": 5.002185344696045,
-      "eval_runtime": 186.3494,
-      "eval_samples_per_second": 96.652,
-      "eval_steps_per_second": 6.042,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22761812986432187,
+      "eval_loss": 5.02394962310791,
+      "eval_runtime": 186.5051,
+      "eval_samples_per_second": 96.571,
+      "eval_steps_per_second": 6.037,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 1.307389259338379,
-      "learning_rate": 0.000593858420428833,
-      "loss": 5.042,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 1.2229199409484863,
+      "learning_rate": 0.0005938478143550997,
+      "loss": 5.0464,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 1.1008410453796387,
-      "learning_rate": 0.0005935351793987716,
-      "loss": 4.9912,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.1812303066253662,
+      "learning_rate": 0.0005935240151106314,
+      "loss": 5.0196,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 1.0474328994750977,
-      "learning_rate": 0.0005932119383687103,
-      "loss": 4.9593,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 0.9059402942657471,
+      "learning_rate": 0.0005932002158661629,
+      "loss": 4.9889,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 1.0157822370529175,
-      "learning_rate": 0.0005928886973386488,
-      "loss": 4.9448,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 1.0551512241363525,
+      "learning_rate": 0.0005928764166216945,
+      "loss": 4.9478,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 1.1797782182693481,
-      "learning_rate": 0.0005925654563085874,
-      "loss": 4.9042,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 0.8280609250068665,
+      "learning_rate": 0.000592552617377226,
+      "loss": 4.9016,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 0.9304313659667969,
-      "learning_rate": 0.000592242215278526,
-      "loss": 4.9019,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 0.9362272024154663,
+      "learning_rate": 0.0005922288181327577,
+      "loss": 4.8703,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 0.9265602827072144,
-      "learning_rate": 0.0005919189742484645,
-      "loss": 4.8623,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 0.9991732239723206,
+      "learning_rate": 0.0005919050188882893,
+      "loss": 4.8564,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 1.1393800973892212,
-      "learning_rate": 0.0005915957332184032,
-      "loss": 4.8209,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 0.897462785243988,
+      "learning_rate": 0.0005915812196438207,
+      "loss": 4.8445,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 0.7940781116485596,
-      "learning_rate": 0.0005912724921883417,
-      "loss": 4.8361,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 0.9689521789550781,
+      "learning_rate": 0.0005912574203993524,
+      "loss": 4.7958,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 0.8800358772277832,
-      "learning_rate": 0.0005909492511582803,
-      "loss": 4.7892,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 1.1575392484664917,
+      "learning_rate": 0.0005909336211548839,
+      "loss": 4.8112,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 1.0024454593658447,
-      "learning_rate": 0.0005906260101282189,
-      "loss": 4.7359,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 1.140886902809143,
+      "learning_rate": 0.0005906098219104155,
+      "loss": 4.7551,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 1.371899962425232,
-      "learning_rate": 0.0005903027690981575,
-      "loss": 4.7626,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 1.2990275621414185,
+      "learning_rate": 0.000590286022665947,
+      "loss": 4.7634,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 1.2752752304077148,
-      "learning_rate": 0.000589979528068096,
-      "loss": 4.7085,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 0.8967852592468262,
+      "learning_rate": 0.0005899622234214787,
+      "loss": 4.7092,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 0.8268405199050903,
-      "learning_rate": 0.0005896562870380347,
-      "loss": 4.704,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.9230934977531433,
+      "learning_rate": 0.0005896384241770102,
+      "loss": 4.7052,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 1.1426270008087158,
-      "learning_rate": 0.0005893330460079732,
-      "loss": 4.675,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 0.9519078731536865,
+      "learning_rate": 0.0005893146249325418,
+      "loss": 4.6664,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 0.826781690120697,
-      "learning_rate": 0.0005890098049779118,
-      "loss": 4.6422,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 1.0994844436645508,
+      "learning_rate": 0.0005889908256880733,
+      "loss": 4.6757,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 0.9465463757514954,
-      "learning_rate": 0.0005886865639478504,
-      "loss": 4.6417,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 1.0593208074569702,
+      "learning_rate": 0.0005886670264436049,
+      "loss": 4.6452,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 1.1180988550186157,
-      "learning_rate": 0.0005883633229177889,
-      "loss": 4.6105,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 0.8123473525047302,
+      "learning_rate": 0.0005883432271991365,
+      "loss": 4.6351,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 0.9736685752868652,
-      "learning_rate": 0.0005880400818877276,
-      "loss": 4.5893,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 0.8100857734680176,
+      "learning_rate": 0.0005880194279546681,
+      "loss": 4.6017,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 0.8578049540519714,
-      "learning_rate": 0.0005877168408576662,
-      "loss": 4.5663,
+      "epoch": 0.215633423180593,
+      "grad_norm": 0.9699437618255615,
+      "learning_rate": 0.0005876956287101996,
+      "loss": 4.5892,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.2720923645160964,
-      "eval_loss": 4.499114036560059,
-      "eval_runtime": 186.2853,
-      "eval_samples_per_second": 96.685,
-      "eval_steps_per_second": 6.044,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.2705247206507091,
+      "eval_loss": 4.5065999031066895,
+      "eval_runtime": 181.8723,
+      "eval_samples_per_second": 99.031,
+      "eval_steps_per_second": 6.191,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 0.9600580930709839,
-      "learning_rate": 0.0005873935998276048,
-      "loss": 4.5855,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.9187719821929932,
+      "learning_rate": 0.0005873718294657312,
+      "loss": 4.5658,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 0.9652285575866699,
-      "learning_rate": 0.0005870703587975433,
-      "loss": 4.5373,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 0.8715267777442932,
+      "learning_rate": 0.0005870480302212628,
+      "loss": 4.5514,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 0.984366774559021,
-      "learning_rate": 0.0005867471177674818,
-      "loss": 4.5194,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 1.0863162279129028,
+      "learning_rate": 0.0005867242309767943,
+      "loss": 4.5263,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 0.9824601411819458,
-      "learning_rate": 0.0005864238767374205,
-      "loss": 4.5149,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 1.0446217060089111,
+      "learning_rate": 0.0005864004317323259,
+      "loss": 4.5041,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 0.9533531665802002,
-      "learning_rate": 0.0005861006357073591,
-      "loss": 4.4764,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 0.8052617311477661,
+      "learning_rate": 0.0005860766324878575,
+      "loss": 4.4915,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 0.7990318536758423,
-      "learning_rate": 0.0005857773946772977,
-      "loss": 4.4778,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 0.8702593445777893,
+      "learning_rate": 0.000585752833243389,
+      "loss": 4.5015,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.8323132991790771,
-      "learning_rate": 0.0005854541536472362,
-      "loss": 4.4508,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.7712243795394897,
+      "learning_rate": 0.0005854290339989206,
+      "loss": 4.4548,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 0.7075672745704651,
-      "learning_rate": 0.0005851309126171749,
-      "loss": 4.4423,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 0.8878899812698364,
+      "learning_rate": 0.0005851052347544521,
+      "loss": 4.4596,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 0.7002449631690979,
-      "learning_rate": 0.0005848076715871134,
-      "loss": 4.4339,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 0.9322397708892822,
+      "learning_rate": 0.0005847814355099838,
+      "loss": 4.4375,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 0.9258910417556763,
-      "learning_rate": 0.000584484430557052,
-      "loss": 4.4174,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 1.0136586427688599,
+      "learning_rate": 0.0005844576362655154,
+      "loss": 4.4259,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.87129145860672,
-      "learning_rate": 0.0005841611895269906,
-      "loss": 4.392,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.6975424885749817,
+      "learning_rate": 0.0005841338370210469,
+      "loss": 4.4078,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 0.8697466254234314,
-      "learning_rate": 0.0005838379484969291,
-      "loss": 4.3902,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 0.7953416109085083,
+      "learning_rate": 0.0005838100377765785,
+      "loss": 4.3812,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.9642153382301331,
-      "learning_rate": 0.0005835147074668678,
-      "loss": 4.3637,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 0.8708193302154541,
+      "learning_rate": 0.0005834862385321101,
+      "loss": 4.391,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.8309049010276794,
-      "learning_rate": 0.0005831914664368063,
-      "loss": 4.3664,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 1.0261796712875366,
+      "learning_rate": 0.0005831624392876417,
+      "loss": 4.3931,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.7619785070419312,
-      "learning_rate": 0.0005828682254067449,
-      "loss": 4.3657,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.8319907784461975,
+      "learning_rate": 0.0005828386400431731,
+      "loss": 4.3539,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 0.8297452926635742,
-      "learning_rate": 0.0005825449843766835,
-      "loss": 4.3396,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.7017777562141418,
+      "learning_rate": 0.0005825148407987048,
+      "loss": 4.3239,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 0.8317713737487793,
-      "learning_rate": 0.0005822217433466221,
-      "loss": 4.3176,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.7562263011932373,
+      "learning_rate": 0.0005821910415542363,
+      "loss": 4.3339,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 0.6838580965995789,
-      "learning_rate": 0.0005818985023165607,
-      "loss": 4.3277,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.9018781185150146,
+      "learning_rate": 0.0005818672423097679,
+      "loss": 4.3281,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.8009113669395447,
-      "learning_rate": 0.0005815752612864992,
-      "loss": 4.3202,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.9005151987075806,
+      "learning_rate": 0.0005815434430652994,
+      "loss": 4.3118,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.9078367352485657,
-      "learning_rate": 0.0005812520202564378,
-      "loss": 4.3055,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.7664577960968018,
+      "learning_rate": 0.0005812196438208311,
+      "loss": 4.3097,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.29884085839692875,
-      "eval_loss": 4.233619689941406,
-      "eval_runtime": 186.2849,
-      "eval_samples_per_second": 96.685,
-      "eval_steps_per_second": 6.045,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.2979642468980415,
+      "eval_loss": 4.245328426361084,
+      "eval_runtime": 181.1547,
+      "eval_samples_per_second": 99.423,
+      "eval_steps_per_second": 6.216,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.7016154527664185,
-      "learning_rate": 0.0005809287792263764,
-      "loss": 4.3064,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 0.6678867340087891,
+      "learning_rate": 0.0005808958445763626,
+      "loss": 4.3007,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.6988784670829773,
-      "learning_rate": 0.0005806055381963151,
-      "loss": 4.2912,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.6981791257858276,
+      "learning_rate": 0.0005805720453318942,
+      "loss": 4.307,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.8818886876106262,
-      "learning_rate": 0.0005802822971662536,
-      "loss": 4.2921,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.7250864505767822,
+      "learning_rate": 0.0005802482460874257,
+      "loss": 4.2657,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.7029647827148438,
-      "learning_rate": 0.0005799590561361922,
-      "loss": 4.2781,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.6579429507255554,
+      "learning_rate": 0.0005799244468429573,
+      "loss": 4.2598,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 0.8836530447006226,
-      "learning_rate": 0.0005796358151061307,
-      "loss": 4.2607,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 0.8516393303871155,
+      "learning_rate": 0.0005796006475984889,
+      "loss": 4.2574,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.6623589396476746,
-      "learning_rate": 0.0005793125740760694,
-      "loss": 4.2624,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.7543065547943115,
+      "learning_rate": 0.0005792768483540205,
+      "loss": 4.2443,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.7654575705528259,
-      "learning_rate": 0.0005789893330460079,
-      "loss": 4.2521,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.6784524321556091,
+      "learning_rate": 0.000578953049109552,
+      "loss": 4.2587,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.714368462562561,
-      "learning_rate": 0.0005786660920159465,
-      "loss": 4.2421,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.7622239589691162,
+      "learning_rate": 0.0005786292498650836,
+      "loss": 4.2311,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.6406934857368469,
-      "learning_rate": 0.0005783428509858851,
-      "loss": 4.2245,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.6954922080039978,
+      "learning_rate": 0.0005783054506206152,
+      "loss": 4.2116,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.7343611717224121,
-      "learning_rate": 0.0005780196099558237,
-      "loss": 4.1957,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.7494664788246155,
+      "learning_rate": 0.0005779816513761467,
+      "loss": 4.2204,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.6550223231315613,
-      "learning_rate": 0.0005776963689257623,
-      "loss": 4.2137,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.6824067831039429,
+      "learning_rate": 0.0005776578521316782,
+      "loss": 4.2146,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.6633121371269226,
-      "learning_rate": 0.0005773731278957008,
-      "loss": 4.2116,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.694776713848114,
+      "learning_rate": 0.0005773340528872099,
+      "loss": 4.2154,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.7366774678230286,
-      "learning_rate": 0.0005770498868656394,
-      "loss": 4.1917,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 0.7665195465087891,
+      "learning_rate": 0.0005770102536427414,
+      "loss": 4.1868,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.6168131232261658,
-      "learning_rate": 0.000576726645835578,
-      "loss": 4.2027,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.7928882837295532,
+      "learning_rate": 0.000576686454398273,
+      "loss": 4.1913,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.6847689151763916,
-      "learning_rate": 0.0005764034048055167,
-      "loss": 4.2069,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.68477463722229,
+      "learning_rate": 0.0005763626551538045,
+      "loss": 4.1944,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.835010290145874,
-      "learning_rate": 0.0005760801637754552,
-      "loss": 4.1715,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.7075636982917786,
+      "learning_rate": 0.0005760388559093362,
+      "loss": 4.1857,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.6226333975791931,
-      "learning_rate": 0.0005757569227453937,
-      "loss": 4.1648,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.6924806237220764,
+      "learning_rate": 0.0005757150566648678,
+      "loss": 4.171,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.7558175921440125,
-      "learning_rate": 0.0005754336817153324,
-      "loss": 4.1649,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 0.7602837085723877,
+      "learning_rate": 0.0005753912574203993,
+      "loss": 4.1712,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 0.6441270709037781,
-      "learning_rate": 0.0005751104406852709,
-      "loss": 4.1689,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.6978543996810913,
+      "learning_rate": 0.0005750674581759309,
+      "loss": 4.156,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 0.7551788091659546,
-      "learning_rate": 0.0005747871996552096,
-      "loss": 4.1503,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.7027019262313843,
+      "learning_rate": 0.0005747436589314624,
+      "loss": 4.1596,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.3127350637319811,
-      "eval_loss": 4.089627742767334,
-      "eval_runtime": 186.522,
-      "eval_samples_per_second": 96.562,
-      "eval_steps_per_second": 6.037,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.3133712263901349,
+      "eval_loss": 4.08456563949585,
+      "eval_runtime": 180.9817,
+      "eval_samples_per_second": 99.518,
+      "eval_steps_per_second": 6.222,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.714053750038147,
-      "learning_rate": 0.0005744639586251481,
-      "loss": 4.1586,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.6210776567459106,
+      "learning_rate": 0.0005744198596869941,
+      "loss": 4.1477,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.78158038854599,
-      "learning_rate": 0.0005741407175950867,
-      "loss": 4.1577,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.7034844160079956,
+      "learning_rate": 0.0005740960604425255,
+      "loss": 4.1506,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.7412471771240234,
-      "learning_rate": 0.0005738174765650253,
-      "loss": 4.1318,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.6764259338378906,
+      "learning_rate": 0.0005737722611980572,
+      "loss": 4.1507,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.6652634739875793,
-      "learning_rate": 0.0005734942355349638,
-      "loss": 4.1742,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 0.6597635746002197,
+      "learning_rate": 0.0005734484619535887,
+      "loss": 4.1352,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.686331570148468,
-      "learning_rate": 0.0005731709945049025,
-      "loss": 4.134,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.6612351536750793,
+      "learning_rate": 0.0005731246627091203,
+      "loss": 4.1199,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.6156903505325317,
-      "learning_rate": 0.000572847753474841,
-      "loss": 4.1383,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.8773025274276733,
+      "learning_rate": 0.0005728008634646518,
+      "loss": 4.1169,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.6592285633087158,
-      "learning_rate": 0.0005725245124447796,
-      "loss": 4.118,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.5631827116012573,
+      "learning_rate": 0.0005724770642201835,
+      "loss": 4.1141,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.7174545526504517,
-      "learning_rate": 0.0005722012714147182,
-      "loss": 4.1234,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.6348029971122742,
+      "learning_rate": 0.000572153264975715,
+      "loss": 4.1142,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 0.7114279866218567,
-      "learning_rate": 0.0005718780303846568,
-      "loss": 4.1467,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.804291307926178,
+      "learning_rate": 0.0005718294657312466,
+      "loss": 4.0981,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.6440300941467285,
-      "learning_rate": 0.0005715547893545953,
-      "loss": 4.0963,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.6766175031661987,
+      "learning_rate": 0.0005715056664867781,
+      "loss": 4.1086,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.5948485732078552,
-      "learning_rate": 0.000571231548324534,
-      "loss": 4.0778,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.7049275040626526,
+      "learning_rate": 0.0005711818672423097,
+      "loss": 4.0973,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.6361920833587646,
-      "learning_rate": 0.0005709083072944725,
-      "loss": 4.0789,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.7659531831741333,
+      "learning_rate": 0.0005708580679978413,
+      "loss": 4.0796,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.7733972668647766,
-      "learning_rate": 0.0005705850662644111,
-      "loss": 4.0968,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.6955700516700745,
+      "learning_rate": 0.0005705342687533729,
+      "loss": 4.075,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 1.04631769657135,
-      "learning_rate": 0.0005702618252343497,
-      "loss": 4.088,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.7187655568122864,
+      "learning_rate": 0.0005702104695089044,
+      "loss": 4.091,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.6549464464187622,
-      "learning_rate": 0.0005699385842042882,
-      "loss": 4.0913,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.6598371863365173,
+      "learning_rate": 0.000569886670264436,
+      "loss": 4.0778,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.7663791179656982,
-      "learning_rate": 0.0005696153431742269,
-      "loss": 4.0813,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.7967696189880371,
+      "learning_rate": 0.0005695628710199675,
+      "loss": 4.0734,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.6914228796958923,
-      "learning_rate": 0.0005692921021441655,
-      "loss": 4.0521,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.6475524306297302,
+      "learning_rate": 0.0005692390717754991,
+      "loss": 4.0637,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.7587223649024963,
-      "learning_rate": 0.0005689688611141041,
-      "loss": 4.0619,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.6172209978103638,
+      "learning_rate": 0.0005689152725310306,
+      "loss": 4.0501,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.6404293775558472,
-      "learning_rate": 0.0005686456200840426,
-      "loss": 4.058,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.7210772037506104,
+      "learning_rate": 0.0005685914732865623,
+      "loss": 4.0673,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.6166594624519348,
-      "learning_rate": 0.0005683223790539811,
-      "loss": 4.0557,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.552850067615509,
+      "learning_rate": 0.0005682676740420939,
+      "loss": 4.0439,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.321851692936943,
-      "eval_loss": 3.9909260272979736,
-      "eval_runtime": 186.9401,
-      "eval_samples_per_second": 96.346,
-      "eval_steps_per_second": 6.023,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.3217686821306527,
+      "eval_loss": 3.9859416484832764,
+      "eval_runtime": 180.9638,
+      "eval_samples_per_second": 99.528,
+      "eval_steps_per_second": 6.222,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.6536020040512085,
-      "learning_rate": 0.0005679991380239198,
-      "loss": 4.0541,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.6001549959182739,
+      "learning_rate": 0.0005679438747976254,
+      "loss": 4.0537,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.6692348122596741,
-      "learning_rate": 0.0005676758969938584,
-      "loss": 4.0365,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.6376379132270813,
+      "learning_rate": 0.000567620075553157,
+      "loss": 4.044,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.6620268225669861,
-      "learning_rate": 0.000567352655963797,
-      "loss": 4.0695,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.7252948880195618,
+      "learning_rate": 0.0005672962763086886,
+      "loss": 4.0449,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.6851233243942261,
-      "learning_rate": 0.0005670294149337355,
-      "loss": 4.0485,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.666740894317627,
+      "learning_rate": 0.0005669724770642202,
+      "loss": 4.0641,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.6668241620063782,
-      "learning_rate": 0.0005667061739036742,
-      "loss": 4.0364,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.5618765950202942,
+      "learning_rate": 0.0005666486778197517,
+      "loss": 4.0408,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.6462230086326599,
-      "learning_rate": 0.0005663829328736127,
-      "loss": 4.0343,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.6255050301551819,
+      "learning_rate": 0.0005663248785752833,
+      "loss": 4.0277,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6770803928375244,
-      "learning_rate": 0.0005660596918435512,
-      "loss": 4.0289,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.7014107704162598,
+      "learning_rate": 0.0005660010793308148,
+      "loss": 4.0227,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.6875442862510681,
-      "learning_rate": 0.0005657364508134899,
-      "loss": 4.0597,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.6181480288505554,
+      "learning_rate": 0.0005656772800863465,
+      "loss": 4.0328,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.6067531108856201,
-      "learning_rate": 0.0005654132097834284,
-      "loss": 4.0212,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.5627080798149109,
+      "learning_rate": 0.0005653534808418779,
+      "loss": 4.0321,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.5920923352241516,
-      "learning_rate": 0.0005650899687533671,
-      "loss": 4.0218,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.7584900856018066,
+      "learning_rate": 0.0005650296815974096,
+      "loss": 4.0151,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.6050952076911926,
-      "learning_rate": 0.0005647667277233056,
-      "loss": 4.0055,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6583033204078674,
+      "learning_rate": 0.0005647058823529411,
+      "loss": 4.0035,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.6834766864776611,
-      "learning_rate": 0.0005644434866932442,
-      "loss": 4.005,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.5877638459205627,
+      "learning_rate": 0.0005643820831084727,
+      "loss": 4.0206,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.6527867317199707,
-      "learning_rate": 0.0005641202456631828,
-      "loss": 4.0084,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.715157687664032,
+      "learning_rate": 0.0005640582838640042,
+      "loss": 3.9998,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.6814901232719421,
-      "learning_rate": 0.0005637970046331214,
-      "loss": 3.9991,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.5580232739448547,
+      "learning_rate": 0.0005637344846195358,
+      "loss": 3.9973,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.6742534637451172,
-      "learning_rate": 0.00056347376360306,
-      "loss": 4.0083,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.605633020401001,
+      "learning_rate": 0.0005634106853750674,
+      "loss": 3.9747,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.5482516884803772,
-      "learning_rate": 0.0005631505225729985,
-      "loss": 4.0034,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.5568682551383972,
+      "learning_rate": 0.000563086886130599,
+      "loss": 3.9812,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.6242794990539551,
-      "learning_rate": 0.0005628272815429371,
-      "loss": 4.0115,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.6361333131790161,
+      "learning_rate": 0.0005627630868861305,
+      "loss": 3.9911,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.6518068909645081,
-      "learning_rate": 0.0005625040405128757,
-      "loss": 3.9988,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.7065743803977966,
+      "learning_rate": 0.0005624392876416621,
+      "loss": 3.9895,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.6518497467041016,
-      "learning_rate": 0.0005621807994828143,
-      "loss": 3.9782,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.6945791840553284,
+      "learning_rate": 0.0005621154883971937,
+      "loss": 3.9774,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.5246138572692871,
-      "learning_rate": 0.0005618575584527529,
-      "loss": 3.9995,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.6124924421310425,
+      "learning_rate": 0.0005617916891527253,
+      "loss": 3.9905,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.3279617880831903,
-      "eval_loss": 3.920464038848877,
-      "eval_runtime": 186.8661,
-      "eval_samples_per_second": 96.385,
-      "eval_steps_per_second": 6.026,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32887360311772945,
+      "eval_loss": 3.911766529083252,
+      "eval_runtime": 181.0508,
+      "eval_samples_per_second": 99.48,
+      "eval_steps_per_second": 6.219,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.6844974160194397,
-      "learning_rate": 0.0005615343174226915,
-      "loss": 3.9804,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.6067901253700256,
+      "learning_rate": 0.0005614678899082568,
+      "loss": 3.9623,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.594265341758728,
-      "learning_rate": 0.00056121107639263,
-      "loss": 3.9785,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.5927957892417908,
+      "learning_rate": 0.0005611440906637884,
+      "loss": 3.9946,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.6252731680870056,
-      "learning_rate": 0.0005608878353625687,
-      "loss": 3.9893,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.7454538345336914,
+      "learning_rate": 0.00056082029141932,
+      "loss": 3.9779,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.7016366720199585,
-      "learning_rate": 0.0005605645943325072,
-      "loss": 3.9647,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.6399766206741333,
+      "learning_rate": 0.0005604964921748515,
+      "loss": 3.9493,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.5966526865959167,
-      "learning_rate": 0.0005602413533024458,
-      "loss": 3.9618,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.6017284393310547,
+      "learning_rate": 0.000560172692930383,
+      "loss": 3.9692,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.6206498146057129,
-      "learning_rate": 0.0005599181122723844,
-      "loss": 3.9637,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.6948426365852356,
+      "learning_rate": 0.0005598488936859147,
+      "loss": 3.9722,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.7185539603233337,
-      "learning_rate": 0.000559594871242323,
-      "loss": 3.9644,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.5732848048210144,
+      "learning_rate": 0.0005595250944414463,
+      "loss": 3.9587,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.6993795037269592,
-      "learning_rate": 0.0005592716302122616,
-      "loss": 3.9493,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.645921528339386,
+      "learning_rate": 0.0005592012951969778,
+      "loss": 3.965,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.5701709985733032,
-      "learning_rate": 0.0005589483891822001,
-      "loss": 3.959,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.6667640805244446,
+      "learning_rate": 0.0005588839719373988,
+      "loss": 3.9617,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.6983979344367981,
-      "learning_rate": 0.0005586316129727399,
-      "loss": 3.9529,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6847909092903137,
+      "learning_rate": 0.0005585601726929303,
+      "loss": 3.9563,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.5747671127319336,
-      "learning_rate": 0.0005583083719426786,
-      "loss": 3.9294,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.6105998754501343,
+      "learning_rate": 0.0005582363734484619,
+      "loss": 3.955,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.663241446018219,
-      "learning_rate": 0.0005579851309126171,
-      "loss": 3.9599,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.6470130085945129,
+      "learning_rate": 0.0005579125742039935,
+      "loss": 3.9544,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.6497787237167358,
-      "learning_rate": 0.0005576618898825558,
-      "loss": 3.9514,
+      "epoch": 0.7169811320754716,
+      "grad_norm": 0.6885042786598206,
+      "learning_rate": 0.0005575887749595251,
+      "loss": 3.9664,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.5732184052467346,
-      "learning_rate": 0.0005573386488524943,
-      "loss": 3.9494,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.6374613642692566,
+      "learning_rate": 0.0005572649757150566,
+      "loss": 3.932,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6865854859352112,
-      "learning_rate": 0.0005570154078224328,
-      "loss": 3.938,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.57664954662323,
+      "learning_rate": 0.0005569411764705882,
+      "loss": 3.9552,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.59453946352005,
-      "learning_rate": 0.0005566921667923715,
-      "loss": 3.9561,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.5588705539703369,
+      "learning_rate": 0.0005566173772261198,
+      "loss": 3.9305,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.6086866855621338,
-      "learning_rate": 0.00055636892576231,
-      "loss": 3.9318,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.5841559171676636,
+      "learning_rate": 0.0005562935779816513,
+      "loss": 3.9317,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.6675519943237305,
-      "learning_rate": 0.0005560456847322487,
-      "loss": 3.9383,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.5759849548339844,
+      "learning_rate": 0.0005559697787371828,
+      "loss": 3.9313,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.5952894687652588,
-      "learning_rate": 0.0005557224437021872,
-      "loss": 3.9334,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.604414165019989,
+      "learning_rate": 0.0005556459794927145,
+      "loss": 3.934,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.6614987850189209,
-      "learning_rate": 0.0005553992026721258,
-      "loss": 3.9245,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.6301313638687134,
+      "learning_rate": 0.000555322180248246,
+      "loss": 3.9186,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.33301393006078805,
-      "eval_loss": 3.8641064167022705,
-      "eval_runtime": 187.1185,
-      "eval_samples_per_second": 96.255,
-      "eval_steps_per_second": 6.018,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.3338135066622148,
+      "eval_loss": 3.8569676876068115,
+      "eval_runtime": 181.0082,
+      "eval_samples_per_second": 99.504,
+      "eval_steps_per_second": 6.221,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.7127641439437866,
-      "learning_rate": 0.0005550759616420644,
-      "loss": 3.9228,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.7266548275947571,
+      "learning_rate": 0.0005549983810037776,
+      "loss": 3.9297,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.5868001580238342,
-      "learning_rate": 0.000554752720612003,
-      "loss": 3.9265,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 1.0155426263809204,
+      "learning_rate": 0.0005546745817593091,
+      "loss": 3.905,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.6019564270973206,
-      "learning_rate": 0.0005544294795819415,
-      "loss": 3.9187,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.6490169167518616,
+      "learning_rate": 0.0005543507825148408,
+      "loss": 3.9306,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.6453304290771484,
-      "learning_rate": 0.0005541062385518801,
-      "loss": 3.9091,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.632588267326355,
+      "learning_rate": 0.0005540269832703723,
+      "loss": 3.9036,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.6741933226585388,
-      "learning_rate": 0.0005537829975218188,
-      "loss": 3.9152,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.6417953968048096,
+      "learning_rate": 0.0005537031840259039,
+      "loss": 3.8946,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.6336858868598938,
-      "learning_rate": 0.0005534597564917573,
-      "loss": 3.9138,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.5551266670227051,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9065,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.616560697555542,
-      "learning_rate": 0.0005531365154616959,
-      "loss": 3.9238,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.6416473388671875,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9074,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.6457054018974304,
-      "learning_rate": 0.0005528132744316344,
-      "loss": 3.905,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.5541009902954102,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.9056,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.6403147578239441,
-      "learning_rate": 0.0005524900334015731,
-      "loss": 3.9104,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.6023843288421631,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.8815,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.5391979813575745,
-      "learning_rate": 0.0005521667923715117,
-      "loss": 3.9187,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.5644687414169312,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.8985,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.5619264841079712,
-      "learning_rate": 0.0005518435513414502,
-      "loss": 3.901,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.6028684973716736,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.8963,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.5828893184661865,
-      "learning_rate": 0.0005515203103113888,
-      "loss": 3.9057,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.594016969203949,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.8843,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.5648857951164246,
-      "learning_rate": 0.0005511970692813274,
-      "loss": 3.9157,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.523352861404419,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.8911,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.5570770502090454,
-      "learning_rate": 0.000550873828251266,
-      "loss": 3.9079,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.5669171214103699,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.887,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.5684058666229248,
-      "learning_rate": 0.0005505505872212045,
-      "loss": 3.903,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.6337490677833557,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.8891,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.616293728351593,
-      "learning_rate": 0.0005502273461911432,
-      "loss": 3.8889,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.647167444229126,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.8713,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.6040659546852112,
-      "learning_rate": 0.0005499041051610817,
-      "loss": 3.8847,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.5458323359489441,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8783,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6027542948722839,
-      "learning_rate": 0.0005495808641310204,
-      "loss": 3.8935,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.6074749231338501,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8739,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.5909377336502075,
-      "learning_rate": 0.0005492576231009589,
-      "loss": 3.8891,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.6446775794029236,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.903,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.561553418636322,
-      "learning_rate": 0.0005489343820708974,
-      "loss": 3.8868,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.6206774115562439,
+      "learning_rate": 0.0005488461953588775,
+      "loss": 3.8513,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.3372976788157617,
-      "eval_loss": 3.8179571628570557,
-      "eval_runtime": 187.0682,
-      "eval_samples_per_second": 96.28,
-      "eval_steps_per_second": 6.019,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.3382809874504828,
+      "eval_loss": 3.812793254852295,
+      "eval_runtime": 180.7789,
+      "eval_samples_per_second": 99.63,
+      "eval_steps_per_second": 6.229,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.723307192325592,
-      "learning_rate": 0.0005486111410408361,
-      "loss": 3.8613,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.6646156311035156,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.892,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.6052539348602295,
-      "learning_rate": 0.0005482879000107746,
-      "loss": 3.8715,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.719357430934906,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.8686,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.533970057964325,
-      "learning_rate": 0.0005479646589807133,
-      "loss": 3.8757,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5545136332511902,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.8735,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.6301079392433167,
-      "learning_rate": 0.0005476414179506518,
-      "loss": 3.8598,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.6142967939376831,
+      "learning_rate": 0.0005475509983810037,
+      "loss": 3.8676,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.564543604850769,
-      "learning_rate": 0.0005473181769205904,
-      "loss": 3.8967,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.5893920660018921,
+      "learning_rate": 0.0005472271991365352,
+      "loss": 3.864,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.5990196466445923,
-      "learning_rate": 0.000546994935890529,
-      "loss": 3.8545,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.534954309463501,
+      "learning_rate": 0.0005469033998920669,
+      "loss": 3.8838,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.5828101634979248,
-      "learning_rate": 0.0005466716948604677,
-      "loss": 3.8602,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.5580828785896301,
+      "learning_rate": 0.0005465796006475984,
+      "loss": 3.8537,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.582362949848175,
-      "learning_rate": 0.0005463484538304062,
-      "loss": 3.8579,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.598829984664917,
+      "learning_rate": 0.00054625580140313,
+      "loss": 3.8601,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.582258939743042,
-      "learning_rate": 0.0005460252128003447,
-      "loss": 3.8713,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.5943628549575806,
+      "learning_rate": 0.0005459320021586615,
+      "loss": 3.8536,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.6159566044807434,
-      "learning_rate": 0.0005457019717702833,
-      "loss": 3.8811,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.6090864539146423,
+      "learning_rate": 0.0005456082029141932,
+      "loss": 3.8608,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.5610488057136536,
-      "learning_rate": 0.0005453787307402219,
-      "loss": 3.8651,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.5677192211151123,
+      "learning_rate": 0.0005452844036697248,
+      "loss": 3.8552,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.620211124420166,
-      "learning_rate": 0.0005450554897101605,
-      "loss": 3.8796,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.5808280110359192,
+      "learning_rate": 0.0005449670804101457,
+      "loss": 3.8601,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.565957248210907,
-      "learning_rate": 0.0005447322486800991,
-      "loss": 3.8582,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.6455959677696228,
+      "learning_rate": 0.0005446432811656773,
+      "loss": 3.8387,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.838001012802124,
-      "learning_rate": 0.0005444090076500377,
-      "loss": 3.851,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.5884677171707153,
+      "learning_rate": 0.0005443194819212088,
+      "loss": 3.8465,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.587591826915741,
-      "learning_rate": 0.0005440857666199763,
-      "loss": 3.8579,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5414503812789917,
+      "learning_rate": 0.0005439956826767404,
+      "loss": 3.843,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.5734500288963318,
-      "learning_rate": 0.0005437625255899148,
-      "loss": 3.8511,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.4871487617492676,
+      "learning_rate": 0.000543671883432272,
+      "loss": 3.8446,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.5970901250839233,
-      "learning_rate": 0.0005434392845598534,
-      "loss": 3.857,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.568858802318573,
+      "learning_rate": 0.0005433480841878035,
+      "loss": 3.8443,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.5941752791404724,
-      "learning_rate": 0.000543116043529792,
-      "loss": 3.8406,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.5653600096702576,
+      "learning_rate": 0.000543024284943335,
+      "loss": 3.8336,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.6333903670310974,
-      "learning_rate": 0.0005427928024997306,
-      "loss": 3.8459,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.5542508363723755,
+      "learning_rate": 0.0005427004856988667,
+      "loss": 3.8267,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.5914638042449951,
-      "learning_rate": 0.0005424760262902704,
-      "loss": 3.8631,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.6457640528678894,
+      "learning_rate": 0.0005423766864543982,
+      "loss": 3.8522,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.34091190847602265,
-      "eval_loss": 3.7817656993865967,
-      "eval_runtime": 186.6777,
-      "eval_samples_per_second": 96.482,
-      "eval_steps_per_second": 6.032,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.34194128593517703,
+      "eval_loss": 3.775505781173706,
+      "eval_runtime": 181.162,
+      "eval_samples_per_second": 99.419,
+      "eval_steps_per_second": 6.215,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.5284610986709595,
-      "learning_rate": 0.000542152785260209,
-      "loss": 3.8596,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.6142376065254211,
+      "learning_rate": 0.0005420528872099298,
+      "loss": 3.8671,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5833525061607361,
-      "learning_rate": 0.0005418295442301476,
-      "loss": 3.8324,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.5045973062515259,
+      "learning_rate": 0.0005417290879654613,
+      "loss": 3.8223,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.5118765234947205,
-      "learning_rate": 0.0005415063032000861,
-      "loss": 3.815,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.5798270106315613,
+      "learning_rate": 0.000541405288720993,
+      "loss": 3.839,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.5589722990989685,
-      "learning_rate": 0.0005411830621700248,
-      "loss": 3.8491,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5274220108985901,
+      "learning_rate": 0.0005410814894765245,
+      "loss": 3.8256,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.5750828385353088,
-      "learning_rate": 0.0005408598211399633,
-      "loss": 3.8439,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.5438529849052429,
+      "learning_rate": 0.0005407576902320561,
+      "loss": 3.8413,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.558642566204071,
-      "learning_rate": 0.0005405365801099019,
-      "loss": 3.8156,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5718055963516235,
+      "learning_rate": 0.0005404338909875876,
+      "loss": 3.7926,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.611592710018158,
-      "learning_rate": 0.0005402133390798405,
-      "loss": 3.7884,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.5577304363250732,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.7564,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.5523668527603149,
-      "learning_rate": 0.0005398965628703803,
-      "loss": 3.7556,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.5996473431587219,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.7529,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.548882007598877,
-      "learning_rate": 0.0005395733218403189,
-      "loss": 3.7664,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.5260946750640869,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.7505,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.5738718509674072,
-      "learning_rate": 0.0005392500808102575,
-      "loss": 3.7784,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.5556830763816833,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.7543,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.5679486989974976,
-      "learning_rate": 0.000538926839780196,
-      "loss": 3.7616,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.5205579400062561,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7655,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.5814436674118042,
-      "learning_rate": 0.0005386035987501346,
-      "loss": 3.7849,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.5281539559364319,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.7635,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.5564053654670715,
-      "learning_rate": 0.0005382803577200732,
-      "loss": 3.7783,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.6079031229019165,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.7686,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 0.594978392124176,
-      "learning_rate": 0.0005379571166900119,
-      "loss": 3.7647,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.5303264856338501,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.7532,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.5973650217056274,
-      "learning_rate": 0.0005376338756599504,
-      "loss": 3.76,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.5694045424461365,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.752,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.5629571080207825,
-      "learning_rate": 0.0005373106346298889,
-      "loss": 3.7808,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.5944398045539856,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.7777,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.575493335723877,
-      "learning_rate": 0.0005369873935998276,
-      "loss": 3.7722,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.5400773286819458,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.7781,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.6311414241790771,
-      "learning_rate": 0.0005366641525697661,
-      "loss": 3.785,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.5470821857452393,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7695,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.6355499029159546,
-      "learning_rate": 0.0005363409115397048,
-      "loss": 3.7613,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.5295310020446777,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7809,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.6226446032524109,
-      "learning_rate": 0.0005360176705096433,
-      "loss": 3.7481,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.5389605760574341,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.7584,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.3436170394239398,
-      "eval_loss": 3.750166893005371,
-      "eval_runtime": 186.9397,
-      "eval_samples_per_second": 96.347,
-      "eval_steps_per_second": 6.023,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.3450310481059574,
+      "eval_loss": 3.7437524795532227,
+      "eval_runtime": 180.8256,
+      "eval_samples_per_second": 99.604,
+      "eval_steps_per_second": 6.227,
       "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.3611359903744e+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:f4230fa49919b7aea86f6140dc459013369e427bf495a44fe88a1583abc7937c
+oid sha256:5043f49bcfa85802f5886b9ee9d71c580e35305511f078e1e0af539b4d0fde60
 size 5304