Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ec302ecd369b52dd09d862f2670f9b9bb168926d0e86791c661a9e174fb7d29
+oid sha256:3b917b8a493fe9d01f84befccc355e9cbc738fe64162d3172bd4efaf02b0109f
 size 503128704

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac2e668ccd17c9cb1f3d3b51366a5454efdc9d1f08d19822da00f8645edde579
+oid sha256:035c3f5cc71309784445e555cae0331c395e13a058c7c598237d2e168e9a1433
 size 1006351290

checkpoint-10000/rng_state.pth

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

checkpoint-10000/scheduler.pt

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

checkpoint-10000/trainer_state.json

@@ -1,7 +1,7 @@
 {
-  "best_metric": 3.7484071254730225,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_8397/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.74597430229187,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_low_10_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.7851003408432007,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 3.1336257457733154,
       "learning_rate": 0.0003,
-      "loss": 8.477,
+      "loss": 8.6314,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 2.35661244392395,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 2.658242702484131,
       "learning_rate": 0.0006,
-      "loss": 6.8545,
+      "loss": 6.8671,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 0.9086544513702393,
-      "learning_rate": 0.0005996767589699385,
-      "loss": 6.4483,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 1.1952481269836426,
+      "learning_rate": 0.0005996762007555315,
+      "loss": 6.4603,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 1.912062406539917,
-      "learning_rate": 0.0005993535179398771,
-      "loss": 6.2283,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.203603982925415,
+      "learning_rate": 0.000599352401511063,
+      "loss": 6.2349,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.995934009552002,
-      "learning_rate": 0.0005990302769098158,
-      "loss": 6.0812,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 1.2833564281463623,
+      "learning_rate": 0.0005990286022665946,
+      "loss": 6.0801,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 1.1780064105987549,
-      "learning_rate": 0.0005987070358797543,
-      "loss": 5.9232,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 2.3096957206726074,
+      "learning_rate": 0.0005987048030221263,
+      "loss": 5.9778,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 2.9962923526763916,
-      "learning_rate": 0.0005983837948496929,
-      "loss": 5.8634,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 2.2097551822662354,
+      "learning_rate": 0.0005983810037776578,
+      "loss": 5.8676,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 1.3674250841140747,
-      "learning_rate": 0.0005980605538196314,
-      "loss": 5.7794,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 0.8685315847396851,
+      "learning_rate": 0.0005980572045331894,
+      "loss": 5.8243,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.066927433013916,
-      "learning_rate": 0.0005977373127895701,
-      "loss": 5.7021,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 1.4664156436920166,
+      "learning_rate": 0.0005977334052887209,
+      "loss": 5.7513,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 0.8705481886863708,
-      "learning_rate": 0.0005974140717595086,
-      "loss": 5.6249,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 1.3060191869735718,
+      "learning_rate": 0.0005974096060442526,
+      "loss": 5.652,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 1.392909288406372,
-      "learning_rate": 0.0005970908307294472,
-      "loss": 5.5428,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 1.5025136470794678,
+      "learning_rate": 0.0005970858067997841,
+      "loss": 5.5726,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.5083731412887573,
-      "learning_rate": 0.0005967675896993858,
-      "loss": 5.4861,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.3774222135543823,
+      "learning_rate": 0.0005967620075553157,
+      "loss": 5.4955,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 1.1489932537078857,
-      "learning_rate": 0.0005964443486693243,
-      "loss": 5.4241,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 1.5685064792633057,
+      "learning_rate": 0.0005964382083108472,
+      "loss": 5.432,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 1.363460659980774,
-      "learning_rate": 0.000596121107639263,
-      "loss": 5.3553,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 1.7662345170974731,
+      "learning_rate": 0.0005961144090663788,
+      "loss": 5.3811,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 1.1569222211837769,
-      "learning_rate": 0.0005957978666092015,
-      "loss": 5.3117,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.2064779996871948,
+      "learning_rate": 0.0005957906098219104,
+      "loss": 5.3048,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 1.1893969774246216,
-      "learning_rate": 0.0005954746255791401,
-      "loss": 5.2536,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 1.833495020866394,
+      "learning_rate": 0.0005954668105774419,
+      "loss": 5.2572,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.257501482963562,
-      "learning_rate": 0.0005951513845490787,
-      "loss": 5.19,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 1.3901647329330444,
+      "learning_rate": 0.0005951430113329735,
+      "loss": 5.2198,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.0504266023635864,
-      "learning_rate": 0.0005948281435190174,
-      "loss": 5.1544,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 1.3915237188339233,
+      "learning_rate": 0.0005948192120885051,
+      "loss": 5.1653,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 1.2635871171951294,
-      "learning_rate": 0.0005945049024889559,
-      "loss": 5.1161,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 1.2137936353683472,
+      "learning_rate": 0.0005944954128440366,
+      "loss": 5.1415,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.4006985425949097,
-      "learning_rate": 0.0005941816614588944,
-      "loss": 5.0692,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 1.2188574075698853,
+      "learning_rate": 0.0005941716135995682,
+      "loss": 5.0946,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.2266931678303572,
-      "eval_loss": 5.018672466278076,
-      "eval_runtime": 184.614,
-      "eval_samples_per_second": 97.56,
-      "eval_steps_per_second": 6.099,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22748676852295419,
+      "eval_loss": 5.0213775634765625,
+      "eval_runtime": 184.2134,
+      "eval_samples_per_second": 97.772,
+      "eval_steps_per_second": 6.112,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 0.9685474634170532,
-      "learning_rate": 0.000593858420428833,
-      "loss": 5.029,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 1.2033261060714722,
+      "learning_rate": 0.0005938478143550997,
+      "loss": 5.046,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 0.9787606596946716,
-      "learning_rate": 0.0005935351793987716,
-      "loss": 5.0003,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.0172090530395508,
+      "learning_rate": 0.0005935240151106314,
+      "loss": 5.0218,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 1.4381256103515625,
-      "learning_rate": 0.0005932119383687103,
-      "loss": 4.9923,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 0.8544518947601318,
+      "learning_rate": 0.0005932002158661629,
+      "loss": 4.9949,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 0.8574581146240234,
-      "learning_rate": 0.0005928886973386488,
-      "loss": 4.949,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 1.6217145919799805,
+      "learning_rate": 0.0005928764166216945,
+      "loss": 4.952,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 0.989743709564209,
-      "learning_rate": 0.0005925654563085874,
-      "loss": 4.9299,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 0.8311410546302795,
+      "learning_rate": 0.000592552617377226,
+      "loss": 4.906,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 0.7719516158103943,
-      "learning_rate": 0.000592242215278526,
-      "loss": 4.9024,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 0.9451544284820557,
+      "learning_rate": 0.0005922288181327577,
+      "loss": 4.8771,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 1.163643717765808,
-      "learning_rate": 0.0005919189742484645,
-      "loss": 4.8709,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 0.8622845411300659,
+      "learning_rate": 0.0005919050188882893,
+      "loss": 4.8598,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 1.3706718683242798,
-      "learning_rate": 0.0005915957332184032,
-      "loss": 4.8521,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 1.031618356704712,
+      "learning_rate": 0.0005915812196438207,
+      "loss": 4.8463,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 1.0607285499572754,
-      "learning_rate": 0.0005912724921883417,
-      "loss": 4.814,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 0.607681393623352,
+      "learning_rate": 0.0005912574203993524,
+      "loss": 4.7983,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 1.1078616380691528,
-      "learning_rate": 0.0005909492511582803,
-      "loss": 4.795,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 0.8897637724876404,
+      "learning_rate": 0.0005909336211548839,
+      "loss": 4.8123,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 1.0800732374191284,
-      "learning_rate": 0.0005906260101282189,
-      "loss": 4.7628,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 1.3211703300476074,
+      "learning_rate": 0.0005906098219104155,
+      "loss": 4.7648,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 0.9528633952140808,
-      "learning_rate": 0.0005903027690981575,
-      "loss": 4.7699,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 1.2380629777908325,
+      "learning_rate": 0.000590286022665947,
+      "loss": 4.7642,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 1.056601643562317,
-      "learning_rate": 0.000589979528068096,
-      "loss": 4.7457,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 1.0963958501815796,
+      "learning_rate": 0.0005899622234214787,
+      "loss": 4.7164,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 0.9697463512420654,
-      "learning_rate": 0.0005896562870380347,
-      "loss": 4.715,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.8847112655639648,
+      "learning_rate": 0.0005896384241770102,
+      "loss": 4.705,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 0.8505885601043701,
-      "learning_rate": 0.0005893330460079732,
-      "loss": 4.6724,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 1.0140421390533447,
+      "learning_rate": 0.0005893146249325418,
+      "loss": 4.6709,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 1.270241141319275,
-      "learning_rate": 0.0005890098049779118,
-      "loss": 4.6784,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 0.8086866736412048,
+      "learning_rate": 0.0005889908256880733,
+      "loss": 4.6769,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 0.8811225295066833,
-      "learning_rate": 0.0005886865639478504,
-      "loss": 4.6724,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 1.0084503889083862,
+      "learning_rate": 0.0005886670264436049,
+      "loss": 4.6488,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 1.407916784286499,
-      "learning_rate": 0.0005883633229177889,
-      "loss": 4.6474,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 0.9606472253799438,
+      "learning_rate": 0.0005883432271991365,
+      "loss": 4.6366,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 0.8455577492713928,
-      "learning_rate": 0.0005880400818877276,
-      "loss": 4.5873,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 0.9299365282058716,
+      "learning_rate": 0.0005880194279546681,
+      "loss": 4.6035,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 1.058401346206665,
-      "learning_rate": 0.0005877168408576662,
-      "loss": 4.5797,
+      "epoch": 0.215633423180593,
+      "grad_norm": 0.8814052939414978,
+      "learning_rate": 0.0005876956287101996,
+      "loss": 4.5902,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.2688791726647588,
-      "eval_loss": 4.5188069343566895,
-      "eval_runtime": 184.1498,
-      "eval_samples_per_second": 97.806,
-      "eval_steps_per_second": 6.115,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.2701532364272714,
+      "eval_loss": 4.513090133666992,
+      "eval_runtime": 183.4776,
+      "eval_samples_per_second": 98.165,
+      "eval_steps_per_second": 6.137,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 1.1513112783432007,
-      "learning_rate": 0.0005873935998276048,
-      "loss": 4.585,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.8556138277053833,
+      "learning_rate": 0.0005873718294657312,
+      "loss": 4.5682,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 0.8390672206878662,
-      "learning_rate": 0.0005870703587975433,
-      "loss": 4.563,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 0.7709503769874573,
+      "learning_rate": 0.0005870480302212628,
+      "loss": 4.5529,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 0.7547805309295654,
-      "learning_rate": 0.0005867471177674818,
-      "loss": 4.5412,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 0.9734002351760864,
+      "learning_rate": 0.0005867242309767943,
+      "loss": 4.5305,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 0.7424968481063843,
-      "learning_rate": 0.0005864238767374205,
-      "loss": 4.5178,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 0.9718648791313171,
+      "learning_rate": 0.0005864004317323259,
+      "loss": 4.5104,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 0.8238902688026428,
-      "learning_rate": 0.0005861006357073591,
-      "loss": 4.5121,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 0.8877135515213013,
+      "learning_rate": 0.0005860766324878575,
+      "loss": 4.5008,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 0.8803598284721375,
-      "learning_rate": 0.0005857773946772977,
-      "loss": 4.4803,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 1.0457501411437988,
+      "learning_rate": 0.000585752833243389,
+      "loss": 4.506,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.7315595746040344,
-      "learning_rate": 0.0005854541536472362,
-      "loss": 4.4587,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.9116196632385254,
+      "learning_rate": 0.0005854290339989206,
+      "loss": 4.4571,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 0.9177563786506653,
-      "learning_rate": 0.0005851309126171749,
-      "loss": 4.4501,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 0.8825926780700684,
+      "learning_rate": 0.0005851052347544521,
+      "loss": 4.4633,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 0.8450412750244141,
-      "learning_rate": 0.0005848076715871134,
-      "loss": 4.4449,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 1.026060700416565,
+      "learning_rate": 0.0005847814355099838,
+      "loss": 4.4411,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 1.0020413398742676,
-      "learning_rate": 0.000584484430557052,
-      "loss": 4.4178,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 0.8668956160545349,
+      "learning_rate": 0.0005844576362655154,
+      "loss": 4.4259,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.8538787364959717,
-      "learning_rate": 0.0005841611895269906,
-      "loss": 4.3898,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.722511351108551,
+      "learning_rate": 0.0005841338370210469,
+      "loss": 4.4106,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 1.2234236001968384,
-      "learning_rate": 0.0005838379484969291,
-      "loss": 4.4115,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 0.7992992997169495,
+      "learning_rate": 0.0005838100377765785,
+      "loss": 4.3837,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.876475989818573,
-      "learning_rate": 0.0005835147074668678,
-      "loss": 4.3804,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 0.9671723246574402,
+      "learning_rate": 0.0005834862385321101,
+      "loss": 4.3964,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.8736183047294617,
-      "learning_rate": 0.0005831914664368063,
-      "loss": 4.3792,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 0.732172966003418,
+      "learning_rate": 0.0005831624392876417,
+      "loss": 4.3955,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.7667163014411926,
-      "learning_rate": 0.0005828682254067449,
-      "loss": 4.3574,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.8559433221817017,
+      "learning_rate": 0.0005828386400431731,
+      "loss": 4.3577,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 0.8317923545837402,
-      "learning_rate": 0.0005825449843766835,
-      "loss": 4.3536,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.7236409783363342,
+      "learning_rate": 0.0005825148407987048,
+      "loss": 4.3305,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 0.9385647177696228,
-      "learning_rate": 0.0005822217433466221,
-      "loss": 4.3414,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.6667165160179138,
+      "learning_rate": 0.0005821910415542363,
+      "loss": 4.3355,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 1.0671179294586182,
-      "learning_rate": 0.0005818985023165607,
-      "loss": 4.3359,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.902946412563324,
+      "learning_rate": 0.0005818672423097679,
+      "loss": 4.3297,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.8247517943382263,
-      "learning_rate": 0.0005815752612864992,
-      "loss": 4.3098,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.9877454042434692,
+      "learning_rate": 0.0005815434430652994,
+      "loss": 4.3163,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.6964085102081299,
-      "learning_rate": 0.0005812520202564378,
-      "loss": 4.32,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.8606247305870056,
+      "learning_rate": 0.0005812196438208311,
+      "loss": 4.3128,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.2985042517504795,
-      "eval_loss": 4.2403564453125,
-      "eval_runtime": 184.0775,
-      "eval_samples_per_second": 97.845,
-      "eval_steps_per_second": 6.117,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.29751920466955345,
+      "eval_loss": 4.249802112579346,
+      "eval_runtime": 183.342,
+      "eval_samples_per_second": 98.237,
+      "eval_steps_per_second": 6.142,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.769616961479187,
-      "learning_rate": 0.0005809287792263764,
-      "loss": 4.3057,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 0.7559620141983032,
+      "learning_rate": 0.0005808958445763626,
+      "loss": 4.3053,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.793465256690979,
-      "learning_rate": 0.0005806055381963151,
-      "loss": 4.2695,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.6390863656997681,
+      "learning_rate": 0.0005805720453318942,
+      "loss": 4.3121,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.7056701183319092,
-      "learning_rate": 0.0005802822971662536,
-      "loss": 4.2748,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.6653624176979065,
+      "learning_rate": 0.0005802482460874257,
+      "loss": 4.2696,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.8943589329719543,
-      "learning_rate": 0.0005799590561361922,
-      "loss": 4.2823,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.8212940692901611,
+      "learning_rate": 0.0005799244468429573,
+      "loss": 4.2671,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 0.7527636289596558,
-      "learning_rate": 0.0005796358151061307,
-      "loss": 4.2694,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 0.8165138959884644,
+      "learning_rate": 0.0005796006475984889,
+      "loss": 4.2653,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.6932998299598694,
-      "learning_rate": 0.0005793125740760694,
-      "loss": 4.2636,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.6768609881401062,
+      "learning_rate": 0.0005792768483540205,
+      "loss": 4.2454,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.6939517259597778,
-      "learning_rate": 0.0005789893330460079,
-      "loss": 4.256,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.6882625818252563,
+      "learning_rate": 0.000578953049109552,
+      "loss": 4.2689,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.7336264252662659,
-      "learning_rate": 0.0005786660920159465,
-      "loss": 4.2421,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.6341257691383362,
+      "learning_rate": 0.0005786292498650836,
+      "loss": 4.239,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.7286413311958313,
-      "learning_rate": 0.0005783428509858851,
-      "loss": 4.2092,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.6768348813056946,
+      "learning_rate": 0.0005783054506206152,
+      "loss": 4.2121,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.7822768092155457,
-      "learning_rate": 0.0005780196099558237,
-      "loss": 4.2272,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.7023494243621826,
+      "learning_rate": 0.0005779816513761467,
+      "loss": 4.2231,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.7599762082099915,
-      "learning_rate": 0.0005776963689257623,
-      "loss": 4.201,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.8483933806419373,
+      "learning_rate": 0.0005776578521316782,
+      "loss": 4.222,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.798646092414856,
-      "learning_rate": 0.0005773731278957008,
-      "loss": 4.2062,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.5900453925132751,
+      "learning_rate": 0.0005773340528872099,
+      "loss": 4.2163,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.728539228439331,
-      "learning_rate": 0.0005770498868656394,
-      "loss": 4.186,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 0.7810307741165161,
+      "learning_rate": 0.0005770102536427414,
+      "loss": 4.1918,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.8191288113594055,
-      "learning_rate": 0.000576726645835578,
-      "loss": 4.1849,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.8469997048377991,
+      "learning_rate": 0.000576686454398273,
+      "loss": 4.197,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.6519265174865723,
-      "learning_rate": 0.0005764034048055167,
-      "loss": 4.1823,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.7127343416213989,
+      "learning_rate": 0.0005763626551538045,
+      "loss": 4.1987,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.9170729517936707,
-      "learning_rate": 0.0005760801637754552,
-      "loss": 4.1929,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.656541109085083,
+      "learning_rate": 0.0005760388559093362,
+      "loss": 4.1915,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.7417013049125671,
-      "learning_rate": 0.0005757569227453937,
-      "loss": 4.1778,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.7397179007530212,
+      "learning_rate": 0.0005757150566648678,
+      "loss": 4.1739,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.7433168888092041,
-      "learning_rate": 0.0005754336817153324,
-      "loss": 4.1817,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 0.8393912315368652,
+      "learning_rate": 0.0005753912574203993,
+      "loss": 4.1698,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 0.8834530711174011,
-      "learning_rate": 0.0005751104406852709,
-      "loss": 4.1561,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.6189626455307007,
+      "learning_rate": 0.0005750674581759309,
+      "loss": 4.1613,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 0.6912859082221985,
-      "learning_rate": 0.0005747871996552096,
-      "loss": 4.1791,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.7884703874588013,
+      "learning_rate": 0.0005747436589314624,
+      "loss": 4.1606,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.3112565152346017,
-      "eval_loss": 4.094341278076172,
-      "eval_runtime": 184.0591,
-      "eval_samples_per_second": 97.854,
-      "eval_steps_per_second": 6.118,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.31285447325568927,
+      "eval_loss": 4.08674430847168,
+      "eval_runtime": 183.3668,
+      "eval_samples_per_second": 98.224,
+      "eval_steps_per_second": 6.141,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.6573261618614197,
-      "learning_rate": 0.0005744639586251481,
-      "loss": 4.1698,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.6317925453186035,
+      "learning_rate": 0.0005744198596869941,
+      "loss": 4.1518,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.6453205347061157,
-      "learning_rate": 0.0005741407175950867,
-      "loss": 4.1569,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.6720824241638184,
+      "learning_rate": 0.0005740960604425255,
+      "loss": 4.1545,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.7522885799407959,
-      "learning_rate": 0.0005738174765650253,
-      "loss": 4.1499,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.6672471165657043,
+      "learning_rate": 0.0005737722611980572,
+      "loss": 4.156,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.7057735919952393,
-      "learning_rate": 0.0005734942355349638,
-      "loss": 4.1518,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 0.7288913130760193,
+      "learning_rate": 0.0005734484619535887,
+      "loss": 4.1415,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.9186234474182129,
-      "learning_rate": 0.0005731709945049025,
-      "loss": 4.1445,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.7252816557884216,
+      "learning_rate": 0.0005731246627091203,
+      "loss": 4.1244,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.664431631565094,
-      "learning_rate": 0.000572847753474841,
-      "loss": 4.1155,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.7662931680679321,
+      "learning_rate": 0.0005728008634646518,
+      "loss": 4.1203,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.8200898170471191,
-      "learning_rate": 0.0005725245124447796,
-      "loss": 4.1346,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.6179137825965881,
+      "learning_rate": 0.0005724770642201835,
+      "loss": 4.1194,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.7778466939926147,
-      "learning_rate": 0.0005722012714147182,
-      "loss": 4.1089,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.6079455614089966,
+      "learning_rate": 0.000572153264975715,
+      "loss": 4.1212,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 0.6436388492584229,
-      "learning_rate": 0.0005718780303846568,
-      "loss": 4.1102,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.7912569642066956,
+      "learning_rate": 0.0005718294657312466,
+      "loss": 4.1041,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.8299694061279297,
-      "learning_rate": 0.0005715547893545953,
-      "loss": 4.1175,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.7254657745361328,
+      "learning_rate": 0.0005715056664867781,
+      "loss": 4.1114,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.6865853071212769,
-      "learning_rate": 0.000571231548324534,
-      "loss": 4.1037,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.6330100297927856,
+      "learning_rate": 0.0005711818672423097,
+      "loss": 4.1031,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.6101646423339844,
-      "learning_rate": 0.0005709083072944725,
-      "loss": 4.1056,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.6744422316551208,
+      "learning_rate": 0.0005708580679978413,
+      "loss": 4.0872,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.6010715961456299,
-      "learning_rate": 0.0005705850662644111,
-      "loss": 4.0895,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.6740829944610596,
+      "learning_rate": 0.0005705342687533729,
+      "loss": 4.0814,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 0.7732595205307007,
-      "learning_rate": 0.0005702618252343497,
-      "loss": 4.122,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.6551686525344849,
+      "learning_rate": 0.0005702104695089044,
+      "loss": 4.0952,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.7084074020385742,
-      "learning_rate": 0.0005699385842042882,
-      "loss": 4.0893,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.6149501800537109,
+      "learning_rate": 0.000569886670264436,
+      "loss": 4.0841,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.8309442400932312,
-      "learning_rate": 0.0005696153431742269,
-      "loss": 4.0823,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.7598328590393066,
+      "learning_rate": 0.0005695628710199675,
+      "loss": 4.0795,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.6847308278083801,
-      "learning_rate": 0.0005692921021441655,
-      "loss": 4.0971,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.8017769455909729,
+      "learning_rate": 0.0005692390717754991,
+      "loss": 4.0672,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.6202206015586853,
-      "learning_rate": 0.0005689688611141041,
-      "loss": 4.0665,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.6483253240585327,
+      "learning_rate": 0.0005689152725310306,
+      "loss": 4.0518,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.743624746799469,
-      "learning_rate": 0.0005686456200840426,
-      "loss": 4.0618,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.5981343984603882,
+      "learning_rate": 0.0005685914732865623,
+      "loss": 4.0654,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.574941873550415,
-      "learning_rate": 0.0005683223790539811,
-      "loss": 4.0643,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.5546260476112366,
+      "learning_rate": 0.0005682676740420939,
+      "loss": 4.0498,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.32120618613043717,
-      "eval_loss": 3.9959421157836914,
-      "eval_runtime": 184.0345,
-      "eval_samples_per_second": 97.868,
-      "eval_steps_per_second": 6.118,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.3213393945708977,
+      "eval_loss": 3.9915289878845215,
+      "eval_runtime": 183.5232,
+      "eval_samples_per_second": 98.14,
+      "eval_steps_per_second": 6.135,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.6600099802017212,
-      "learning_rate": 0.0005679991380239198,
-      "loss": 4.0475,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.6202845573425293,
+      "learning_rate": 0.0005679438747976254,
+      "loss": 4.0588,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.5597075819969177,
-      "learning_rate": 0.0005676758969938584,
-      "loss": 4.0342,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.6187790036201477,
+      "learning_rate": 0.000567620075553157,
+      "loss": 4.0535,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.6935561895370483,
-      "learning_rate": 0.000567352655963797,
-      "loss": 4.0344,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.6761106848716736,
+      "learning_rate": 0.0005672962763086886,
+      "loss": 4.0486,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.6177208423614502,
-      "learning_rate": 0.0005670294149337355,
-      "loss": 4.0424,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.6477122902870178,
+      "learning_rate": 0.0005669724770642202,
+      "loss": 4.0688,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.6166762113571167,
-      "learning_rate": 0.0005667061739036742,
-      "loss": 4.0621,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.6381343603134155,
+      "learning_rate": 0.0005666486778197517,
+      "loss": 4.0425,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.6113119125366211,
-      "learning_rate": 0.0005663829328736127,
-      "loss": 4.0435,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.7205365896224976,
+      "learning_rate": 0.0005663248785752833,
+      "loss": 4.0339,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6283344626426697,
-      "learning_rate": 0.0005660596918435512,
-      "loss": 4.018,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.607728898525238,
+      "learning_rate": 0.0005660010793308148,
+      "loss": 4.0281,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.6031971573829651,
-      "learning_rate": 0.0005657364508134899,
-      "loss": 4.0429,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.6393865346908569,
+      "learning_rate": 0.0005656772800863465,
+      "loss": 4.0357,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.7106824517250061,
-      "learning_rate": 0.0005654132097834284,
-      "loss": 4.0108,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.5362489819526672,
+      "learning_rate": 0.0005653534808418779,
+      "loss": 4.0376,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.6639676094055176,
-      "learning_rate": 0.0005650899687533671,
-      "loss": 4.0068,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.7597047686576843,
+      "learning_rate": 0.0005650296815974096,
+      "loss": 4.0189,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.7186005115509033,
-      "learning_rate": 0.0005647667277233056,
-      "loss": 4.0193,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6043763160705566,
+      "learning_rate": 0.0005647058823529411,
+      "loss": 4.0072,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.6553114056587219,
-      "learning_rate": 0.0005644434866932442,
-      "loss": 4.0244,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.5756456255912781,
+      "learning_rate": 0.0005643820831084727,
+      "loss": 4.0261,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.6312011480331421,
-      "learning_rate": 0.0005641202456631828,
-      "loss": 4.0089,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.6020849347114563,
+      "learning_rate": 0.0005640582838640042,
+      "loss": 4.0036,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.5797494053840637,
-      "learning_rate": 0.0005637970046331214,
-      "loss": 4.013,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.5935539603233337,
+      "learning_rate": 0.0005637344846195358,
+      "loss": 4.0049,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.5721137523651123,
-      "learning_rate": 0.00056347376360306,
-      "loss": 3.9972,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.632419228553772,
+      "learning_rate": 0.0005634106853750674,
+      "loss": 3.9797,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.59420245885849,
-      "learning_rate": 0.0005631505225729985,
-      "loss": 4.0314,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.6064315438270569,
+      "learning_rate": 0.000563086886130599,
+      "loss": 3.987,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.6400176882743835,
-      "learning_rate": 0.0005628272815429371,
-      "loss": 3.9964,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.6663802862167358,
+      "learning_rate": 0.0005627630868861305,
+      "loss": 3.998,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.6309003829956055,
-      "learning_rate": 0.0005625040405128757,
-      "loss": 4.0015,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.7193619608879089,
+      "learning_rate": 0.0005624392876416621,
+      "loss": 3.9981,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.639087975025177,
-      "learning_rate": 0.0005621807994828143,
-      "loss": 3.974,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.6985192894935608,
+      "learning_rate": 0.0005621154883971937,
+      "loss": 3.9837,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.6096267700195312,
-      "learning_rate": 0.0005618575584527529,
-      "loss": 3.9902,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.603473424911499,
+      "learning_rate": 0.0005617916891527253,
+      "loss": 3.9961,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.3283240368111638,
-      "eval_loss": 3.9176361560821533,
-      "eval_runtime": 183.9745,
-      "eval_samples_per_second": 97.899,
-      "eval_steps_per_second": 6.12,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32797178414886924,
+      "eval_loss": 3.919186592102051,
+      "eval_runtime": 183.7761,
+      "eval_samples_per_second": 98.005,
+      "eval_steps_per_second": 6.127,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.6264605522155762,
-      "learning_rate": 0.0005615343174226915,
-      "loss": 3.9795,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.6135157942771912,
+      "learning_rate": 0.0005614678899082568,
+      "loss": 3.9665,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.5533555150032043,
-      "learning_rate": 0.00056121107639263,
-      "loss": 3.9717,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.6000591516494751,
+      "learning_rate": 0.0005611440906637884,
+      "loss": 4.0003,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.5940308570861816,
-      "learning_rate": 0.0005608878353625687,
-      "loss": 3.9721,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.680533230304718,
+      "learning_rate": 0.00056082029141932,
+      "loss": 3.9766,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.6304638385772705,
-      "learning_rate": 0.0005605645943325072,
-      "loss": 3.9876,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.5540286302566528,
+      "learning_rate": 0.0005604964921748515,
+      "loss": 3.9517,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.595482587814331,
-      "learning_rate": 0.0005602413533024458,
-      "loss": 3.9483,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.5996313691139221,
+      "learning_rate": 0.0005601791689152725,
+      "loss": 3.9725,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.6096006631851196,
-      "learning_rate": 0.0005599181122723844,
-      "loss": 3.9689,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.7453248500823975,
+      "learning_rate": 0.000559855369670804,
+      "loss": 3.9768,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.5732763409614563,
-      "learning_rate": 0.000559594871242323,
-      "loss": 3.9743,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.5699926614761353,
+      "learning_rate": 0.0005595315704263357,
+      "loss": 3.9628,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.6785142421722412,
-      "learning_rate": 0.0005592716302122616,
-      "loss": 3.9582,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.6436143517494202,
+      "learning_rate": 0.0005592077711818672,
+      "loss": 3.9691,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.5796311497688293,
-      "learning_rate": 0.0005589483891822001,
-      "loss": 3.9449,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.7009570002555847,
+      "learning_rate": 0.0005588839719373988,
+      "loss": 3.9677,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.6873115301132202,
-      "learning_rate": 0.0005586251481521387,
-      "loss": 3.9594,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6285167932510376,
+      "learning_rate": 0.0005585601726929303,
+      "loss": 3.9591,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.5725986957550049,
-      "learning_rate": 0.0005583019071220773,
-      "loss": 3.9353,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.6780668497085571,
+      "learning_rate": 0.0005582363734484619,
+      "loss": 3.9582,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.521824836730957,
-      "learning_rate": 0.000557978666092016,
-      "loss": 3.9524,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.6572884321212769,
+      "learning_rate": 0.0005579125742039935,
+      "loss": 3.9617,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.5745775103569031,
-      "learning_rate": 0.0005576554250619545,
-      "loss": 3.9594,
+      "epoch": 0.7169811320754716,
+      "grad_norm": 0.6639407873153687,
+      "learning_rate": 0.0005575887749595251,
+      "loss": 3.969,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.620051383972168,
-      "learning_rate": 0.000557332184031893,
-      "loss": 3.9509,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.6634917259216309,
+      "learning_rate": 0.0005572649757150566,
+      "loss": 3.9328,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6912485361099243,
-      "learning_rate": 0.0005570089430018317,
-      "loss": 3.9476,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.5997982621192932,
+      "learning_rate": 0.0005569411764705882,
+      "loss": 3.9606,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.6365768909454346,
-      "learning_rate": 0.0005566857019717702,
-      "loss": 3.9378,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.607100248336792,
+      "learning_rate": 0.0005566173772261198,
+      "loss": 3.9348,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.618370532989502,
-      "learning_rate": 0.0005563624609417089,
-      "loss": 3.942,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.6579897999763489,
+      "learning_rate": 0.0005562935779816513,
+      "loss": 3.9334,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.5419732332229614,
-      "learning_rate": 0.0005560392199116474,
-      "loss": 3.9099,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.5454487800598145,
+      "learning_rate": 0.0005559697787371828,
+      "loss": 3.9363,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.5781886577606201,
-      "learning_rate": 0.000555715978881586,
-      "loss": 3.9445,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.5797833800315857,
+      "learning_rate": 0.0005556459794927145,
+      "loss": 3.9399,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.6141578555107117,
-      "learning_rate": 0.0005553927378515246,
-      "loss": 3.942,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.6542974710464478,
+      "learning_rate": 0.000555322180248246,
+      "loss": 3.9223,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.3333785691522934,
-      "eval_loss": 3.861785411834717,
-      "eval_runtime": 184.1728,
-      "eval_samples_per_second": 97.794,
-      "eval_steps_per_second": 6.114,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.3333549914756377,
+      "eval_loss": 3.862924814224243,
+      "eval_runtime": 183.1959,
+      "eval_samples_per_second": 98.315,
+      "eval_steps_per_second": 6.146,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.6214739680290222,
-      "learning_rate": 0.0005550694968214631,
-      "loss": 3.9291,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.7430106401443481,
+      "learning_rate": 0.0005549983810037776,
+      "loss": 3.9343,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.6426515579223633,
-      "learning_rate": 0.0005547462557914018,
-      "loss": 3.9034,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 0.803980827331543,
+      "learning_rate": 0.0005546745817593091,
+      "loss": 3.9069,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.5256863236427307,
-      "learning_rate": 0.0005544230147613403,
-      "loss": 3.9213,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.6478641033172607,
+      "learning_rate": 0.0005543507825148408,
+      "loss": 3.9316,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.6033873558044434,
-      "learning_rate": 0.0005540997737312789,
-      "loss": 3.935,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.561348557472229,
+      "learning_rate": 0.0005540269832703723,
+      "loss": 3.9072,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.6033143997192383,
-      "learning_rate": 0.0005537829975218188,
-      "loss": 3.9269,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.5793742537498474,
+      "learning_rate": 0.0005537031840259039,
+      "loss": 3.8978,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.6376200318336487,
-      "learning_rate": 0.0005534597564917573,
-      "loss": 3.9195,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.5496376752853394,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9132,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.563334047794342,
-      "learning_rate": 0.0005531365154616959,
-      "loss": 3.9358,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.6028301119804382,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9107,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.6669442653656006,
-      "learning_rate": 0.0005528132744316344,
-      "loss": 3.8961,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.5068923830986023,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.9076,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.5889538526535034,
-      "learning_rate": 0.0005524900334015731,
-      "loss": 3.9166,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.5969982743263245,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.8866,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.5355585217475891,
-      "learning_rate": 0.0005521667923715117,
-      "loss": 3.9003,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.6282092332839966,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.9013,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.6089932918548584,
-      "learning_rate": 0.0005518435513414502,
-      "loss": 3.9116,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.525468111038208,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.897,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.5592775940895081,
-      "learning_rate": 0.0005515203103113888,
-      "loss": 3.9084,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.5574877262115479,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.8903,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.5718680024147034,
-      "learning_rate": 0.0005511970692813274,
-      "loss": 3.9068,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.5548422932624817,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.8969,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.5790997743606567,
-      "learning_rate": 0.000550873828251266,
-      "loss": 3.9004,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.5981481075286865,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.89,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.6181998252868652,
-      "learning_rate": 0.0005505505872212045,
-      "loss": 3.8947,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.6007484793663025,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.8924,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.5165712833404541,
-      "learning_rate": 0.0005502273461911432,
-      "loss": 3.8927,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.6717764139175415,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.875,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.5599433183670044,
-      "learning_rate": 0.0005499041051610817,
-      "loss": 3.8939,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.5835329294204712,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8799,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6335838437080383,
-      "learning_rate": 0.0005495808641310204,
-      "loss": 3.9013,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.6127477288246155,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8755,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.6030533909797668,
-      "learning_rate": 0.0005492576231009589,
-      "loss": 3.8842,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.6754341125488281,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.9054,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.5869680047035217,
-      "learning_rate": 0.0005489343820708974,
-      "loss": 3.8727,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.5952799916267395,
+      "learning_rate": 0.0005488461953588775,
+      "loss": 3.8538,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.33745522550309276,
-      "eval_loss": 3.8181793689727783,
-      "eval_runtime": 184.0506,
-      "eval_samples_per_second": 97.859,
-      "eval_steps_per_second": 6.118,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.33814625786959285,
+      "eval_loss": 3.8158950805664062,
+      "eval_runtime": 183.3754,
+      "eval_samples_per_second": 98.219,
+      "eval_steps_per_second": 6.14,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.613333523273468,
-      "learning_rate": 0.0005486111410408361,
-      "loss": 3.8842,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.6873990297317505,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.8946,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.5983001589775085,
-      "learning_rate": 0.0005482879000107746,
-      "loss": 3.8791,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.6469279527664185,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.8733,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.599665105342865,
-      "learning_rate": 0.0005479646589807133,
-      "loss": 3.8743,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5836634039878845,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.877,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.5956316590309143,
-      "learning_rate": 0.0005476414179506518,
-      "loss": 3.8628,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.6095578074455261,
+      "learning_rate": 0.0005475509983810037,
+      "loss": 3.8722,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.6256473064422607,
-      "learning_rate": 0.0005473181769205904,
-      "loss": 3.878,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.6191880106925964,
+      "learning_rate": 0.0005472271991365352,
+      "loss": 3.8682,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.5864694118499756,
-      "learning_rate": 0.000546994935890529,
-      "loss": 3.8907,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.5735878944396973,
+      "learning_rate": 0.0005469033998920669,
+      "loss": 3.8862,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.5615338683128357,
-      "learning_rate": 0.0005466716948604677,
-      "loss": 3.885,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.5448331236839294,
+      "learning_rate": 0.0005465860766324878,
+      "loss": 3.8566,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.6215041875839233,
-      "learning_rate": 0.0005463484538304062,
-      "loss": 3.8651,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.6453812122344971,
+      "learning_rate": 0.0005462622773880194,
+      "loss": 3.8632,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.6117379069328308,
-      "learning_rate": 0.0005460252128003447,
-      "loss": 3.8568,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.5335873365402222,
+      "learning_rate": 0.000545938478143551,
+      "loss": 3.8573,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.6314567923545837,
-      "learning_rate": 0.0005457019717702833,
-      "loss": 3.8692,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.6051322221755981,
+      "learning_rate": 0.0005456146788990825,
+      "loss": 3.8621,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.612455427646637,
-      "learning_rate": 0.0005453787307402219,
-      "loss": 3.8427,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.5870794057846069,
+      "learning_rate": 0.000545290879654614,
+      "loss": 3.8578,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.611771285533905,
-      "learning_rate": 0.0005450554897101605,
-      "loss": 3.8701,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.5650087594985962,
+      "learning_rate": 0.0005449670804101457,
+      "loss": 3.8659,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.5838083028793335,
-      "learning_rate": 0.0005447322486800991,
-      "loss": 3.8609,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.6055552363395691,
+      "learning_rate": 0.0005446432811656773,
+      "loss": 3.8431,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.6281047463417053,
-      "learning_rate": 0.0005444090076500377,
-      "loss": 3.8696,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.5563592314720154,
+      "learning_rate": 0.0005443194819212088,
+      "loss": 3.8494,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.5830518007278442,
-      "learning_rate": 0.0005440857666199763,
-      "loss": 3.8657,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5335939526557922,
+      "learning_rate": 0.0005439956826767404,
+      "loss": 3.8466,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.6259497404098511,
-      "learning_rate": 0.0005437625255899148,
-      "loss": 3.8393,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.4980167746543884,
+      "learning_rate": 0.000543671883432272,
+      "loss": 3.8477,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.5686277151107788,
-      "learning_rate": 0.0005434392845598534,
-      "loss": 3.8414,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.5666930079460144,
+      "learning_rate": 0.0005433480841878035,
+      "loss": 3.8461,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.6480411291122437,
-      "learning_rate": 0.000543116043529792,
-      "loss": 3.8488,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.5867534875869751,
+      "learning_rate": 0.000543024284943335,
+      "loss": 3.8358,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.5571646690368652,
-      "learning_rate": 0.0005427928024997306,
-      "loss": 3.856,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.5597561597824097,
+      "learning_rate": 0.0005427004856988667,
+      "loss": 3.8282,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.5905200839042664,
-      "learning_rate": 0.0005424695614696692,
-      "loss": 3.8671,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.6189829111099243,
+      "learning_rate": 0.0005423766864543982,
+      "loss": 3.855,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.3412404748087736,
-      "eval_loss": 3.778561592102051,
-      "eval_runtime": 183.7431,
-      "eval_samples_per_second": 98.023,
-      "eval_steps_per_second": 6.128,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.34114985830033634,
+      "eval_loss": 3.7791531085968018,
+      "eval_runtime": 183.546,
+      "eval_samples_per_second": 98.128,
+      "eval_steps_per_second": 6.135,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.5285218954086304,
-      "learning_rate": 0.0005421463204396078,
-      "loss": 3.8317,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.618819534778595,
+      "learning_rate": 0.0005420528872099298,
+      "loss": 3.8669,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5539981722831726,
-      "learning_rate": 0.0005418230794095463,
-      "loss": 3.8446,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.5420499444007874,
+      "learning_rate": 0.0005417290879654613,
+      "loss": 3.825,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.6069391965866089,
-      "learning_rate": 0.000541499838379485,
-      "loss": 3.8397,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.5963202118873596,
+      "learning_rate": 0.000541405288720993,
+      "loss": 3.8383,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.651209831237793,
-      "learning_rate": 0.0005411765973494235,
-      "loss": 3.8296,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5878642201423645,
+      "learning_rate": 0.0005410814894765245,
+      "loss": 3.8279,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.578993558883667,
-      "learning_rate": 0.0005408533563193621,
-      "loss": 3.8121,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.613294243812561,
+      "learning_rate": 0.0005407576902320561,
+      "loss": 3.8456,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.6293763518333435,
-      "learning_rate": 0.0005405301152893007,
-      "loss": 3.8359,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5522181391716003,
+      "learning_rate": 0.0005404338909875876,
+      "loss": 3.7951,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.6404456496238708,
-      "learning_rate": 0.0005402068742592392,
-      "loss": 3.7673,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.5585771799087524,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.76,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.6690871119499207,
-      "learning_rate": 0.0005398836332291779,
-      "loss": 3.7674,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.5968648791313171,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.7544,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.6160586476325989,
-      "learning_rate": 0.0005395603921991164,
-      "loss": 3.7658,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.540739893913269,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.7534,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.567518949508667,
-      "learning_rate": 0.0005392436159896562,
-      "loss": 3.7742,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.5469689965248108,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.7604,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.63124018907547,
-      "learning_rate": 0.0005389203749595948,
-      "loss": 3.7726,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.5681180953979492,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7678,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.6264467239379883,
-      "learning_rate": 0.0005385971339295334,
-      "loss": 3.7611,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.570727527141571,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.7674,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.575022280216217,
-      "learning_rate": 0.000538273892899472,
-      "loss": 3.7788,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.6408295631408691,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.772,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 0.6051644682884216,
-      "learning_rate": 0.0005379506518694106,
-      "loss": 3.773,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.5871164798736572,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.756,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.5552962422370911,
-      "learning_rate": 0.0005376274108393491,
-      "loss": 3.7698,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.565994381904602,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.7535,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.5538464188575745,
-      "learning_rate": 0.0005373041698092877,
-      "loss": 3.7835,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.545239269733429,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.7803,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.5775489807128906,
-      "learning_rate": 0.0005369809287792263,
-      "loss": 3.7586,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.5473268032073975,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.7813,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.5624511241912842,
-      "learning_rate": 0.000536657687749165,
-      "loss": 3.7688,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.5545540452003479,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7726,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.5958260297775269,
-      "learning_rate": 0.0005363344467191035,
-      "loss": 3.7579,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.5510271787643433,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7838,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.610848069190979,
-      "learning_rate": 0.000536011205689042,
-      "loss": 3.7672,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.5303595662117004,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.7611,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.3445822030263958,
-      "eval_loss": 3.7484071254730225,
-      "eval_runtime": 184.2126,
-      "eval_samples_per_second": 97.773,
-      "eval_steps_per_second": 6.113,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.3447603937624116,
+      "eval_loss": 3.74597430229187,
+      "eval_runtime": 183.5528,
+      "eval_samples_per_second": 98.124,
+      "eval_steps_per_second": 6.134,
       "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.3609792151552e+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:de806f958cda4fd28e35b926e99af957f1556e2a4665381fb905a2ed66ce40ee
+oid sha256:7b9d5b6c5c7663e985dfae6b703a61070de8231ae2c433ba9a41cd2404588dde
 size 5304