Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

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

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e0f101edaa3ad96a8bbd7ac9a13ce880d9d78035b4e5bbb6c072e47c8a51bf8d
+oid sha256:04e25b8abe13a08865f75dcb856f158a1b740be7181141fe1f8efbca53d42a30
 size 1006351290

checkpoint-10000/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a34276a8a29f0c48d7e41083b96ab749687f5287caf6c8c7c1fdb2d9f99139c8
+oid sha256:1abdd4c7c045dcb1f8a6404bddc83e4806c8d3e3173eb1017f9b082b3c161e85
 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.748054027557373,
-  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_495/checkpoint-10000",
-  "epoch": 1.0763104079216446,
+  "best_metric": 3.7513272762298584,
+  "best_model_checkpoint": "/scratch/cl5625/exceptions/models/100M_low_2000_495/checkpoint-10000",
+  "epoch": 1.0781671159029649,
   "eval_steps": 1000,
   "global_step": 10000,
   "is_hyper_param_search": false,
@@ -9,1498 +9,1498 @@
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.005381552039608223,
-      "grad_norm": 6.547796249389648,
+      "epoch": 0.005390835579514825,
+      "grad_norm": 1.1914114952087402,
       "learning_rate": 0.0003,
-      "loss": 8.6491,
+      "loss": 8.5161,
       "step": 50
     },
     {
-      "epoch": 0.010763104079216447,
-      "grad_norm": 5.357585906982422,
+      "epoch": 0.01078167115902965,
+      "grad_norm": 1.799468755722046,
       "learning_rate": 0.0006,
-      "loss": 6.9125,
+      "loss": 6.8979,
       "step": 100
     },
     {
-      "epoch": 0.01614465611882467,
-      "grad_norm": 2.4043188095092773,
-      "learning_rate": 0.0005996767589699385,
-      "loss": 6.5144,
+      "epoch": 0.016172506738544475,
+      "grad_norm": 2.2392101287841797,
+      "learning_rate": 0.0005996762007555315,
+      "loss": 6.4959,
       "step": 150
     },
     {
-      "epoch": 0.021526208158432893,
-      "grad_norm": 1.361128330230713,
-      "learning_rate": 0.0005993535179398771,
-      "loss": 6.2176,
+      "epoch": 0.0215633423180593,
+      "grad_norm": 1.4015636444091797,
+      "learning_rate": 0.000599352401511063,
+      "loss": 6.2756,
       "step": 200
     },
     {
-      "epoch": 0.026907760198041114,
-      "grad_norm": 1.5913020372390747,
-      "learning_rate": 0.0005990302769098158,
-      "loss": 6.0702,
+      "epoch": 0.026954177897574125,
+      "grad_norm": 1.116059422492981,
+      "learning_rate": 0.0005990286022665946,
+      "loss": 6.096,
       "step": 250
     },
     {
-      "epoch": 0.03228931223764934,
-      "grad_norm": 1.3549662828445435,
-      "learning_rate": 0.0005987070358797543,
-      "loss": 5.983,
+      "epoch": 0.03234501347708895,
+      "grad_norm": 0.8914016485214233,
+      "learning_rate": 0.0005987048030221263,
+      "loss": 5.9781,
       "step": 300
     },
     {
-      "epoch": 0.03767086427725756,
-      "grad_norm": 1.7182697057724,
-      "learning_rate": 0.0005983837948496929,
-      "loss": 5.9004,
+      "epoch": 0.03773584905660377,
+      "grad_norm": 0.8895103931427002,
+      "learning_rate": 0.0005983810037776578,
+      "loss": 5.8764,
       "step": 350
     },
     {
-      "epoch": 0.04305241631686579,
-      "grad_norm": 0.9793596863746643,
-      "learning_rate": 0.0005980605538196314,
-      "loss": 5.8284,
+      "epoch": 0.0431266846361186,
+      "grad_norm": 1.2583591938018799,
+      "learning_rate": 0.0005980572045331894,
+      "loss": 5.8176,
       "step": 400
     },
     {
-      "epoch": 0.048433968356474004,
-      "grad_norm": 1.1408363580703735,
-      "learning_rate": 0.0005977373127895701,
-      "loss": 5.7264,
+      "epoch": 0.04851752021563342,
+      "grad_norm": 1.4108319282531738,
+      "learning_rate": 0.0005977334052887209,
+      "loss": 5.7312,
       "step": 450
     },
     {
-      "epoch": 0.05381552039608223,
-      "grad_norm": 1.4277925491333008,
-      "learning_rate": 0.0005974140717595086,
-      "loss": 5.6398,
+      "epoch": 0.05390835579514825,
+      "grad_norm": 2.0755622386932373,
+      "learning_rate": 0.0005974096060442526,
+      "loss": 5.678,
       "step": 500
     },
     {
-      "epoch": 0.05919707243569045,
-      "grad_norm": 2.2396180629730225,
-      "learning_rate": 0.0005970908307294472,
-      "loss": 5.574,
+      "epoch": 0.05929919137466307,
+      "grad_norm": 1.6820226907730103,
+      "learning_rate": 0.0005970858067997841,
+      "loss": 5.5596,
       "step": 550
     },
     {
-      "epoch": 0.06457862447529868,
-      "grad_norm": 1.143998622894287,
-      "learning_rate": 0.0005967675896993858,
-      "loss": 5.5309,
+      "epoch": 0.0646900269541779,
+      "grad_norm": 1.341705322265625,
+      "learning_rate": 0.0005967620075553157,
+      "loss": 5.5285,
       "step": 600
     },
     {
-      "epoch": 0.0699601765149069,
-      "grad_norm": 1.3793718814849854,
-      "learning_rate": 0.0005964443486693243,
-      "loss": 5.4287,
+      "epoch": 0.07008086253369272,
+      "grad_norm": 1.4363597631454468,
+      "learning_rate": 0.0005964382083108472,
+      "loss": 5.4234,
       "step": 650
     },
     {
-      "epoch": 0.07534172855451512,
-      "grad_norm": 0.9985382556915283,
-      "learning_rate": 0.000596121107639263,
-      "loss": 5.3816,
+      "epoch": 0.07547169811320754,
+      "grad_norm": 1.237207055091858,
+      "learning_rate": 0.0005961144090663788,
+      "loss": 5.3992,
       "step": 700
     },
     {
-      "epoch": 0.08072328059412334,
-      "grad_norm": 1.0711445808410645,
-      "learning_rate": 0.0005957978666092015,
-      "loss": 5.3184,
+      "epoch": 0.08086253369272237,
+      "grad_norm": 1.1934465169906616,
+      "learning_rate": 0.0005957906098219104,
+      "loss": 5.3224,
       "step": 750
     },
     {
-      "epoch": 0.08610483263373157,
-      "grad_norm": 1.4245736598968506,
-      "learning_rate": 0.0005954746255791401,
-      "loss": 5.2668,
+      "epoch": 0.0862533692722372,
+      "grad_norm": 1.3680514097213745,
+      "learning_rate": 0.0005954668105774419,
+      "loss": 5.2688,
       "step": 800
     },
     {
-      "epoch": 0.09148638467333979,
-      "grad_norm": 1.3065918684005737,
-      "learning_rate": 0.0005951513845490787,
-      "loss": 5.2295,
+      "epoch": 0.09164420485175202,
+      "grad_norm": 0.9270910620689392,
+      "learning_rate": 0.0005951430113329735,
+      "loss": 5.2368,
       "step": 850
     },
     {
-      "epoch": 0.09686793671294801,
-      "grad_norm": 1.2478379011154175,
-      "learning_rate": 0.0005948281435190174,
-      "loss": 5.1634,
+      "epoch": 0.09703504043126684,
+      "grad_norm": 1.0416382551193237,
+      "learning_rate": 0.0005948192120885051,
+      "loss": 5.1523,
       "step": 900
     },
     {
-      "epoch": 0.10224948875255624,
-      "grad_norm": 0.9259421229362488,
-      "learning_rate": 0.0005945049024889559,
-      "loss": 5.1204,
+      "epoch": 0.10242587601078167,
+      "grad_norm": 1.4154733419418335,
+      "learning_rate": 0.0005944954128440366,
+      "loss": 5.1503,
       "step": 950
     },
     {
-      "epoch": 0.10763104079216446,
-      "grad_norm": 1.41254460811615,
-      "learning_rate": 0.0005941816614588944,
-      "loss": 5.0846,
+      "epoch": 0.1078167115902965,
+      "grad_norm": 0.8388579487800598,
+      "learning_rate": 0.0005941716135995682,
+      "loss": 5.0913,
       "step": 1000
     },
     {
-      "epoch": 0.10763104079216446,
-      "eval_accuracy": 0.22689776121811187,
-      "eval_loss": 5.023209095001221,
-      "eval_runtime": 183.3261,
-      "eval_samples_per_second": 98.246,
-      "eval_steps_per_second": 6.142,
+      "epoch": 0.1078167115902965,
+      "eval_accuracy": 0.22793463572652548,
+      "eval_loss": 5.018302917480469,
+      "eval_runtime": 184.0417,
+      "eval_samples_per_second": 97.864,
+      "eval_steps_per_second": 6.118,
       "step": 1000
     },
     {
-      "epoch": 0.11301259283177269,
-      "grad_norm": 1.1088929176330566,
-      "learning_rate": 0.000593858420428833,
-      "loss": 5.0586,
+      "epoch": 0.11320754716981132,
+      "grad_norm": 1.170982837677002,
+      "learning_rate": 0.0005938478143550997,
+      "loss": 5.0503,
       "step": 1050
     },
     {
-      "epoch": 0.1183941448713809,
-      "grad_norm": 1.2266565561294556,
-      "learning_rate": 0.0005935351793987716,
-      "loss": 5.0344,
+      "epoch": 0.11859838274932614,
+      "grad_norm": 1.429085373878479,
+      "learning_rate": 0.0005935240151106314,
+      "loss": 5.0139,
       "step": 1100
     },
     {
-      "epoch": 0.12377569691098914,
-      "grad_norm": 1.1834657192230225,
-      "learning_rate": 0.0005932119383687103,
-      "loss": 5.0128,
+      "epoch": 0.12398921832884097,
+      "grad_norm": 1.1020185947418213,
+      "learning_rate": 0.0005932002158661629,
+      "loss": 4.9911,
       "step": 1150
     },
     {
-      "epoch": 0.12915724895059735,
-      "grad_norm": 1.5358068943023682,
-      "learning_rate": 0.0005928886973386488,
-      "loss": 4.9492,
+      "epoch": 0.1293800539083558,
+      "grad_norm": 1.4084248542785645,
+      "learning_rate": 0.0005928764166216945,
+      "loss": 4.9719,
       "step": 1200
     },
     {
-      "epoch": 0.13453880099020557,
-      "grad_norm": 0.8692229390144348,
-      "learning_rate": 0.0005925654563085874,
-      "loss": 4.9173,
+      "epoch": 0.1347708894878706,
+      "grad_norm": 0.8585777282714844,
+      "learning_rate": 0.000592552617377226,
+      "loss": 4.9451,
       "step": 1250
     },
     {
-      "epoch": 0.1399203530298138,
-      "grad_norm": 0.8401411175727844,
-      "learning_rate": 0.000592242215278526,
-      "loss": 4.8852,
+      "epoch": 0.14016172506738545,
+      "grad_norm": 1.0741046667099,
+      "learning_rate": 0.0005922288181327577,
+      "loss": 4.8902,
       "step": 1300
     },
     {
-      "epoch": 0.14530190506942203,
-      "grad_norm": 1.0361953973770142,
-      "learning_rate": 0.0005919189742484645,
-      "loss": 4.8926,
+      "epoch": 0.14555256064690028,
+      "grad_norm": 1.410202145576477,
+      "learning_rate": 0.0005919050188882893,
+      "loss": 4.8716,
       "step": 1350
     },
     {
-      "epoch": 0.15068345710903025,
-      "grad_norm": 0.9142393469810486,
-      "learning_rate": 0.0005915957332184032,
-      "loss": 4.8369,
+      "epoch": 0.1509433962264151,
+      "grad_norm": 1.1008950471878052,
+      "learning_rate": 0.0005915812196438207,
+      "loss": 4.8494,
       "step": 1400
     },
     {
-      "epoch": 0.15606500914863847,
-      "grad_norm": 1.1336075067520142,
-      "learning_rate": 0.0005912724921883417,
-      "loss": 4.8169,
+      "epoch": 0.15633423180592992,
+      "grad_norm": 1.0464845895767212,
+      "learning_rate": 0.0005912574203993524,
+      "loss": 4.8259,
       "step": 1450
     },
     {
-      "epoch": 0.16144656118824668,
-      "grad_norm": 1.0692909955978394,
-      "learning_rate": 0.0005909492511582803,
-      "loss": 4.8005,
+      "epoch": 0.16172506738544473,
+      "grad_norm": 1.071110486984253,
+      "learning_rate": 0.0005909336211548839,
+      "loss": 4.812,
       "step": 1500
     },
     {
-      "epoch": 0.1668281132278549,
-      "grad_norm": 0.807711124420166,
-      "learning_rate": 0.0005906260101282189,
-      "loss": 4.7786,
+      "epoch": 0.16711590296495957,
+      "grad_norm": 0.9221274256706238,
+      "learning_rate": 0.0005906098219104155,
+      "loss": 4.7706,
       "step": 1550
     },
     {
-      "epoch": 0.17220966526746315,
-      "grad_norm": 1.2913539409637451,
-      "learning_rate": 0.0005903027690981575,
-      "loss": 4.7333,
+      "epoch": 0.1725067385444744,
+      "grad_norm": 0.9727822542190552,
+      "learning_rate": 0.000590286022665947,
+      "loss": 4.781,
       "step": 1600
     },
     {
-      "epoch": 0.17759121730707136,
-      "grad_norm": 1.1676956415176392,
-      "learning_rate": 0.000589979528068096,
-      "loss": 4.7357,
+      "epoch": 0.1778975741239892,
+      "grad_norm": 0.882673442363739,
+      "learning_rate": 0.0005899622234214787,
+      "loss": 4.7578,
       "step": 1650
     },
     {
-      "epoch": 0.18297276934667958,
-      "grad_norm": 0.9552448987960815,
-      "learning_rate": 0.0005896562870380347,
-      "loss": 4.7186,
+      "epoch": 0.18328840970350405,
+      "grad_norm": 0.9065280556678772,
+      "learning_rate": 0.0005896384241770102,
+      "loss": 4.6787,
       "step": 1700
     },
     {
-      "epoch": 0.1883543213862878,
-      "grad_norm": 0.9495003819465637,
-      "learning_rate": 0.0005893330460079732,
-      "loss": 4.7037,
+      "epoch": 0.18867924528301888,
+      "grad_norm": 0.9698596000671387,
+      "learning_rate": 0.0005893146249325418,
+      "loss": 4.6947,
       "step": 1750
     },
     {
-      "epoch": 0.19373587342589602,
-      "grad_norm": 0.7337987422943115,
-      "learning_rate": 0.0005890098049779118,
-      "loss": 4.6579,
+      "epoch": 0.1940700808625337,
+      "grad_norm": 0.9533225893974304,
+      "learning_rate": 0.0005889908256880733,
+      "loss": 4.6862,
       "step": 1800
     },
     {
-      "epoch": 0.19911742546550426,
-      "grad_norm": 1.1409964561462402,
-      "learning_rate": 0.0005886865639478504,
-      "loss": 4.6561,
+      "epoch": 0.19946091644204852,
+      "grad_norm": 0.9267098903656006,
+      "learning_rate": 0.0005886670264436049,
+      "loss": 4.6382,
       "step": 1850
     },
     {
-      "epoch": 0.20449897750511248,
-      "grad_norm": 0.9477244019508362,
-      "learning_rate": 0.0005883633229177889,
-      "loss": 4.6466,
+      "epoch": 0.20485175202156333,
+      "grad_norm": 0.9240582585334778,
+      "learning_rate": 0.0005883432271991365,
+      "loss": 4.6348,
       "step": 1900
     },
     {
-      "epoch": 0.2098805295447207,
-      "grad_norm": 0.7901616096496582,
-      "learning_rate": 0.0005880400818877276,
-      "loss": 4.6147,
+      "epoch": 0.21024258760107817,
+      "grad_norm": 0.9578183889389038,
+      "learning_rate": 0.0005880194279546681,
+      "loss": 4.5974,
       "step": 1950
     },
     {
-      "epoch": 0.2152620815843289,
-      "grad_norm": 0.9213060736656189,
-      "learning_rate": 0.0005877168408576662,
-      "loss": 4.5929,
+      "epoch": 0.215633423180593,
+      "grad_norm": 1.1578387022018433,
+      "learning_rate": 0.0005876956287101996,
+      "loss": 4.5778,
       "step": 2000
     },
     {
-      "epoch": 0.2152620815843289,
-      "eval_accuracy": 0.2689594671488537,
-      "eval_loss": 4.525809288024902,
-      "eval_runtime": 181.8026,
-      "eval_samples_per_second": 99.069,
-      "eval_steps_per_second": 6.194,
+      "epoch": 0.215633423180593,
+      "eval_accuracy": 0.2701308539323816,
+      "eval_loss": 4.515415668487549,
+      "eval_runtime": 183.5731,
+      "eval_samples_per_second": 98.114,
+      "eval_steps_per_second": 6.134,
       "step": 2000
     },
     {
-      "epoch": 0.22064363362393713,
-      "grad_norm": 0.9790673851966858,
-      "learning_rate": 0.0005873935998276048,
-      "loss": 4.5608,
+      "epoch": 0.2210242587601078,
+      "grad_norm": 0.877032458782196,
+      "learning_rate": 0.0005873718294657312,
+      "loss": 4.5901,
       "step": 2050
     },
     {
-      "epoch": 0.22602518566354537,
-      "grad_norm": 0.6953191161155701,
-      "learning_rate": 0.0005870703587975433,
-      "loss": 4.5888,
+      "epoch": 0.22641509433962265,
+      "grad_norm": 1.0118193626403809,
+      "learning_rate": 0.0005870480302212628,
+      "loss": 4.5545,
       "step": 2100
     },
     {
-      "epoch": 0.2314067377031536,
-      "grad_norm": 0.8427320718765259,
-      "learning_rate": 0.0005867471177674818,
-      "loss": 4.548,
+      "epoch": 0.23180592991913745,
+      "grad_norm": 1.136588454246521,
+      "learning_rate": 0.0005867242309767943,
+      "loss": 4.5007,
       "step": 2150
     },
     {
-      "epoch": 0.2367882897427618,
-      "grad_norm": 1.2038559913635254,
-      "learning_rate": 0.0005864238767374205,
-      "loss": 4.5136,
+      "epoch": 0.2371967654986523,
+      "grad_norm": 0.9212758541107178,
+      "learning_rate": 0.0005864004317323259,
+      "loss": 4.5444,
       "step": 2200
     },
     {
-      "epoch": 0.24216984178237003,
-      "grad_norm": 0.8505734205245972,
-      "learning_rate": 0.0005861006357073591,
-      "loss": 4.5178,
+      "epoch": 0.24258760107816713,
+      "grad_norm": 0.806296706199646,
+      "learning_rate": 0.0005860766324878575,
+      "loss": 4.4748,
       "step": 2250
     },
     {
-      "epoch": 0.24755139382197827,
-      "grad_norm": 0.7332435846328735,
-      "learning_rate": 0.0005857773946772977,
-      "loss": 4.5166,
+      "epoch": 0.24797843665768193,
+      "grad_norm": 1.0752389430999756,
+      "learning_rate": 0.000585752833243389,
+      "loss": 4.4568,
       "step": 2300
     },
     {
-      "epoch": 0.2529329458615865,
-      "grad_norm": 0.9289398193359375,
-      "learning_rate": 0.0005854541536472362,
-      "loss": 4.4692,
+      "epoch": 0.25336927223719674,
+      "grad_norm": 0.9796498417854309,
+      "learning_rate": 0.0005854290339989206,
+      "loss": 4.4524,
       "step": 2350
     },
     {
-      "epoch": 0.2583144979011947,
-      "grad_norm": 0.7615149617195129,
-      "learning_rate": 0.0005851309126171749,
-      "loss": 4.4647,
+      "epoch": 0.2587601078167116,
+      "grad_norm": 1.1429473161697388,
+      "learning_rate": 0.0005851052347544521,
+      "loss": 4.4477,
       "step": 2400
     },
     {
-      "epoch": 0.2636960499408029,
-      "grad_norm": 1.141018033027649,
-      "learning_rate": 0.0005848076715871134,
-      "loss": 4.4292,
+      "epoch": 0.2641509433962264,
+      "grad_norm": 0.8981454372406006,
+      "learning_rate": 0.0005847814355099838,
+      "loss": 4.4505,
       "step": 2450
     },
     {
-      "epoch": 0.26907760198041114,
-      "grad_norm": 0.8744039535522461,
-      "learning_rate": 0.000584484430557052,
-      "loss": 4.4525,
+      "epoch": 0.2695417789757412,
+      "grad_norm": 0.8447709679603577,
+      "learning_rate": 0.0005844576362655154,
+      "loss": 4.4434,
       "step": 2500
     },
     {
-      "epoch": 0.27445915402001936,
-      "grad_norm": 0.7360879778862,
-      "learning_rate": 0.0005841611895269906,
-      "loss": 4.4223,
+      "epoch": 0.2749326145552561,
+      "grad_norm": 0.8942362666130066,
+      "learning_rate": 0.0005841338370210469,
+      "loss": 4.4048,
       "step": 2550
     },
     {
-      "epoch": 0.2798407060596276,
-      "grad_norm": 0.6995299458503723,
-      "learning_rate": 0.0005838379484969291,
-      "loss": 4.4034,
+      "epoch": 0.2803234501347709,
+      "grad_norm": 0.9772809743881226,
+      "learning_rate": 0.0005838100377765785,
+      "loss": 4.3967,
       "step": 2600
     },
     {
-      "epoch": 0.2852222580992358,
-      "grad_norm": 0.8300929665565491,
-      "learning_rate": 0.0005835147074668678,
-      "loss": 4.4013,
+      "epoch": 0.2857142857142857,
+      "grad_norm": 1.3033642768859863,
+      "learning_rate": 0.0005834862385321101,
+      "loss": 4.3923,
       "step": 2650
     },
     {
-      "epoch": 0.29060381013884407,
-      "grad_norm": 0.8035259246826172,
-      "learning_rate": 0.0005831914664368063,
-      "loss": 4.3722,
+      "epoch": 0.29110512129380056,
+      "grad_norm": 0.8010821342468262,
+      "learning_rate": 0.0005831624392876417,
+      "loss": 4.3871,
       "step": 2700
     },
     {
-      "epoch": 0.2959853621784523,
-      "grad_norm": 0.8101586699485779,
-      "learning_rate": 0.0005828682254067449,
-      "loss": 4.3752,
+      "epoch": 0.29649595687331537,
+      "grad_norm": 0.8844598531723022,
+      "learning_rate": 0.0005828386400431731,
+      "loss": 4.3466,
       "step": 2750
     },
     {
-      "epoch": 0.3013669142180605,
-      "grad_norm": 0.9427844285964966,
-      "learning_rate": 0.0005825449843766835,
-      "loss": 4.3694,
+      "epoch": 0.3018867924528302,
+      "grad_norm": 0.7504172921180725,
+      "learning_rate": 0.0005825148407987048,
+      "loss": 4.3869,
       "step": 2800
     },
     {
-      "epoch": 0.3067484662576687,
-      "grad_norm": 0.7412756085395813,
-      "learning_rate": 0.0005822217433466221,
-      "loss": 4.3536,
+      "epoch": 0.30727762803234504,
+      "grad_norm": 0.9708079695701599,
+      "learning_rate": 0.0005821910415542363,
+      "loss": 4.3577,
       "step": 2850
     },
     {
-      "epoch": 0.31213001829727693,
-      "grad_norm": 0.7558133602142334,
-      "learning_rate": 0.0005818985023165607,
-      "loss": 4.3376,
+      "epoch": 0.31266846361185985,
+      "grad_norm": 0.7660669088363647,
+      "learning_rate": 0.0005818672423097679,
+      "loss": 4.3357,
       "step": 2900
     },
     {
-      "epoch": 0.31751157033688515,
-      "grad_norm": 0.8248081803321838,
-      "learning_rate": 0.0005815752612864992,
-      "loss": 4.3552,
+      "epoch": 0.31805929919137466,
+      "grad_norm": 0.7895600199699402,
+      "learning_rate": 0.0005815434430652994,
+      "loss": 4.3345,
       "step": 2950
     },
     {
-      "epoch": 0.32289312237649337,
-      "grad_norm": 0.7637624740600586,
-      "learning_rate": 0.0005812520202564378,
-      "loss": 4.3137,
+      "epoch": 0.32345013477088946,
+      "grad_norm": 0.6386252045631409,
+      "learning_rate": 0.0005812196438208311,
+      "loss": 4.3308,
       "step": 3000
     },
     {
-      "epoch": 0.32289312237649337,
-      "eval_accuracy": 0.29744379956541017,
-      "eval_loss": 4.242817401885986,
-      "eval_runtime": 181.6642,
-      "eval_samples_per_second": 99.144,
-      "eval_steps_per_second": 6.198,
+      "epoch": 0.32345013477088946,
+      "eval_accuracy": 0.29829140074325095,
+      "eval_loss": 4.236527919769287,
+      "eval_runtime": 183.658,
+      "eval_samples_per_second": 98.068,
+      "eval_steps_per_second": 6.131,
       "step": 3000
     },
     {
-      "epoch": 0.3282746744161016,
-      "grad_norm": 0.7341495156288147,
-      "learning_rate": 0.0005809287792263764,
-      "loss": 4.3128,
+      "epoch": 0.3288409703504043,
+      "grad_norm": 1.0621103048324585,
+      "learning_rate": 0.0005808958445763626,
+      "loss": 4.3108,
       "step": 3050
     },
     {
-      "epoch": 0.3336562264557098,
-      "grad_norm": 0.7144924998283386,
-      "learning_rate": 0.0005806055381963151,
-      "loss": 4.293,
+      "epoch": 0.33423180592991913,
+      "grad_norm": 0.7970667481422424,
+      "learning_rate": 0.0005805720453318942,
+      "loss": 4.3062,
       "step": 3100
     },
     {
-      "epoch": 0.3390377784953181,
-      "grad_norm": 0.7501051425933838,
-      "learning_rate": 0.0005802822971662536,
-      "loss": 4.2715,
+      "epoch": 0.33962264150943394,
+      "grad_norm": 0.6979541778564453,
+      "learning_rate": 0.0005802482460874257,
+      "loss": 4.2719,
       "step": 3150
     },
     {
-      "epoch": 0.3444193305349263,
-      "grad_norm": 0.6868528723716736,
-      "learning_rate": 0.0005799590561361922,
-      "loss": 4.2802,
+      "epoch": 0.3450134770889488,
+      "grad_norm": 0.8238575458526611,
+      "learning_rate": 0.0005799244468429573,
+      "loss": 4.277,
       "step": 3200
     },
     {
-      "epoch": 0.3498008825745345,
-      "grad_norm": 0.8597863912582397,
-      "learning_rate": 0.0005796358151061307,
-      "loss": 4.2787,
+      "epoch": 0.3504043126684636,
+      "grad_norm": 0.7516454458236694,
+      "learning_rate": 0.0005796006475984889,
+      "loss": 4.2512,
       "step": 3250
     },
     {
-      "epoch": 0.35518243461414273,
-      "grad_norm": 0.7811806201934814,
-      "learning_rate": 0.0005793125740760694,
-      "loss": 4.2557,
+      "epoch": 0.3557951482479784,
+      "grad_norm": 0.7323547601699829,
+      "learning_rate": 0.0005792768483540205,
+      "loss": 4.278,
       "step": 3300
     },
     {
-      "epoch": 0.36056398665375095,
-      "grad_norm": 0.641840398311615,
-      "learning_rate": 0.0005789893330460079,
-      "loss": 4.2406,
+      "epoch": 0.3611859838274933,
+      "grad_norm": 0.7122869491577148,
+      "learning_rate": 0.000578953049109552,
+      "loss": 4.2551,
       "step": 3350
     },
     {
-      "epoch": 0.36594553869335916,
-      "grad_norm": 0.860731840133667,
-      "learning_rate": 0.0005786660920159465,
-      "loss": 4.2436,
+      "epoch": 0.3665768194070081,
+      "grad_norm": 0.8707606196403503,
+      "learning_rate": 0.0005786292498650836,
+      "loss": 4.2444,
       "step": 3400
     },
     {
-      "epoch": 0.3713270907329674,
-      "grad_norm": 0.8146591186523438,
-      "learning_rate": 0.0005783428509858851,
-      "loss": 4.2515,
+      "epoch": 0.3719676549865229,
+      "grad_norm": 0.7068629860877991,
+      "learning_rate": 0.0005783054506206152,
+      "loss": 4.2353,
       "step": 3450
     },
     {
-      "epoch": 0.3767086427725756,
-      "grad_norm": 0.7766924500465393,
-      "learning_rate": 0.0005780196099558237,
-      "loss": 4.2245,
+      "epoch": 0.37735849056603776,
+      "grad_norm": 0.6897825002670288,
+      "learning_rate": 0.0005779816513761467,
+      "loss": 4.2143,
       "step": 3500
     },
     {
-      "epoch": 0.3820901948121838,
-      "grad_norm": 0.6820563077926636,
-      "learning_rate": 0.0005776963689257623,
-      "loss": 4.2161,
+      "epoch": 0.38274932614555257,
+      "grad_norm": 0.7283990383148193,
+      "learning_rate": 0.0005776578521316782,
+      "loss": 4.2169,
       "step": 3550
     },
     {
-      "epoch": 0.38747174685179203,
-      "grad_norm": 0.7251272797584534,
-      "learning_rate": 0.0005773731278957008,
-      "loss": 4.2113,
+      "epoch": 0.3881401617250674,
+      "grad_norm": 0.8579701781272888,
+      "learning_rate": 0.0005773340528872099,
+      "loss": 4.2211,
       "step": 3600
     },
     {
-      "epoch": 0.3928532988914003,
-      "grad_norm": 0.7628197073936462,
-      "learning_rate": 0.0005770498868656394,
-      "loss": 4.2133,
+      "epoch": 0.3935309973045822,
+      "grad_norm": 1.0900028944015503,
+      "learning_rate": 0.0005770102536427414,
+      "loss": 4.2298,
       "step": 3650
     },
     {
-      "epoch": 0.3982348509310085,
-      "grad_norm": 0.7888264656066895,
-      "learning_rate": 0.000576726645835578,
-      "loss": 4.1951,
+      "epoch": 0.39892183288409705,
+      "grad_norm": 0.8825750946998596,
+      "learning_rate": 0.000576686454398273,
+      "loss": 4.1957,
       "step": 3700
     },
     {
-      "epoch": 0.40361640297061674,
-      "grad_norm": 0.7610321640968323,
-      "learning_rate": 0.0005764034048055167,
-      "loss": 4.2038,
+      "epoch": 0.40431266846361186,
+      "grad_norm": 0.7539132833480835,
+      "learning_rate": 0.0005763626551538045,
+      "loss": 4.1938,
       "step": 3750
     },
     {
-      "epoch": 0.40899795501022496,
-      "grad_norm": 0.7846929430961609,
-      "learning_rate": 0.0005760801637754552,
-      "loss": 4.1858,
+      "epoch": 0.40970350404312667,
+      "grad_norm": 0.642201840877533,
+      "learning_rate": 0.0005760388559093362,
+      "loss": 4.2143,
       "step": 3800
     },
     {
-      "epoch": 0.4143795070498332,
-      "grad_norm": 0.7305687665939331,
-      "learning_rate": 0.0005757569227453937,
-      "loss": 4.1908,
+      "epoch": 0.41509433962264153,
+      "grad_norm": 0.8492645621299744,
+      "learning_rate": 0.0005757150566648678,
+      "loss": 4.1957,
       "step": 3850
     },
     {
-      "epoch": 0.4197610590894414,
-      "grad_norm": 0.8204297423362732,
-      "learning_rate": 0.0005754336817153324,
-      "loss": 4.1887,
+      "epoch": 0.42048517520215634,
+      "grad_norm": 1.0330305099487305,
+      "learning_rate": 0.0005753912574203993,
+      "loss": 4.1613,
       "step": 3900
     },
     {
-      "epoch": 0.4251426111290496,
-      "grad_norm": 1.0344388484954834,
-      "learning_rate": 0.0005751104406852709,
-      "loss": 4.1589,
+      "epoch": 0.42587601078167114,
+      "grad_norm": 0.640697717666626,
+      "learning_rate": 0.0005750674581759309,
+      "loss": 4.1614,
       "step": 3950
     },
     {
-      "epoch": 0.4305241631686578,
-      "grad_norm": 0.7527312636375427,
-      "learning_rate": 0.0005747871996552096,
-      "loss": 4.1807,
+      "epoch": 0.431266846361186,
+      "grad_norm": 0.8644952774047852,
+      "learning_rate": 0.0005747436589314624,
+      "loss": 4.1661,
       "step": 4000
     },
     {
-      "epoch": 0.4305241631686578,
-      "eval_accuracy": 0.31194559184912113,
-      "eval_loss": 4.095859050750732,
-      "eval_runtime": 181.6682,
-      "eval_samples_per_second": 99.142,
-      "eval_steps_per_second": 6.198,
+      "epoch": 0.431266846361186,
+      "eval_accuracy": 0.31251438971682993,
+      "eval_loss": 4.095582485198975,
+      "eval_runtime": 183.6127,
+      "eval_samples_per_second": 98.092,
+      "eval_steps_per_second": 6.132,
       "step": 4000
     },
     {
-      "epoch": 0.43590571520826604,
-      "grad_norm": 0.7800742387771606,
-      "learning_rate": 0.0005744639586251481,
-      "loss": 4.1596,
+      "epoch": 0.4366576819407008,
+      "grad_norm": 0.7124435901641846,
+      "learning_rate": 0.0005744198596869941,
+      "loss": 4.1653,
       "step": 4050
     },
     {
-      "epoch": 0.44128726724787426,
-      "grad_norm": 0.7501474618911743,
-      "learning_rate": 0.0005741407175950867,
-      "loss": 4.1408,
+      "epoch": 0.4420485175202156,
+      "grad_norm": 0.6204947233200073,
+      "learning_rate": 0.0005740960604425255,
+      "loss": 4.1683,
       "step": 4100
     },
     {
-      "epoch": 0.44666881928748253,
-      "grad_norm": 0.7201575040817261,
-      "learning_rate": 0.0005738174765650253,
-      "loss": 4.1604,
+      "epoch": 0.4474393530997305,
+      "grad_norm": 0.6831852793693542,
+      "learning_rate": 0.0005737722611980572,
+      "loss": 4.1514,
       "step": 4150
     },
     {
-      "epoch": 0.45205037132709075,
-      "grad_norm": 0.5777507424354553,
-      "learning_rate": 0.0005734942355349638,
-      "loss": 4.1428,
+      "epoch": 0.4528301886792453,
+      "grad_norm": 1.1595909595489502,
+      "learning_rate": 0.0005734484619535887,
+      "loss": 4.1467,
       "step": 4200
     },
     {
-      "epoch": 0.45743192336669897,
-      "grad_norm": 0.7474133372306824,
-      "learning_rate": 0.0005731709945049025,
-      "loss": 4.1435,
+      "epoch": 0.4582210242587601,
+      "grad_norm": 0.5590158104896545,
+      "learning_rate": 0.0005731246627091203,
+      "loss": 4.138,
       "step": 4250
     },
     {
-      "epoch": 0.4628134754063072,
-      "grad_norm": 0.665733814239502,
-      "learning_rate": 0.000572847753474841,
-      "loss": 4.1553,
+      "epoch": 0.4636118598382749,
+      "grad_norm": 0.6859606504440308,
+      "learning_rate": 0.0005728008634646518,
+      "loss": 4.1304,
       "step": 4300
     },
     {
-      "epoch": 0.4681950274459154,
-      "grad_norm": 0.7177342176437378,
-      "learning_rate": 0.0005725245124447796,
-      "loss": 4.1161,
+      "epoch": 0.46900269541778977,
+      "grad_norm": 0.7505568861961365,
+      "learning_rate": 0.0005724770642201835,
+      "loss": 4.1381,
       "step": 4350
     },
     {
-      "epoch": 0.4735765794855236,
-      "grad_norm": 0.7497685551643372,
-      "learning_rate": 0.0005722012714147182,
-      "loss": 4.1077,
+      "epoch": 0.4743935309973046,
+      "grad_norm": 0.687965452671051,
+      "learning_rate": 0.000572153264975715,
+      "loss": 4.1213,
       "step": 4400
     },
     {
-      "epoch": 0.47895813152513184,
-      "grad_norm": 0.7238962054252625,
-      "learning_rate": 0.0005718780303846568,
-      "loss": 4.0956,
+      "epoch": 0.4797843665768194,
+      "grad_norm": 0.7188898921012878,
+      "learning_rate": 0.0005718294657312466,
+      "loss": 4.1262,
       "step": 4450
     },
     {
-      "epoch": 0.48433968356474005,
-      "grad_norm": 0.754342794418335,
-      "learning_rate": 0.0005715547893545953,
-      "loss": 4.1189,
+      "epoch": 0.48517520215633425,
+      "grad_norm": 0.6583627462387085,
+      "learning_rate": 0.0005715056664867781,
+      "loss": 4.1096,
       "step": 4500
     },
     {
-      "epoch": 0.48972123560434827,
-      "grad_norm": 0.641211986541748,
-      "learning_rate": 0.000571231548324534,
-      "loss": 4.1009,
+      "epoch": 0.49056603773584906,
+      "grad_norm": 0.579518735408783,
+      "learning_rate": 0.0005711818672423097,
+      "loss": 4.0965,
       "step": 4550
     },
     {
-      "epoch": 0.49510278764395654,
-      "grad_norm": 0.902698278427124,
-      "learning_rate": 0.0005709083072944725,
-      "loss": 4.0739,
+      "epoch": 0.49595687331536387,
+      "grad_norm": 0.7784225940704346,
+      "learning_rate": 0.0005708580679978413,
+      "loss": 4.1068,
       "step": 4600
     },
     {
-      "epoch": 0.5004843396835648,
-      "grad_norm": 0.6036228537559509,
-      "learning_rate": 0.0005705850662644111,
-      "loss": 4.0997,
+      "epoch": 0.5013477088948787,
+      "grad_norm": 0.6951068639755249,
+      "learning_rate": 0.0005705342687533729,
+      "loss": 4.0859,
       "step": 4650
     },
     {
-      "epoch": 0.505865891723173,
-      "grad_norm": 0.6570672988891602,
-      "learning_rate": 0.0005702618252343497,
-      "loss": 4.0842,
+      "epoch": 0.5067385444743935,
+      "grad_norm": 0.7131955623626709,
+      "learning_rate": 0.0005702104695089044,
+      "loss": 4.0954,
       "step": 4700
     },
     {
-      "epoch": 0.5112474437627812,
-      "grad_norm": 0.733204185962677,
-      "learning_rate": 0.0005699385842042882,
-      "loss": 4.0988,
+      "epoch": 0.5121293800539084,
+      "grad_norm": 0.5950631499290466,
+      "learning_rate": 0.000569886670264436,
+      "loss": 4.0742,
       "step": 4750
     },
     {
-      "epoch": 0.5166289958023894,
-      "grad_norm": 0.6296361088752747,
-      "learning_rate": 0.0005696153431742269,
-      "loss": 4.0862,
+      "epoch": 0.5175202156334232,
+      "grad_norm": 0.6306304931640625,
+      "learning_rate": 0.0005695628710199675,
+      "loss": 4.0879,
       "step": 4800
     },
     {
-      "epoch": 0.5220105478419976,
-      "grad_norm": 0.6667340993881226,
-      "learning_rate": 0.0005692921021441655,
-      "loss": 4.0714,
+      "epoch": 0.522911051212938,
+      "grad_norm": 0.6062384843826294,
+      "learning_rate": 0.0005692390717754991,
+      "loss": 4.0816,
       "step": 4850
     },
     {
-      "epoch": 0.5273920998816058,
-      "grad_norm": 0.6166004538536072,
-      "learning_rate": 0.0005689688611141041,
-      "loss": 4.073,
+      "epoch": 0.5283018867924528,
+      "grad_norm": 0.6436238884925842,
+      "learning_rate": 0.0005689152725310306,
+      "loss": 4.0905,
       "step": 4900
     },
     {
-      "epoch": 0.5327736519212141,
-      "grad_norm": 0.6546381711959839,
-      "learning_rate": 0.0005686456200840426,
-      "loss": 4.0827,
+      "epoch": 0.5336927223719676,
+      "grad_norm": 0.6756132245063782,
+      "learning_rate": 0.0005685914732865623,
+      "loss": 4.0624,
       "step": 4950
     },
     {
-      "epoch": 0.5381552039608223,
-      "grad_norm": 0.6323681473731995,
-      "learning_rate": 0.0005683223790539811,
-      "loss": 4.0535,
+      "epoch": 0.5390835579514824,
+      "grad_norm": 0.6142243146896362,
+      "learning_rate": 0.0005682676740420939,
+      "loss": 4.0541,
       "step": 5000
     },
     {
-      "epoch": 0.5381552039608223,
-      "eval_accuracy": 0.3216463389789736,
-      "eval_loss": 3.9916508197784424,
-      "eval_runtime": 181.8071,
-      "eval_samples_per_second": 99.067,
-      "eval_steps_per_second": 6.193,
+      "epoch": 0.5390835579514824,
+      "eval_accuracy": 0.3212785489537216,
+      "eval_loss": 3.998600721359253,
+      "eval_runtime": 183.6277,
+      "eval_samples_per_second": 98.084,
+      "eval_steps_per_second": 6.132,
       "step": 5000
     },
     {
-      "epoch": 0.5435367560004305,
-      "grad_norm": 0.6683503985404968,
-      "learning_rate": 0.0005679991380239198,
-      "loss": 4.0387,
+      "epoch": 0.5444743935309974,
+      "grad_norm": 0.5935885310173035,
+      "learning_rate": 0.0005679438747976254,
+      "loss": 4.0488,
       "step": 5050
     },
     {
-      "epoch": 0.5489183080400387,
-      "grad_norm": 0.6250083446502686,
-      "learning_rate": 0.0005676758969938584,
-      "loss": 4.0595,
+      "epoch": 0.5498652291105122,
+      "grad_norm": 0.8128730654716492,
+      "learning_rate": 0.000567620075553157,
+      "loss": 4.0534,
       "step": 5100
     },
     {
-      "epoch": 0.5542998600796469,
-      "grad_norm": 0.6272826194763184,
-      "learning_rate": 0.000567352655963797,
-      "loss": 4.0677,
+      "epoch": 0.555256064690027,
+      "grad_norm": 0.6337621808052063,
+      "learning_rate": 0.0005672962763086886,
+      "loss": 4.0576,
       "step": 5150
     },
     {
-      "epoch": 0.5596814121192552,
-      "grad_norm": 0.6267666220664978,
-      "learning_rate": 0.0005670294149337355,
-      "loss": 4.0499,
+      "epoch": 0.5606469002695418,
+      "grad_norm": 0.6212897896766663,
+      "learning_rate": 0.0005669724770642202,
+      "loss": 4.0352,
       "step": 5200
     },
     {
-      "epoch": 0.5650629641588634,
-      "grad_norm": 0.6394859552383423,
-      "learning_rate": 0.0005667061739036742,
-      "loss": 4.0364,
+      "epoch": 0.5660377358490566,
+      "grad_norm": 0.6357078552246094,
+      "learning_rate": 0.0005666486778197517,
+      "loss": 4.0372,
       "step": 5250
     },
     {
-      "epoch": 0.5704445161984716,
-      "grad_norm": 0.6560896039009094,
-      "learning_rate": 0.0005663829328736127,
-      "loss": 4.0384,
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.6911345720291138,
+      "learning_rate": 0.0005663248785752833,
+      "loss": 4.0203,
       "step": 5300
     },
     {
-      "epoch": 0.5758260682380799,
-      "grad_norm": 0.6860178709030151,
-      "learning_rate": 0.0005660596918435512,
-      "loss": 4.0365,
+      "epoch": 0.5768194070080862,
+      "grad_norm": 0.6729469895362854,
+      "learning_rate": 0.0005660010793308148,
+      "loss": 4.0349,
       "step": 5350
     },
     {
-      "epoch": 0.5812076202776881,
-      "grad_norm": 0.6861039400100708,
-      "learning_rate": 0.0005657364508134899,
-      "loss": 4.033,
+      "epoch": 0.5822102425876011,
+      "grad_norm": 0.693757951259613,
+      "learning_rate": 0.0005656772800863465,
+      "loss": 4.0253,
       "step": 5400
     },
     {
-      "epoch": 0.5865891723172963,
-      "grad_norm": 0.6428167819976807,
-      "learning_rate": 0.0005654132097834284,
-      "loss": 4.0161,
+      "epoch": 0.5876010781671159,
+      "grad_norm": 0.707956075668335,
+      "learning_rate": 0.0005653534808418779,
+      "loss": 4.036,
       "step": 5450
     },
     {
-      "epoch": 0.5919707243569046,
-      "grad_norm": 0.6751614212989807,
-      "learning_rate": 0.0005650899687533671,
-      "loss": 4.0278,
+      "epoch": 0.5929919137466307,
+      "grad_norm": 0.6174004077911377,
+      "learning_rate": 0.0005650296815974096,
+      "loss": 4.0375,
       "step": 5500
     },
     {
-      "epoch": 0.5973522763965128,
-      "grad_norm": 0.6935444474220276,
-      "learning_rate": 0.0005647667277233056,
-      "loss": 4.0194,
+      "epoch": 0.5983827493261455,
+      "grad_norm": 0.6651358604431152,
+      "learning_rate": 0.0005647058823529411,
+      "loss": 4.0422,
       "step": 5550
     },
     {
-      "epoch": 0.602733828436121,
-      "grad_norm": 0.700750470161438,
-      "learning_rate": 0.0005644434866932442,
-      "loss": 4.028,
+      "epoch": 0.6037735849056604,
+      "grad_norm": 0.7006116509437561,
+      "learning_rate": 0.0005643820831084727,
+      "loss": 4.017,
       "step": 5600
     },
     {
-      "epoch": 0.6081153804757292,
-      "grad_norm": 0.641048789024353,
-      "learning_rate": 0.0005641202456631828,
-      "loss": 4.0257,
+      "epoch": 0.6091644204851752,
+      "grad_norm": 0.5895059704780579,
+      "learning_rate": 0.0005640582838640042,
+      "loss": 4.0083,
       "step": 5650
     },
     {
-      "epoch": 0.6134969325153374,
-      "grad_norm": 0.6311193108558655,
-      "learning_rate": 0.0005637970046331214,
-      "loss": 4.005,
+      "epoch": 0.6145552560646901,
+      "grad_norm": 0.6189951300621033,
+      "learning_rate": 0.0005637344846195358,
+      "loss": 4.0139,
       "step": 5700
     },
     {
-      "epoch": 0.6188784845549457,
-      "grad_norm": 0.7258860468864441,
-      "learning_rate": 0.00056347376360306,
-      "loss": 4.0176,
+      "epoch": 0.6199460916442049,
+      "grad_norm": 0.6917612552642822,
+      "learning_rate": 0.0005634106853750674,
+      "loss": 4.0242,
       "step": 5750
     },
     {
-      "epoch": 0.6242600365945539,
-      "grad_norm": 0.8383513689041138,
-      "learning_rate": 0.0005631505225729985,
-      "loss": 4.0076,
+      "epoch": 0.6253369272237197,
+      "grad_norm": 0.6454603672027588,
+      "learning_rate": 0.000563086886130599,
+      "loss": 4.0121,
       "step": 5800
     },
     {
-      "epoch": 0.6296415886341621,
-      "grad_norm": 0.5887354612350464,
-      "learning_rate": 0.0005628272815429371,
-      "loss": 3.9825,
+      "epoch": 0.6307277628032345,
+      "grad_norm": 0.6462289094924927,
+      "learning_rate": 0.0005627630868861305,
+      "loss": 4.0228,
       "step": 5850
     },
     {
-      "epoch": 0.6350231406737703,
-      "grad_norm": 0.6457569599151611,
-      "learning_rate": 0.0005625040405128757,
-      "loss": 3.9889,
+      "epoch": 0.6361185983827493,
+      "grad_norm": 0.6403717398643494,
+      "learning_rate": 0.0005624392876416621,
+      "loss": 3.9963,
       "step": 5900
     },
     {
-      "epoch": 0.6404046927133785,
-      "grad_norm": 0.6390018463134766,
-      "learning_rate": 0.0005621807994828143,
-      "loss": 3.9978,
+      "epoch": 0.6415094339622641,
+      "grad_norm": 0.6747262477874756,
+      "learning_rate": 0.0005621154883971937,
+      "loss": 3.9897,
       "step": 5950
     },
     {
-      "epoch": 0.6457862447529867,
-      "grad_norm": 0.6403166651725769,
-      "learning_rate": 0.0005618575584527529,
-      "loss": 3.9888,
+      "epoch": 0.6469002695417789,
+      "grad_norm": 0.5862438082695007,
+      "learning_rate": 0.0005617916891527253,
+      "loss": 3.9993,
       "step": 6000
     },
     {
-      "epoch": 0.6457862447529867,
-      "eval_accuracy": 0.3272947680048972,
-      "eval_loss": 3.9194672107696533,
-      "eval_runtime": 181.325,
-      "eval_samples_per_second": 99.33,
-      "eval_steps_per_second": 6.21,
+      "epoch": 0.6469002695417789,
+      "eval_accuracy": 0.32746252806368276,
+      "eval_loss": 3.923083782196045,
+      "eval_runtime": 183.6666,
+      "eval_samples_per_second": 98.064,
+      "eval_steps_per_second": 6.131,
       "step": 6000
     },
     {
-      "epoch": 0.651167796792595,
-      "grad_norm": 0.6269046664237976,
-      "learning_rate": 0.0005615343174226915,
-      "loss": 3.9825,
+      "epoch": 0.6522911051212938,
+      "grad_norm": 0.7054334878921509,
+      "learning_rate": 0.0005614678899082568,
+      "loss": 3.9704,
       "step": 6050
     },
     {
-      "epoch": 0.6565493488322032,
-      "grad_norm": 0.7226353883743286,
-      "learning_rate": 0.00056121107639263,
-      "loss": 3.9832,
+      "epoch": 0.6576819407008087,
+      "grad_norm": 0.7405375242233276,
+      "learning_rate": 0.0005611440906637884,
+      "loss": 4.0026,
       "step": 6100
     },
     {
-      "epoch": 0.6619309008718114,
-      "grad_norm": 0.5846033692359924,
-      "learning_rate": 0.0005608878353625687,
-      "loss": 3.9839,
+      "epoch": 0.6630727762803235,
+      "grad_norm": 0.629364013671875,
+      "learning_rate": 0.00056082029141932,
+      "loss": 3.985,
       "step": 6150
     },
     {
-      "epoch": 0.6673124529114196,
-      "grad_norm": 0.6530674695968628,
-      "learning_rate": 0.0005605645943325072,
-      "loss": 3.9759,
+      "epoch": 0.6684636118598383,
+      "grad_norm": 0.6291308403015137,
+      "learning_rate": 0.0005604964921748515,
+      "loss": 3.966,
       "step": 6200
     },
     {
-      "epoch": 0.6726940049510278,
-      "grad_norm": 0.6534646153450012,
-      "learning_rate": 0.0005602413533024458,
-      "loss": 3.9751,
+      "epoch": 0.6738544474393531,
+      "grad_norm": 0.6362651586532593,
+      "learning_rate": 0.000560172692930383,
+      "loss": 3.9815,
       "step": 6250
     },
     {
-      "epoch": 0.6780755569906362,
-      "grad_norm": 0.6664760112762451,
-      "learning_rate": 0.0005599181122723844,
-      "loss": 3.9406,
+      "epoch": 0.6792452830188679,
+      "grad_norm": 0.7001793384552002,
+      "learning_rate": 0.0005598488936859147,
+      "loss": 3.9965,
       "step": 6300
     },
     {
-      "epoch": 0.6834571090302444,
-      "grad_norm": 0.6543918251991272,
-      "learning_rate": 0.000559594871242323,
-      "loss": 3.9741,
+      "epoch": 0.6846361185983828,
+      "grad_norm": 0.5869341492652893,
+      "learning_rate": 0.0005595250944414463,
+      "loss": 3.9604,
       "step": 6350
     },
     {
-      "epoch": 0.6888386610698526,
-      "grad_norm": 0.6636673808097839,
-      "learning_rate": 0.0005592716302122616,
-      "loss": 3.975,
+      "epoch": 0.6900269541778976,
+      "grad_norm": 0.6543765068054199,
+      "learning_rate": 0.0005592012951969778,
+      "loss": 3.965,
       "step": 6400
     },
     {
-      "epoch": 0.6942202131094608,
-      "grad_norm": 0.7524754405021667,
-      "learning_rate": 0.0005589483891822001,
-      "loss": 3.9804,
+      "epoch": 0.6954177897574124,
+      "grad_norm": 0.6181590557098389,
+      "learning_rate": 0.0005588774959525094,
+      "loss": 3.951,
       "step": 6450
     },
     {
-      "epoch": 0.699601765149069,
-      "grad_norm": 0.6153088808059692,
-      "learning_rate": 0.0005586251481521387,
-      "loss": 3.9682,
+      "epoch": 0.7008086253369272,
+      "grad_norm": 0.6935147643089294,
+      "learning_rate": 0.000558553696708041,
+      "loss": 3.9631,
       "step": 6500
     },
     {
-      "epoch": 0.7049833171886772,
-      "grad_norm": 0.62467360496521,
-      "learning_rate": 0.0005583019071220773,
-      "loss": 3.9476,
+      "epoch": 0.706199460916442,
+      "grad_norm": 0.5719255208969116,
+      "learning_rate": 0.0005582298974635726,
+      "loss": 3.988,
       "step": 6550
     },
     {
-      "epoch": 0.7103648692282855,
-      "grad_norm": 0.6360256671905518,
-      "learning_rate": 0.000557978666092016,
-      "loss": 3.9448,
+      "epoch": 0.7115902964959568,
+      "grad_norm": 0.7059009671211243,
+      "learning_rate": 0.0005579060982191041,
+      "loss": 3.9421,
       "step": 6600
     },
     {
-      "epoch": 0.7157464212678937,
-      "grad_norm": 0.5992259979248047,
-      "learning_rate": 0.0005576554250619545,
-      "loss": 3.9452,
+      "epoch": 0.7169811320754716,
+      "grad_norm": 0.6006096005439758,
+      "learning_rate": 0.0005575822989746357,
+      "loss": 3.977,
       "step": 6650
     },
     {
-      "epoch": 0.7211279733075019,
-      "grad_norm": 0.6049088835716248,
-      "learning_rate": 0.000557332184031893,
-      "loss": 3.9372,
+      "epoch": 0.7223719676549866,
+      "grad_norm": 0.5939664244651794,
+      "learning_rate": 0.0005572584997301672,
+      "loss": 3.9475,
       "step": 6700
     },
     {
-      "epoch": 0.7265095253471101,
-      "grad_norm": 0.6222875118255615,
-      "learning_rate": 0.0005570089430018317,
-      "loss": 3.95,
+      "epoch": 0.7277628032345014,
+      "grad_norm": 0.5215754508972168,
+      "learning_rate": 0.0005569347004856989,
+      "loss": 3.9544,
       "step": 6750
     },
     {
-      "epoch": 0.7318910773867183,
-      "grad_norm": 0.5336934924125671,
-      "learning_rate": 0.0005566857019717702,
-      "loss": 3.9299,
+      "epoch": 0.7331536388140162,
+      "grad_norm": 0.6375124454498291,
+      "learning_rate": 0.0005566109012412303,
+      "loss": 3.9424,
       "step": 6800
     },
     {
-      "epoch": 0.7372726294263265,
-      "grad_norm": 0.6668237447738647,
-      "learning_rate": 0.0005563624609417089,
-      "loss": 3.9269,
+      "epoch": 0.738544474393531,
+      "grad_norm": 0.6478970050811768,
+      "learning_rate": 0.000556287101996762,
+      "loss": 3.9226,
       "step": 6850
     },
     {
-      "epoch": 0.7426541814659348,
-      "grad_norm": 0.5827189087867737,
-      "learning_rate": 0.0005560392199116474,
-      "loss": 3.934,
+      "epoch": 0.7439353099730458,
+      "grad_norm": 0.773593544960022,
+      "learning_rate": 0.0005559633027522935,
+      "loss": 3.9479,
       "step": 6900
     },
     {
-      "epoch": 0.748035733505543,
-      "grad_norm": 0.6519905924797058,
-      "learning_rate": 0.000555715978881586,
-      "loss": 3.9101,
+      "epoch": 0.7493261455525606,
+      "grad_norm": 0.6449713110923767,
+      "learning_rate": 0.0005556395035078251,
+      "loss": 3.9454,
       "step": 6950
     },
     {
-      "epoch": 0.7534172855451512,
-      "grad_norm": 0.6796578764915466,
-      "learning_rate": 0.0005553927378515246,
-      "loss": 3.9305,
+      "epoch": 0.7547169811320755,
+      "grad_norm": 0.6649606823921204,
+      "learning_rate": 0.0005553157042633566,
+      "loss": 3.9313,
       "step": 7000
     },
     {
-      "epoch": 0.7534172855451512,
-      "eval_accuracy": 0.33403450663602946,
-      "eval_loss": 3.864957571029663,
-      "eval_runtime": 181.9746,
-      "eval_samples_per_second": 98.975,
-      "eval_steps_per_second": 6.188,
+      "epoch": 0.7547169811320755,
+      "eval_accuracy": 0.33284540943178775,
+      "eval_loss": 3.866408348083496,
+      "eval_runtime": 183.7096,
+      "eval_samples_per_second": 98.041,
+      "eval_steps_per_second": 6.129,
       "step": 7000
     },
     {
-      "epoch": 0.7587988375847594,
-      "grad_norm": 0.561911940574646,
-      "learning_rate": 0.0005550694968214631,
-      "loss": 3.9368,
+      "epoch": 0.7601078167115903,
+      "grad_norm": 0.6450538635253906,
+      "learning_rate": 0.0005549919050188882,
+      "loss": 3.9229,
       "step": 7050
     },
     {
-      "epoch": 0.7641803896243676,
-      "grad_norm": 0.6652414798736572,
-      "learning_rate": 0.0005547462557914018,
-      "loss": 3.9173,
+      "epoch": 0.7654986522911051,
+      "grad_norm": 0.6038523316383362,
+      "learning_rate": 0.0005546681057744198,
+      "loss": 3.9345,
       "step": 7100
     },
     {
-      "epoch": 0.7695619416639758,
-      "grad_norm": 0.6089035868644714,
-      "learning_rate": 0.0005544230147613403,
-      "loss": 3.9405,
+      "epoch": 0.77088948787062,
+      "grad_norm": 0.5474239587783813,
+      "learning_rate": 0.0005543443065299514,
+      "loss": 3.9337,
       "step": 7150
     },
     {
-      "epoch": 0.7749434937035841,
-      "grad_norm": 0.6177041530609131,
-      "learning_rate": 0.0005540997737312789,
-      "loss": 3.9035,
+      "epoch": 0.7762803234501348,
+      "grad_norm": 0.623572826385498,
+      "learning_rate": 0.000554020507285483,
+      "loss": 3.9346,
       "step": 7200
     },
     {
-      "epoch": 0.7803250457431924,
-      "grad_norm": 0.679482638835907,
-      "learning_rate": 0.0005537765327012175,
-      "loss": 3.9234,
+      "epoch": 0.7816711590296496,
+      "grad_norm": 0.6270178556442261,
+      "learning_rate": 0.0005536967080410145,
+      "loss": 3.8924,
       "step": 7250
     },
     {
-      "epoch": 0.7857065977828006,
-      "grad_norm": 0.6637169122695923,
-      "learning_rate": 0.0005534532916711561,
-      "loss": 3.9177,
+      "epoch": 0.7870619946091644,
+      "grad_norm": 0.5980831980705261,
+      "learning_rate": 0.0005533793847814354,
+      "loss": 3.9315,
       "step": 7300
     },
     {
-      "epoch": 0.7910881498224088,
-      "grad_norm": 0.6377207636833191,
-      "learning_rate": 0.0005531300506410946,
-      "loss": 3.9106,
+      "epoch": 0.7924528301886793,
+      "grad_norm": 0.5830099582672119,
+      "learning_rate": 0.000553055585536967,
+      "loss": 3.9168,
       "step": 7350
     },
     {
-      "epoch": 0.796469701862017,
-      "grad_norm": 0.6485190987586975,
-      "learning_rate": 0.0005528068096110332,
-      "loss": 3.9204,
+      "epoch": 0.7978436657681941,
+      "grad_norm": 0.7148050665855408,
+      "learning_rate": 0.0005527317862924987,
+      "loss": 3.9162,
       "step": 7400
     },
     {
-      "epoch": 0.8018512539016253,
-      "grad_norm": 0.571081817150116,
-      "learning_rate": 0.0005524835685809719,
-      "loss": 3.8924,
+      "epoch": 0.8032345013477089,
+      "grad_norm": 0.5602888464927673,
+      "learning_rate": 0.0005524079870480301,
+      "loss": 3.9059,
       "step": 7450
     },
     {
-      "epoch": 0.8072328059412335,
-      "grad_norm": 0.677717387676239,
-      "learning_rate": 0.0005521603275509104,
-      "loss": 3.9122,
+      "epoch": 0.8086253369272237,
+      "grad_norm": 0.6142837405204773,
+      "learning_rate": 0.0005520841878035618,
+      "loss": 3.8923,
       "step": 7500
     },
     {
-      "epoch": 0.8126143579808417,
-      "grad_norm": 0.5621068477630615,
-      "learning_rate": 0.000551837086520849,
-      "loss": 3.8824,
+      "epoch": 0.8140161725067385,
+      "grad_norm": 0.6284271478652954,
+      "learning_rate": 0.0005517603885590933,
+      "loss": 3.922,
       "step": 7550
     },
     {
-      "epoch": 0.8179959100204499,
-      "grad_norm": 0.5969014167785645,
-      "learning_rate": 0.0005515138454907875,
-      "loss": 3.8977,
+      "epoch": 0.8194070080862533,
+      "grad_norm": 0.621902346611023,
+      "learning_rate": 0.0005514365893146249,
+      "loss": 3.923,
       "step": 7600
     },
     {
-      "epoch": 0.8233774620600581,
-      "grad_norm": 0.6785698533058167,
-      "learning_rate": 0.0005511906044607262,
-      "loss": 3.8885,
+      "epoch": 0.8247978436657682,
+      "grad_norm": 0.6389454007148743,
+      "learning_rate": 0.0005511127900701564,
+      "loss": 3.8969,
       "step": 7650
     },
     {
-      "epoch": 0.8287590140996663,
-      "grad_norm": 0.8015792369842529,
-      "learning_rate": 0.0005508673634306648,
-      "loss": 3.9005,
+      "epoch": 0.8301886792452831,
+      "grad_norm": 0.6034095287322998,
+      "learning_rate": 0.000550788990825688,
+      "loss": 3.9015,
       "step": 7700
     },
     {
-      "epoch": 0.8341405661392746,
-      "grad_norm": 0.6184539198875427,
-      "learning_rate": 0.0005505441224006034,
-      "loss": 3.9204,
+      "epoch": 0.8355795148247979,
+      "grad_norm": 0.5711124539375305,
+      "learning_rate": 0.0005504651915812196,
+      "loss": 3.8822,
       "step": 7750
     },
     {
-      "epoch": 0.8395221181788828,
-      "grad_norm": 0.6219672560691833,
-      "learning_rate": 0.0005502208813705419,
-      "loss": 3.8908,
+      "epoch": 0.8409703504043127,
+      "grad_norm": 0.7393220663070679,
+      "learning_rate": 0.0005501413923367512,
+      "loss": 3.9127,
       "step": 7800
     },
     {
-      "epoch": 0.844903670218491,
-      "grad_norm": 0.5572973489761353,
-      "learning_rate": 0.0005498976403404805,
-      "loss": 3.8861,
+      "epoch": 0.8463611859838275,
+      "grad_norm": 0.6161301136016846,
+      "learning_rate": 0.0005498175930922827,
+      "loss": 3.8893,
       "step": 7850
     },
     {
-      "epoch": 0.8502852222580992,
-      "grad_norm": 0.6005538105964661,
-      "learning_rate": 0.0005495743993104191,
-      "loss": 3.8814,
+      "epoch": 0.8517520215633423,
+      "grad_norm": 0.5497040748596191,
+      "learning_rate": 0.0005494937938478143,
+      "loss": 3.8928,
       "step": 7900
     },
     {
-      "epoch": 0.8556667742977074,
-      "grad_norm": 0.6196568012237549,
-      "learning_rate": 0.0005492511582803576,
-      "loss": 3.8699,
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.6119050979614258,
+      "learning_rate": 0.0005491699946033459,
+      "loss": 3.8886,
       "step": 7950
     },
     {
-      "epoch": 0.8610483263373157,
-      "grad_norm": 0.5852003693580627,
-      "learning_rate": 0.0005489279172502963,
-      "loss": 3.8504,
+      "epoch": 0.862533692722372,
+      "grad_norm": 0.6763274073600769,
+      "learning_rate": 0.0005488461953588775,
+      "loss": 3.8759,
       "step": 8000
     },
     {
-      "epoch": 0.8610483263373157,
-      "eval_accuracy": 0.33771957797914537,
-      "eval_loss": 3.815176248550415,
-      "eval_runtime": 181.7741,
-      "eval_samples_per_second": 99.085,
-      "eval_steps_per_second": 6.195,
+      "epoch": 0.862533692722372,
+      "eval_accuracy": 0.33728507508077527,
+      "eval_loss": 3.8209762573242188,
+      "eval_runtime": 183.5848,
+      "eval_samples_per_second": 98.107,
+      "eval_steps_per_second": 6.133,
       "step": 8000
     },
     {
-      "epoch": 0.8664298783769239,
-      "grad_norm": 0.6063625812530518,
-      "learning_rate": 0.0005486176058614372,
-      "loss": 3.8921,
+      "epoch": 0.8679245283018868,
+      "grad_norm": 0.5846931338310242,
+      "learning_rate": 0.000548522396114409,
+      "loss": 3.8861,
       "step": 8050
     },
     {
-      "epoch": 0.8718114304165321,
-      "grad_norm": 0.5820379257202148,
-      "learning_rate": 0.0005482943648313759,
-      "loss": 3.8736,
+      "epoch": 0.8733153638814016,
+      "grad_norm": 0.6224812269210815,
+      "learning_rate": 0.0005481985968699406,
+      "loss": 3.8722,
       "step": 8100
     },
     {
-      "epoch": 0.8771929824561403,
-      "grad_norm": 0.6044704914093018,
-      "learning_rate": 0.0005479711238013145,
-      "loss": 3.8832,
+      "epoch": 0.8787061994609164,
+      "grad_norm": 0.5953459739685059,
+      "learning_rate": 0.0005478747976254721,
+      "loss": 3.882,
       "step": 8150
     },
     {
-      "epoch": 0.8825745344957485,
-      "grad_norm": 0.5778374671936035,
-      "learning_rate": 0.0005476478827712531,
-      "loss": 3.8815,
+      "epoch": 0.8840970350404312,
+      "grad_norm": 0.5575414299964905,
+      "learning_rate": 0.0005475509983810037,
+      "loss": 3.8713,
       "step": 8200
     },
     {
-      "epoch": 0.8879560865353568,
-      "grad_norm": 0.5325333476066589,
-      "learning_rate": 0.0005473246417411916,
-      "loss": 3.8749,
+      "epoch": 0.889487870619946,
+      "grad_norm": 0.6434709429740906,
+      "learning_rate": 0.0005472271991365352,
+      "loss": 3.8591,
       "step": 8250
     },
     {
-      "epoch": 0.8933376385749651,
-      "grad_norm": 0.618618905544281,
-      "learning_rate": 0.0005470014007111302,
-      "loss": 3.8779,
+      "epoch": 0.894878706199461,
+      "grad_norm": 0.6556322574615479,
+      "learning_rate": 0.0005469033998920669,
+      "loss": 3.8837,
       "step": 8300
     },
     {
-      "epoch": 0.8987191906145733,
-      "grad_norm": 0.7097580432891846,
-      "learning_rate": 0.0005466781596810688,
-      "loss": 3.8664,
+      "epoch": 0.9002695417789758,
+      "grad_norm": 0.6208202242851257,
+      "learning_rate": 0.0005465796006475984,
+      "loss": 3.8659,
       "step": 8350
     },
     {
-      "epoch": 0.9041007426541815,
-      "grad_norm": 0.6304284334182739,
-      "learning_rate": 0.0005463549186510073,
-      "loss": 3.859,
+      "epoch": 0.9056603773584906,
+      "grad_norm": 0.6528877019882202,
+      "learning_rate": 0.00054625580140313,
+      "loss": 3.878,
       "step": 8400
     },
     {
-      "epoch": 0.9094822946937897,
-      "grad_norm": 0.5668020248413086,
-      "learning_rate": 0.000546031677620946,
-      "loss": 3.8624,
+      "epoch": 0.9110512129380054,
+      "grad_norm": 0.6114417314529419,
+      "learning_rate": 0.0005459320021586615,
+      "loss": 3.8875,
       "step": 8450
     },
     {
-      "epoch": 0.9148638467333979,
-      "grad_norm": 0.5498570203781128,
-      "learning_rate": 0.0005457084365908845,
-      "loss": 3.8593,
+      "epoch": 0.9164420485175202,
+      "grad_norm": 0.6060699224472046,
+      "learning_rate": 0.0005456082029141932,
+      "loss": 3.8798,
       "step": 8500
     },
     {
-      "epoch": 0.9202453987730062,
-      "grad_norm": 0.5575296878814697,
-      "learning_rate": 0.0005453851955608232,
-      "loss": 3.8623,
+      "epoch": 0.921832884097035,
+      "grad_norm": 0.6428260207176208,
+      "learning_rate": 0.0005452844036697248,
+      "loss": 3.8613,
       "step": 8550
     },
     {
-      "epoch": 0.9256269508126144,
-      "grad_norm": 0.5803061127662659,
-      "learning_rate": 0.0005450619545307617,
-      "loss": 3.8653,
+      "epoch": 0.9272237196765498,
+      "grad_norm": 0.565753698348999,
+      "learning_rate": 0.0005449606044252563,
+      "loss": 3.8561,
       "step": 8600
     },
     {
-      "epoch": 0.9310085028522226,
-      "grad_norm": 0.6923871636390686,
-      "learning_rate": 0.0005447387135007003,
-      "loss": 3.8628,
+      "epoch": 0.9326145552560647,
+      "grad_norm": 0.607820451259613,
+      "learning_rate": 0.0005446368051807879,
+      "loss": 3.8477,
       "step": 8650
     },
     {
-      "epoch": 0.9363900548918308,
-      "grad_norm": 0.6441589593887329,
-      "learning_rate": 0.0005444154724706389,
-      "loss": 3.8624,
+      "epoch": 0.9380053908355795,
+      "grad_norm": 0.5893778800964355,
+      "learning_rate": 0.0005443130059363194,
+      "loss": 3.8519,
       "step": 8700
     },
     {
-      "epoch": 0.941771606931439,
-      "grad_norm": 0.6883953809738159,
-      "learning_rate": 0.0005440922314405775,
-      "loss": 3.8515,
+      "epoch": 0.9433962264150944,
+      "grad_norm": 0.5969862341880798,
+      "learning_rate": 0.0005439892066918511,
+      "loss": 3.8529,
       "step": 8750
     },
     {
-      "epoch": 0.9471531589710472,
-      "grad_norm": 0.6100162863731384,
-      "learning_rate": 0.0005437689904105161,
-      "loss": 3.8368,
+      "epoch": 0.9487870619946092,
+      "grad_norm": 0.5738219022750854,
+      "learning_rate": 0.0005436654074473825,
+      "loss": 3.8464,
       "step": 8800
     },
     {
-      "epoch": 0.9525347110106555,
-      "grad_norm": 0.5788195133209229,
-      "learning_rate": 0.0005434457493804546,
-      "loss": 3.8312,
+      "epoch": 0.954177897574124,
+      "grad_norm": 0.5586745738983154,
+      "learning_rate": 0.0005433416082029142,
+      "loss": 3.8343,
       "step": 8850
     },
     {
-      "epoch": 0.9579162630502637,
-      "grad_norm": 0.5563709735870361,
-      "learning_rate": 0.0005431225083503932,
-      "loss": 3.8517,
+      "epoch": 0.9595687331536388,
+      "grad_norm": 0.5607208013534546,
+      "learning_rate": 0.0005430178089584457,
+      "loss": 3.8526,
       "step": 8900
     },
     {
-      "epoch": 0.9632978150898719,
-      "grad_norm": 0.6155171990394592,
-      "learning_rate": 0.0005427992673203318,
-      "loss": 3.8583,
+      "epoch": 0.9649595687331537,
+      "grad_norm": 0.5701066851615906,
+      "learning_rate": 0.0005426940097139773,
+      "loss": 3.8585,
       "step": 8950
     },
     {
-      "epoch": 0.9686793671294801,
-      "grad_norm": 0.5446226596832275,
-      "learning_rate": 0.0005424760262902704,
-      "loss": 3.832,
+      "epoch": 0.9703504043126685,
+      "grad_norm": 0.6304426193237305,
+      "learning_rate": 0.0005423702104695088,
+      "loss": 3.8469,
       "step": 9000
     },
     {
-      "epoch": 0.9686793671294801,
-      "eval_accuracy": 0.34148950722764443,
-      "eval_loss": 3.7837822437286377,
-      "eval_runtime": 181.5161,
-      "eval_samples_per_second": 99.225,
-      "eval_steps_per_second": 6.203,
+      "epoch": 0.9703504043126685,
+      "eval_accuracy": 0.3409309227313902,
+      "eval_loss": 3.782121419906616,
+      "eval_runtime": 183.7985,
+      "eval_samples_per_second": 97.993,
+      "eval_steps_per_second": 6.126,
       "step": 9000
     },
     {
-      "epoch": 0.9740609191690883,
-      "grad_norm": 0.6641152501106262,
-      "learning_rate": 0.000542152785260209,
-      "loss": 3.8364,
+      "epoch": 0.9757412398921833,
+      "grad_norm": 0.5790097713470459,
+      "learning_rate": 0.0005420464112250404,
+      "loss": 3.8395,
       "step": 9050
     },
     {
-      "epoch": 0.9794424712086965,
-      "grad_norm": 0.5310914516448975,
-      "learning_rate": 0.0005418295442301476,
-      "loss": 3.8395,
+      "epoch": 0.9811320754716981,
+      "grad_norm": 0.6304307579994202,
+      "learning_rate": 0.000541722611980572,
+      "loss": 3.854,
       "step": 9100
     },
     {
-      "epoch": 0.9848240232483048,
-      "grad_norm": 0.6875292062759399,
-      "learning_rate": 0.0005415063032000861,
-      "loss": 3.837,
+      "epoch": 0.9865229110512129,
+      "grad_norm": 0.5811077952384949,
+      "learning_rate": 0.0005413988127361036,
+      "loss": 3.8277,
       "step": 9150
     },
     {
-      "epoch": 0.9902055752879131,
-      "grad_norm": 0.6055516600608826,
-      "learning_rate": 0.0005411830621700248,
-      "loss": 3.838,
+      "epoch": 0.9919137466307277,
+      "grad_norm": 0.5954909324645996,
+      "learning_rate": 0.0005410750134916351,
+      "loss": 3.8239,
       "step": 9200
     },
     {
-      "epoch": 0.9955871273275213,
-      "grad_norm": 0.586495041847229,
-      "learning_rate": 0.0005408598211399633,
-      "loss": 3.8258,
+      "epoch": 0.9973045822102425,
+      "grad_norm": 0.6167082786560059,
+      "learning_rate": 0.0005407512142471667,
+      "loss": 3.838,
       "step": 9250
     },
     {
-      "epoch": 1.0009686793671295,
-      "grad_norm": 0.5521350502967834,
-      "learning_rate": 0.0005405365801099019,
-      "loss": 3.8056,
+      "epoch": 1.0026954177897573,
+      "grad_norm": 0.5655587911605835,
+      "learning_rate": 0.0005404274150026983,
+      "loss": 3.8113,
       "step": 9300
     },
     {
-      "epoch": 1.0063502314067376,
-      "grad_norm": 0.543850302696228,
-      "learning_rate": 0.0005402133390798405,
-      "loss": 3.7712,
+      "epoch": 1.0080862533692723,
+      "grad_norm": 0.559786856174469,
+      "learning_rate": 0.0005401100917431192,
+      "loss": 3.777,
       "step": 9350
     },
     {
-      "epoch": 1.011731783446346,
-      "grad_norm": 0.6066914200782776,
-      "learning_rate": 0.000539890098049779,
-      "loss": 3.7626,
+      "epoch": 1.013477088948787,
+      "grad_norm": 0.6333407759666443,
+      "learning_rate": 0.0005397862924986508,
+      "loss": 3.7711,
       "step": 9400
     },
     {
-      "epoch": 1.017113335485954,
-      "grad_norm": 0.5474905967712402,
-      "learning_rate": 0.0005395668570197177,
-      "loss": 3.7664,
+      "epoch": 1.0188679245283019,
+      "grad_norm": 0.6588902473449707,
+      "learning_rate": 0.0005394624932541824,
+      "loss": 3.7722,
       "step": 9450
     },
     {
-      "epoch": 1.0224948875255624,
-      "grad_norm": 0.6281277537345886,
-      "learning_rate": 0.0005392436159896562,
-      "loss": 3.7591,
+      "epoch": 1.0242587601078168,
+      "grad_norm": 0.6027793884277344,
+      "learning_rate": 0.0005391386940097139,
+      "loss": 3.7697,
       "step": 9500
     },
     {
-      "epoch": 1.0278764395651705,
-      "grad_norm": 0.5506016612052917,
-      "learning_rate": 0.0005389203749595948,
-      "loss": 3.7604,
+      "epoch": 1.0296495956873315,
+      "grad_norm": 0.532487690448761,
+      "learning_rate": 0.0005388148947652455,
+      "loss": 3.7684,
       "step": 9550
     },
     {
-      "epoch": 1.0332579916047788,
-      "grad_norm": 0.5733294486999512,
-      "learning_rate": 0.0005385971339295334,
-      "loss": 3.7693,
+      "epoch": 1.0350404312668464,
+      "grad_norm": 0.7084537148475647,
+      "learning_rate": 0.000538491095520777,
+      "loss": 3.757,
       "step": 9600
     },
     {
-      "epoch": 1.0386395436443872,
-      "grad_norm": 0.6508961915969849,
-      "learning_rate": 0.000538273892899472,
-      "loss": 3.7602,
+      "epoch": 1.0404312668463611,
+      "grad_norm": 0.5634706020355225,
+      "learning_rate": 0.0005381672962763086,
+      "loss": 3.7978,
       "step": 9650
     },
     {
-      "epoch": 1.0440210956839953,
-      "grad_norm": 0.8109516501426697,
-      "learning_rate": 0.0005379506518694106,
-      "loss": 3.7589,
+      "epoch": 1.045822102425876,
+      "grad_norm": 0.6227463483810425,
+      "learning_rate": 0.0005378434970318403,
+      "loss": 3.7834,
       "step": 9700
     },
     {
-      "epoch": 1.0494026477236036,
-      "grad_norm": 0.5636641979217529,
-      "learning_rate": 0.0005376274108393491,
-      "loss": 3.7685,
+      "epoch": 1.0512129380053907,
+      "grad_norm": 0.5373621582984924,
+      "learning_rate": 0.0005375196977873718,
+      "loss": 3.7813,
       "step": 9750
     },
     {
-      "epoch": 1.0547841997632117,
-      "grad_norm": 0.5453770160675049,
-      "learning_rate": 0.0005373041698092877,
-      "loss": 3.748,
+      "epoch": 1.0566037735849056,
+      "grad_norm": 0.5735413432121277,
+      "learning_rate": 0.0005371958985429034,
+      "loss": 3.791,
       "step": 9800
     },
     {
-      "epoch": 1.06016575180282,
-      "grad_norm": 0.5409913063049316,
-      "learning_rate": 0.0005369809287792263,
-      "loss": 3.7585,
+      "epoch": 1.0619946091644206,
+      "grad_norm": 0.6301229596138,
+      "learning_rate": 0.0005368720992984349,
+      "loss": 3.7878,
       "step": 9850
     },
     {
-      "epoch": 1.0655473038424281,
-      "grad_norm": 0.5672662854194641,
-      "learning_rate": 0.000536657687749165,
-      "loss": 3.7782,
+      "epoch": 1.0673854447439353,
+      "grad_norm": 0.5798570513725281,
+      "learning_rate": 0.0005365483000539665,
+      "loss": 3.7761,
       "step": 9900
     },
     {
-      "epoch": 1.0709288558820365,
-      "grad_norm": 0.6657660007476807,
-      "learning_rate": 0.0005363344467191035,
-      "loss": 3.7581,
+      "epoch": 1.0727762803234502,
+      "grad_norm": 0.6208717823028564,
+      "learning_rate": 0.0005362245008094981,
+      "loss": 3.7714,
       "step": 9950
     },
     {
-      "epoch": 1.0763104079216446,
-      "grad_norm": 0.6501979231834412,
-      "learning_rate": 0.000536011205689042,
-      "loss": 3.7729,
+      "epoch": 1.0781671159029649,
+      "grad_norm": 0.6318506002426147,
+      "learning_rate": 0.0005359007015650297,
+      "loss": 3.7655,
       "step": 10000
     },
     {
-      "epoch": 1.0763104079216446,
-      "eval_accuracy": 0.3445393937885969,
-      "eval_loss": 3.748054027557373,
-      "eval_runtime": 180.5408,
-      "eval_samples_per_second": 99.761,
-      "eval_steps_per_second": 6.237,
+      "epoch": 1.0781671159029649,
+      "eval_accuracy": 0.34419757180353255,
+      "eval_loss": 3.7513272762298584,
+      "eval_runtime": 183.4398,
+      "eval_samples_per_second": 98.185,
+      "eval_steps_per_second": 6.138,
       "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:f11599afc4c0758396f0e41fe533161cd5207f21434bf81e841ef073d767ad4b
+oid sha256:e59f92209bd56ce90383c284fca2492d24984283c65cccf575b530309245403e
 size 5304