diff --git "a/trainer_state.json" "b/trainer_state.json"
--- "a/trainer_state.json"
+++ "b/trainer_state.json"
@@ -1,8313 +1,983 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 1.9996613039796782,
+  "epoch": 0.999627004848937,
   "eval_steps": 500,
-  "global_step": 5904,
+  "global_step": 670,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.0,
-      "grad_norm": 0.671875,
-      "learning_rate": 3.38409475465313e-07,
-      "loss": 2.4832,
+      "grad_norm": 0.337890625,
+      "learning_rate": 9.477611940298507e-07,
+      "loss": 2.637,
       "step": 1
     },
     {
-      "epoch": 0.0,
-      "grad_norm": 0.546875,
-      "learning_rate": 1.6920473773265652e-06,
-      "loss": 2.5283,
+      "epoch": 0.01,
+      "grad_norm": 0.349609375,
+      "learning_rate": 4.738805970149253e-06,
+      "loss": 2.6461,
       "step": 5
     },
     {
-      "epoch": 0.0,
-      "grad_norm": 0.71484375,
-      "learning_rate": 3.3840947546531303e-06,
-      "loss": 2.5147,
+      "epoch": 0.01,
+      "grad_norm": 0.32421875,
+      "learning_rate": 9.477611940298506e-06,
+      "loss": 2.6439,
       "step": 10
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.8671875,
-      "learning_rate": 5.076142131979695e-06,
-      "loss": 2.5183,
+      "epoch": 0.02,
+      "grad_norm": 0.35546875,
+      "learning_rate": 1.4216417910447761e-05,
+      "loss": 2.642,
       "step": 15
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.88671875,
-      "learning_rate": 6.768189509306261e-06,
-      "loss": 2.5027,
+      "epoch": 0.03,
+      "grad_norm": 0.365234375,
+      "learning_rate": 1.895522388059701e-05,
+      "loss": 2.6207,
       "step": 20
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.76953125,
-      "learning_rate": 8.460236886632826e-06,
-      "loss": 2.5038,
+      "epoch": 0.04,
+      "grad_norm": 0.353515625,
+      "learning_rate": 2.369402985074627e-05,
+      "loss": 2.6062,
       "step": 25
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.59375,
-      "learning_rate": 1.015228426395939e-05,
-      "loss": 2.4691,
+      "epoch": 0.04,
+      "grad_norm": 0.291015625,
+      "learning_rate": 2.8432835820895522e-05,
+      "loss": 2.5733,
       "step": 30
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.5546875,
-      "learning_rate": 1.1844331641285957e-05,
-      "loss": 2.4495,
+      "epoch": 0.05,
+      "grad_norm": 0.255859375,
+      "learning_rate": 3.317164179104477e-05,
+      "loss": 2.531,
       "step": 35
     },
     {
-      "epoch": 0.01,
-      "grad_norm": 0.470703125,
-      "learning_rate": 1.3536379018612521e-05,
-      "loss": 2.4366,
+      "epoch": 0.06,
+      "grad_norm": 0.259765625,
+      "learning_rate": 3.791044776119402e-05,
+      "loss": 2.495,
       "step": 40
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.349609375,
-      "learning_rate": 1.5228426395939088e-05,
-      "loss": 2.4702,
+      "epoch": 0.07,
+      "grad_norm": 0.177734375,
+      "learning_rate": 4.2649253731343286e-05,
+      "loss": 2.4618,
       "step": 45
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.259765625,
-      "learning_rate": 1.6920473773265652e-05,
-      "loss": 2.4238,
+      "epoch": 0.07,
+      "grad_norm": 0.1494140625,
+      "learning_rate": 4.738805970149254e-05,
+      "loss": 2.4374,
       "step": 50
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.2265625,
-      "learning_rate": 1.8612521150592218e-05,
-      "loss": 2.402,
+      "epoch": 0.08,
+      "grad_norm": 0.1513671875,
+      "learning_rate": 5.2126865671641794e-05,
+      "loss": 2.4205,
       "step": 55
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.19140625,
-      "learning_rate": 2.030456852791878e-05,
-      "loss": 2.3977,
+      "epoch": 0.09,
+      "grad_norm": 0.15234375,
+      "learning_rate": 5.6865671641791044e-05,
+      "loss": 2.3846,
       "step": 60
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.199661590524535e-05,
-      "loss": 2.3896,
+      "epoch": 0.1,
+      "grad_norm": 0.1298828125,
+      "learning_rate": 6.16044776119403e-05,
+      "loss": 2.3675,
       "step": 65
     },
     {
-      "epoch": 0.02,
-      "grad_norm": 0.17578125,
-      "learning_rate": 2.3688663282571914e-05,
-      "loss": 2.4104,
+      "epoch": 0.1,
+      "grad_norm": 0.10888671875,
+      "learning_rate": 6.349612195786573e-05,
+      "loss": 2.3648,
       "step": 70
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 2.5380710659898476e-05,
-      "loss": 2.3876,
+      "epoch": 0.11,
+      "grad_norm": 0.138671875,
+      "learning_rate": 6.3472426242142e-05,
+      "loss": 2.3394,
       "step": 75
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 2.7072758037225043e-05,
-      "loss": 2.4002,
+      "epoch": 0.12,
+      "grad_norm": 0.09912109375,
+      "learning_rate": 6.342720533773385e-05,
+      "loss": 2.3241,
       "step": 80
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 2.876480541455161e-05,
-      "loss": 2.3435,
+      "epoch": 0.13,
+      "grad_norm": 0.083984375,
+      "learning_rate": 6.336048992919527e-05,
+      "loss": 2.3123,
       "step": 85
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.1806640625,
-      "learning_rate": 3.0456852791878175e-05,
-      "loss": 2.3393,
+      "epoch": 0.13,
+      "grad_norm": 0.09228515625,
+      "learning_rate": 6.327232528613285e-05,
+      "loss": 2.3201,
       "step": 90
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 3.214890016920474e-05,
-      "loss": 2.3178,
+      "epoch": 0.14,
+      "grad_norm": 0.0908203125,
+      "learning_rate": 6.316277123248829e-05,
+      "loss": 2.2999,
       "step": 95
     },
     {
-      "epoch": 0.03,
-      "grad_norm": 0.150390625,
-      "learning_rate": 3.3840947546531304e-05,
-      "loss": 2.3124,
+      "epoch": 0.15,
+      "grad_norm": 0.07958984375,
+      "learning_rate": 6.303190210594489e-05,
+      "loss": 2.3114,
       "step": 100
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 3.553299492385787e-05,
-      "loss": 2.305,
+      "epoch": 0.16,
+      "grad_norm": 0.0712890625,
+      "learning_rate": 6.287980670748592e-05,
+      "loss": 2.3002,
       "step": 105
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.138671875,
-      "learning_rate": 3.7225042301184437e-05,
-      "loss": 2.3204,
+      "epoch": 0.16,
+      "grad_norm": 0.072265625,
+      "learning_rate": 6.270658824113884e-05,
+      "loss": 2.2732,
       "step": 110
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 3.8917089678510996e-05,
-      "loss": 2.293,
+      "epoch": 0.17,
+      "grad_norm": 0.07275390625,
+      "learning_rate": 6.251236424394651e-05,
+      "loss": 2.2867,
       "step": 115
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 4.060913705583756e-05,
-      "loss": 2.3071,
+      "epoch": 0.18,
+      "grad_norm": 0.0732421875,
+      "learning_rate": 6.229726650621257e-05,
+      "loss": 2.2718,
       "step": 120
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.138671875,
-      "learning_rate": 4.230118443316413e-05,
-      "loss": 2.273,
+      "epoch": 0.19,
+      "grad_norm": 0.0751953125,
+      "learning_rate": 6.20614409820754e-05,
+      "loss": 2.274,
       "step": 125
     },
     {
-      "epoch": 0.04,
-      "grad_norm": 0.134765625,
-      "learning_rate": 4.39932318104907e-05,
-      "loss": 2.2847,
+      "epoch": 0.19,
+      "grad_norm": 0.072265625,
+      "learning_rate": 6.180504769047129e-05,
+      "loss": 2.2688,
       "step": 130
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.134765625,
-      "learning_rate": 4.568527918781726e-05,
-      "loss": 2.2559,
+      "epoch": 0.2,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 6.152826060655387e-05,
+      "loss": 2.272,
       "step": 135
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.1328125,
-      "learning_rate": 4.737732656514383e-05,
-      "loss": 2.2525,
+      "epoch": 0.21,
+      "grad_norm": 0.07861328125,
+      "learning_rate": 6.123126754364366e-05,
+      "loss": 2.2807,
       "step": 140
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.1298828125,
-      "learning_rate": 4.906937394247039e-05,
-      "loss": 2.2716,
+      "epoch": 0.22,
+      "grad_norm": 0.072265625,
+      "learning_rate": 6.091427002578765e-05,
+      "loss": 2.2519,
       "step": 145
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.1318359375,
-      "learning_rate": 5.076142131979695e-05,
-      "loss": 2.2682,
+      "epoch": 0.22,
+      "grad_norm": 0.0732421875,
+      "learning_rate": 6.057748315101562e-05,
+      "loss": 2.2648,
       "step": 150
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.1279296875,
-      "learning_rate": 5.245346869712352e-05,
-      "loss": 2.2575,
+      "epoch": 0.23,
+      "grad_norm": 0.07177734375,
+      "learning_rate": 6.0221135445385774e-05,
+      "loss": 2.2739,
       "step": 155
     },
     {
-      "epoch": 0.05,
-      "grad_norm": 0.130859375,
-      "learning_rate": 5.4145516074450085e-05,
-      "loss": 2.2448,
+      "epoch": 0.24,
+      "grad_norm": 0.07470703125,
+      "learning_rate": 5.984546870791885e-05,
+      "loss": 2.2764,
       "step": 160
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.1318359375,
-      "learning_rate": 5.583756345177665e-05,
-      "loss": 2.211,
+      "epoch": 0.25,
+      "grad_norm": 0.0751953125,
+      "learning_rate": 5.945073784652589e-05,
+      "loss": 2.2514,
       "step": 165
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 5.752961082910322e-05,
-      "loss": 2.2194,
+      "epoch": 0.25,
+      "grad_norm": 0.078125,
+      "learning_rate": 5.9037210705040984e-05,
+      "loss": 2.2544,
       "step": 170
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.13671875,
-      "learning_rate": 5.9221658206429784e-05,
-      "loss": 2.2511,
+      "epoch": 0.26,
+      "grad_norm": 0.07470703125,
+      "learning_rate": 5.860516788147634e-05,
+      "loss": 2.2479,
       "step": 175
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.1357421875,
-      "learning_rate": 6.091370558375635e-05,
-      "loss": 2.2315,
+      "epoch": 0.27,
+      "grad_norm": 0.07763671875,
+      "learning_rate": 5.815490253762313e-05,
+      "loss": 2.253,
       "step": 180
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.142578125,
-      "learning_rate": 6.26057529610829e-05,
-      "loss": 2.2351,
+      "epoch": 0.28,
+      "grad_norm": 0.080078125,
+      "learning_rate": 5.7686720200127084e-05,
+      "loss": 2.2528,
       "step": 185
     },
     {
-      "epoch": 0.06,
-      "grad_norm": 0.140625,
-      "learning_rate": 6.429780033840948e-05,
-      "loss": 2.2085,
+      "epoch": 0.28,
+      "grad_norm": 0.0791015625,
+      "learning_rate": 5.7200938553174043e-05,
+      "loss": 2.253,
       "step": 190
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.140625,
-      "learning_rate": 6.598984771573604e-05,
-      "loss": 2.2269,
+      "epoch": 0.29,
+      "grad_norm": 0.078125,
+      "learning_rate": 5.669788722292595e-05,
+      "loss": 2.2571,
       "step": 195
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 6.768189509306261e-05,
-      "loss": 2.2031,
+      "epoch": 0.3,
+      "grad_norm": 0.0791015625,
+      "learning_rate": 5.617790755385372e-05,
+      "loss": 2.2666,
       "step": 200
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.14453125,
-      "learning_rate": 6.937394247038918e-05,
-      "loss": 2.212,
+      "epoch": 0.31,
+      "grad_norm": 0.07958984375,
+      "learning_rate": 5.5641352377118605e-05,
+      "loss": 2.2601,
       "step": 205
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.1376953125,
-      "learning_rate": 7.106598984771574e-05,
-      "loss": 2.2063,
+      "epoch": 0.31,
+      "grad_norm": 0.07861328125,
+      "learning_rate": 5.508858577115933e-05,
+      "loss": 2.2471,
       "step": 210
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 7.275803722504231e-05,
-      "loss": 2.2315,
+      "epoch": 0.32,
+      "grad_norm": 0.080078125,
+      "learning_rate": 5.451998281464741e-05,
+      "loss": 2.2523,
       "step": 215
     },
     {
-      "epoch": 0.07,
-      "grad_norm": 0.142578125,
-      "learning_rate": 7.445008460236887e-05,
-      "loss": 2.2089,
+      "epoch": 0.33,
+      "grad_norm": 0.07958984375,
+      "learning_rate": 5.393592933197822e-05,
+      "loss": 2.244,
       "step": 220
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 7.614213197969543e-05,
-      "loss": 2.2178,
+      "epoch": 0.34,
+      "grad_norm": 0.0791015625,
+      "learning_rate": 5.333682163147071e-05,
+      "loss": 2.242,
       "step": 225
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 7.783417935702199e-05,
-      "loss": 2.2294,
+      "epoch": 0.34,
+      "grad_norm": 0.0771484375,
+      "learning_rate": 5.2723066236453086e-05,
+      "loss": 2.2519,
       "step": 230
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.150390625,
-      "learning_rate": 7.952622673434857e-05,
-      "loss": 2.2275,
+      "epoch": 0.35,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 5.209507960941733e-05,
+      "loss": 2.2521,
       "step": 235
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.150390625,
-      "learning_rate": 8.121827411167512e-05,
-      "loss": 2.1849,
+      "epoch": 0.36,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 5.145328786942933e-05,
+      "loss": 2.2525,
       "step": 240
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.15234375,
-      "learning_rate": 8.29103214890017e-05,
-      "loss": 2.1791,
+      "epoch": 0.37,
+      "grad_norm": 0.07958984375,
+      "learning_rate": 5.0798126502986685e-05,
+      "loss": 2.2423,
       "step": 245
     },
     {
-      "epoch": 0.08,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 8.460236886632826e-05,
-      "loss": 2.2004,
+      "epoch": 0.37,
+      "grad_norm": 0.080078125,
+      "learning_rate": 5.013004006852019e-05,
+      "loss": 2.2444,
       "step": 250
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 8.629441624365483e-05,
-      "loss": 2.23,
+      "epoch": 0.38,
+      "grad_norm": 0.08154296875,
+      "learning_rate": 4.944948189473962e-05,
+      "loss": 2.2562,
       "step": 255
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 8.79864636209814e-05,
-      "loss": 2.2296,
+      "epoch": 0.39,
+      "grad_norm": 0.08203125,
+      "learning_rate": 4.875691377302846e-05,
+      "loss": 2.2443,
       "step": 260
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 8.967851099830795e-05,
-      "loss": 2.2362,
+      "epoch": 0.4,
+      "grad_norm": 0.0859375,
+      "learning_rate": 4.805280564409623e-05,
+      "loss": 2.2492,
       "step": 265
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 9.137055837563452e-05,
-      "loss": 2.2202,
+      "epoch": 0.4,
+      "grad_norm": 0.08154296875,
+      "learning_rate": 4.7337635279101233e-05,
+      "loss": 2.2402,
       "step": 270
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.15234375,
-      "learning_rate": 9.306260575296108e-05,
-      "loss": 2.1897,
+      "epoch": 0.41,
+      "grad_norm": 0.08203125,
+      "learning_rate": 4.661188795545985e-05,
+      "loss": 2.2402,
       "step": 275
     },
     {
-      "epoch": 0.09,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 9.475465313028765e-05,
-      "loss": 2.1788,
+      "epoch": 0.42,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 4.5876056127562524e-05,
+      "loss": 2.2464,
       "step": 280
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 9.644670050761421e-05,
-      "loss": 2.1784,
+      "epoch": 0.43,
+      "grad_norm": 0.0859375,
+      "learning_rate": 4.5130639092619825e-05,
+      "loss": 2.2518,
       "step": 285
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.150390625,
-      "learning_rate": 9.813874788494079e-05,
-      "loss": 2.1903,
+      "epoch": 0.43,
+      "grad_norm": 0.083984375,
+      "learning_rate": 4.437614265186536e-05,
+      "loss": 2.2491,
       "step": 290
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 9.983079526226735e-05,
-      "loss": 2.1736,
+      "epoch": 0.44,
+      "grad_norm": 0.08154296875,
+      "learning_rate": 4.361307876734529e-05,
+      "loss": 2.2474,
       "step": 295
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001015228426395939,
-      "loss": 2.1937,
+      "epoch": 0.45,
+      "grad_norm": 0.083984375,
+      "learning_rate": 4.2841965214527606e-05,
+      "loss": 2.2634,
       "step": 300
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00010321489001692048,
-      "loss": 2.1986,
+      "epoch": 0.46,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 4.206332523096655e-05,
+      "loss": 2.2433,
       "step": 305
     },
     {
-      "epoch": 0.1,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00010490693739424704,
-      "loss": 2.1945,
+      "epoch": 0.46,
+      "grad_norm": 0.083984375,
+      "learning_rate": 4.127768716126082e-05,
+      "loss": 2.2468,
       "step": 310
     },
     {
-      "epoch": 0.11,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00010659898477157362,
-      "loss": 2.2217,
+      "epoch": 0.47,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 4.0485584098546456e-05,
+      "loss": 2.2399,
       "step": 315
     },
     {
-      "epoch": 0.11,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00010829103214890017,
-      "loss": 2.2081,
+      "epoch": 0.48,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 3.968755352276755e-05,
+      "loss": 2.2431,
       "step": 320
     },
     {
-      "epoch": 0.11,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00010998307952622673,
-      "loss": 2.1782,
+      "epoch": 0.48,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 3.888413693597025e-05,
+      "loss": 2.2309,
       "step": 325
     },
     {
-      "epoch": 0.11,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001116751269035533,
-      "loss": 2.1853,
+      "epoch": 0.49,
+      "grad_norm": 0.08740234375,
+      "learning_rate": 3.8075879494867705e-05,
+      "loss": 2.2298,
       "step": 330
     },
     {
-      "epoch": 0.11,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00011336717428087986,
-      "loss": 2.1776,
+      "epoch": 0.5,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 3.726332964092504e-05,
+      "loss": 2.2443,
       "step": 335
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00011505922165820644,
-      "loss": 2.1847,
+      "epoch": 0.51,
+      "grad_norm": 0.09033203125,
+      "learning_rate": 3.644703872821547e-05,
+      "loss": 2.2481,
       "step": 340
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.000116751269035533,
-      "loss": 2.2162,
+      "epoch": 0.51,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 3.5627560649300175e-05,
+      "loss": 2.2375,
       "step": 345
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00011844331641285957,
-      "loss": 2.1783,
+      "epoch": 0.52,
+      "grad_norm": 0.087890625,
+      "learning_rate": 3.4805451459385544e-05,
+      "loss": 2.2368,
       "step": 350
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00012013536379018613,
-      "loss": 2.1957,
+      "epoch": 0.53,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 3.398126899901305e-05,
+      "loss": 2.2359,
       "step": 355
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.0001218274111675127,
-      "loss": 2.1976,
+      "epoch": 0.54,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 3.31555725155377e-05,
+      "loss": 2.2472,
       "step": 360
     },
     {
-      "epoch": 0.12,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00012351945854483927,
-      "loss": 2.2132,
+      "epoch": 0.54,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 3.232892228365181e-05,
+      "loss": 2.2511,
       "step": 365
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.0001252115059221658,
-      "loss": 2.1745,
+      "epoch": 0.55,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 3.15018792252118e-05,
+      "loss": 2.2273,
       "step": 370
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001269035532994924,
-      "loss": 2.1744,
+      "epoch": 0.56,
+      "grad_norm": 0.08740234375,
+      "learning_rate": 3.067500452862575e-05,
+      "loss": 2.2391,
       "step": 375
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012859560067681895,
-      "loss": 2.1901,
+      "epoch": 0.57,
+      "grad_norm": 0.08447265625,
+      "learning_rate": 2.984885926806012e-05,
+      "loss": 2.2424,
       "step": 380
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00013028764805414554,
-      "loss": 2.2184,
+      "epoch": 0.57,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 2.9024004022724027e-05,
+      "loss": 2.2435,
       "step": 385
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00013197969543147207,
-      "loss": 2.1992,
+      "epoch": 0.58,
+      "grad_norm": 0.087890625,
+      "learning_rate": 2.8200998496489373e-05,
+      "loss": 2.223,
       "step": 390
     },
     {
-      "epoch": 0.13,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00013367174280879866,
-      "loss": 2.1905,
+      "epoch": 0.59,
+      "grad_norm": 0.08447265625,
+      "learning_rate": 2.7380401138104845e-05,
+      "loss": 2.2235,
       "step": 395
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00013536379018612522,
-      "loss": 2.1837,
+      "epoch": 0.6,
+      "grad_norm": 0.08984375,
+      "learning_rate": 2.656276876226166e-05,
+      "loss": 2.2379,
       "step": 400
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013705583756345178,
-      "loss": 2.2044,
+      "epoch": 0.6,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 2.5748656171768127e-05,
+      "loss": 2.2232,
       "step": 405
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00013874788494077836,
-      "loss": 2.2061,
+      "epoch": 0.61,
+      "grad_norm": 0.0888671875,
+      "learning_rate": 2.4938615781089216e-05,
+      "loss": 2.2492,
       "step": 410
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001404399323181049,
-      "loss": 2.1866,
+      "epoch": 0.62,
+      "grad_norm": 0.08447265625,
+      "learning_rate": 2.413319724150689e-05,
+      "loss": 2.2429,
       "step": 415
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00014213197969543148,
-      "loss": 2.1918,
+      "epoch": 0.63,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 2.3332947068155364e-05,
+      "loss": 2.2486,
       "step": 420
     },
     {
-      "epoch": 0.14,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014382402707275804,
-      "loss": 2.1879,
+      "epoch": 0.63,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 2.2538408269184268e-05,
+      "loss": 2.2415,
       "step": 425
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00014551607445008463,
-      "loss": 2.1818,
+      "epoch": 0.64,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 2.1750119977301616e-05,
+      "loss": 2.2568,
       "step": 430
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00014720812182741116,
-      "loss": 2.164,
+      "epoch": 0.65,
+      "grad_norm": 0.0859375,
+      "learning_rate": 2.096861708394641e-05,
+      "loss": 2.235,
       "step": 435
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014890016920473775,
-      "loss": 2.1927,
+      "epoch": 0.66,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 2.0194429876339054e-05,
+      "loss": 2.2364,
       "step": 440
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.0001505922165820643,
-      "loss": 2.1951,
+      "epoch": 0.66,
+      "grad_norm": 0.0859375,
+      "learning_rate": 1.9428083677656066e-05,
+      "loss": 2.2439,
       "step": 445
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.166015625,
-      "learning_rate": 0.00015228426395939087,
-      "loss": 2.1963,
+      "epoch": 0.67,
+      "grad_norm": 0.0859375,
+      "learning_rate": 1.8670098490573132e-05,
+      "loss": 2.238,
       "step": 450
     },
     {
-      "epoch": 0.15,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00015397631133671742,
-      "loss": 2.1982,
+      "epoch": 0.68,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 1.792098864441825e-05,
+      "loss": 2.2325,
       "step": 455
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00015566835871404398,
-      "loss": 2.1704,
+      "epoch": 0.69,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 1.7181262446174615e-05,
+      "loss": 2.2342,
       "step": 460
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015736040609137057,
-      "loss": 2.1999,
+      "epoch": 0.69,
+      "grad_norm": 0.08740234375,
+      "learning_rate": 1.6451421835570044e-05,
+      "loss": 2.2262,
       "step": 465
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015905245346869713,
-      "loss": 2.1524,
+      "epoch": 0.7,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 1.5731962044486608e-05,
+      "loss": 2.2577,
       "step": 470
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00016074450084602372,
-      "loss": 2.1904,
+      "epoch": 0.71,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 1.5023371260922157e-05,
+      "loss": 2.2481,
       "step": 475
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00016243654822335025,
-      "loss": 2.1741,
+      "epoch": 0.72,
+      "grad_norm": 0.083984375,
+      "learning_rate": 1.4326130297731294e-05,
+      "loss": 2.229,
       "step": 480
     },
     {
-      "epoch": 0.16,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00016412859560067684,
-      "loss": 2.2073,
+      "epoch": 0.72,
+      "grad_norm": 0.08251953125,
+      "learning_rate": 1.3640712266370778e-05,
+      "loss": 2.2365,
       "step": 485
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001658206429780034,
-      "loss": 2.1577,
+      "epoch": 0.73,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 1.2967582255870662e-05,
+      "loss": 2.2276,
       "step": 490
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00016751269035532995,
-      "loss": 2.1828,
+      "epoch": 0.74,
+      "grad_norm": 0.08203125,
+      "learning_rate": 1.2307197017249163e-05,
+      "loss": 2.245,
       "step": 495
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00016920473773265651,
-      "loss": 2.1915,
+      "epoch": 0.75,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 1.166000465358504e-05,
+      "loss": 2.2327,
       "step": 500
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00017089678510998307,
-      "loss": 2.1873,
+      "epoch": 0.75,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 1.1026444315958248e-05,
+      "loss": 2.2257,
       "step": 505
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00017258883248730966,
-      "loss": 2.1937,
+      "epoch": 0.76,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 1.0406945905464832e-05,
+      "loss": 2.2355,
       "step": 510
     },
     {
-      "epoch": 0.17,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00017428087986463622,
-      "loss": 2.1888,
+      "epoch": 0.77,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 9.801929781508377e-06,
+      "loss": 2.2366,
       "step": 515
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001759729272419628,
-      "loss": 2.2072,
+      "epoch": 0.78,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 9.211806476565995e-06,
+      "loss": 2.2307,
       "step": 520
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00017766497461928934,
-      "loss": 2.2001,
+      "epoch": 0.78,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 8.63697641762235e-06,
+      "loss": 2.2298,
       "step": 525
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001793570219966159,
-      "loss": 2.193,
+      "epoch": 0.79,
+      "grad_norm": 0.0830078125,
+      "learning_rate": 8.077829654460684e-06,
+      "loss": 2.24,
       "step": 530
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00018104906937394248,
-      "loss": 2.1653,
+      "epoch": 0.8,
+      "grad_norm": 0.0869140625,
+      "learning_rate": 7.534745594995376e-06,
+      "loss": 2.2349,
       "step": 535
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00018274111675126904,
-      "loss": 2.191,
+      "epoch": 0.81,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 7.00809274782543e-06,
+      "loss": 2.2522,
       "step": 540
     },
     {
-      "epoch": 0.18,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001844331641285956,
-      "loss": 2.2017,
+      "epoch": 0.81,
+      "grad_norm": 0.087890625,
+      "learning_rate": 6.498228472183709e-06,
+      "loss": 2.2253,
       "step": 545
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00018612521150592216,
-      "loss": 2.1921,
+      "epoch": 0.82,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 6.00549873545155e-06,
+      "loss": 2.2319,
       "step": 550
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018781725888324875,
-      "loss": 2.1688,
+      "epoch": 0.83,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 5.530237878403297e-06,
+      "loss": 2.23,
       "step": 555
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001895093062605753,
-      "loss": 2.1696,
+      "epoch": 0.84,
+      "grad_norm": 0.08203125,
+      "learning_rate": 5.0727683883399965e-06,
+      "loss": 2.2366,
       "step": 560
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019120135363790187,
-      "loss": 2.1755,
+      "epoch": 0.84,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 4.63340068026638e-06,
+      "loss": 2.238,
       "step": 565
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00019289340101522843,
-      "loss": 2.2083,
+      "epoch": 0.85,
+      "grad_norm": 0.087890625,
+      "learning_rate": 4.212432886259269e-06,
+      "loss": 2.2238,
       "step": 570
     },
     {
-      "epoch": 0.19,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.000194585448392555,
-      "loss": 2.1612,
+      "epoch": 0.86,
+      "grad_norm": 0.0859375,
+      "learning_rate": 3.8101506531707373e-06,
+      "loss": 2.2308,
       "step": 575
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019627749576988157,
-      "loss": 2.1808,
+      "epoch": 0.87,
+      "grad_norm": 0.0859375,
+      "learning_rate": 3.426826948802979e-06,
+      "loss": 2.2305,
       "step": 580
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00019796954314720813,
-      "loss": 2.1557,
+      "epoch": 0.87,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 3.0627218766865635e-06,
+      "loss": 2.2364,
       "step": 585
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001996615905245347,
-      "loss": 2.1605,
+      "epoch": 0.88,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 2.7180824995877015e-06,
+      "loss": 2.2435,
       "step": 590
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00019999972028877317,
-      "loss": 2.1787,
+      "epoch": 0.89,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 2.3931426718643005e-06,
+      "loss": 2.2347,
       "step": 595
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019999858396459598,
-      "loss": 2.1948,
+      "epoch": 0.9,
+      "grad_norm": 0.08251953125,
+      "learning_rate": 2.0881228807845587e-06,
+      "loss": 2.2589,
       "step": 600
     },
     {
-      "epoch": 0.2,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019999657355544167,
-      "loss": 2.1524,
+      "epoch": 0.9,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 1.8032300969157738e-06,
+      "loss": 2.2323,
       "step": 605
     },
     {
-      "epoch": 0.21,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019999368907888313,
-      "loss": 2.1582,
+      "epoch": 0.91,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 1.5386576336848828e-06,
+      "loss": 2.2285,
       "step": 610
     },
     {
-      "epoch": 0.21,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001999899305601336,
-      "loss": 2.2172,
+      "epoch": 0.92,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 1.2945850162060271e-06,
+      "loss": 2.2368,
       "step": 615
     },
     {
-      "epoch": 0.21,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001999852980320461,
-      "loss": 2.1962,
+      "epoch": 0.93,
+      "grad_norm": 0.08251953125,
+      "learning_rate": 1.0711778594641267e-06,
+      "loss": 2.2417,
       "step": 620
     },
     {
-      "epoch": 0.21,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001999797915351135,
-      "loss": 2.1669,
+      "epoch": 0.93,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 8.685877559371869e-07,
+      "loss": 2.2355,
       "step": 625
     },
     {
-      "epoch": 0.21,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019997341111746791,
-      "loss": 2.1439,
+      "epoch": 0.94,
+      "grad_norm": 0.08544921875,
+      "learning_rate": 6.869521727335145e-07,
+      "loss": 2.2285,
       "step": 630
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019996615683488039,
-      "loss": 2.1757,
+      "epoch": 0.95,
+      "grad_norm": 0.08447265625,
+      "learning_rate": 5.26394358313689e-07,
+      "loss": 2.2262,
       "step": 635
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019995802875076042,
-      "loss": 2.1692,
+      "epoch": 0.95,
+      "grad_norm": 0.08154296875,
+      "learning_rate": 3.8702325886057816e-07,
+      "loss": 2.2289,
       "step": 640
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001999490269361554,
-      "loss": 2.1709,
+      "epoch": 0.96,
+      "grad_norm": 0.08642578125,
+      "learning_rate": 2.6893344435409695e-07,
+      "loss": 2.2428,
       "step": 645
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019993915146974996,
-      "loss": 2.1642,
+      "epoch": 0.97,
+      "grad_norm": 0.08740234375,
+      "learning_rate": 1.722050444009702e-07,
+      "loss": 2.2248,
       "step": 650
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019992840243786525,
-      "loss": 2.1699,
+      "epoch": 0.98,
+      "grad_norm": 0.08349609375,
+      "learning_rate": 9.690369386293054e-08,
+      "loss": 2.2297,
       "step": 655
     },
     {
-      "epoch": 0.22,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001999167799344583,
-      "loss": 2.1677,
+      "epoch": 0.98,
+      "grad_norm": 0.0849609375,
+      "learning_rate": 4.308048832030403e-08,
+      "loss": 2.2344,
       "step": 660
     },
     {
-      "epoch": 0.23,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001999042840611211,
-      "loss": 2.195,
+      "epoch": 0.99,
+      "grad_norm": 0.0859375,
+      "learning_rate": 1.0771949401241265e-08,
+      "loss": 2.2296,
       "step": 665
     },
     {
-      "epoch": 0.23,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019989091492707975,
-      "loss": 2.1986,
+      "epoch": 1.0,
+      "grad_norm": 0.087890625,
+      "learning_rate": 0.0,
+      "loss": 2.2321,
       "step": 670
     },
     {
-      "epoch": 0.23,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001998766726491935,
-      "loss": 2.1512,
-      "step": 675
-    },
-    {
-      "epoch": 0.23,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019986155735195372,
-      "loss": 2.1755,
-      "step": 680
-    },
-    {
-      "epoch": 0.23,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001998455691674828,
-      "loss": 2.1832,
-      "step": 685
-    },
-    {
-      "epoch": 0.23,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 0.00019982870823553308,
-      "loss": 2.1831,
-      "step": 690
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019981097470348548,
-      "loss": 2.1756,
-      "step": 695
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019979236872634838,
-      "loss": 2.1821,
-      "step": 700
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001997728904667561,
-      "loss": 2.193,
-      "step": 705
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.140625,
-      "learning_rate": 0.00019975254009496762,
-      "loss": 2.1561,
-      "step": 710
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.140625,
-      "learning_rate": 0.00019973131778886497,
-      "loss": 2.1596,
-      "step": 715
-    },
-    {
-      "epoch": 0.24,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019970922373395176,
-      "loss": 2.1796,
-      "step": 720
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019968625812335158,
-      "loss": 2.1946,
-      "step": 725
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019966242115780617,
-      "loss": 2.1836,
-      "step": 730
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019963771304567387,
-      "loss": 2.1734,
-      "step": 735
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019961213400292762,
-      "loss": 2.1611,
-      "step": 740
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00019958568425315314,
-      "loss": 2.188,
-      "step": 745
-    },
-    {
-      "epoch": 0.25,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.000199558364027547,
-      "loss": 2.1382,
-      "step": 750
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00019953017356491457,
-      "loss": 2.1972,
-      "step": 755
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001995011131116679,
-      "loss": 2.1808,
-      "step": 760
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019947118292182377,
-      "loss": 2.1648,
-      "step": 765
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019944038325700103,
-      "loss": 2.171,
-      "step": 770
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019940871438641882,
-      "loss": 2.1709,
-      "step": 775
-    },
-    {
-      "epoch": 0.26,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019937617658689384,
-      "loss": 2.1942,
-      "step": 780
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 0.0001993427701428382,
-      "loss": 2.1225,
-      "step": 785
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001993084953462567,
-      "loss": 2.1786,
-      "step": 790
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019927335249674447,
-      "loss": 2.1774,
-      "step": 795
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019923734190148418,
-      "loss": 2.1731,
-      "step": 800
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001992004638752435,
-      "loss": 2.1626,
-      "step": 805
-    },
-    {
-      "epoch": 0.27,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001991627187403723,
-      "loss": 2.1665,
-      "step": 810
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001991241068267998,
-      "loss": 2.1873,
-      "step": 815
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019908462847203172,
-      "loss": 2.1638,
-      "step": 820
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.0001990442840211473,
-      "loss": 2.1676,
-      "step": 825
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019900307382679638,
-      "loss": 2.1379,
-      "step": 830
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.1376953125,
-      "learning_rate": 0.00019896099824919604,
-      "loss": 2.1727,
-      "step": 835
-    },
-    {
-      "epoch": 0.28,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019891805765612794,
-      "loss": 2.1587,
-      "step": 840
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001988742524229346,
-      "loss": 2.1581,
-      "step": 845
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019882958293251636,
-      "loss": 2.1801,
-      "step": 850
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.138671875,
-      "learning_rate": 0.00019878404957532814,
-      "loss": 2.1448,
-      "step": 855
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019873765274937578,
-      "loss": 2.1585,
-      "step": 860
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019869039286021271,
-      "loss": 2.1749,
-      "step": 865
-    },
-    {
-      "epoch": 0.29,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019864227032093633,
-      "loss": 2.1607,
-      "step": 870
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019859328555218455,
-      "loss": 2.1621,
-      "step": 875
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001985434389821319,
-      "loss": 2.1608,
-      "step": 880
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019849273104648592,
-      "loss": 2.1601,
-      "step": 885
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019844116218848334,
-      "loss": 2.1498,
-      "step": 890
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001983887328588862,
-      "loss": 2.1895,
-      "step": 895
-    },
-    {
-      "epoch": 0.3,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019833544351597788,
-      "loss": 2.1391,
-      "step": 900
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001982812946255591,
-      "loss": 2.1489,
-      "step": 905
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001982262866609439,
-      "loss": 2.1337,
-      "step": 910
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019817042010295544,
-      "loss": 2.1387,
-      "step": 915
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019811369543992183,
-      "loss": 2.1662,
-      "step": 920
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001980561131676718,
-      "loss": 2.1467,
-      "step": 925
-    },
-    {
-      "epoch": 0.31,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001979976737895305,
-      "loss": 2.1823,
-      "step": 930
-    },
-    {
-      "epoch": 0.32,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019793837781631506,
-      "loss": 2.1764,
-      "step": 935
-    },
-    {
-      "epoch": 0.32,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001978782257663299,
-      "loss": 2.1521,
-      "step": 940
-    },
-    {
-      "epoch": 0.32,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019781721816536257,
-      "loss": 2.1694,
-      "step": 945
-    },
-    {
-      "epoch": 0.32,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019775535554667886,
-      "loss": 2.1781,
-      "step": 950
-    },
-    {
-      "epoch": 0.32,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019769263845101828,
-      "loss": 2.1894,
-      "step": 955
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 0.00019762906742658935,
-      "loss": 2.1465,
-      "step": 960
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00019756464302906465,
-      "loss": 2.1602,
-      "step": 965
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001974993658215762,
-      "loss": 2.161,
-      "step": 970
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019743323637471028,
-      "loss": 2.1644,
-      "step": 975
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.140625,
-      "learning_rate": 0.00019736625526650269,
-      "loss": 2.1671,
-      "step": 980
-    },
-    {
-      "epoch": 0.33,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001972984230824335,
-      "loss": 2.2012,
-      "step": 985
-    },
-    {
-      "epoch": 0.34,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019722974041542203,
-      "loss": 2.1562,
-      "step": 990
-    },
-    {
-      "epoch": 0.34,
-      "grad_norm": 0.1396484375,
-      "learning_rate": 0.00019716020786582169,
-      "loss": 2.1351,
-      "step": 995
+      "epoch": 1.0,
+      "eval_loss": 2.2356441020965576,
+      "eval_runtime": 186.994,
+      "eval_samples_per_second": 25.787,
+      "eval_steps_per_second": 3.225,
+      "step": 670
     },
     {
-      "epoch": 0.34,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001970898260414146,
-      "loss": 2.1662,
-      "step": 1000
-    },
-    {
-      "epoch": 0.34,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00019701859555740648,
-      "loss": 2.1601,
-      "step": 1005
-    },
-    {
-      "epoch": 0.34,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00019694651703642104,
-      "loss": 2.1325,
-      "step": 1010
-    },
-    {
-      "epoch": 0.34,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001968735911084948,
-      "loss": 2.1698,
-      "step": 1015
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001967998184110713,
-      "loss": 2.1379,
-      "step": 1020
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00019672519958899583,
-      "loss": 2.1447,
-      "step": 1025
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019664973529450946,
-      "loss": 2.1684,
-      "step": 1030
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019657342618724358,
-      "loss": 2.1496,
-      "step": 1035
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019649627293421413,
-      "loss": 2.1179,
-      "step": 1040
-    },
-    {
-      "epoch": 0.35,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019641827620981564,
-      "loss": 2.1262,
-      "step": 1045
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001963394366958154,
-      "loss": 2.1477,
-      "step": 1050
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019625975508134755,
-      "loss": 2.1596,
-      "step": 1055
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019617923206290692,
-      "loss": 2.165,
-      "step": 1060
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00019609786834434313,
-      "loss": 2.1671,
-      "step": 1065
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019601566463685425,
-      "loss": 2.1638,
-      "step": 1070
-    },
-    {
-      "epoch": 0.36,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019593262165898076,
-      "loss": 2.1579,
-      "step": 1075
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001958487401365991,
-      "loss": 2.1628,
-      "step": 1080
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019576402080291545,
-      "loss": 2.1963,
-      "step": 1085
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019567846439845927,
-      "loss": 2.1792,
-      "step": 1090
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019559207167107684,
-      "loss": 2.1701,
-      "step": 1095
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019550484337592464,
-      "loss": 2.1511,
-      "step": 1100
-    },
-    {
-      "epoch": 0.37,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019541678027546296,
-      "loss": 2.1653,
-      "step": 1105
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.140625,
-      "learning_rate": 0.00019532788313944904,
-      "loss": 2.1728,
-      "step": 1110
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019523815274493031,
-      "loss": 2.1703,
-      "step": 1115
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00019514758987623784,
-      "loss": 2.1723,
-      "step": 1120
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019505619532497926,
-      "loss": 2.174,
-      "step": 1125
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019496396989003193,
-      "loss": 2.1431,
-      "step": 1130
-    },
-    {
-      "epoch": 0.38,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019487091437753594,
-      "loss": 2.158,
-      "step": 1135
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019477702960088702,
-      "loss": 2.1763,
-      "step": 1140
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001946823163807296,
-      "loss": 2.1655,
-      "step": 1145
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019458677554494932,
-      "loss": 2.1646,
-      "step": 1150
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001944904079286662,
-      "loss": 2.1502,
-      "step": 1155
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019439321437422695,
-      "loss": 2.1562,
-      "step": 1160
-    },
-    {
-      "epoch": 0.39,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00019429519573119794,
-      "loss": 2.1559,
-      "step": 1165
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00019419635285635746,
-      "loss": 2.1707,
-      "step": 1170
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001940966866136885,
-      "loss": 2.1393,
-      "step": 1175
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019399619787437104,
-      "loss": 2.1421,
-      "step": 1180
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001938948875167745,
-      "loss": 2.1468,
-      "step": 1185
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019379275642645002,
-      "loss": 2.1617,
-      "step": 1190
-    },
-    {
-      "epoch": 0.4,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001936898054961228,
-      "loss": 2.1828,
-      "step": 1195
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00019358603562568416,
-      "loss": 2.1237,
-      "step": 1200
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001934814477221838,
-      "loss": 2.1469,
-      "step": 1205
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.140625,
-      "learning_rate": 0.0001933760426998218,
-      "loss": 2.1826,
-      "step": 1210
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001932698214799407,
-      "loss": 2.1484,
-      "step": 1215
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001931627849910174,
-      "loss": 2.1428,
-      "step": 1220
-    },
-    {
-      "epoch": 0.41,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00019305493416865493,
-      "loss": 2.1281,
-      "step": 1225
-    },
-    {
-      "epoch": 0.42,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019294626995557457,
-      "loss": 2.161,
-      "step": 1230
-    },
-    {
-      "epoch": 0.42,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00019283679330160726,
-      "loss": 2.1455,
-      "step": 1235
-    },
-    {
-      "epoch": 0.42,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001927265051636856,
-      "loss": 2.1554,
-      "step": 1240
-    },
-    {
-      "epoch": 0.42,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019261540650583522,
-      "loss": 2.1855,
-      "step": 1245
-    },
-    {
-      "epoch": 0.42,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.00019250349829916661,
-      "loss": 2.1403,
-      "step": 1250
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001923907815218664,
-      "loss": 2.1797,
-      "step": 1255
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.1416015625,
-      "learning_rate": 0.00019227725715918897,
-      "loss": 2.1369,
-      "step": 1260
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00019216292620344777,
-      "loss": 2.1737,
-      "step": 1265
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00019204778965400667,
-      "loss": 2.1471,
-      "step": 1270
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001919318485172712,
-      "loss": 2.1757,
-      "step": 1275
-    },
-    {
-      "epoch": 0.43,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00019181510380667977,
-      "loss": 2.1649,
-      "step": 1280
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001916975565426948,
-      "loss": 2.1699,
-      "step": 1285
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019157920775279383,
-      "loss": 2.1468,
-      "step": 1290
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001914600584714605,
-      "loss": 2.1455,
-      "step": 1295
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001913401097401755,
-      "loss": 2.1362,
-      "step": 1300
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019121936260740752,
-      "loss": 2.176,
-      "step": 1305
-    },
-    {
-      "epoch": 0.44,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.0001910978181286041,
-      "loss": 2.1609,
-      "step": 1310
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00019097547736618228,
-      "loss": 2.1566,
-      "step": 1315
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001908523413895194,
-      "loss": 2.1392,
-      "step": 1320
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001907284112749438,
-      "loss": 2.1392,
-      "step": 1325
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00019060368810572539,
-      "loss": 2.1563,
-      "step": 1330
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00019047817297206598,
-      "loss": 2.1203,
-      "step": 1335
-    },
-    {
-      "epoch": 0.45,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00019035186697109011,
-      "loss": 2.1533,
-      "step": 1340
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.0001902247712068352,
-      "loss": 2.1503,
-      "step": 1345
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001900968867902419,
-      "loss": 2.1798,
-      "step": 1350
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001899682148391446,
-      "loss": 2.1541,
-      "step": 1355
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00018983875647826136,
-      "loss": 2.1286,
-      "step": 1360
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018970851283918428,
-      "loss": 2.1471,
-      "step": 1365
-    },
-    {
-      "epoch": 0.46,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00018957748506036957,
-      "loss": 2.137,
-      "step": 1370
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018944567428712765,
-      "loss": 2.1514,
-      "step": 1375
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001893130816716129,
-      "loss": 2.1667,
-      "step": 1380
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00018917970837281392,
-      "loss": 2.1724,
-      "step": 1385
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018904555555654317,
-      "loss": 2.1847,
-      "step": 1390
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001889106243954269,
-      "loss": 2.157,
-      "step": 1395
-    },
-    {
-      "epoch": 0.47,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018877491606889476,
-      "loss": 2.1312,
-      "step": 1400
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001886384317631697,
-      "loss": 2.1694,
-      "step": 1405
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018850117267125738,
-      "loss": 2.1772,
-      "step": 1410
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00018836313999293593,
-      "loss": 2.1599,
-      "step": 1415
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00018822433493474532,
-      "loss": 2.1282,
-      "step": 1420
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001880847587099769,
-      "loss": 2.1316,
-      "step": 1425
-    },
-    {
-      "epoch": 0.48,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018794441253866274,
-      "loss": 2.17,
-      "step": 1430
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018780329764756505,
-      "loss": 2.1501,
-      "step": 1435
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018766141527016533,
-      "loss": 2.1335,
-      "step": 1440
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00018751876664665367,
-      "loss": 2.1353,
-      "step": 1445
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00018737535302391795,
-      "loss": 2.1778,
-      "step": 1450
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018723117565553284,
-      "loss": 2.1515,
-      "step": 1455
-    },
-    {
-      "epoch": 0.49,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018708623580174889,
-      "loss": 2.1433,
-      "step": 1460
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00018694053472948156,
-      "loss": 2.1386,
-      "step": 1465
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00018679407371230002,
-      "loss": 2.1572,
-      "step": 1470
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00018664685403041619,
-      "loss": 2.1624,
-      "step": 1475
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.142578125,
-      "learning_rate": 0.0001864988769706734,
-      "loss": 2.162,
-      "step": 1480
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001863501438265352,
-      "loss": 2.1513,
-      "step": 1485
-    },
-    {
-      "epoch": 0.5,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018620065589807413,
-      "loss": 2.165,
-      "step": 1490
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00018605041449196012,
-      "loss": 2.18,
-      "step": 1495
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.14453125,
-      "learning_rate": 0.00018589942092144942,
-      "loss": 2.1805,
-      "step": 1500
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00018574767650637278,
-      "loss": 2.114,
-      "step": 1505
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001855951825731241,
-      "loss": 2.1331,
-      "step": 1510
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.1435546875,
-      "learning_rate": 0.00018544194045464886,
-      "loss": 2.1368,
-      "step": 1515
-    },
-    {
-      "epoch": 0.51,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018528795149043236,
-      "loss": 2.1579,
-      "step": 1520
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018513321702648807,
-      "loss": 2.1433,
-      "step": 1525
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001849777384153458,
-      "loss": 2.1542,
-      "step": 1530
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00018482151701604003,
-      "loss": 2.163,
-      "step": 1535
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00018466455419409786,
-      "loss": 2.1541,
-      "step": 1540
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001845068513215271,
-      "loss": 2.143,
-      "step": 1545
-    },
-    {
-      "epoch": 0.52,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018434840977680453,
-      "loss": 2.1377,
-      "step": 1550
-    },
-    {
-      "epoch": 0.53,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018418923094486338,
-      "loss": 2.1539,
-      "step": 1555
-    },
-    {
-      "epoch": 0.53,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00018402931621708165,
-      "loss": 2.161,
-      "step": 1560
-    },
-    {
-      "epoch": 0.53,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018386866699126973,
-      "loss": 2.1789,
-      "step": 1565
-    },
-    {
-      "epoch": 0.53,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018370728467165828,
-      "loss": 2.1419,
-      "step": 1570
-    },
-    {
-      "epoch": 0.53,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001835451706688859,
-      "loss": 2.1748,
-      "step": 1575
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001833823263999867,
-      "loss": 2.127,
-      "step": 1580
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018321875328837828,
-      "loss": 2.132,
-      "step": 1585
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018305445276384875,
-      "loss": 2.1438,
-      "step": 1590
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018288942626254473,
-      "loss": 2.1572,
-      "step": 1595
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018272367522695844,
-      "loss": 2.1464,
-      "step": 1600
-    },
-    {
-      "epoch": 0.54,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.00018255720110591533,
-      "loss": 2.157,
-      "step": 1605
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001823900053545613,
-      "loss": 2.1392,
-      "step": 1610
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00018222208943434999,
-      "loss": 2.1497,
-      "step": 1615
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018205345481302998,
-      "loss": 2.1474,
-      "step": 1620
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001818841029646321,
-      "loss": 2.153,
-      "step": 1625
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00018171403536945628,
-      "loss": 2.1696,
-      "step": 1630
-    },
-    {
-      "epoch": 0.55,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018154325351405897,
-      "loss": 2.1849,
-      "step": 1635
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00018137175889123978,
-      "loss": 2.1445,
-      "step": 1640
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001811995530000287,
-      "loss": 2.1402,
-      "step": 1645
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018102663734567283,
-      "loss": 2.1413,
-      "step": 1650
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001808530134396234,
-      "loss": 2.1554,
-      "step": 1655
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00018067868279952236,
-      "loss": 2.1374,
-      "step": 1660
-    },
-    {
-      "epoch": 0.56,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001805036469491892,
-      "loss": 2.167,
-      "step": 1665
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00018032790741860763,
-      "loss": 2.1369,
-      "step": 1670
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00018015146574391233,
-      "loss": 2.1283,
-      "step": 1675
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00017997432346737524,
-      "loss": 2.1505,
-      "step": 1680
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017979648213739232,
-      "loss": 2.1338,
-      "step": 1685
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017961794330846994,
-      "loss": 2.1555,
-      "step": 1690
-    },
-    {
-      "epoch": 0.57,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00017943870854121124,
-      "loss": 2.1543,
-      "step": 1695
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001792587794023027,
-      "loss": 2.1429,
-      "step": 1700
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017907815746450004,
-      "loss": 2.1174,
-      "step": 1705
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001788968443066149,
-      "loss": 2.1252,
-      "step": 1710
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017871484151350077,
-      "loss": 2.1074,
-      "step": 1715
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017853215067603926,
-      "loss": 2.1919,
-      "step": 1720
-    },
-    {
-      "epoch": 0.58,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017834877339112612,
-      "loss": 2.126,
-      "step": 1725
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001781647112616574,
-      "loss": 2.1564,
-      "step": 1730
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001779799658965153,
-      "loss": 2.137,
-      "step": 1735
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00017779453891055412,
-      "loss": 2.1714,
-      "step": 1740
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00017760843192458626,
-      "loss": 2.1456,
-      "step": 1745
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00017742164656536798,
-      "loss": 2.1576,
-      "step": 1750
-    },
-    {
-      "epoch": 0.59,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00017723418446558516,
-      "loss": 2.1461,
-      "step": 1755
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00017704604726383904,
-      "loss": 2.1383,
-      "step": 1760
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017685723660463193,
-      "loss": 2.13,
-      "step": 1765
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00017666775413835282,
-      "loss": 2.1463,
-      "step": 1770
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001764776015212629,
-      "loss": 2.1076,
-      "step": 1775
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001762867804154812,
-      "loss": 2.1439,
-      "step": 1780
-    },
-    {
-      "epoch": 0.6,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017609529248896997,
-      "loss": 2.1802,
-      "step": 1785
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00017590313941552002,
-      "loss": 2.1747,
-      "step": 1790
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00017571032287473642,
-      "loss": 2.1198,
-      "step": 1795
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00017551684455202336,
-      "loss": 2.1279,
-      "step": 1800
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017532270613856976,
-      "loss": 2.1531,
-      "step": 1805
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017512790933133437,
-      "loss": 2.1273,
-      "step": 1810
-    },
-    {
-      "epoch": 0.61,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001749324558330309,
-      "loss": 2.1526,
-      "step": 1815
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.0001747363473521132,
-      "loss": 2.1484,
-      "step": 1820
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00017453958560276038,
-      "loss": 2.1475,
-      "step": 1825
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017434217230486164,
-      "loss": 2.1606,
-      "step": 1830
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001741441091840014,
-      "loss": 2.1659,
-      "step": 1835
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00017394539797144413,
-      "loss": 2.1168,
-      "step": 1840
-    },
-    {
-      "epoch": 0.62,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00017374604040411935,
-      "loss": 2.1317,
-      "step": 1845
-    },
-    {
-      "epoch": 0.63,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00017354603822460621,
-      "loss": 2.1322,
-      "step": 1850
-    },
-    {
-      "epoch": 0.63,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00017334539318111856,
-      "loss": 2.1417,
-      "step": 1855
-    },
-    {
-      "epoch": 0.63,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00017314410702748932,
-      "loss": 2.1343,
-      "step": 1860
-    },
-    {
-      "epoch": 0.63,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00017294218152315546,
-      "loss": 2.1471,
-      "step": 1865
-    },
-    {
-      "epoch": 0.63,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017273961843314252,
-      "loss": 2.147,
-      "step": 1870
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.0001725364195280491,
-      "loss": 2.1529,
-      "step": 1875
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00017233258658403138,
-      "loss": 2.1462,
-      "step": 1880
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001721281213827878,
-      "loss": 2.1401,
-      "step": 1885
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017192302571154331,
-      "loss": 2.1614,
-      "step": 1890
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00017171730136303364,
-      "loss": 2.1488,
-      "step": 1895
-    },
-    {
-      "epoch": 0.64,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00017151095013548994,
-      "loss": 2.1445,
-      "step": 1900
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00017130397383262284,
-      "loss": 2.125,
-      "step": 1905
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.1455078125,
-      "learning_rate": 0.0001710963742636067,
-      "loss": 2.1351,
-      "step": 1910
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017088815324306392,
-      "loss": 2.1285,
-      "step": 1915
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00017067931259104885,
-      "loss": 2.1599,
-      "step": 1920
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00017046985413303215,
-      "loss": 2.122,
-      "step": 1925
-    },
-    {
-      "epoch": 0.65,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00017025977969988465,
-      "loss": 2.1373,
-      "step": 1930
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00017004909112786144,
-      "loss": 2.1298,
-      "step": 1935
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001698377902585857,
-      "loss": 2.1379,
-      "step": 1940
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00016962587893903276,
-      "loss": 2.1501,
-      "step": 1945
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001694133590215139,
-      "loss": 2.1611,
-      "step": 1950
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00016920023236366002,
-      "loss": 2.1135,
-      "step": 1955
-    },
-    {
-      "epoch": 0.66,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00016898650082840572,
-      "loss": 2.1337,
-      "step": 1960
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00016877216628397257,
-      "loss": 2.1258,
-      "step": 1965
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001685572306038532,
-      "loss": 2.1085,
-      "step": 1970
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.0001683416956667947,
-      "loss": 2.1549,
-      "step": 1975
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001681255633567823,
-      "loss": 2.1572,
-      "step": 1980
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00016790883556302272,
-      "loss": 2.152,
-      "step": 1985
-    },
-    {
-      "epoch": 0.67,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00016769151417992791,
-      "loss": 2.1427,
-      "step": 1990
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00016747360110709838,
-      "loss": 2.1405,
-      "step": 1995
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00016725509824930645,
-      "loss": 2.1496,
-      "step": 2000
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001670360075164799,
-      "loss": 2.1588,
-      "step": 2005
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00016681633082368498,
-      "loss": 2.1569,
-      "step": 2010
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00016659607009110984,
-      "loss": 2.1429,
-      "step": 2015
-    },
-    {
-      "epoch": 0.68,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00016637522724404774,
-      "loss": 2.1548,
-      "step": 2020
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00016615380421288018,
-      "loss": 2.1186,
-      "step": 2025
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00016593180293306001,
-      "loss": 2.1406,
-      "step": 2030
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001657092253450945,
-      "loss": 2.1324,
-      "step": 2035
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00016548607339452853,
-      "loss": 2.1107,
-      "step": 2040
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00016526234903192733,
-      "loss": 2.1533,
-      "step": 2045
-    },
-    {
-      "epoch": 0.69,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00016503805421285968,
-      "loss": 2.1516,
-      "step": 2050
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00016481319089788063,
-      "loss": 2.1283,
-      "step": 2055
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00016458776105251447,
-      "loss": 2.1586,
-      "step": 2060
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.0001643617666472376,
-      "loss": 2.1291,
-      "step": 2065
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00016413520965746097,
-      "loss": 2.1661,
-      "step": 2070
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001639080920635134,
-      "loss": 2.1415,
-      "step": 2075
-    },
-    {
-      "epoch": 0.7,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.0001636804158506237,
-      "loss": 2.1533,
-      "step": 2080
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00016345218300890357,
-      "loss": 2.1246,
-      "step": 2085
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00016322339553333034,
-      "loss": 2.1427,
-      "step": 2090
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00016299405542372924,
-      "loss": 2.1526,
-      "step": 2095
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00016276416468475607,
-      "loss": 2.1265,
-      "step": 2100
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00016253372532587976,
-      "loss": 2.1505,
-      "step": 2105
-    },
-    {
-      "epoch": 0.71,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001623027393613646,
-      "loss": 2.1563,
-      "step": 2110
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00016207120881025282,
-      "loss": 2.1499,
-      "step": 2115
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.0001618391356963468,
-      "loss": 2.1095,
-      "step": 2120
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00016160652204819158,
-      "loss": 2.144,
-      "step": 2125
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001613733698990568,
-      "loss": 2.1447,
-      "step": 2130
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00016113968128691933,
-      "loss": 2.1404,
-      "step": 2135
-    },
-    {
-      "epoch": 0.72,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00016090545825444506,
-      "loss": 2.162,
-      "step": 2140
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.1494140625,
-      "learning_rate": 0.00016067070284897137,
-      "loss": 2.1605,
-      "step": 2145
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.000160435417122489,
-      "loss": 2.1262,
-      "step": 2150
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00016019960313162434,
-      "loss": 2.1194,
-      "step": 2155
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001599632629376212,
-      "loss": 2.1531,
-      "step": 2160
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00015972639860632292,
-      "loss": 2.1829,
-      "step": 2165
-    },
-    {
-      "epoch": 0.73,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00015948901220815445,
-      "loss": 2.164,
-      "step": 2170
-    },
-    {
-      "epoch": 0.74,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00015925110581810394,
-      "loss": 2.1432,
-      "step": 2175
-    },
-    {
-      "epoch": 0.74,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00015901268151570491,
-      "loss": 2.1862,
-      "step": 2180
-    },
-    {
-      "epoch": 0.74,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.0001587737413850178,
-      "loss": 2.1313,
-      "step": 2185
-    },
-    {
-      "epoch": 0.74,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015853428751461202,
-      "loss": 2.1317,
-      "step": 2190
-    },
-    {
-      "epoch": 0.74,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015829432199754756,
-      "loss": 2.1345,
-      "step": 2195
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001580538469313566,
-      "loss": 2.1354,
-      "step": 2200
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00015781286441802534,
-      "loss": 2.135,
-      "step": 2205
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00015757137656397557,
-      "loss": 2.1041,
-      "step": 2210
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001573293854800462,
-      "loss": 2.1074,
-      "step": 2215
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00015708689328147493,
-      "loss": 2.1311,
-      "step": 2220
-    },
-    {
-      "epoch": 0.75,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00015684390208787962,
-      "loss": 2.143,
-      "step": 2225
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001566004140232399,
-      "loss": 2.1528,
-      "step": 2230
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.1474609375,
-      "learning_rate": 0.00015635643121587848,
-      "loss": 2.1641,
-      "step": 2235
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00015611195579844265,
-      "loss": 2.1366,
-      "step": 2240
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015586698990788554,
-      "loss": 2.1539,
-      "step": 2245
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.1484375,
-      "learning_rate": 0.00015562153568544752,
-      "loss": 2.1306,
-      "step": 2250
-    },
-    {
-      "epoch": 0.76,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015537559527663744,
-      "loss": 2.1832,
-      "step": 2255
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015512917083121397,
-      "loss": 2.1523,
-      "step": 2260
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015488226450316664,
-      "loss": 2.1115,
-      "step": 2265
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00015463487845069707,
-      "loss": 2.1527,
-      "step": 2270
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00015438701483620027,
-      "loss": 2.1486,
-      "step": 2275
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015413867582624553,
-      "loss": 2.1297,
-      "step": 2280
-    },
-    {
-      "epoch": 0.77,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015388986359155758,
-      "loss": 2.1407,
-      "step": 2285
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.0001536405803069975,
-      "loss": 2.1033,
-      "step": 2290
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00015339082815154394,
-      "loss": 2.1388,
-      "step": 2295
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00015314060930827393,
-      "loss": 2.1755,
-      "step": 2300
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.0001528899259643437,
-      "loss": 2.1607,
-      "step": 2305
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00015263878031096975,
-      "loss": 2.1443,
-      "step": 2310
-    },
-    {
-      "epoch": 0.78,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015238717454340957,
-      "loss": 2.1384,
-      "step": 2315
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00015213511086094254,
-      "loss": 2.1205,
-      "step": 2320
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015188259146685064,
-      "loss": 2.1265,
-      "step": 2325
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001516296185683992,
-      "loss": 2.1564,
-      "step": 2330
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015137619437681767,
-      "loss": 2.1287,
-      "step": 2335
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00015112232110728015,
-      "loss": 2.1516,
-      "step": 2340
-    },
-    {
-      "epoch": 0.79,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00015086800097888624,
-      "loss": 2.1425,
-      "step": 2345
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00015061323621464134,
-      "loss": 2.1345,
-      "step": 2350
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00015035802904143762,
-      "loss": 2.1628,
-      "step": 2355
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001501023816900342,
-      "loss": 2.1679,
-      "step": 2360
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014984629639503785,
-      "loss": 2.1477,
-      "step": 2365
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.0001495897753948833,
-      "loss": 2.1486,
-      "step": 2370
-    },
-    {
-      "epoch": 0.8,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00014933282093181383,
-      "loss": 2.1509,
-      "step": 2375
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014907543525186166,
-      "loss": 2.1244,
-      "step": 2380
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00014881762060482814,
-      "loss": 2.123,
-      "step": 2385
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00014855937924426434,
-      "loss": 2.1334,
-      "step": 2390
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014830071342745112,
-      "loss": 2.1473,
-      "step": 2395
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00014804162541537955,
-      "loss": 2.1544,
-      "step": 2400
-    },
-    {
-      "epoch": 0.81,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014778211747273114,
-      "loss": 2.1675,
-      "step": 2405
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.146484375,
-      "learning_rate": 0.00014752219186785784,
-      "loss": 2.1268,
-      "step": 2410
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001472618508727626,
-      "loss": 2.1661,
-      "step": 2415
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00014700109676307914,
-      "loss": 2.1348,
-      "step": 2420
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.0001467399318180522,
-      "loss": 2.1568,
-      "step": 2425
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001464783583205177,
-      "loss": 2.1437,
-      "step": 2430
-    },
-    {
-      "epoch": 0.82,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.0001462163785568826,
-      "loss": 2.1418,
-      "step": 2435
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00014595399481710515,
-      "loss": 2.1295,
-      "step": 2440
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00014569120939467465,
-      "loss": 2.1444,
-      "step": 2445
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014542802458659152,
-      "loss": 2.1483,
-      "step": 2450
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.0001451644426933472,
-      "loss": 2.1281,
-      "step": 2455
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00014490046601890405,
-      "loss": 2.1445,
-      "step": 2460
-    },
-    {
-      "epoch": 0.83,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014463609687067526,
-      "loss": 2.1529,
-      "step": 2465
-    },
-    {
-      "epoch": 0.84,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014437133755950448,
-      "loss": 2.154,
-      "step": 2470
-    },
-    {
-      "epoch": 0.84,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00014410619039964586,
-      "loss": 2.1333,
-      "step": 2475
-    },
-    {
-      "epoch": 0.84,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00014384065770874373,
-      "loss": 2.1382,
-      "step": 2480
-    },
-    {
-      "epoch": 0.84,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014357474180781232,
-      "loss": 2.1293,
-      "step": 2485
-    },
-    {
-      "epoch": 0.84,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014330844502121547,
-      "loss": 2.1312,
-      "step": 2490
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00014304176967664637,
-      "loss": 2.1036,
-      "step": 2495
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001427747181051071,
-      "loss": 2.127,
-      "step": 2500
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014250729264088843,
-      "loss": 2.1564,
-      "step": 2505
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00014223949562154929,
-      "loss": 2.1459,
-      "step": 2510
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00014197132938789629,
-      "loss": 2.1222,
-      "step": 2515
-    },
-    {
-      "epoch": 0.85,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.0001417027962839634,
-      "loss": 2.1501,
-      "step": 2520
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00014143389865699132,
-      "loss": 2.1567,
-      "step": 2525
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00014116463885740723,
-      "loss": 2.1289,
-      "step": 2530
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00014089501923880384,
-      "loss": 2.1546,
-      "step": 2535
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00014062504215791905,
-      "loss": 2.1307,
-      "step": 2540
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00014035470997461548,
-      "loss": 2.1206,
-      "step": 2545
-    },
-    {
-      "epoch": 0.86,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00014008402505185952,
-      "loss": 2.1112,
-      "step": 2550
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.000139812989755701,
-      "loss": 2.1233,
-      "step": 2555
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00013954160645525217,
-      "loss": 2.1254,
-      "step": 2560
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00013926987752266735,
-      "loss": 2.1362,
-      "step": 2565
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.0001389978053331219,
-      "loss": 2.1494,
-      "step": 2570
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00013872539226479172,
-      "loss": 2.139,
-      "step": 2575
-    },
-    {
-      "epoch": 0.87,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00013845264069883216,
-      "loss": 2.1307,
-      "step": 2580
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013817955301935743,
-      "loss": 2.1316,
-      "step": 2585
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001379061316134198,
-      "loss": 2.1408,
-      "step": 2590
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00013763237887098843,
-      "loss": 2.1299,
-      "step": 2595
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001373582971849289,
-      "loss": 2.1311,
-      "step": 2600
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00013708388895098192,
-      "loss": 2.1357,
-      "step": 2605
-    },
-    {
-      "epoch": 0.88,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013680915656774265,
-      "loss": 2.1451,
-      "step": 2610
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00013653410243663952,
-      "loss": 2.1327,
-      "step": 2615
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00013625872896191345,
-      "loss": 2.1629,
-      "step": 2620
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.0001359830385505967,
-      "loss": 2.1211,
-      "step": 2625
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00013570703361249188,
-      "loss": 2.1541,
-      "step": 2630
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00013543071656015084,
-      "loss": 2.122,
-      "step": 2635
-    },
-    {
-      "epoch": 0.89,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001351540898088536,
-      "loss": 2.1544,
-      "step": 2640
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00013487715577658726,
-      "loss": 2.1587,
-      "step": 2645
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.150390625,
-      "learning_rate": 0.00013459991688402492,
-      "loss": 2.1158,
-      "step": 2650
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00013432237555450444,
-      "loss": 2.1499,
-      "step": 2655
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00013404453421400714,
-      "loss": 2.152,
-      "step": 2660
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013376639529113688,
-      "loss": 2.1234,
-      "step": 2665
-    },
-    {
-      "epoch": 0.9,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00013348796121709862,
-      "loss": 2.1329,
-      "step": 2670
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00013320923442567727,
-      "loss": 2.1402,
-      "step": 2675
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00013293021735321628,
-      "loss": 2.1526,
-      "step": 2680
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00013265091243859652,
-      "loss": 2.1237,
-      "step": 2685
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013237132212321487,
-      "loss": 2.1443,
-      "step": 2690
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001320914488509629,
-      "loss": 2.1558,
-      "step": 2695
-    },
-    {
-      "epoch": 0.91,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00013181129506820545,
-      "loss": 2.1194,
-      "step": 2700
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001315308632237593,
-      "loss": 2.1201,
-      "step": 2705
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.00013125015576887186,
-      "loss": 2.1523,
-      "step": 2710
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00013096917515719952,
-      "loss": 2.1443,
-      "step": 2715
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00013068792384478636,
-      "loss": 2.1487,
-      "step": 2720
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00013040640429004267,
-      "loss": 2.1623,
-      "step": 2725
-    },
-    {
-      "epoch": 0.92,
-      "grad_norm": 0.1513671875,
-      "learning_rate": 0.00013012461895372344,
-      "loss": 2.1395,
-      "step": 2730
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00012984257029890683,
-      "loss": 2.1418,
-      "step": 2735
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00012956026079097272,
-      "loss": 2.1475,
-      "step": 2740
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.000129277692897581,
-      "loss": 2.1373,
-      "step": 2745
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012899486908865012,
-      "loss": 2.138,
-      "step": 2750
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001287117918363356,
-      "loss": 2.1417,
-      "step": 2755
-    },
-    {
-      "epoch": 0.93,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00012842846361500816,
-      "loss": 2.1108,
-      "step": 2760
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00012814488690123226,
-      "loss": 2.1408,
-      "step": 2765
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012786106417374455,
-      "loss": 2.1762,
-      "step": 2770
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012757699791343186,
-      "loss": 2.1445,
-      "step": 2775
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00012729269060330999,
-      "loss": 2.1307,
-      "step": 2780
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.0001270081447285015,
-      "loss": 2.1431,
-      "step": 2785
-    },
-    {
-      "epoch": 0.94,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012672336277621442,
-      "loss": 2.1351,
-      "step": 2790
-    },
-    {
-      "epoch": 0.95,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.0001264383472357202,
-      "loss": 2.1395,
-      "step": 2795
-    },
-    {
-      "epoch": 0.95,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.0001261531005983322,
-      "loss": 2.1492,
-      "step": 2800
-    },
-    {
-      "epoch": 0.95,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012586762535738374,
-      "loss": 2.1344,
-      "step": 2805
-    },
-    {
-      "epoch": 0.95,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001255819240082063,
-      "loss": 2.1352,
-      "step": 2810
-    },
-    {
-      "epoch": 0.95,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012529599904810784,
-      "loss": 2.1554,
-      "step": 2815
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00012500985297635088,
-      "loss": 2.125,
-      "step": 2820
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00012472348829413064,
-      "loss": 2.1478,
-      "step": 2825
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00012443690750455326,
-      "loss": 2.116,
-      "step": 2830
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.1533203125,
-      "learning_rate": 0.0001241501131126138,
-      "loss": 2.1667,
-      "step": 2835
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00012386310762517452,
-      "loss": 2.1338,
-      "step": 2840
-    },
-    {
-      "epoch": 0.96,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012357589355094275,
-      "loss": 2.109,
-      "step": 2845
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.0001232884734004491,
-      "loss": 2.1318,
-      "step": 2850
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00012300084968602549,
-      "loss": 2.1117,
-      "step": 2855
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00012271302492178327,
-      "loss": 2.1379,
-      "step": 2860
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00012242500162359105,
-      "loss": 2.122,
-      "step": 2865
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00012213678230905284,
-      "loss": 2.1651,
-      "step": 2870
-    },
-    {
-      "epoch": 0.97,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00012184836949748608,
-      "loss": 2.1159,
-      "step": 2875
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00012155976570989949,
-      "loss": 2.1241,
-      "step": 2880
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.0001212709734689712,
-      "loss": 2.1343,
-      "step": 2885
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012098199529902648,
-      "loss": 2.134,
-      "step": 2890
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.0001206928337260159,
-      "loss": 2.1265,
-      "step": 2895
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00012040349127749313,
-      "loss": 2.1531,
-      "step": 2900
-    },
-    {
-      "epoch": 0.98,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00012011397048259285,
-      "loss": 2.1379,
-      "step": 2905
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00011982427387200867,
-      "loss": 2.1186,
-      "step": 2910
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00011953440397797097,
-      "loss": 2.1584,
-      "step": 2915
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.15625,
-      "learning_rate": 0.00011924436333422489,
-      "loss": 2.1315,
-      "step": 2920
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.000118954154476008,
-      "loss": 2.1224,
-      "step": 2925
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.0001186637799400282,
-      "loss": 2.14,
-      "step": 2930
-    },
-    {
-      "epoch": 0.99,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00011837324226444169,
-      "loss": 2.1388,
-      "step": 2935
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00011808254398883056,
-      "loss": 2.1444,
-      "step": 2940
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.1552734375,
-      "learning_rate": 0.00011779168765418079,
-      "loss": 2.1314,
-      "step": 2945
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00011750067580285988,
-      "loss": 2.1288,
-      "step": 2950
-    },
-    {
-      "epoch": 1.0,
-      "eval_loss": 2.1319899559020996,
-      "eval_runtime": 173.5199,
-      "eval_samples_per_second": 7.653,
-      "eval_steps_per_second": 0.957,
-      "step": 2952
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00011720951097859476,
-      "loss": 2.1113,
-      "step": 2955
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.154296875,
-      "learning_rate": 0.00011691819572644939,
-      "loss": 2.0925,
-      "step": 2960
-    },
-    {
-      "epoch": 1.0,
-      "grad_norm": 0.15234375,
-      "learning_rate": 0.00011662673259280276,
-      "loss": 2.1001,
-      "step": 2965
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00011633512412532637,
-      "loss": 2.1235,
-      "step": 2970
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001160433728729621,
-      "loss": 2.1128,
-      "step": 2975
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.00011575148138589996,
-      "loss": 2.1216,
-      "step": 2980
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.158203125,
-      "learning_rate": 0.0001154594522155557,
-      "loss": 2.1345,
-      "step": 2985
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 0.00011516728791454861,
-      "loss": 2.1045,
-      "step": 2990
-    },
-    {
-      "epoch": 1.01,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00011487499103667904,
-      "loss": 2.0973,
-      "step": 2995
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00011458256413690633,
-      "loss": 2.0934,
-      "step": 3000
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 0.00011429000977132629,
-      "loss": 2.1147,
-      "step": 3005
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00011399733049714884,
-      "loss": 2.1049,
-      "step": 3010
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00011370452887267582,
-      "loss": 2.0878,
-      "step": 3015
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.00011341160745727844,
-      "loss": 2.1226,
-      "step": 3020
-    },
-    {
-      "epoch": 1.02,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00011311856881137506,
-      "loss": 2.1125,
-      "step": 3025
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00011282541549640873,
-      "loss": 2.0914,
-      "step": 3030
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.0001125321500748248,
-      "loss": 2.0832,
-      "step": 3035
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00011223877511004863,
-      "loss": 2.1107,
-      "step": 3040
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00011194529316646293,
-      "loss": 2.1353,
-      "step": 3045
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00011165170680938572,
-      "loss": 2.1065,
-      "step": 3050
-    },
-    {
-      "epoch": 1.03,
-      "grad_norm": 0.1572265625,
-      "learning_rate": 0.00011135801860504749,
-      "loss": 2.1289,
-      "step": 3055
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 0.00011106423112056911,
-      "loss": 2.1507,
-      "step": 3060
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00011077034692393917,
-      "loss": 2.1053,
-      "step": 3065
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00011047636858399169,
-      "loss": 2.1196,
-      "step": 3070
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00011018229867038356,
-      "loss": 2.0859,
-      "step": 3075
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.00010988813975357208,
-      "loss": 2.1273,
-      "step": 3080
-    },
-    {
-      "epoch": 1.04,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00010959389440479264,
-      "loss": 2.1134,
-      "step": 3085
-    },
-    {
-      "epoch": 1.05,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00010929956519603594,
-      "loss": 2.1101,
-      "step": 3090
-    },
-    {
-      "epoch": 1.05,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.00010900515470002595,
-      "loss": 2.1221,
-      "step": 3095
-    },
-    {
-      "epoch": 1.05,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 0.00010871066549019688,
-      "loss": 2.1026,
-      "step": 3100
-    },
-    {
-      "epoch": 1.05,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.0001084161001406712,
-      "loss": 2.1148,
-      "step": 3105
-    },
-    {
-      "epoch": 1.05,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00010812146122623683,
-      "loss": 2.0942,
-      "step": 3110
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00010782675132232474,
-      "loss": 2.1331,
-      "step": 3115
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00010753197300498638,
-      "loss": 2.1119,
-      "step": 3120
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.00010723712885087123,
-      "loss": 2.0959,
-      "step": 3125
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00010694222143720423,
-      "loss": 2.0654,
-      "step": 3130
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00010664725334176331,
-      "loss": 2.1195,
-      "step": 3135
-    },
-    {
-      "epoch": 1.06,
-      "grad_norm": 0.166015625,
-      "learning_rate": 0.00010635222714285676,
-      "loss": 2.1052,
-      "step": 3140
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.16796875,
-      "learning_rate": 0.0001060571454193008,
-      "loss": 2.1174,
-      "step": 3145
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010576201075039696,
-      "loss": 2.0997,
-      "step": 3150
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.166015625,
-      "learning_rate": 0.00010546682571590958,
-      "loss": 2.1202,
-      "step": 3155
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010517159289604324,
-      "loss": 2.1057,
-      "step": 3160
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00010487631487142017,
-      "loss": 2.135,
-      "step": 3165
-    },
-    {
-      "epoch": 1.07,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00010458099422305785,
-      "loss": 2.1585,
-      "step": 3170
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 0.0001042856335323462,
-      "loss": 2.1096,
-      "step": 3175
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00010399023538102522,
-      "loss": 2.1008,
-      "step": 3180
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00010369480235116229,
-      "loss": 2.1126,
-      "step": 3185
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 0.00010339933702512979,
-      "loss": 2.156,
-      "step": 3190
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010310384198558225,
-      "loss": 2.1252,
-      "step": 3195
-    },
-    {
-      "epoch": 1.08,
-      "grad_norm": 0.16015625,
-      "learning_rate": 0.00010280831981543405,
-      "loss": 2.0947,
-      "step": 3200
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010251277309783663,
-      "loss": 2.0965,
-      "step": 3205
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00010221720441615599,
-      "loss": 2.1106,
-      "step": 3210
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 0.00010192161635395026,
-      "loss": 2.0715,
-      "step": 3215
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1591796875,
-      "learning_rate": 0.00010162601149494676,
-      "loss": 2.1249,
-      "step": 3220
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010133039242301985,
-      "loss": 2.0997,
-      "step": 3225
-    },
-    {
-      "epoch": 1.09,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 0.00010103476172216792,
-      "loss": 2.1182,
-      "step": 3230
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.00010073912197649116,
-      "loss": 2.0769,
-      "step": 3235
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.162109375,
-      "learning_rate": 0.0001004434757701688,
-      "loss": 2.1404,
-      "step": 3240
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.1640625,
-      "learning_rate": 0.00010014782568743641,
-      "loss": 2.1354,
-      "step": 3245
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.98521743125636e-05,
-      "loss": 2.1234,
-      "step": 3250
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.955652422983122e-05,
-      "loss": 2.1039,
-      "step": 3255
-    },
-    {
-      "epoch": 1.1,
-      "grad_norm": 0.16796875,
-      "learning_rate": 9.926087802350886e-05,
-      "loss": 2.1161,
-      "step": 3260
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.896523827783207e-05,
-      "loss": 2.1135,
-      "step": 3265
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.1640625,
-      "learning_rate": 9.866960757698017e-05,
-      "loss": 2.1019,
-      "step": 3270
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.162109375,
-      "learning_rate": 9.837398850505324e-05,
-      "loss": 2.1151,
-      "step": 3275
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.162109375,
-      "learning_rate": 9.807838364604978e-05,
-      "loss": 2.1191,
-      "step": 3280
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 9.7782795583844e-05,
-      "loss": 2.1127,
-      "step": 3285
-    },
-    {
-      "epoch": 1.11,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.748722690216341e-05,
-      "loss": 2.1007,
-      "step": 3290
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.719168018456598e-05,
-      "loss": 2.1043,
-      "step": 3295
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 9.689615801441774e-05,
-      "loss": 2.1124,
-      "step": 3300
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1640625,
-      "learning_rate": 9.660066297487022e-05,
-      "loss": 2.126,
-      "step": 3305
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 9.630519764883772e-05,
-      "loss": 2.1014,
-      "step": 3310
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1640625,
-      "learning_rate": 9.600976461897483e-05,
-      "loss": 2.1159,
-      "step": 3315
-    },
-    {
-      "epoch": 1.12,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.571436646765382e-05,
-      "loss": 2.113,
-      "step": 3320
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.162109375,
-      "learning_rate": 9.541900577694217e-05,
-      "loss": 2.1211,
-      "step": 3325
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 9.512368512857984e-05,
-      "loss": 2.1062,
-      "step": 3330
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.482840710395675e-05,
-      "loss": 2.1036,
-      "step": 3335
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.453317428409044e-05,
-      "loss": 2.1095,
-      "step": 3340
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.166015625,
-      "learning_rate": 9.423798924960306e-05,
-      "loss": 2.1065,
-      "step": 3345
-    },
-    {
-      "epoch": 1.13,
-      "grad_norm": 0.16796875,
-      "learning_rate": 9.394285458069923e-05,
-      "loss": 2.1398,
-      "step": 3350
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 9.364777285714324e-05,
-      "loss": 2.1007,
-      "step": 3355
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.33527466582367e-05,
-      "loss": 2.1188,
-      "step": 3360
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.166015625,
-      "learning_rate": 9.30577785627958e-05,
-      "loss": 2.1267,
-      "step": 3365
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.276287114912878e-05,
-      "loss": 2.1252,
-      "step": 3370
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.16796875,
-      "learning_rate": 9.246802699501363e-05,
-      "loss": 2.1334,
-      "step": 3375
-    },
-    {
-      "epoch": 1.14,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 9.217324867767527e-05,
-      "loss": 2.1118,
-      "step": 3380
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.187853877376318e-05,
-      "loss": 2.1104,
-      "step": 3385
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 9.158389985932881e-05,
-      "loss": 2.1238,
-      "step": 3390
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.1640625,
-      "learning_rate": 9.128933450980314e-05,
-      "loss": 2.1228,
-      "step": 3395
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 9.099484529997409e-05,
-      "loss": 2.0859,
-      "step": 3400
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.166015625,
-      "learning_rate": 9.070043480396404e-05,
-      "loss": 2.0926,
-      "step": 3405
-    },
-    {
-      "epoch": 1.15,
-      "grad_norm": 0.1640625,
-      "learning_rate": 9.04061055952074e-05,
-      "loss": 2.1346,
-      "step": 3410
-    },
-    {
-      "epoch": 1.16,
-      "grad_norm": 0.162109375,
-      "learning_rate": 9.011186024642793e-05,
-      "loss": 2.096,
-      "step": 3415
-    },
-    {
-      "epoch": 1.16,
-      "grad_norm": 0.162109375,
-      "learning_rate": 8.981770132961649e-05,
-      "loss": 2.1154,
-      "step": 3420
-    },
-    {
-      "epoch": 1.16,
-      "grad_norm": 0.162109375,
-      "learning_rate": 8.952363141600834e-05,
-      "loss": 2.1073,
-      "step": 3425
-    },
-    {
-      "epoch": 1.16,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 8.922965307606086e-05,
-      "loss": 2.1085,
-      "step": 3430
-    },
-    {
-      "epoch": 1.16,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 8.893576887943094e-05,
-      "loss": 2.0935,
-      "step": 3435
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 8.86419813949525e-05,
-      "loss": 2.0911,
-      "step": 3440
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.166015625,
-      "learning_rate": 8.834829319061431e-05,
-      "loss": 2.1139,
-      "step": 3445
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.805470683353708e-05,
-      "loss": 2.1199,
-      "step": 3450
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 8.77612248899514e-05,
-      "loss": 2.1038,
-      "step": 3455
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 8.746784992517518e-05,
-      "loss": 2.0909,
-      "step": 3460
-    },
-    {
-      "epoch": 1.17,
-      "grad_norm": 0.1640625,
-      "learning_rate": 8.71745845035913e-05,
-      "loss": 2.1016,
-      "step": 3465
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.1640625,
-      "learning_rate": 8.688143118862499e-05,
-      "loss": 2.1213,
-      "step": 3470
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.658839254272157e-05,
-      "loss": 2.1115,
-      "step": 3475
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.169921875,
-      "learning_rate": 8.62954711273242e-05,
-      "loss": 2.1116,
-      "step": 3480
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.162109375,
-      "learning_rate": 8.600266950285117e-05,
-      "loss": 2.1493,
-      "step": 3485
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.570999022867373e-05,
-      "loss": 2.1146,
-      "step": 3490
-    },
-    {
-      "epoch": 1.18,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.541743586309365e-05,
-      "loss": 2.0887,
-      "step": 3495
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.166015625,
-      "learning_rate": 8.512500896332097e-05,
-      "loss": 2.1072,
-      "step": 3500
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.483271208545144e-05,
-      "loss": 2.0959,
-      "step": 3505
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.169921875,
-      "learning_rate": 8.454054778444431e-05,
-      "loss": 2.1305,
-      "step": 3510
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.166015625,
-      "learning_rate": 8.424851861410007e-05,
-      "loss": 2.0803,
-      "step": 3515
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 8.395662712703793e-05,
-      "loss": 2.0963,
-      "step": 3520
-    },
-    {
-      "epoch": 1.19,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.366487587467368e-05,
-      "loss": 2.1136,
-      "step": 3525
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.337326740719726e-05,
-      "loss": 2.0984,
-      "step": 3530
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.1640625,
-      "learning_rate": 8.308180427355062e-05,
-      "loss": 2.1252,
-      "step": 3535
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.1640625,
-      "learning_rate": 8.279048902140528e-05,
-      "loss": 2.1101,
-      "step": 3540
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.24993241971401e-05,
-      "loss": 2.1124,
-      "step": 3545
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 8.220831234581922e-05,
-      "loss": 2.1135,
-      "step": 3550
-    },
-    {
-      "epoch": 1.2,
-      "grad_norm": 0.166015625,
-      "learning_rate": 8.191745601116947e-05,
-      "loss": 2.1134,
-      "step": 3555
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.16015625,
-      "learning_rate": 8.162675773555836e-05,
-      "loss": 2.098,
-      "step": 3560
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.162109375,
-      "learning_rate": 8.133622005997181e-05,
-      "loss": 2.095,
-      "step": 3565
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.104584552399204e-05,
-      "loss": 2.1298,
-      "step": 3570
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.166015625,
-      "learning_rate": 8.075563666577515e-05,
-      "loss": 2.1018,
-      "step": 3575
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 8.046559602202901e-05,
-      "loss": 2.1074,
-      "step": 3580
-    },
-    {
-      "epoch": 1.21,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.017572612799135e-05,
-      "loss": 2.1308,
-      "step": 3585
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.162109375,
-      "learning_rate": 7.988602951740717e-05,
-      "loss": 2.1215,
-      "step": 3590
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.959650872250688e-05,
-      "loss": 2.1063,
-      "step": 3595
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 7.930716627398412e-05,
-      "loss": 2.1156,
-      "step": 3600
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.901800470097355e-05,
-      "loss": 2.1169,
-      "step": 3605
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.872902653102884e-05,
-      "loss": 2.0942,
-      "step": 3610
-    },
-    {
-      "epoch": 1.22,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.84402342901005e-05,
-      "loss": 2.1382,
-      "step": 3615
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.815163050251395e-05,
-      "loss": 2.1234,
-      "step": 3620
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 7.786321769094717e-05,
-      "loss": 2.1158,
-      "step": 3625
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.7574998376409e-05,
-      "loss": 2.1071,
-      "step": 3630
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.728697507821674e-05,
-      "loss": 2.1165,
-      "step": 3635
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.699915031397452e-05,
-      "loss": 2.1276,
-      "step": 3640
-    },
-    {
-      "epoch": 1.23,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.671152659955096e-05,
-      "loss": 2.12,
-      "step": 3645
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 7.642410644905726e-05,
-      "loss": 2.1144,
-      "step": 3650
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.162109375,
-      "learning_rate": 7.613689237482551e-05,
-      "loss": 2.1058,
-      "step": 3655
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 7.584988688738622e-05,
-      "loss": 2.0953,
-      "step": 3660
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 7.556309249544678e-05,
-      "loss": 2.1459,
-      "step": 3665
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.527651170586936e-05,
-      "loss": 2.1061,
-      "step": 3670
-    },
-    {
-      "epoch": 1.24,
-      "grad_norm": 0.1640625,
-      "learning_rate": 7.499014702364913e-05,
-      "loss": 2.1119,
-      "step": 3675
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.470400095189219e-05,
-      "loss": 2.1395,
-      "step": 3680
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.1640625,
-      "learning_rate": 7.44180759917937e-05,
-      "loss": 2.0965,
-      "step": 3685
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 7.413237464261627e-05,
-      "loss": 2.1035,
-      "step": 3690
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.38468994016678e-05,
-      "loss": 2.1184,
-      "step": 3695
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 7.356165276427983e-05,
-      "loss": 2.1213,
-      "step": 3700
-    },
-    {
-      "epoch": 1.25,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 7.327663722378561e-05,
-      "loss": 2.0926,
-      "step": 3705
-    },
-    {
-      "epoch": 1.26,
-      "grad_norm": 0.1640625,
-      "learning_rate": 7.299185527149853e-05,
-      "loss": 2.1019,
-      "step": 3710
-    },
-    {
-      "epoch": 1.26,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.270730939669006e-05,
-      "loss": 2.0903,
-      "step": 3715
-    },
-    {
-      "epoch": 1.26,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.242300208656814e-05,
-      "loss": 2.1252,
-      "step": 3720
-    },
-    {
-      "epoch": 1.26,
-      "grad_norm": 0.162109375,
-      "learning_rate": 7.213893582625548e-05,
-      "loss": 2.1357,
-      "step": 3725
-    },
-    {
-      "epoch": 1.26,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.185511309876775e-05,
-      "loss": 2.0908,
-      "step": 3730
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.166015625,
-      "learning_rate": 7.157153638499188e-05,
-      "loss": 2.0689,
-      "step": 3735
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.1640625,
-      "learning_rate": 7.128820816366442e-05,
-      "loss": 2.1063,
-      "step": 3740
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.100513091134989e-05,
-      "loss": 2.1295,
-      "step": 3745
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.173828125,
-      "learning_rate": 7.072230710241905e-05,
-      "loss": 2.1386,
-      "step": 3750
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.043973920902729e-05,
-      "loss": 2.1216,
-      "step": 3755
-    },
-    {
-      "epoch": 1.27,
-      "grad_norm": 0.1640625,
-      "learning_rate": 7.015742970109317e-05,
-      "loss": 2.1305,
-      "step": 3760
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 6.98753810462766e-05,
-      "loss": 2.1098,
-      "step": 3765
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.959359570995738e-05,
-      "loss": 2.1271,
-      "step": 3770
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.931207615521366e-05,
-      "loss": 2.1041,
-      "step": 3775
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.903082484280053e-05,
-      "loss": 2.114,
-      "step": 3780
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.874984423112819e-05,
-      "loss": 2.1364,
-      "step": 3785
-    },
-    {
-      "epoch": 1.28,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 6.84691367762407e-05,
-      "loss": 2.117,
-      "step": 3790
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.818870493179458e-05,
-      "loss": 2.0934,
-      "step": 3795
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.790855114903714e-05,
-      "loss": 2.1207,
-      "step": 3800
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.1640625,
-      "learning_rate": 6.762867787678512e-05,
-      "loss": 2.121,
-      "step": 3805
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.73490875614035e-05,
-      "loss": 2.1113,
-      "step": 3810
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.171875,
-      "learning_rate": 6.706978264678376e-05,
-      "loss": 2.1088,
-      "step": 3815
-    },
-    {
-      "epoch": 1.29,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 6.679076557432278e-05,
-      "loss": 2.1221,
-      "step": 3820
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.651203878290139e-05,
-      "loss": 2.1239,
-      "step": 3825
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.623360470886314e-05,
-      "loss": 2.1229,
-      "step": 3830
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.59554657859929e-05,
-      "loss": 2.0984,
-      "step": 3835
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.567762444549558e-05,
-      "loss": 2.1042,
-      "step": 3840
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 6.540008311597507e-05,
-      "loss": 2.1203,
-      "step": 3845
-    },
-    {
-      "epoch": 1.3,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.512284422341275e-05,
-      "loss": 2.0983,
-      "step": 3850
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.166015625,
-      "learning_rate": 6.484591019114646e-05,
-      "loss": 2.112,
-      "step": 3855
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.456928343984919e-05,
-      "loss": 2.1393,
-      "step": 3860
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.429296638750814e-05,
-      "loss": 2.1089,
-      "step": 3865
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.401696144940332e-05,
-      "loss": 2.0942,
-      "step": 3870
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.374127103808654e-05,
-      "loss": 2.1108,
-      "step": 3875
-    },
-    {
-      "epoch": 1.31,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.34658975633605e-05,
-      "loss": 2.1063,
-      "step": 3880
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.319084343225738e-05,
-      "loss": 2.1309,
-      "step": 3885
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.291611104901812e-05,
-      "loss": 2.1287,
-      "step": 3890
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.264170281507111e-05,
-      "loss": 2.1196,
-      "step": 3895
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.166015625,
-      "learning_rate": 6.236762112901158e-05,
-      "loss": 2.1202,
-      "step": 3900
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.209386838658024e-05,
-      "loss": 2.1264,
-      "step": 3905
-    },
-    {
-      "epoch": 1.32,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.182044698064256e-05,
-      "loss": 2.1083,
-      "step": 3910
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.166015625,
-      "learning_rate": 6.154735930116786e-05,
-      "loss": 2.1208,
-      "step": 3915
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.12746077352083e-05,
-      "loss": 2.0995,
-      "step": 3920
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 6.1002194666878106e-05,
-      "loss": 2.1249,
-      "step": 3925
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.0730122477332675e-05,
-      "loss": 2.1024,
-      "step": 3930
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.045839354474786e-05,
-      "loss": 2.09,
-      "step": 3935
-    },
-    {
-      "epoch": 1.33,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 6.0187010244299046e-05,
-      "loss": 2.0995,
-      "step": 3940
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.9915974948140474e-05,
-      "loss": 2.101,
-      "step": 3945
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.964529002538455e-05,
-      "loss": 2.0827,
-      "step": 3950
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.937495784208096e-05,
-      "loss": 2.1324,
-      "step": 3955
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 5.910498076119622e-05,
-      "loss": 2.1116,
-      "step": 3960
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.883536114259277e-05,
-      "loss": 2.0813,
-      "step": 3965
-    },
-    {
-      "epoch": 1.34,
-      "grad_norm": 0.166015625,
-      "learning_rate": 5.8566101343008687e-05,
-      "loss": 2.1031,
-      "step": 3970
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 5.829720371603664e-05,
-      "loss": 2.1116,
-      "step": 3975
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.802867061210375e-05,
-      "loss": 2.1094,
-      "step": 3980
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.776050437845075e-05,
-      "loss": 2.1008,
-      "step": 3985
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 5.749270735911158e-05,
-      "loss": 2.1074,
-      "step": 3990
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.7225281894892935e-05,
-      "loss": 2.1045,
-      "step": 3995
-    },
-    {
-      "epoch": 1.35,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.695823032335366e-05,
-      "loss": 2.1119,
-      "step": 4000
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.669155497878454e-05,
-      "loss": 2.1185,
-      "step": 4005
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 5.642525819218769e-05,
-      "loss": 2.0826,
-      "step": 4010
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.166015625,
-      "learning_rate": 5.6159342291256254e-05,
-      "loss": 2.0952,
-      "step": 4015
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.589380960035417e-05,
-      "loss": 2.1267,
-      "step": 4020
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.1640625,
-      "learning_rate": 5.562866244049557e-05,
-      "loss": 2.1282,
-      "step": 4025
-    },
-    {
-      "epoch": 1.36,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.53639031293248e-05,
-      "loss": 2.1114,
-      "step": 4030
-    },
-    {
-      "epoch": 1.37,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 5.509953398109594e-05,
-      "loss": 2.1173,
-      "step": 4035
-    },
-    {
-      "epoch": 1.37,
-      "grad_norm": 0.166015625,
-      "learning_rate": 5.483555730665282e-05,
-      "loss": 2.0859,
-      "step": 4040
-    },
-    {
-      "epoch": 1.37,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.457197541340853e-05,
-      "loss": 2.1138,
-      "step": 4045
-    },
-    {
-      "epoch": 1.37,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.4308790605325364e-05,
-      "loss": 2.1407,
-      "step": 4050
-    },
-    {
-      "epoch": 1.37,
-      "grad_norm": 0.166015625,
-      "learning_rate": 5.404600518289487e-05,
-      "loss": 2.1156,
-      "step": 4055
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.166015625,
-      "learning_rate": 5.3783621443117414e-05,
-      "loss": 2.1036,
-      "step": 4060
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 5.352164167948233e-05,
-      "loss": 2.1022,
-      "step": 4065
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.326006818194782e-05,
-      "loss": 2.1037,
-      "step": 4070
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.2998903236920895e-05,
-      "loss": 2.1189,
-      "step": 4075
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.171875,
-      "learning_rate": 5.273814912723742e-05,
-      "loss": 2.1133,
-      "step": 4080
-    },
-    {
-      "epoch": 1.38,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.247780813214214e-05,
-      "loss": 2.1261,
-      "step": 4085
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.171875,
-      "learning_rate": 5.221788252726889e-05,
-      "loss": 2.1114,
-      "step": 4090
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.195837458462045e-05,
-      "loss": 2.1119,
-      "step": 4095
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.16992865725489e-05,
-      "loss": 2.125,
-      "step": 4100
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 5.14406207557357e-05,
-      "loss": 2.0831,
-      "step": 4105
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 5.11823793951719e-05,
-      "loss": 2.1304,
-      "step": 4110
-    },
-    {
-      "epoch": 1.39,
-      "grad_norm": 0.17578125,
-      "learning_rate": 5.092456474813841e-05,
-      "loss": 2.1066,
-      "step": 4115
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.1640625,
-      "learning_rate": 5.066717906818618e-05,
-      "loss": 2.1206,
-      "step": 4120
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.1611328125,
-      "learning_rate": 5.041022460511673e-05,
-      "loss": 2.1133,
-      "step": 4125
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.015370360496219e-05,
-      "loss": 2.1384,
-      "step": 4130
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.989761830996581e-05,
-      "loss": 2.1042,
-      "step": 4135
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.1640625,
-      "learning_rate": 4.9641970958562366e-05,
-      "loss": 2.1241,
-      "step": 4140
-    },
-    {
-      "epoch": 1.4,
-      "grad_norm": 0.171875,
-      "learning_rate": 4.938676378535866e-05,
-      "loss": 2.1237,
-      "step": 4145
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.1787109375,
-      "learning_rate": 4.913199902111385e-05,
-      "loss": 2.1187,
-      "step": 4150
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.8877678892719866e-05,
-      "loss": 2.0729,
-      "step": 4155
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.171875,
-      "learning_rate": 4.862380562318236e-05,
-      "loss": 2.1087,
-      "step": 4160
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.837038143160082e-05,
-      "loss": 2.118,
-      "step": 4165
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.811740853314939e-05,
-      "loss": 2.118,
-      "step": 4170
-    },
-    {
-      "epoch": 1.41,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.786488913905745e-05,
-      "loss": 2.1102,
-      "step": 4175
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.7612825456590435e-05,
-      "loss": 2.0915,
-      "step": 4180
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.736121968903027e-05,
-      "loss": 2.1321,
-      "step": 4185
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.7110074035656316e-05,
-      "loss": 2.0898,
-      "step": 4190
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.685939069172609e-05,
-      "loss": 2.1212,
-      "step": 4195
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 4.6609171848456066e-05,
-      "loss": 2.0965,
-      "step": 4200
-    },
-    {
-      "epoch": 1.42,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 4.6359419693002534e-05,
-      "loss": 2.095,
-      "step": 4205
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 4.611013640844245e-05,
-      "loss": 2.118,
-      "step": 4210
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.5861324173754484e-05,
-      "loss": 2.1134,
-      "step": 4215
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.561298516379974e-05,
-      "loss": 2.1027,
-      "step": 4220
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.5365121549302914e-05,
-      "loss": 2.0939,
-      "step": 4225
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.5117735496833415e-05,
-      "loss": 2.1279,
-      "step": 4230
-    },
-    {
-      "epoch": 1.43,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.487082916878606e-05,
-      "loss": 2.1078,
-      "step": 4235
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 4.4624404723362576e-05,
-      "loss": 2.1143,
-      "step": 4240
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 4.437846431455249e-05,
-      "loss": 2.0946,
-      "step": 4245
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.171875,
-      "learning_rate": 4.4133010092114494e-05,
-      "loss": 2.1203,
-      "step": 4250
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.166015625,
-      "learning_rate": 4.3888044201557376e-05,
-      "loss": 2.1112,
-      "step": 4255
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 4.36435687841215e-05,
-      "loss": 2.0873,
-      "step": 4260
-    },
-    {
-      "epoch": 1.44,
-      "grad_norm": 0.173828125,
-      "learning_rate": 4.3399585976760105e-05,
-      "loss": 2.1056,
-      "step": 4265
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.3156097912120385e-05,
-      "loss": 2.101,
-      "step": 4270
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 4.29131067185251e-05,
-      "loss": 2.1302,
-      "step": 4275
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.2670614519953834e-05,
-      "loss": 2.0862,
-      "step": 4280
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.242862343602447e-05,
-      "loss": 2.0792,
-      "step": 4285
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.21871355819747e-05,
-      "loss": 2.112,
-      "step": 4290
-    },
-    {
-      "epoch": 1.45,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 4.19461530686434e-05,
-      "loss": 2.1081,
-      "step": 4295
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.177734375,
-      "learning_rate": 4.170567800245244e-05,
-      "loss": 2.1193,
-      "step": 4300
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.1465712485387966e-05,
-      "loss": 2.1102,
-      "step": 4305
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.1226258614982214e-05,
-      "loss": 2.1125,
-      "step": 4310
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.171875,
-      "learning_rate": 4.0987318484295135e-05,
-      "loss": 2.103,
-      "step": 4315
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 4.074889418189608e-05,
-      "loss": 2.1172,
-      "step": 4320
-    },
-    {
-      "epoch": 1.46,
-      "grad_norm": 0.1640625,
-      "learning_rate": 4.051098779184559e-05,
-      "loss": 2.0825,
-      "step": 4325
-    },
-    {
-      "epoch": 1.47,
-      "grad_norm": 0.1640625,
-      "learning_rate": 4.0273601393677064e-05,
-      "loss": 2.1347,
-      "step": 4330
-    },
-    {
-      "epoch": 1.47,
-      "grad_norm": 0.171875,
-      "learning_rate": 4.0036737062378823e-05,
-      "loss": 2.1057,
-      "step": 4335
-    },
-    {
-      "epoch": 1.47,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.980039686837568e-05,
-      "loss": 2.1365,
-      "step": 4340
-    },
-    {
-      "epoch": 1.47,
-      "grad_norm": 0.171875,
-      "learning_rate": 3.956458287751097e-05,
-      "loss": 2.1176,
-      "step": 4345
-    },
-    {
-      "epoch": 1.47,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.932929715102863e-05,
-      "loss": 2.0994,
-      "step": 4350
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 3.9094541745554946e-05,
-      "loss": 2.0961,
-      "step": 4355
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 3.8860318713080725e-05,
-      "loss": 2.1128,
-      "step": 4360
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.8626630100943196e-05,
-      "loss": 2.1145,
-      "step": 4365
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.169921875,
-      "learning_rate": 3.8393477951808444e-05,
-      "loss": 2.1021,
-      "step": 4370
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.816086430365321e-05,
-      "loss": 2.1185,
-      "step": 4375
-    },
-    {
-      "epoch": 1.48,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.7928791189747195e-05,
-      "loss": 2.1078,
-      "step": 4380
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.769726063863541e-05,
-      "loss": 2.1133,
-      "step": 4385
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 3.746627467412026e-05,
-      "loss": 2.1013,
-      "step": 4390
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.723583531524394e-05,
-      "loss": 2.115,
-      "step": 4395
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.700594457627079e-05,
-      "loss": 2.1374,
-      "step": 4400
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.6776604466669686e-05,
-      "loss": 2.1082,
-      "step": 4405
-    },
-    {
-      "epoch": 1.49,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 3.654781699109645e-05,
-      "loss": 2.1171,
-      "step": 4410
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.631958414937633e-05,
-      "loss": 2.1055,
-      "step": 4415
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.166015625,
-      "learning_rate": 3.609190793648661e-05,
-      "loss": 2.1367,
-      "step": 4420
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.171875,
-      "learning_rate": 3.586479034253902e-05,
-      "loss": 2.1161,
-      "step": 4425
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.563823335276244e-05,
-      "loss": 2.1004,
-      "step": 4430
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.541223894748553e-05,
-      "loss": 2.0959,
-      "step": 4435
-    },
-    {
-      "epoch": 1.5,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.51868091021194e-05,
-      "loss": 2.1085,
-      "step": 4440
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.496194578714036e-05,
-      "loss": 2.1207,
-      "step": 4445
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.473765096807269e-05,
-      "loss": 2.1181,
-      "step": 4450
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.166015625,
-      "learning_rate": 3.45139266054715e-05,
-      "loss": 2.1069,
-      "step": 4455
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.429077465490551e-05,
-      "loss": 2.1213,
-      "step": 4460
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 3.406819706694003e-05,
-      "loss": 2.0987,
-      "step": 4465
-    },
-    {
-      "epoch": 1.51,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 3.3846195787119814e-05,
-      "loss": 2.0996,
-      "step": 4470
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.362477275595225e-05,
-      "loss": 2.0933,
-      "step": 4475
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.171875,
-      "learning_rate": 3.340392990889018e-05,
-      "loss": 2.0977,
-      "step": 4480
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 3.3183669176315045e-05,
-      "loss": 2.1336,
-      "step": 4485
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.166015625,
-      "learning_rate": 3.296399248352012e-05,
-      "loss": 2.0963,
-      "step": 4490
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.2744901750693556e-05,
-      "loss": 2.0951,
-      "step": 4495
-    },
-    {
-      "epoch": 1.52,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 3.2526398892901654e-05,
-      "loss": 2.0882,
-      "step": 4500
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.169921875,
-      "learning_rate": 3.2308485820072075e-05,
-      "loss": 2.1044,
-      "step": 4505
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.2091164436977294e-05,
-      "loss": 2.1091,
-      "step": 4510
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 3.187443664321773e-05,
-      "loss": 2.0628,
-      "step": 4515
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.165830433320531e-05,
-      "loss": 2.1029,
-      "step": 4520
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.162109375,
-      "learning_rate": 3.144276939614683e-05,
-      "loss": 2.1297,
-      "step": 4525
-    },
-    {
-      "epoch": 1.53,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.122783371602747e-05,
-      "loss": 2.1324,
-      "step": 4530
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.166015625,
-      "learning_rate": 3.101349917159433e-05,
-      "loss": 2.1366,
-      "step": 4535
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.079976763633996e-05,
-      "loss": 2.1136,
-      "step": 4540
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.058664097848612e-05,
-      "loss": 2.1107,
-      "step": 4545
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.171875,
-      "learning_rate": 3.0374121060967255e-05,
-      "loss": 2.1024,
-      "step": 4550
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.0162209741414304e-05,
-      "loss": 2.1044,
-      "step": 4555
-    },
-    {
-      "epoch": 1.54,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.9950908872138584e-05,
-      "loss": 2.0755,
-      "step": 4560
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.9740220300115386e-05,
-      "loss": 2.1191,
-      "step": 4565
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.9530145866967895e-05,
-      "loss": 2.1415,
-      "step": 4570
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 2.9320687408951162e-05,
-      "loss": 2.0845,
-      "step": 4575
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.9111846756936113e-05,
-      "loss": 2.118,
-      "step": 4580
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.8903625736393304e-05,
-      "loss": 2.0881,
-      "step": 4585
-    },
-    {
-      "epoch": 1.55,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.8696026167377155e-05,
-      "loss": 2.1497,
-      "step": 4590
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 2.8489049864510054e-05,
-      "loss": 2.1046,
-      "step": 4595
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.8282698636966375e-05,
-      "loss": 2.139,
-      "step": 4600
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.8076974288456726e-05,
-      "loss": 2.0716,
-      "step": 4605
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.78718786172122e-05,
-      "loss": 2.1047,
-      "step": 4610
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.7667413415968635e-05,
-      "loss": 2.0918,
-      "step": 4615
-    },
-    {
-      "epoch": 1.56,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.7463580471950943e-05,
-      "loss": 2.1356,
-      "step": 4620
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 2.7260381566857473e-05,
-      "loss": 2.0997,
-      "step": 4625
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.7057818476844533e-05,
-      "loss": 2.0962,
-      "step": 4630
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.68558929725107e-05,
-      "loss": 2.095,
-      "step": 4635
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.6654606818881465e-05,
-      "loss": 2.1209,
-      "step": 4640
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.1640625,
-      "learning_rate": 2.645396177539379e-05,
-      "loss": 2.1148,
-      "step": 4645
-    },
-    {
-      "epoch": 1.57,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 2.6253959595880673e-05,
-      "loss": 2.1075,
-      "step": 4650
-    },
-    {
-      "epoch": 1.58,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.6054602028555885e-05,
-      "loss": 2.1129,
-      "step": 4655
-    },
-    {
-      "epoch": 1.58,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.585589081599862e-05,
-      "loss": 2.1114,
-      "step": 4660
-    },
-    {
-      "epoch": 1.58,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.5657827695138372e-05,
-      "loss": 2.1017,
-      "step": 4665
-    },
-    {
-      "epoch": 1.58,
-      "grad_norm": 0.177734375,
-      "learning_rate": 2.546041439723963e-05,
-      "loss": 2.1269,
-      "step": 4670
-    },
-    {
-      "epoch": 1.58,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.5263652647886803e-05,
-      "loss": 2.1149,
-      "step": 4675
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.5067544166969114e-05,
-      "loss": 2.1177,
-      "step": 4680
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.487209066866565e-05,
-      "loss": 2.0814,
-      "step": 4685
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.467729386143025e-05,
-      "loss": 2.1058,
-      "step": 4690
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.4483155447976657e-05,
-      "loss": 2.1239,
-      "step": 4695
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.42896771252636e-05,
-      "loss": 2.0943,
-      "step": 4700
-    },
-    {
-      "epoch": 1.59,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.4096860584479974e-05,
-      "loss": 2.135,
-      "step": 4705
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.390470751103008e-05,
-      "loss": 2.0901,
-      "step": 4710
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.37132195845188e-05,
-      "loss": 2.1333,
-      "step": 4715
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.3522398478737108e-05,
-      "loss": 2.0971,
-      "step": 4720
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.3332245861647206e-05,
-      "loss": 2.0869,
-      "step": 4725
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 2.3142763395368095e-05,
-      "loss": 2.0907,
-      "step": 4730
-    },
-    {
-      "epoch": 1.6,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.295395273616099e-05,
-      "loss": 2.1023,
-      "step": 4735
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.2765815534414868e-05,
-      "loss": 2.0932,
-      "step": 4740
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 2.257835343463205e-05,
-      "loss": 2.0842,
-      "step": 4745
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.173828125,
-      "learning_rate": 2.239156807541375e-05,
-      "loss": 2.1045,
-      "step": 4750
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.173828125,
-      "learning_rate": 2.22054610894459e-05,
-      "loss": 2.1369,
-      "step": 4755
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.1806640625,
-      "learning_rate": 2.202003410348473e-05,
-      "loss": 2.0935,
-      "step": 4760
-    },
-    {
-      "epoch": 1.61,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.1835288738342596e-05,
-      "loss": 2.1332,
-      "step": 4765
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.1651226608873877e-05,
-      "loss": 2.0987,
-      "step": 4770
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.1467849323960797e-05,
-      "loss": 2.136,
-      "step": 4775
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 2.128515848649929e-05,
-      "loss": 2.1241,
-      "step": 4780
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.1103155693385136e-05,
-      "loss": 2.137,
-      "step": 4785
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 2.092184253549998e-05,
-      "loss": 2.1204,
-      "step": 4790
-    },
-    {
-      "epoch": 1.62,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.074122059769733e-05,
-      "loss": 2.0799,
-      "step": 4795
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.0561291458788733e-05,
-      "loss": 2.1012,
-      "step": 4800
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.0382056691530084e-05,
-      "loss": 2.1381,
-      "step": 4805
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.02035178626077e-05,
-      "loss": 2.1153,
-      "step": 4810
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.002567653262479e-05,
-      "loss": 2.0965,
-      "step": 4815
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.1640625,
-      "learning_rate": 1.984853425608769e-05,
-      "loss": 2.1355,
-      "step": 4820
-    },
-    {
-      "epoch": 1.63,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.9672092581392375e-05,
-      "loss": 2.1273,
-      "step": 4825
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.9496353050810843e-05,
-      "loss": 2.1183,
-      "step": 4830
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 1.9321317200477653e-05,
-      "loss": 2.1448,
-      "step": 4835
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.91469865603766e-05,
-      "loss": 2.0943,
-      "step": 4840
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.8973362654327175e-05,
-      "loss": 2.117,
-      "step": 4845
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.166015625,
-      "learning_rate": 1.8800446999971346e-05,
-      "loss": 2.1076,
-      "step": 4850
-    },
-    {
-      "epoch": 1.64,
-      "grad_norm": 0.17578125,
-      "learning_rate": 1.8628241108760268e-05,
-      "loss": 2.0966,
-      "step": 4855
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.845674648594108e-05,
-      "loss": 2.1312,
-      "step": 4860
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.828596463054375e-05,
-      "loss": 2.0882,
-      "step": 4865
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.8115897035367934e-05,
-      "loss": 2.0856,
-      "step": 4870
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.7946545186970022e-05,
-      "loss": 2.1279,
-      "step": 4875
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.7777910565650024e-05,
-      "loss": 2.0998,
-      "step": 4880
-    },
-    {
-      "epoch": 1.65,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.760999464543869e-05,
-      "loss": 2.0973,
-      "step": 4885
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.7442798894084655e-05,
-      "loss": 2.1295,
-      "step": 4890
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.7276324773041565e-05,
-      "loss": 2.1326,
-      "step": 4895
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.7110573737455295e-05,
-      "loss": 2.1087,
-      "step": 4900
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.694554723615126e-05,
-      "loss": 2.1013,
-      "step": 4905
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.6781246711621744e-05,
-      "loss": 2.1268,
-      "step": 4910
-    },
-    {
-      "epoch": 1.66,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 1.6617673600013296e-05,
-      "loss": 2.1262,
-      "step": 4915
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.645482933111412e-05,
-      "loss": 2.0923,
-      "step": 4920
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.6292715328341712e-05,
-      "loss": 2.0922,
-      "step": 4925
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.6131333008730277e-05,
-      "loss": 2.1355,
-      "step": 4930
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.5970683782918374e-05,
-      "loss": 2.108,
-      "step": 4935
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.5810769055136644e-05,
-      "loss": 2.1115,
-      "step": 4940
-    },
-    {
-      "epoch": 1.67,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.56515902231955e-05,
-      "loss": 2.1024,
-      "step": 4945
-    },
-    {
-      "epoch": 1.68,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.5493148678472903e-05,
-      "loss": 2.1268,
-      "step": 4950
-    },
-    {
-      "epoch": 1.68,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.533544580590217e-05,
-      "loss": 2.1084,
-      "step": 4955
-    },
-    {
-      "epoch": 1.68,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.5178482983959985e-05,
-      "loss": 2.0965,
-      "step": 4960
-    },
-    {
-      "epoch": 1.68,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.5022261584654207e-05,
-      "loss": 2.1289,
-      "step": 4965
-    },
-    {
-      "epoch": 1.68,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.4866782973511962e-05,
-      "loss": 2.0724,
-      "step": 4970
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.4712048509567634e-05,
-      "loss": 2.1429,
-      "step": 4975
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.4558059545351143e-05,
-      "loss": 2.1328,
-      "step": 4980
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.4404817426875938e-05,
-      "loss": 2.0846,
-      "step": 4985
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.4252323493627251e-05,
-      "loss": 2.1035,
-      "step": 4990
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.4100579078550613e-05,
-      "loss": 2.0901,
-      "step": 4995
-    },
-    {
-      "epoch": 1.69,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 1.3949585508039886e-05,
-      "loss": 2.1229,
-      "step": 5000
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.3799344101925904e-05,
-      "loss": 2.1345,
-      "step": 5005
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.177734375,
-      "learning_rate": 1.3649856173464781e-05,
-      "loss": 2.1067,
-      "step": 5010
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.3501123029326601e-05,
-      "loss": 2.1274,
-      "step": 5015
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.3353145969583813e-05,
-      "loss": 2.1117,
-      "step": 5020
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.3205926287699988e-05,
-      "loss": 2.1139,
-      "step": 5025
-    },
-    {
-      "epoch": 1.7,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.3059465270518468e-05,
-      "loss": 2.1131,
-      "step": 5030
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.2913764198251132e-05,
-      "loss": 2.1569,
-      "step": 5035
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.276882434446719e-05,
-      "loss": 2.1104,
-      "step": 5040
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.2624646976082066e-05,
-      "loss": 2.1107,
-      "step": 5045
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.2481233353346344e-05,
-      "loss": 2.1026,
-      "step": 5050
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.2338584729834701e-05,
-      "loss": 2.0853,
-      "step": 5055
-    },
-    {
-      "epoch": 1.71,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.2196702352434953e-05,
-      "loss": 2.1188,
-      "step": 5060
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.205558746133727e-05,
-      "loss": 2.1077,
-      "step": 5065
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.1915241290023115e-05,
-      "loss": 2.097,
-      "step": 5070
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.1775665065254704e-05,
-      "loss": 2.1289,
-      "step": 5075
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.1636860007064076e-05,
-      "loss": 2.082,
-      "step": 5080
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.1498827328742623e-05,
-      "loss": 2.108,
-      "step": 5085
-    },
-    {
-      "epoch": 1.72,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.1361568236830323e-05,
-      "loss": 2.1289,
-      "step": 5090
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.166015625,
-      "learning_rate": 1.122508393110524e-05,
-      "loss": 2.1245,
-      "step": 5095
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.1089375604573116e-05,
-      "loss": 2.0823,
-      "step": 5100
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.0954444443456824e-05,
-      "loss": 2.0788,
-      "step": 5105
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.0820291627186107e-05,
-      "loss": 2.1059,
-      "step": 5110
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.1640625,
-      "learning_rate": 1.0686918328387118e-05,
-      "loss": 2.1058,
-      "step": 5115
-    },
-    {
-      "epoch": 1.73,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.0554325712872381e-05,
-      "loss": 2.1066,
-      "step": 5120
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.0422514939630424e-05,
-      "loss": 2.1088,
-      "step": 5125
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.0291487160815726e-05,
-      "loss": 2.1371,
-      "step": 5130
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.0161243521738661e-05,
-      "loss": 2.1256,
-      "step": 5135
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 1.003178516085541e-05,
-      "loss": 2.1213,
-      "step": 5140
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.169921875,
-      "learning_rate": 9.903113209758096e-06,
-      "loss": 2.1239,
-      "step": 5145
-    },
-    {
-      "epoch": 1.74,
-      "grad_norm": 0.16796875,
-      "learning_rate": 9.775228793164826e-06,
-      "loss": 2.0883,
-      "step": 5150
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 9.6481330289099e-06,
-      "loss": 2.0986,
-      "step": 5155
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 9.521827027934038e-06,
-      "loss": 2.1189,
-      "step": 5160
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 9.396311894274645e-06,
-      "loss": 2.1267,
-      "step": 5165
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 9.271588725056201e-06,
-      "loss": 2.1026,
-      "step": 5170
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.169921875,
-      "learning_rate": 9.147658610480625e-06,
-      "loss": 2.097,
-      "step": 5175
-    },
-    {
-      "epoch": 1.75,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 9.024522633817756e-06,
-      "loss": 2.1243,
-      "step": 5180
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.169921875,
-      "learning_rate": 8.90218187139591e-06,
-      "loss": 2.1412,
-      "step": 5185
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.780637392592495e-06,
-      "loss": 2.1046,
-      "step": 5190
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.659890259824532e-06,
-      "loss": 2.1173,
-      "step": 5195
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.1640625,
-      "learning_rate": 8.53994152853952e-06,
-      "loss": 2.1192,
-      "step": 5200
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.420792247206177e-06,
-      "loss": 2.1264,
-      "step": 5205
-    },
-    {
-      "epoch": 1.76,
-      "grad_norm": 0.16796875,
-      "learning_rate": 8.302443457305209e-06,
-      "loss": 2.0895,
-      "step": 5210
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 8.184896193320246e-06,
-      "loss": 2.1079,
-      "step": 5215
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.068151482728802e-06,
-      "loss": 2.0993,
-      "step": 5220
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.952210345993339e-06,
-      "loss": 2.1263,
-      "step": 5225
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.837073796552241e-06,
-      "loss": 2.0958,
-      "step": 5230
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.72274284081106e-06,
-      "loss": 2.1071,
-      "step": 5235
-    },
-    {
-      "epoch": 1.77,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.609218478133628e-06,
-      "loss": 2.1182,
-      "step": 5240
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 7.4965017008334245e-06,
-      "loss": 2.0934,
-      "step": 5245
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.169921875,
-      "learning_rate": 7.3845934941648046e-06,
-      "loss": 2.1125,
-      "step": 5250
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.2734948363144206e-06,
-      "loss": 2.0973,
-      "step": 5255
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.16796875,
-      "learning_rate": 7.163206698392744e-06,
-      "loss": 2.1164,
-      "step": 5260
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 7.0537300444254435e-06,
-      "loss": 2.1232,
-      "step": 5265
-    },
-    {
-      "epoch": 1.78,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.945065831345077e-06,
-      "loss": 2.1179,
-      "step": 5270
-    },
-    {
-      "epoch": 1.79,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 6.837215008982633e-06,
-      "loss": 2.1167,
-      "step": 5275
-    },
-    {
-      "epoch": 1.79,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.7301785200593046e-06,
-      "loss": 2.1473,
-      "step": 5280
-    },
-    {
-      "epoch": 1.79,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 6.623957300178207e-06,
-      "loss": 2.0955,
-      "step": 5285
-    },
-    {
-      "epoch": 1.79,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.518552277816215e-06,
-      "loss": 2.128,
-      "step": 5290
-    },
-    {
-      "epoch": 1.79,
-      "grad_norm": 0.169921875,
-      "learning_rate": 6.413964374315851e-06,
-      "loss": 2.1291,
-      "step": 5295
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.16796875,
-      "learning_rate": 6.31019450387721e-06,
-      "loss": 2.103,
-      "step": 5300
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.17578125,
-      "learning_rate": 6.207243573549959e-06,
-      "loss": 2.1046,
-      "step": 5305
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.1051124832254944e-06,
-      "loss": 2.0974,
-      "step": 5310
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.003802125628966e-06,
-      "loss": 2.1328,
-      "step": 5315
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 5.903313386311527e-06,
-      "loss": 2.0939,
-      "step": 5320
-    },
-    {
-      "epoch": 1.8,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.803647143642554e-06,
-      "loss": 2.1194,
-      "step": 5325
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.704804268802077e-06,
-      "loss": 2.1237,
-      "step": 5330
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 5.606785625773048e-06,
-      "loss": 2.1483,
-      "step": 5335
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.17578125,
-      "learning_rate": 5.5095920713338134e-06,
-      "loss": 2.14,
-      "step": 5340
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.17578125,
-      "learning_rate": 5.413224455050692e-06,
-      "loss": 2.1279,
-      "step": 5345
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.3176836192704414e-06,
-      "loss": 2.1356,
-      "step": 5350
-    },
-    {
-      "epoch": 1.81,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.222970399112981e-06,
-      "loss": 2.1192,
-      "step": 5355
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.12908562246408e-06,
-      "loss": 2.1321,
-      "step": 5360
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.171875,
-      "learning_rate": 5.036030109968082e-06,
-      "loss": 2.1285,
-      "step": 5365
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 4.9438046750207465e-06,
-      "loss": 2.117,
-      "step": 5370
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.8524101237621635e-06,
-      "loss": 2.1062,
-      "step": 5375
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.7618472550696954e-06,
-      "loss": 2.1149,
-      "step": 5380
-    },
-    {
-      "epoch": 1.82,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 4.67211686055099e-06,
-      "loss": 2.1403,
-      "step": 5385
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 4.583219724537046e-06,
-      "loss": 2.1146,
-      "step": 5390
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.495156624075347e-06,
-      "loss": 2.1065,
-      "step": 5395
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.169921875,
-      "learning_rate": 4.407928328923194e-06,
-      "loss": 2.1347,
-      "step": 5400
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 4.321535601540738e-06,
-      "loss": 2.1006,
-      "step": 5405
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.2359791970845496e-06,
-      "loss": 2.0962,
-      "step": 5410
-    },
-    {
-      "epoch": 1.83,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 4.1512598634009025e-06,
-      "loss": 2.0732,
-      "step": 5415
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.067378341019257e-06,
-      "loss": 2.1268,
-      "step": 5420
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 3.984335363145752e-06,
-      "loss": 2.1028,
-      "step": 5425
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 3.902131655656893e-06,
-      "loss": 2.0817,
-      "step": 5430
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.820767937093095e-06,
-      "loss": 2.1152,
-      "step": 5435
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.166015625,
-      "learning_rate": 3.740244918652469e-06,
-      "loss": 2.1147,
-      "step": 5440
-    },
-    {
-      "epoch": 1.84,
-      "grad_norm": 0.169921875,
-      "learning_rate": 3.6605633041846053e-06,
-      "loss": 2.0832,
-      "step": 5445
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.581723790184377e-06,
-      "loss": 2.1154,
-      "step": 5450
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 3.503727065785878e-06,
-      "loss": 2.1344,
-      "step": 5455
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.173828125,
-      "learning_rate": 3.4265738127564286e-06,
-      "loss": 2.1045,
-      "step": 5460
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.171875,
-      "learning_rate": 3.350264705490569e-06,
-      "loss": 2.1143,
-      "step": 5465
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 3.2748004110041863e-06,
-      "loss": 2.1074,
-      "step": 5470
-    },
-    {
-      "epoch": 1.85,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 3.2001815889286856e-06,
-      "loss": 2.0837,
-      "step": 5475
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.126408891505217e-06,
-      "loss": 2.0874,
-      "step": 5480
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.0534829635789684e-06,
-      "loss": 2.1232,
-      "step": 5485
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.9814044425935606e-06,
-      "loss": 2.1074,
-      "step": 5490
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.910173958585416e-06,
-      "loss": 2.082,
-      "step": 5495
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.8397921341783317e-06,
-      "loss": 2.0771,
-      "step": 5500
-    },
-    {
-      "epoch": 1.86,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 2.770259584577972e-06,
-      "loss": 2.1078,
-      "step": 5505
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 2.7015769175665063e-06,
-      "loss": 2.0803,
-      "step": 5510
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.633744733497312e-06,
-      "loss": 2.1227,
-      "step": 5515
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.1767578125,
-      "learning_rate": 2.5667636252897143e-06,
-      "loss": 2.1089,
-      "step": 5520
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 2.5006341784238107e-06,
-      "loss": 2.0958,
-      "step": 5525
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.435356970935354e-06,
-      "loss": 2.1027,
-      "step": 5530
-    },
-    {
-      "epoch": 1.87,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.370932573410667e-06,
-      "loss": 2.1082,
-      "step": 5535
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.3073615489817235e-06,
-      "loss": 2.1316,
-      "step": 5540
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.244644453321165e-06,
-      "loss": 2.0948,
-      "step": 5545
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 2.1827818346374482e-06,
-      "loss": 2.1408,
-      "step": 5550
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.121774233670104e-06,
-      "loss": 2.1094,
-      "step": 5555
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.0616221836849638e-06,
-      "loss": 2.117,
-      "step": 5560
-    },
-    {
-      "epoch": 1.88,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 2.0023262104694852e-06,
-      "loss": 2.1125,
-      "step": 5565
-    },
-    {
-      "epoch": 1.89,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.9438868323282124e-06,
-      "loss": 2.1178,
-      "step": 5570
-    },
-    {
-      "epoch": 1.89,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.8863045600782003e-06,
-      "loss": 2.1206,
-      "step": 5575
-    },
-    {
-      "epoch": 1.89,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 1.8295798970445754e-06,
-      "loss": 2.1231,
-      "step": 5580
-    },
-    {
-      "epoch": 1.89,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 1.7737133390561046e-06,
-      "loss": 2.1121,
-      "step": 5585
-    },
-    {
-      "epoch": 1.89,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.7187053744409098e-06,
-      "loss": 2.0981,
-      "step": 5590
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.6645564840221396e-06,
-      "loss": 2.1042,
-      "step": 5595
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.6112671411138036e-06,
-      "loss": 2.0861,
-      "step": 5600
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 1.5588378115166669e-06,
-      "loss": 2.0713,
-      "step": 5605
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.5072689535141072e-06,
-      "loss": 2.1116,
-      "step": 5610
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.4565610178681299e-06,
-      "loss": 2.1319,
-      "step": 5615
-    },
-    {
-      "epoch": 1.9,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.4067144478154604e-06,
-      "loss": 2.1159,
-      "step": 5620
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.3577296790636684e-06,
-      "loss": 2.1301,
-      "step": 5625
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 1.3096071397873056e-06,
-      "loss": 2.0978,
-      "step": 5630
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.166015625,
-      "learning_rate": 1.2623472506242184e-06,
-      "loss": 2.1322,
-      "step": 5635
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.2159504246718522e-06,
-      "loss": 2.1104,
-      "step": 5640
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.1704170674836313e-06,
-      "loss": 2.1059,
-      "step": 5645
-    },
-    {
-      "epoch": 1.91,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 1.125747577065428e-06,
-      "loss": 2.1137,
-      "step": 5650
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.0819423438720665e-06,
-      "loss": 2.1135,
-      "step": 5655
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 1.0390017508039473e-06,
-      "loss": 2.1428,
-      "step": 5660
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.17578125,
-      "learning_rate": 9.969261732036605e-07,
-      "loss": 2.1313,
-      "step": 5665
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.557159788526892e-07,
-      "loss": 2.115,
-      "step": 5670
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 9.153715279682784e-07,
-      "loss": 2.1337,
-      "step": 5675
-    },
-    {
-      "epoch": 1.92,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 8.75893173200204e-07,
-      "loss": 2.107,
-      "step": 5680
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.169921875,
-      "learning_rate": 8.372812596277091e-07,
-      "loss": 2.1094,
-      "step": 5685
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.99536124756517e-07,
-      "loss": 2.093,
-      "step": 5690
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.171875,
-      "learning_rate": 7.62658098515856e-07,
-      "loss": 2.0956,
-      "step": 5695
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 7.266475032555619e-07,
-      "loss": 2.12,
-      "step": 5700
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 6.915046537433023e-07,
-      "loss": 2.1025,
-      "step": 5705
-    },
-    {
-      "epoch": 1.93,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 6.572298571618118e-07,
-      "loss": 2.1076,
-      "step": 5710
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 6.238234131061616e-07,
-      "loss": 2.0958,
-      "step": 5715
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 5.912856135812051e-07,
-      "loss": 2.102,
-      "step": 5720
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.16796875,
-      "learning_rate": 5.596167429989807e-07,
-      "loss": 2.1054,
-      "step": 5725
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.288170781762469e-07,
-      "loss": 2.1115,
-      "step": 5730
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 4.98886888332073e-07,
-      "loss": 2.1091,
-      "step": 5735
-    },
-    {
-      "epoch": 1.94,
-      "grad_norm": 0.16796875,
-      "learning_rate": 4.698264350854409e-07,
-      "loss": 2.1136,
-      "step": 5740
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.1767578125,
-      "learning_rate": 4.416359724530139e-07,
-      "loss": 2.1142,
-      "step": 5745
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 4.143157468468717e-07,
-      "loss": 2.1154,
-      "step": 5750
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 3.878659970724008e-07,
-      "loss": 2.1069,
-      "step": 5755
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.16796875,
-      "learning_rate": 3.622869543261298e-07,
-      "loss": 2.0822,
-      "step": 5760
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.1748046875,
-      "learning_rate": 3.3757884219383085e-07,
-      "loss": 2.1078,
-      "step": 5765
-    },
-    {
-      "epoch": 1.95,
-      "grad_norm": 0.1728515625,
-      "learning_rate": 3.1374187664844346e-07,
-      "loss": 2.115,
-      "step": 5770
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.16796875,
-      "learning_rate": 2.907762660482538e-07,
-      "loss": 2.1022,
-      "step": 5775
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.6868221113505175e-07,
-      "loss": 2.1222,
-      "step": 5780
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.474599050323989e-07,
-      "loss": 2.1228,
-      "step": 5785
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.271095332438966e-07,
-      "loss": 2.0893,
-      "step": 5790
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.166015625,
-      "learning_rate": 2.076312736516206e-07,
-      "loss": 2.1281,
-      "step": 5795
-    },
-    {
-      "epoch": 1.96,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.890252965145112e-07,
-      "loss": 2.0929,
-      "step": 5800
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.166015625,
-      "learning_rate": 1.7129176446692984e-07,
-      "loss": 2.0959,
-      "step": 5805
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.5443083251720503e-07,
-      "loss": 2.133,
-      "step": 5810
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.384426480462997e-07,
-      "loss": 2.0971,
-      "step": 5815
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.2332735080651248e-07,
-      "loss": 2.0809,
-      "step": 5820
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.171875,
-      "learning_rate": 1.0908507292026748e-07,
-      "loss": 2.133,
-      "step": 5825
-    },
-    {
-      "epoch": 1.97,
-      "grad_norm": 0.1650390625,
-      "learning_rate": 9.571593887891528e-08,
-      "loss": 2.1076,
-      "step": 5830
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.171875,
-      "learning_rate": 8.322006554171146e-08,
-      "loss": 2.1422,
-      "step": 5835
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.1669921875,
-      "learning_rate": 7.159756213476199e-08,
-      "loss": 2.0922,
-      "step": 5840
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.1640625,
-      "learning_rate": 6.084853025005721e-08,
-      "loss": 2.1117,
-      "step": 5845
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.169921875,
-      "learning_rate": 5.0973063844605986e-08,
-      "loss": 2.0981,
-      "step": 5850
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.1708984375,
-      "learning_rate": 4.1971249239591834e-08,
-      "loss": 2.0941,
-      "step": 5855
-    },
-    {
-      "epoch": 1.98,
-      "grad_norm": 0.1640625,
-      "learning_rate": 3.384316511964025e-08,
-      "loss": 2.1075,
-      "step": 5860
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 2.658888253211922e-08,
-      "loss": 2.107,
-      "step": 5865
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.171875,
-      "learning_rate": 2.0208464886517508e-08,
-      "loss": 2.105,
-      "step": 5870
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.16796875,
-      "learning_rate": 1.4701967953911766e-08,
-      "loss": 2.1027,
-      "step": 5875
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.1689453125,
-      "learning_rate": 1.0069439866422503e-08,
-      "loss": 2.1107,
-      "step": 5880
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.171875,
-      "learning_rate": 6.3109211168699275e-09,
-      "loss": 2.1134,
-      "step": 5885
-    },
-    {
-      "epoch": 1.99,
-      "grad_norm": 0.1630859375,
-      "learning_rate": 3.4264445583631622e-09,
-      "loss": 2.1139,
-      "step": 5890
-    },
-    {
-      "epoch": 2.0,
-      "grad_norm": 0.169921875,
-      "learning_rate": 1.4160354040448908e-09,
-      "loss": 2.0934,
-      "step": 5895
-    },
-    {
-      "epoch": 2.0,
-      "grad_norm": 0.169921875,
-      "learning_rate": 2.7971122683601023e-10,
-      "loss": 2.1251,
-      "step": 5900
-    },
-    {
-      "epoch": 2.0,
-      "eval_loss": 2.12707781791687,
-      "eval_runtime": 173.3977,
-      "eval_samples_per_second": 7.659,
-      "eval_steps_per_second": 0.957,
-      "step": 5904
-    },
-    {
-      "epoch": 2.0,
-      "step": 5904,
-      "total_flos": 6.077264558911652e+17,
-      "train_loss": 2.1395558682560596,
-      "train_runtime": 23055.2544,
-      "train_samples_per_second": 2.049,
-      "train_steps_per_second": 0.256
+      "epoch": 1.0,
+      "step": 670,
+      "total_flos": 3.038889647317975e+17,
+      "train_loss": 2.275535515884855,
+      "train_runtime": 10859.5951,
+      "train_samples_per_second": 7.898,
+      "train_steps_per_second": 0.062
     }
   ],
   "logging_steps": 5,
-  "max_steps": 5904,
+  "max_steps": 670,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 2,
+  "num_train_epochs": 1,
   "save_steps": 100,
-  "total_flos": 6.077264558911652e+17,
-  "train_batch_size": 4,
+  "total_flos": 3.038889647317975e+17,
+  "train_batch_size": 32,
   "trial_name": null,
   "trial_params": null
 }