diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,4234 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.790812618113098,
+  "eval_steps": 500,
+  "global_step": 600,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.004651838930077046,
+      "grad_norm": 0.49046942591667175,
+      "learning_rate": 0.0,
+      "loss": 1.9232,
+      "step": 1
+    },
+    {
+      "epoch": 0.009303677860154092,
+      "grad_norm": 0.4551486074924469,
+      "learning_rate": 3.0769230769230774e-06,
+      "loss": 1.8513,
+      "step": 2
+    },
+    {
+      "epoch": 0.013955516790231139,
+      "grad_norm": 0.47721266746520996,
+      "learning_rate": 6.153846153846155e-06,
+      "loss": 1.9152,
+      "step": 3
+    },
+    {
+      "epoch": 0.018607355720308184,
+      "grad_norm": 0.48268425464630127,
+      "learning_rate": 9.230769230769232e-06,
+      "loss": 1.9277,
+      "step": 4
+    },
+    {
+      "epoch": 0.02325919465038523,
+      "grad_norm": 0.47317737340927124,
+      "learning_rate": 1.230769230769231e-05,
+      "loss": 1.8662,
+      "step": 5
+    },
+    {
+      "epoch": 0.027911033580462277,
+      "grad_norm": 0.4764399826526642,
+      "learning_rate": 1.5384615384615387e-05,
+      "loss": 1.9053,
+      "step": 6
+    },
+    {
+      "epoch": 0.032562872510539326,
+      "grad_norm": 0.4838545024394989,
+      "learning_rate": 1.8461538461538465e-05,
+      "loss": 1.8085,
+      "step": 7
+    },
+    {
+      "epoch": 0.03721471144061637,
+      "grad_norm": 0.47017717361450195,
+      "learning_rate": 2.1538461538461542e-05,
+      "loss": 1.7872,
+      "step": 8
+    },
+    {
+      "epoch": 0.041866550370693416,
+      "grad_norm": 0.5008382797241211,
+      "learning_rate": 2.461538461538462e-05,
+      "loss": 1.8718,
+      "step": 9
+    },
+    {
+      "epoch": 0.04651838930077046,
+      "grad_norm": 0.5195590853691101,
+      "learning_rate": 2.7692307692307694e-05,
+      "loss": 1.8715,
+      "step": 10
+    },
+    {
+      "epoch": 0.051170228230847506,
+      "grad_norm": 0.5609027147293091,
+      "learning_rate": 3.0769230769230774e-05,
+      "loss": 1.8675,
+      "step": 11
+    },
+    {
+      "epoch": 0.055822067160924554,
+      "grad_norm": 0.5538550019264221,
+      "learning_rate": 3.384615384615385e-05,
+      "loss": 1.8735,
+      "step": 12
+    },
+    {
+      "epoch": 0.060473906091001596,
+      "grad_norm": 0.5431339144706726,
+      "learning_rate": 3.692307692307693e-05,
+      "loss": 1.8222,
+      "step": 13
+    },
+    {
+      "epoch": 0.06512574502107865,
+      "grad_norm": 0.6020752191543579,
+      "learning_rate": 4e-05,
+      "loss": 1.8907,
+      "step": 14
+    },
+    {
+      "epoch": 0.06977758395115569,
+      "grad_norm": 0.6046931743621826,
+      "learning_rate": 4.3076923076923084e-05,
+      "loss": 1.8166,
+      "step": 15
+    },
+    {
+      "epoch": 0.07442942288123273,
+      "grad_norm": 0.6009289622306824,
+      "learning_rate": 4.615384615384616e-05,
+      "loss": 1.835,
+      "step": 16
+    },
+    {
+      "epoch": 0.07908126181130978,
+      "grad_norm": 0.5840696692466736,
+      "learning_rate": 4.923076923076924e-05,
+      "loss": 1.8054,
+      "step": 17
+    },
+    {
+      "epoch": 0.08373310074138683,
+      "grad_norm": 0.5175262093544006,
+      "learning_rate": 5.230769230769231e-05,
+      "loss": 1.6161,
+      "step": 18
+    },
+    {
+      "epoch": 0.08838493967146388,
+      "grad_norm": 0.5089394450187683,
+      "learning_rate": 5.538461538461539e-05,
+      "loss": 1.7065,
+      "step": 19
+    },
+    {
+      "epoch": 0.09303677860154091,
+      "grad_norm": 0.4888729155063629,
+      "learning_rate": 5.846153846153847e-05,
+      "loss": 1.615,
+      "step": 20
+    },
+    {
+      "epoch": 0.09768861753161796,
+      "grad_norm": 0.4274122416973114,
+      "learning_rate": 6.153846153846155e-05,
+      "loss": 1.5828,
+      "step": 21
+    },
+    {
+      "epoch": 0.10234045646169501,
+      "grad_norm": 0.4088641107082367,
+      "learning_rate": 6.461538461538462e-05,
+      "loss": 1.6295,
+      "step": 22
+    },
+    {
+      "epoch": 0.10699229539177206,
+      "grad_norm": 0.3996894955635071,
+      "learning_rate": 6.76923076923077e-05,
+      "loss": 1.6303,
+      "step": 23
+    },
+    {
+      "epoch": 0.11164413432184911,
+      "grad_norm": 0.41219764947891235,
+      "learning_rate": 7.076923076923078e-05,
+      "loss": 1.5787,
+      "step": 24
+    },
+    {
+      "epoch": 0.11629597325192616,
+      "grad_norm": 0.400268018245697,
+      "learning_rate": 7.384615384615386e-05,
+      "loss": 1.5245,
+      "step": 25
+    },
+    {
+      "epoch": 0.12094781218200319,
+      "grad_norm": 0.4030200242996216,
+      "learning_rate": 7.692307692307693e-05,
+      "loss": 1.5006,
+      "step": 26
+    },
+    {
+      "epoch": 0.12559965111208024,
+      "grad_norm": 0.39891666173934937,
+      "learning_rate": 8e-05,
+      "loss": 1.5109,
+      "step": 27
+    },
+    {
+      "epoch": 0.1302514900421573,
+      "grad_norm": 0.3856427073478699,
+      "learning_rate": 8.307692307692309e-05,
+      "loss": 1.4708,
+      "step": 28
+    },
+    {
+      "epoch": 0.13490332897223434,
+      "grad_norm": 0.37282320857048035,
+      "learning_rate": 8.615384615384617e-05,
+      "loss": 1.4338,
+      "step": 29
+    },
+    {
+      "epoch": 0.13955516790231137,
+      "grad_norm": 0.3495185673236847,
+      "learning_rate": 8.923076923076924e-05,
+      "loss": 1.4543,
+      "step": 30
+    },
+    {
+      "epoch": 0.14420700683238843,
+      "grad_norm": 0.3242153227329254,
+      "learning_rate": 9.230769230769232e-05,
+      "loss": 1.4181,
+      "step": 31
+    },
+    {
+      "epoch": 0.14885884576246547,
+      "grad_norm": 0.3061245381832123,
+      "learning_rate": 9.53846153846154e-05,
+      "loss": 1.3795,
+      "step": 32
+    },
+    {
+      "epoch": 0.15351068469254253,
+      "grad_norm": 0.3043881058692932,
+      "learning_rate": 9.846153846153848e-05,
+      "loss": 1.3648,
+      "step": 33
+    },
+    {
+      "epoch": 0.15816252362261957,
+      "grad_norm": 0.28627124428749084,
+      "learning_rate": 0.00010153846153846153,
+      "loss": 1.368,
+      "step": 34
+    },
+    {
+      "epoch": 0.1628143625526966,
+      "grad_norm": 0.28764569759368896,
+      "learning_rate": 0.00010461538461538463,
+      "loss": 1.3213,
+      "step": 35
+    },
+    {
+      "epoch": 0.16746620148277366,
+      "grad_norm": 0.2773509919643402,
+      "learning_rate": 0.0001076923076923077,
+      "loss": 1.3,
+      "step": 36
+    },
+    {
+      "epoch": 0.1721180404128507,
+      "grad_norm": 0.2724374532699585,
+      "learning_rate": 0.00011076923076923077,
+      "loss": 1.2917,
+      "step": 37
+    },
+    {
+      "epoch": 0.17676987934292776,
+      "grad_norm": 0.26691335439682007,
+      "learning_rate": 0.00011384615384615384,
+      "loss": 1.3031,
+      "step": 38
+    },
+    {
+      "epoch": 0.1814217182730048,
+      "grad_norm": 0.2559492588043213,
+      "learning_rate": 0.00011692307692307694,
+      "loss": 1.2803,
+      "step": 39
+    },
+    {
+      "epoch": 0.18607355720308183,
+      "grad_norm": 0.2503993809223175,
+      "learning_rate": 0.00012,
+      "loss": 1.234,
+      "step": 40
+    },
+    {
+      "epoch": 0.1907253961331589,
+      "grad_norm": 0.22476454079151154,
+      "learning_rate": 0.0001230769230769231,
+      "loss": 1.1956,
+      "step": 41
+    },
+    {
+      "epoch": 0.19537723506323593,
+      "grad_norm": 0.22879214584827423,
+      "learning_rate": 0.00012615384615384615,
+      "loss": 1.1645,
+      "step": 42
+    },
+    {
+      "epoch": 0.200029073993313,
+      "grad_norm": 0.23353807628154755,
+      "learning_rate": 0.00012923076923076923,
+      "loss": 1.2069,
+      "step": 43
+    },
+    {
+      "epoch": 0.20468091292339002,
+      "grad_norm": 0.2165878415107727,
+      "learning_rate": 0.0001323076923076923,
+      "loss": 1.1479,
+      "step": 44
+    },
+    {
+      "epoch": 0.2093327518534671,
+      "grad_norm": 0.22984957695007324,
+      "learning_rate": 0.0001353846153846154,
+      "loss": 1.1979,
+      "step": 45
+    },
+    {
+      "epoch": 0.21398459078354412,
+      "grad_norm": 0.2613763213157654,
+      "learning_rate": 0.00013846153846153847,
+      "loss": 1.1338,
+      "step": 46
+    },
+    {
+      "epoch": 0.21863642971362116,
+      "grad_norm": 0.23487871885299683,
+      "learning_rate": 0.00014153846153846156,
+      "loss": 1.1355,
+      "step": 47
+    },
+    {
+      "epoch": 0.22328826864369822,
+      "grad_norm": 0.22157953679561615,
+      "learning_rate": 0.0001446153846153846,
+      "loss": 1.1652,
+      "step": 48
+    },
+    {
+      "epoch": 0.22794010757377525,
+      "grad_norm": 0.2523654103279114,
+      "learning_rate": 0.00014769230769230772,
+      "loss": 1.0605,
+      "step": 49
+    },
+    {
+      "epoch": 0.23259194650385231,
+      "grad_norm": 0.23878046870231628,
+      "learning_rate": 0.00015076923076923077,
+      "loss": 1.0494,
+      "step": 50
+    },
+    {
+      "epoch": 0.23724378543392935,
+      "grad_norm": 0.23895445466041565,
+      "learning_rate": 0.00015384615384615385,
+      "loss": 1.1001,
+      "step": 51
+    },
+    {
+      "epoch": 0.24189562436400638,
+      "grad_norm": 0.2330354005098343,
+      "learning_rate": 0.00015692307692307693,
+      "loss": 1.0466,
+      "step": 52
+    },
+    {
+      "epoch": 0.24654746329408345,
+      "grad_norm": 0.2505215108394623,
+      "learning_rate": 0.00016,
+      "loss": 1.0958,
+      "step": 53
+    },
+    {
+      "epoch": 0.2511993022241605,
+      "grad_norm": 0.2130110114812851,
+      "learning_rate": 0.0001630769230769231,
+      "loss": 1.0403,
+      "step": 54
+    },
+    {
+      "epoch": 0.2558511411542375,
+      "grad_norm": 0.22909505665302277,
+      "learning_rate": 0.00016615384615384617,
+      "loss": 1.035,
+      "step": 55
+    },
+    {
+      "epoch": 0.2605029800843146,
+      "grad_norm": 0.2630414664745331,
+      "learning_rate": 0.00016923076923076923,
+      "loss": 1.0245,
+      "step": 56
+    },
+    {
+      "epoch": 0.26515481901439164,
+      "grad_norm": 0.22364258766174316,
+      "learning_rate": 0.00017230769230769234,
+      "loss": 1.009,
+      "step": 57
+    },
+    {
+      "epoch": 0.2698066579444687,
+      "grad_norm": 0.23399153351783752,
+      "learning_rate": 0.0001753846153846154,
+      "loss": 1.0473,
+      "step": 58
+    },
+    {
+      "epoch": 0.2744584968745457,
+      "grad_norm": 0.22089968621730804,
+      "learning_rate": 0.00017846153846153847,
+      "loss": 0.9691,
+      "step": 59
+    },
+    {
+      "epoch": 0.27911033580462274,
+      "grad_norm": 0.23827041685581207,
+      "learning_rate": 0.00018153846153846155,
+      "loss": 0.9627,
+      "step": 60
+    },
+    {
+      "epoch": 0.28376217473469983,
+      "grad_norm": 0.23421761393547058,
+      "learning_rate": 0.00018461538461538463,
+      "loss": 0.9668,
+      "step": 61
+    },
+    {
+      "epoch": 0.28841401366477687,
+      "grad_norm": 0.25627240538597107,
+      "learning_rate": 0.0001876923076923077,
+      "loss": 0.9729,
+      "step": 62
+    },
+    {
+      "epoch": 0.2930658525948539,
+      "grad_norm": 0.24521057307720184,
+      "learning_rate": 0.0001907692307692308,
+      "loss": 0.9909,
+      "step": 63
+    },
+    {
+      "epoch": 0.29771769152493094,
+      "grad_norm": 0.24802164733409882,
+      "learning_rate": 0.00019384615384615385,
+      "loss": 0.907,
+      "step": 64
+    },
+    {
+      "epoch": 0.302369530455008,
+      "grad_norm": 0.2066604346036911,
+      "learning_rate": 0.00019692307692307696,
+      "loss": 0.9244,
+      "step": 65
+    },
+    {
+      "epoch": 0.30702136938508506,
+      "grad_norm": 0.24043628573417664,
+      "learning_rate": 0.0002,
+      "loss": 0.878,
+      "step": 66
+    },
+    {
+      "epoch": 0.3116732083151621,
+      "grad_norm": 0.24069836735725403,
+      "learning_rate": 0.000199748427672956,
+      "loss": 0.9361,
+      "step": 67
+    },
+    {
+      "epoch": 0.31632504724523913,
+      "grad_norm": 0.24457211792469025,
+      "learning_rate": 0.00019949685534591195,
+      "loss": 0.9777,
+      "step": 68
+    },
+    {
+      "epoch": 0.32097688617531617,
+      "grad_norm": 0.24842187762260437,
+      "learning_rate": 0.00019924528301886794,
+      "loss": 0.9024,
+      "step": 69
+    },
+    {
+      "epoch": 0.3256287251053932,
+      "grad_norm": 0.24496324360370636,
+      "learning_rate": 0.0001989937106918239,
+      "loss": 0.8873,
+      "step": 70
+    },
+    {
+      "epoch": 0.3302805640354703,
+      "grad_norm": 0.23857317864894867,
+      "learning_rate": 0.00019874213836477988,
+      "loss": 0.9328,
+      "step": 71
+    },
+    {
+      "epoch": 0.3349324029655473,
+      "grad_norm": 0.2626417577266693,
+      "learning_rate": 0.00019849056603773587,
+      "loss": 0.9145,
+      "step": 72
+    },
+    {
+      "epoch": 0.33958424189562436,
+      "grad_norm": 0.25375983119010925,
+      "learning_rate": 0.00019823899371069183,
+      "loss": 0.849,
+      "step": 73
+    },
+    {
+      "epoch": 0.3442360808257014,
+      "grad_norm": 0.26692864298820496,
+      "learning_rate": 0.0001979874213836478,
+      "loss": 0.9047,
+      "step": 74
+    },
+    {
+      "epoch": 0.34888791975577843,
+      "grad_norm": 0.2556419372558594,
+      "learning_rate": 0.0001977358490566038,
+      "loss": 0.8856,
+      "step": 75
+    },
+    {
+      "epoch": 0.3535397586858555,
+      "grad_norm": 0.2259674370288849,
+      "learning_rate": 0.00019748427672955975,
+      "loss": 0.8663,
+      "step": 76
+    },
+    {
+      "epoch": 0.35819159761593256,
+      "grad_norm": 0.26506906747817993,
+      "learning_rate": 0.00019723270440251574,
+      "loss": 0.8499,
+      "step": 77
+    },
+    {
+      "epoch": 0.3628434365460096,
+      "grad_norm": 0.2743973135948181,
+      "learning_rate": 0.0001969811320754717,
+      "loss": 0.8647,
+      "step": 78
+    },
+    {
+      "epoch": 0.3674952754760866,
+      "grad_norm": 0.28192541003227234,
+      "learning_rate": 0.00019672955974842768,
+      "loss": 0.8876,
+      "step": 79
+    },
+    {
+      "epoch": 0.37214711440616366,
+      "grad_norm": 0.275380939245224,
+      "learning_rate": 0.00019647798742138367,
+      "loss": 0.8733,
+      "step": 80
+    },
+    {
+      "epoch": 0.37679895333624075,
+      "grad_norm": 0.286601185798645,
+      "learning_rate": 0.00019622641509433963,
+      "loss": 0.8222,
+      "step": 81
+    },
+    {
+      "epoch": 0.3814507922663178,
+      "grad_norm": 0.286024272441864,
+      "learning_rate": 0.0001959748427672956,
+      "loss": 0.8565,
+      "step": 82
+    },
+    {
+      "epoch": 0.3861026311963948,
+      "grad_norm": 0.28871631622314453,
+      "learning_rate": 0.00019572327044025157,
+      "loss": 0.8377,
+      "step": 83
+    },
+    {
+      "epoch": 0.39075447012647185,
+      "grad_norm": 0.2878912091255188,
+      "learning_rate": 0.00019547169811320755,
+      "loss": 0.8302,
+      "step": 84
+    },
+    {
+      "epoch": 0.39540630905654894,
+      "grad_norm": 0.2819346487522125,
+      "learning_rate": 0.00019522012578616354,
+      "loss": 0.8237,
+      "step": 85
+    },
+    {
+      "epoch": 0.400058147986626,
+      "grad_norm": 0.2735177278518677,
+      "learning_rate": 0.0001949685534591195,
+      "loss": 0.7906,
+      "step": 86
+    },
+    {
+      "epoch": 0.404709986916703,
+      "grad_norm": 0.2869543433189392,
+      "learning_rate": 0.00019471698113207548,
+      "loss": 0.8045,
+      "step": 87
+    },
+    {
+      "epoch": 0.40936182584678005,
+      "grad_norm": 0.2559930682182312,
+      "learning_rate": 0.00019446540880503147,
+      "loss": 0.7989,
+      "step": 88
+    },
+    {
+      "epoch": 0.4140136647768571,
+      "grad_norm": 0.3100188076496124,
+      "learning_rate": 0.00019421383647798743,
+      "loss": 0.7979,
+      "step": 89
+    },
+    {
+      "epoch": 0.4186655037069342,
+      "grad_norm": 0.26276376843452454,
+      "learning_rate": 0.0001939622641509434,
+      "loss": 0.8527,
+      "step": 90
+    },
+    {
+      "epoch": 0.4233173426370112,
+      "grad_norm": 0.33543840050697327,
+      "learning_rate": 0.00019371069182389937,
+      "loss": 0.8235,
+      "step": 91
+    },
+    {
+      "epoch": 0.42796918156708824,
+      "grad_norm": 0.29981857538223267,
+      "learning_rate": 0.00019345911949685536,
+      "loss": 0.8094,
+      "step": 92
+    },
+    {
+      "epoch": 0.4326210204971653,
+      "grad_norm": 0.2866899073123932,
+      "learning_rate": 0.00019320754716981134,
+      "loss": 0.818,
+      "step": 93
+    },
+    {
+      "epoch": 0.4372728594272423,
+      "grad_norm": 0.33317920565605164,
+      "learning_rate": 0.0001929559748427673,
+      "loss": 0.8095,
+      "step": 94
+    },
+    {
+      "epoch": 0.4419246983573194,
+      "grad_norm": 0.2851763367652893,
+      "learning_rate": 0.00019270440251572328,
+      "loss": 0.8122,
+      "step": 95
+    },
+    {
+      "epoch": 0.44657653728739644,
+      "grad_norm": 0.3371889591217041,
+      "learning_rate": 0.00019245283018867927,
+      "loss": 0.7764,
+      "step": 96
+    },
+    {
+      "epoch": 0.45122837621747347,
+      "grad_norm": 0.25665178894996643,
+      "learning_rate": 0.00019220125786163523,
+      "loss": 0.8043,
+      "step": 97
+    },
+    {
+      "epoch": 0.4558802151475505,
+      "grad_norm": 0.338375985622406,
+      "learning_rate": 0.0001919496855345912,
+      "loss": 0.8125,
+      "step": 98
+    },
+    {
+      "epoch": 0.46053205407762754,
+      "grad_norm": 0.29257163405418396,
+      "learning_rate": 0.00019169811320754717,
+      "loss": 0.7649,
+      "step": 99
+    },
+    {
+      "epoch": 0.46518389300770463,
+      "grad_norm": 0.3698625862598419,
+      "learning_rate": 0.00019144654088050316,
+      "loss": 0.7995,
+      "step": 100
+    },
+    {
+      "epoch": 0.46983573193778166,
+      "grad_norm": 0.3086777925491333,
+      "learning_rate": 0.00019119496855345914,
+      "loss": 0.7625,
+      "step": 101
+    },
+    {
+      "epoch": 0.4744875708678587,
+      "grad_norm": 0.3038915991783142,
+      "learning_rate": 0.0001909433962264151,
+      "loss": 0.7973,
+      "step": 102
+    },
+    {
+      "epoch": 0.47913940979793573,
+      "grad_norm": 0.31609514355659485,
+      "learning_rate": 0.00019069182389937108,
+      "loss": 0.7634,
+      "step": 103
+    },
+    {
+      "epoch": 0.48379124872801277,
+      "grad_norm": 0.27239060401916504,
+      "learning_rate": 0.00019044025157232704,
+      "loss": 0.7627,
+      "step": 104
+    },
+    {
+      "epoch": 0.48844308765808986,
+      "grad_norm": 0.327146977186203,
+      "learning_rate": 0.00019018867924528303,
+      "loss": 0.7797,
+      "step": 105
+    },
+    {
+      "epoch": 0.4930949265881669,
+      "grad_norm": 0.27846774458885193,
+      "learning_rate": 0.00018993710691823901,
+      "loss": 0.7859,
+      "step": 106
+    },
+    {
+      "epoch": 0.4977467655182439,
+      "grad_norm": 0.3272345960140228,
+      "learning_rate": 0.00018968553459119497,
+      "loss": 0.784,
+      "step": 107
+    },
+    {
+      "epoch": 0.502398604448321,
+      "grad_norm": 0.2832285761833191,
+      "learning_rate": 0.00018943396226415096,
+      "loss": 0.7409,
+      "step": 108
+    },
+    {
+      "epoch": 0.507050443378398,
+      "grad_norm": 0.31071850657463074,
+      "learning_rate": 0.00018918238993710694,
+      "loss": 0.7726,
+      "step": 109
+    },
+    {
+      "epoch": 0.511702282308475,
+      "grad_norm": 0.2981129586696625,
+      "learning_rate": 0.0001889308176100629,
+      "loss": 0.7316,
+      "step": 110
+    },
+    {
+      "epoch": 0.5163541212385521,
+      "grad_norm": 0.2790229320526123,
+      "learning_rate": 0.00018867924528301889,
+      "loss": 0.7958,
+      "step": 111
+    },
+    {
+      "epoch": 0.5210059601686292,
+      "grad_norm": 0.3225451111793518,
+      "learning_rate": 0.00018842767295597484,
+      "loss": 0.7332,
+      "step": 112
+    },
+    {
+      "epoch": 0.5256577990987062,
+      "grad_norm": 0.2896290421485901,
+      "learning_rate": 0.00018817610062893083,
+      "loss": 0.7516,
+      "step": 113
+    },
+    {
+      "epoch": 0.5303096380287833,
+      "grad_norm": 0.3030947148799896,
+      "learning_rate": 0.00018792452830188681,
+      "loss": 0.7661,
+      "step": 114
+    },
+    {
+      "epoch": 0.5349614769588603,
+      "grad_norm": 0.31590983271598816,
+      "learning_rate": 0.00018767295597484277,
+      "loss": 0.7399,
+      "step": 115
+    },
+    {
+      "epoch": 0.5396133158889374,
+      "grad_norm": 0.27730801701545715,
+      "learning_rate": 0.00018742138364779876,
+      "loss": 0.7549,
+      "step": 116
+    },
+    {
+      "epoch": 0.5442651548190144,
+      "grad_norm": 0.3037121593952179,
+      "learning_rate": 0.00018716981132075472,
+      "loss": 0.7483,
+      "step": 117
+    },
+    {
+      "epoch": 0.5489169937490914,
+      "grad_norm": 0.3050958216190338,
+      "learning_rate": 0.0001869182389937107,
+      "loss": 0.7772,
+      "step": 118
+    },
+    {
+      "epoch": 0.5535688326791685,
+      "grad_norm": 0.3435346782207489,
+      "learning_rate": 0.0001866666666666667,
+      "loss": 0.7806,
+      "step": 119
+    },
+    {
+      "epoch": 0.5582206716092455,
+      "grad_norm": 0.33755987882614136,
+      "learning_rate": 0.00018641509433962264,
+      "loss": 0.7574,
+      "step": 120
+    },
+    {
+      "epoch": 0.5628725105393225,
+      "grad_norm": 0.2960521876811981,
+      "learning_rate": 0.00018616352201257863,
+      "loss": 0.7536,
+      "step": 121
+    },
+    {
+      "epoch": 0.5675243494693997,
+      "grad_norm": 0.34236013889312744,
+      "learning_rate": 0.00018591194968553462,
+      "loss": 0.7581,
+      "step": 122
+    },
+    {
+      "epoch": 0.5721761883994767,
+      "grad_norm": 0.27558809518814087,
+      "learning_rate": 0.00018566037735849057,
+      "loss": 0.7436,
+      "step": 123
+    },
+    {
+      "epoch": 0.5768280273295537,
+      "grad_norm": 0.311848908662796,
+      "learning_rate": 0.00018540880503144656,
+      "loss": 0.7833,
+      "step": 124
+    },
+    {
+      "epoch": 0.5814798662596308,
+      "grad_norm": 0.3175119161605835,
+      "learning_rate": 0.00018515723270440252,
+      "loss": 0.7611,
+      "step": 125
+    },
+    {
+      "epoch": 0.5861317051897078,
+      "grad_norm": 0.2740802764892578,
+      "learning_rate": 0.0001849056603773585,
+      "loss": 0.7321,
+      "step": 126
+    },
+    {
+      "epoch": 0.5907835441197848,
+      "grad_norm": 0.3262477219104767,
+      "learning_rate": 0.0001846540880503145,
+      "loss": 0.7358,
+      "step": 127
+    },
+    {
+      "epoch": 0.5954353830498619,
+      "grad_norm": 0.3001246154308319,
+      "learning_rate": 0.00018440251572327045,
+      "loss": 0.7386,
+      "step": 128
+    },
+    {
+      "epoch": 0.6000872219799389,
+      "grad_norm": 0.3316512107849121,
+      "learning_rate": 0.00018415094339622643,
+      "loss": 0.7373,
+      "step": 129
+    },
+    {
+      "epoch": 0.604739060910016,
+      "grad_norm": 0.3032156229019165,
+      "learning_rate": 0.0001838993710691824,
+      "loss": 0.7617,
+      "step": 130
+    },
+    {
+      "epoch": 0.6093908998400931,
+      "grad_norm": 0.29149872064590454,
+      "learning_rate": 0.00018364779874213837,
+      "loss": 0.7426,
+      "step": 131
+    },
+    {
+      "epoch": 0.6140427387701701,
+      "grad_norm": 0.29729750752449036,
+      "learning_rate": 0.00018339622641509436,
+      "loss": 0.7127,
+      "step": 132
+    },
+    {
+      "epoch": 0.6186945777002472,
+      "grad_norm": 0.2744731903076172,
+      "learning_rate": 0.00018314465408805032,
+      "loss": 0.7327,
+      "step": 133
+    },
+    {
+      "epoch": 0.6233464166303242,
+      "grad_norm": 0.285020649433136,
+      "learning_rate": 0.0001828930817610063,
+      "loss": 0.7021,
+      "step": 134
+    },
+    {
+      "epoch": 0.6279982555604012,
+      "grad_norm": 0.3577159345149994,
+      "learning_rate": 0.0001826415094339623,
+      "loss": 0.7087,
+      "step": 135
+    },
+    {
+      "epoch": 0.6326500944904783,
+      "grad_norm": 0.33541566133499146,
+      "learning_rate": 0.00018238993710691825,
+      "loss": 0.7505,
+      "step": 136
+    },
+    {
+      "epoch": 0.6373019334205553,
+      "grad_norm": 0.3366207480430603,
+      "learning_rate": 0.00018213836477987423,
+      "loss": 0.721,
+      "step": 137
+    },
+    {
+      "epoch": 0.6419537723506323,
+      "grad_norm": 0.31512752175331116,
+      "learning_rate": 0.0001818867924528302,
+      "loss": 0.7335,
+      "step": 138
+    },
+    {
+      "epoch": 0.6466056112807094,
+      "grad_norm": 0.29900509119033813,
+      "learning_rate": 0.00018163522012578617,
+      "loss": 0.7627,
+      "step": 139
+    },
+    {
+      "epoch": 0.6512574502107864,
+      "grad_norm": 0.3184279501438141,
+      "learning_rate": 0.00018138364779874216,
+      "loss": 0.7242,
+      "step": 140
+    },
+    {
+      "epoch": 0.6559092891408635,
+      "grad_norm": 0.3222281336784363,
+      "learning_rate": 0.00018113207547169812,
+      "loss": 0.7425,
+      "step": 141
+    },
+    {
+      "epoch": 0.6605611280709406,
+      "grad_norm": 0.3259677588939667,
+      "learning_rate": 0.0001808805031446541,
+      "loss": 0.7335,
+      "step": 142
+    },
+    {
+      "epoch": 0.6652129670010176,
+      "grad_norm": 0.2882985770702362,
+      "learning_rate": 0.0001806289308176101,
+      "loss": 0.7362,
+      "step": 143
+    },
+    {
+      "epoch": 0.6698648059310947,
+      "grad_norm": 0.3377401530742645,
+      "learning_rate": 0.00018037735849056605,
+      "loss": 0.6988,
+      "step": 144
+    },
+    {
+      "epoch": 0.6745166448611717,
+      "grad_norm": 0.30174747109413147,
+      "learning_rate": 0.00018012578616352203,
+      "loss": 0.729,
+      "step": 145
+    },
+    {
+      "epoch": 0.6791684837912487,
+      "grad_norm": 0.31059524416923523,
+      "learning_rate": 0.000179874213836478,
+      "loss": 0.7083,
+      "step": 146
+    },
+    {
+      "epoch": 0.6838203227213258,
+      "grad_norm": 0.3255733847618103,
+      "learning_rate": 0.00017962264150943398,
+      "loss": 0.692,
+      "step": 147
+    },
+    {
+      "epoch": 0.6884721616514028,
+      "grad_norm": 0.3032463788986206,
+      "learning_rate": 0.00017937106918238996,
+      "loss": 0.7152,
+      "step": 148
+    },
+    {
+      "epoch": 0.6931240005814798,
+      "grad_norm": 0.31814324855804443,
+      "learning_rate": 0.00017911949685534592,
+      "loss": 0.7499,
+      "step": 149
+    },
+    {
+      "epoch": 0.6977758395115569,
+      "grad_norm": 0.33511579036712646,
+      "learning_rate": 0.0001788679245283019,
+      "loss": 0.6878,
+      "step": 150
+    },
+    {
+      "epoch": 0.702427678441634,
+      "grad_norm": 0.3108777701854706,
+      "learning_rate": 0.00017861635220125786,
+      "loss": 0.7084,
+      "step": 151
+    },
+    {
+      "epoch": 0.707079517371711,
+      "grad_norm": 0.36597543954849243,
+      "learning_rate": 0.00017836477987421385,
+      "loss": 0.7362,
+      "step": 152
+    },
+    {
+      "epoch": 0.7117313563017881,
+      "grad_norm": 0.29230964183807373,
+      "learning_rate": 0.00017811320754716983,
+      "loss": 0.697,
+      "step": 153
+    },
+    {
+      "epoch": 0.7163831952318651,
+      "grad_norm": 0.3120034635066986,
+      "learning_rate": 0.0001778616352201258,
+      "loss": 0.7347,
+      "step": 154
+    },
+    {
+      "epoch": 0.7210350341619421,
+      "grad_norm": 0.3264995217323303,
+      "learning_rate": 0.00017761006289308178,
+      "loss": 0.6952,
+      "step": 155
+    },
+    {
+      "epoch": 0.7256868730920192,
+      "grad_norm": 0.31232118606567383,
+      "learning_rate": 0.00017735849056603776,
+      "loss": 0.6886,
+      "step": 156
+    },
+    {
+      "epoch": 0.7303387120220962,
+      "grad_norm": 0.2971724271774292,
+      "learning_rate": 0.00017710691823899372,
+      "loss": 0.6752,
+      "step": 157
+    },
+    {
+      "epoch": 0.7349905509521732,
+      "grad_norm": 0.31577828526496887,
+      "learning_rate": 0.0001768553459119497,
+      "loss": 0.7018,
+      "step": 158
+    },
+    {
+      "epoch": 0.7396423898822503,
+      "grad_norm": 0.32843655347824097,
+      "learning_rate": 0.00017660377358490566,
+      "loss": 0.7071,
+      "step": 159
+    },
+    {
+      "epoch": 0.7442942288123273,
+      "grad_norm": 0.2946259677410126,
+      "learning_rate": 0.00017635220125786165,
+      "loss": 0.7507,
+      "step": 160
+    },
+    {
+      "epoch": 0.7489460677424045,
+      "grad_norm": 0.3511378765106201,
+      "learning_rate": 0.00017610062893081763,
+      "loss": 0.6173,
+      "step": 161
+    },
+    {
+      "epoch": 0.7535979066724815,
+      "grad_norm": 0.3054347634315491,
+      "learning_rate": 0.0001758490566037736,
+      "loss": 0.6699,
+      "step": 162
+    },
+    {
+      "epoch": 0.7582497456025585,
+      "grad_norm": 0.3269181549549103,
+      "learning_rate": 0.00017559748427672958,
+      "loss": 0.689,
+      "step": 163
+    },
+    {
+      "epoch": 0.7629015845326356,
+      "grad_norm": 0.35425734519958496,
+      "learning_rate": 0.00017534591194968554,
+      "loss": 0.7083,
+      "step": 164
+    },
+    {
+      "epoch": 0.7675534234627126,
+      "grad_norm": 0.34333133697509766,
+      "learning_rate": 0.00017509433962264152,
+      "loss": 0.7168,
+      "step": 165
+    },
+    {
+      "epoch": 0.7722052623927896,
+      "grad_norm": 0.33856451511383057,
+      "learning_rate": 0.0001748427672955975,
+      "loss": 0.7375,
+      "step": 166
+    },
+    {
+      "epoch": 0.7768571013228667,
+      "grad_norm": 0.31838399171829224,
+      "learning_rate": 0.00017459119496855346,
+      "loss": 0.6647,
+      "step": 167
+    },
+    {
+      "epoch": 0.7815089402529437,
+      "grad_norm": 0.35158535838127136,
+      "learning_rate": 0.00017433962264150945,
+      "loss": 0.682,
+      "step": 168
+    },
+    {
+      "epoch": 0.7861607791830207,
+      "grad_norm": 0.30730119347572327,
+      "learning_rate": 0.00017408805031446543,
+      "loss": 0.7189,
+      "step": 169
+    },
+    {
+      "epoch": 0.7908126181130979,
+      "grad_norm": 0.3460267186164856,
+      "learning_rate": 0.0001738364779874214,
+      "loss": 0.667,
+      "step": 170
+    },
+    {
+      "epoch": 0.7954644570431749,
+      "grad_norm": 0.35215961933135986,
+      "learning_rate": 0.00017358490566037738,
+      "loss": 0.7592,
+      "step": 171
+    },
+    {
+      "epoch": 0.800116295973252,
+      "grad_norm": 0.3169519901275635,
+      "learning_rate": 0.00017333333333333334,
+      "loss": 0.6616,
+      "step": 172
+    },
+    {
+      "epoch": 0.804768134903329,
+      "grad_norm": 0.3051493167877197,
+      "learning_rate": 0.00017308176100628932,
+      "loss": 0.7076,
+      "step": 173
+    },
+    {
+      "epoch": 0.809419973833406,
+      "grad_norm": 0.30809450149536133,
+      "learning_rate": 0.0001728301886792453,
+      "loss": 0.6445,
+      "step": 174
+    },
+    {
+      "epoch": 0.8140718127634831,
+      "grad_norm": 0.30463534593582153,
+      "learning_rate": 0.00017257861635220126,
+      "loss": 0.7266,
+      "step": 175
+    },
+    {
+      "epoch": 0.8187236516935601,
+      "grad_norm": 0.32270652055740356,
+      "learning_rate": 0.00017232704402515725,
+      "loss": 0.733,
+      "step": 176
+    },
+    {
+      "epoch": 0.8233754906236371,
+      "grad_norm": 0.354934424161911,
+      "learning_rate": 0.00017207547169811324,
+      "loss": 0.7081,
+      "step": 177
+    },
+    {
+      "epoch": 0.8280273295537142,
+      "grad_norm": 0.305822491645813,
+      "learning_rate": 0.0001718238993710692,
+      "loss": 0.6822,
+      "step": 178
+    },
+    {
+      "epoch": 0.8326791684837912,
+      "grad_norm": 0.3144387900829315,
+      "learning_rate": 0.00017157232704402518,
+      "loss": 0.66,
+      "step": 179
+    },
+    {
+      "epoch": 0.8373310074138683,
+      "grad_norm": 0.3486148715019226,
+      "learning_rate": 0.00017132075471698114,
+      "loss": 0.731,
+      "step": 180
+    },
+    {
+      "epoch": 0.8419828463439454,
+      "grad_norm": 0.305627703666687,
+      "learning_rate": 0.00017106918238993712,
+      "loss": 0.6876,
+      "step": 181
+    },
+    {
+      "epoch": 0.8466346852740224,
+      "grad_norm": 0.3102213144302368,
+      "learning_rate": 0.0001708176100628931,
+      "loss": 0.7362,
+      "step": 182
+    },
+    {
+      "epoch": 0.8512865242040994,
+      "grad_norm": 0.320419579744339,
+      "learning_rate": 0.00017056603773584907,
+      "loss": 0.6855,
+      "step": 183
+    },
+    {
+      "epoch": 0.8559383631341765,
+      "grad_norm": 0.29344695806503296,
+      "learning_rate": 0.00017031446540880505,
+      "loss": 0.6853,
+      "step": 184
+    },
+    {
+      "epoch": 0.8605902020642535,
+      "grad_norm": 0.37375548481941223,
+      "learning_rate": 0.000170062893081761,
+      "loss": 0.6845,
+      "step": 185
+    },
+    {
+      "epoch": 0.8652420409943306,
+      "grad_norm": 0.3486350476741791,
+      "learning_rate": 0.000169811320754717,
+      "loss": 0.6785,
+      "step": 186
+    },
+    {
+      "epoch": 0.8698938799244076,
+      "grad_norm": 0.31900399923324585,
+      "learning_rate": 0.00016955974842767298,
+      "loss": 0.6788,
+      "step": 187
+    },
+    {
+      "epoch": 0.8745457188544846,
+      "grad_norm": 0.38080456852912903,
+      "learning_rate": 0.00016930817610062894,
+      "loss": 0.6646,
+      "step": 188
+    },
+    {
+      "epoch": 0.8791975577845617,
+      "grad_norm": 0.32579371333122253,
+      "learning_rate": 0.00016905660377358492,
+      "loss": 0.657,
+      "step": 189
+    },
+    {
+      "epoch": 0.8838493967146388,
+      "grad_norm": 0.3697237968444824,
+      "learning_rate": 0.0001688050314465409,
+      "loss": 0.675,
+      "step": 190
+    },
+    {
+      "epoch": 0.8885012356447158,
+      "grad_norm": 0.3532376289367676,
+      "learning_rate": 0.00016855345911949687,
+      "loss": 0.699,
+      "step": 191
+    },
+    {
+      "epoch": 0.8931530745747929,
+      "grad_norm": 0.3699309229850769,
+      "learning_rate": 0.00016830188679245285,
+      "loss": 0.6464,
+      "step": 192
+    },
+    {
+      "epoch": 0.8978049135048699,
+      "grad_norm": 0.3340603709220886,
+      "learning_rate": 0.0001680503144654088,
+      "loss": 0.6738,
+      "step": 193
+    },
+    {
+      "epoch": 0.9024567524349469,
+      "grad_norm": 0.3464495837688446,
+      "learning_rate": 0.0001677987421383648,
+      "loss": 0.6959,
+      "step": 194
+    },
+    {
+      "epoch": 0.907108591365024,
+      "grad_norm": 0.33247748017311096,
+      "learning_rate": 0.00016754716981132078,
+      "loss": 0.6726,
+      "step": 195
+    },
+    {
+      "epoch": 0.911760430295101,
+      "grad_norm": 0.3268505930900574,
+      "learning_rate": 0.00016729559748427674,
+      "loss": 0.6683,
+      "step": 196
+    },
+    {
+      "epoch": 0.916412269225178,
+      "grad_norm": 0.325344443321228,
+      "learning_rate": 0.00016704402515723272,
+      "loss": 0.7064,
+      "step": 197
+    },
+    {
+      "epoch": 0.9210641081552551,
+      "grad_norm": 0.32244226336479187,
+      "learning_rate": 0.00016679245283018868,
+      "loss": 0.664,
+      "step": 198
+    },
+    {
+      "epoch": 0.9257159470853322,
+      "grad_norm": 0.3163536489009857,
+      "learning_rate": 0.00016654088050314467,
+      "loss": 0.678,
+      "step": 199
+    },
+    {
+      "epoch": 0.9303677860154093,
+      "grad_norm": 0.3107887804508209,
+      "learning_rate": 0.00016628930817610065,
+      "loss": 0.6543,
+      "step": 200
+    },
+    {
+      "epoch": 0.9350196249454863,
+      "grad_norm": 0.3148704767227173,
+      "learning_rate": 0.0001660377358490566,
+      "loss": 0.683,
+      "step": 201
+    },
+    {
+      "epoch": 0.9396714638755633,
+      "grad_norm": 0.36367544531822205,
+      "learning_rate": 0.0001657861635220126,
+      "loss": 0.6751,
+      "step": 202
+    },
+    {
+      "epoch": 0.9443233028056404,
+      "grad_norm": 0.3681708872318268,
+      "learning_rate": 0.00016553459119496858,
+      "loss": 0.6564,
+      "step": 203
+    },
+    {
+      "epoch": 0.9489751417357174,
+      "grad_norm": 0.37342405319213867,
+      "learning_rate": 0.00016528301886792454,
+      "loss": 0.6474,
+      "step": 204
+    },
+    {
+      "epoch": 0.9536269806657944,
+      "grad_norm": 0.3597882091999054,
+      "learning_rate": 0.00016503144654088052,
+      "loss": 0.6501,
+      "step": 205
+    },
+    {
+      "epoch": 0.9582788195958715,
+      "grad_norm": 0.3550153970718384,
+      "learning_rate": 0.00016477987421383648,
+      "loss": 0.7065,
+      "step": 206
+    },
+    {
+      "epoch": 0.9629306585259485,
+      "grad_norm": 0.40322139859199524,
+      "learning_rate": 0.00016452830188679247,
+      "loss": 0.6526,
+      "step": 207
+    },
+    {
+      "epoch": 0.9675824974560255,
+      "grad_norm": 0.3350887596607208,
+      "learning_rate": 0.00016427672955974845,
+      "loss": 0.6567,
+      "step": 208
+    },
+    {
+      "epoch": 0.9722343363861027,
+      "grad_norm": 0.3692532777786255,
+      "learning_rate": 0.0001640251572327044,
+      "loss": 0.6301,
+      "step": 209
+    },
+    {
+      "epoch": 0.9768861753161797,
+      "grad_norm": 0.35285553336143494,
+      "learning_rate": 0.0001637735849056604,
+      "loss": 0.6899,
+      "step": 210
+    },
+    {
+      "epoch": 0.9815380142462568,
+      "grad_norm": 0.3626399040222168,
+      "learning_rate": 0.00016352201257861635,
+      "loss": 0.6632,
+      "step": 211
+    },
+    {
+      "epoch": 0.9861898531763338,
+      "grad_norm": 0.3436664640903473,
+      "learning_rate": 0.00016327044025157234,
+      "loss": 0.6441,
+      "step": 212
+    },
+    {
+      "epoch": 0.9908416921064108,
+      "grad_norm": 0.33126094937324524,
+      "learning_rate": 0.00016301886792452833,
+      "loss": 0.6598,
+      "step": 213
+    },
+    {
+      "epoch": 0.9954935310364879,
+      "grad_norm": 0.3500431478023529,
+      "learning_rate": 0.00016276729559748428,
+      "loss": 0.6673,
+      "step": 214
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.30532780289649963,
+      "learning_rate": 0.00016251572327044027,
+      "loss": 0.6418,
+      "step": 215
+    },
+    {
+      "epoch": 1.0046518389300771,
+      "grad_norm": 0.39760303497314453,
+      "learning_rate": 0.00016226415094339625,
+      "loss": 0.6456,
+      "step": 216
+    },
+    {
+      "epoch": 1.009303677860154,
+      "grad_norm": 0.3375679850578308,
+      "learning_rate": 0.0001620125786163522,
+      "loss": 0.6435,
+      "step": 217
+    },
+    {
+      "epoch": 1.0139555167902312,
+      "grad_norm": 0.39267343282699585,
+      "learning_rate": 0.0001617610062893082,
+      "loss": 0.6792,
+      "step": 218
+    },
+    {
+      "epoch": 1.0186073557203081,
+      "grad_norm": 0.3079686760902405,
+      "learning_rate": 0.00016150943396226416,
+      "loss": 0.6422,
+      "step": 219
+    },
+    {
+      "epoch": 1.0232591946503853,
+      "grad_norm": 0.338821142911911,
+      "learning_rate": 0.00016125786163522014,
+      "loss": 0.6521,
+      "step": 220
+    },
+    {
+      "epoch": 1.0279110335804622,
+      "grad_norm": 0.32997390627861023,
+      "learning_rate": 0.00016100628930817613,
+      "loss": 0.6377,
+      "step": 221
+    },
+    {
+      "epoch": 1.0325628725105394,
+      "grad_norm": 0.3223334848880768,
+      "learning_rate": 0.00016075471698113208,
+      "loss": 0.6389,
+      "step": 222
+    },
+    {
+      "epoch": 1.0372147114406163,
+      "grad_norm": 0.3401314616203308,
+      "learning_rate": 0.00016050314465408807,
+      "loss": 0.6605,
+      "step": 223
+    },
+    {
+      "epoch": 1.0418665503706934,
+      "grad_norm": 0.31138718128204346,
+      "learning_rate": 0.00016025157232704405,
+      "loss": 0.6361,
+      "step": 224
+    },
+    {
+      "epoch": 1.0465183893007706,
+      "grad_norm": 0.36233529448509216,
+      "learning_rate": 0.00016,
+      "loss": 0.6261,
+      "step": 225
+    },
+    {
+      "epoch": 1.0511702282308475,
+      "grad_norm": 0.31210434436798096,
+      "learning_rate": 0.000159748427672956,
+      "loss": 0.6734,
+      "step": 226
+    },
+    {
+      "epoch": 1.0558220671609246,
+      "grad_norm": 0.3606509268283844,
+      "learning_rate": 0.00015949685534591196,
+      "loss": 0.6282,
+      "step": 227
+    },
+    {
+      "epoch": 1.0604739060910016,
+      "grad_norm": 0.35877060890197754,
+      "learning_rate": 0.00015924528301886794,
+      "loss": 0.6556,
+      "step": 228
+    },
+    {
+      "epoch": 1.0651257450210787,
+      "grad_norm": 0.3874422609806061,
+      "learning_rate": 0.00015899371069182393,
+      "loss": 0.6795,
+      "step": 229
+    },
+    {
+      "epoch": 1.0697775839511556,
+      "grad_norm": 0.33604809641838074,
+      "learning_rate": 0.00015874213836477989,
+      "loss": 0.6513,
+      "step": 230
+    },
+    {
+      "epoch": 1.0744294228812328,
+      "grad_norm": 0.31052348017692566,
+      "learning_rate": 0.00015849056603773587,
+      "loss": 0.6205,
+      "step": 231
+    },
+    {
+      "epoch": 1.0790812618113097,
+      "grad_norm": 0.39934998750686646,
+      "learning_rate": 0.00015823899371069183,
+      "loss": 0.6602,
+      "step": 232
+    },
+    {
+      "epoch": 1.0837331007413868,
+      "grad_norm": 0.2991245985031128,
+      "learning_rate": 0.00015798742138364781,
+      "loss": 0.6508,
+      "step": 233
+    },
+    {
+      "epoch": 1.0883849396714638,
+      "grad_norm": 0.35071277618408203,
+      "learning_rate": 0.0001577358490566038,
+      "loss": 0.6326,
+      "step": 234
+    },
+    {
+      "epoch": 1.093036778601541,
+      "grad_norm": 0.3201613426208496,
+      "learning_rate": 0.00015748427672955976,
+      "loss": 0.622,
+      "step": 235
+    },
+    {
+      "epoch": 1.097688617531618,
+      "grad_norm": 0.3288263976573944,
+      "learning_rate": 0.00015723270440251574,
+      "loss": 0.6158,
+      "step": 236
+    },
+    {
+      "epoch": 1.102340456461695,
+      "grad_norm": 0.35064592957496643,
+      "learning_rate": 0.00015698113207547173,
+      "loss": 0.666,
+      "step": 237
+    },
+    {
+      "epoch": 1.1069922953917721,
+      "grad_norm": 0.3487169146537781,
+      "learning_rate": 0.00015672955974842769,
+      "loss": 0.6334,
+      "step": 238
+    },
+    {
+      "epoch": 1.111644134321849,
+      "grad_norm": 0.3314840793609619,
+      "learning_rate": 0.00015647798742138367,
+      "loss": 0.6606,
+      "step": 239
+    },
+    {
+      "epoch": 1.1162959732519262,
+      "grad_norm": 0.3044390380382538,
+      "learning_rate": 0.00015622641509433963,
+      "loss": 0.6103,
+      "step": 240
+    },
+    {
+      "epoch": 1.1209478121820031,
+      "grad_norm": 0.3635900616645813,
+      "learning_rate": 0.00015597484276729561,
+      "loss": 0.6672,
+      "step": 241
+    },
+    {
+      "epoch": 1.1255996511120803,
+      "grad_norm": 0.34150922298431396,
+      "learning_rate": 0.0001557232704402516,
+      "loss": 0.6563,
+      "step": 242
+    },
+    {
+      "epoch": 1.1302514900421574,
+      "grad_norm": 0.3509139120578766,
+      "learning_rate": 0.00015547169811320756,
+      "loss": 0.6323,
+      "step": 243
+    },
+    {
+      "epoch": 1.1349033289722343,
+      "grad_norm": 0.32937365770339966,
+      "learning_rate": 0.00015522012578616354,
+      "loss": 0.632,
+      "step": 244
+    },
+    {
+      "epoch": 1.1395551679023113,
+      "grad_norm": 0.3382979929447174,
+      "learning_rate": 0.0001549685534591195,
+      "loss": 0.6527,
+      "step": 245
+    },
+    {
+      "epoch": 1.1442070068323884,
+      "grad_norm": 0.3821505308151245,
+      "learning_rate": 0.0001547169811320755,
+      "loss": 0.648,
+      "step": 246
+    },
+    {
+      "epoch": 1.1488588457624656,
+      "grad_norm": 0.35066843032836914,
+      "learning_rate": 0.00015446540880503147,
+      "loss": 0.6576,
+      "step": 247
+    },
+    {
+      "epoch": 1.1535106846925425,
+      "grad_norm": 0.37199634313583374,
+      "learning_rate": 0.00015421383647798743,
+      "loss": 0.6321,
+      "step": 248
+    },
+    {
+      "epoch": 1.1581625236226196,
+      "grad_norm": 0.34103065729141235,
+      "learning_rate": 0.00015396226415094342,
+      "loss": 0.6196,
+      "step": 249
+    },
+    {
+      "epoch": 1.1628143625526965,
+      "grad_norm": 0.35918059945106506,
+      "learning_rate": 0.0001537106918238994,
+      "loss": 0.6284,
+      "step": 250
+    },
+    {
+      "epoch": 1.1674662014827737,
+      "grad_norm": 0.3341059684753418,
+      "learning_rate": 0.00015345911949685536,
+      "loss": 0.6463,
+      "step": 251
+    },
+    {
+      "epoch": 1.1721180404128506,
+      "grad_norm": 0.3380013108253479,
+      "learning_rate": 0.00015320754716981134,
+      "loss": 0.6453,
+      "step": 252
+    },
+    {
+      "epoch": 1.1767698793429278,
+      "grad_norm": 0.34588825702667236,
+      "learning_rate": 0.0001529559748427673,
+      "loss": 0.6355,
+      "step": 253
+    },
+    {
+      "epoch": 1.181421718273005,
+      "grad_norm": 0.34837037324905396,
+      "learning_rate": 0.0001527044025157233,
+      "loss": 0.5925,
+      "step": 254
+    },
+    {
+      "epoch": 1.1860735572030818,
+      "grad_norm": 0.3584837317466736,
+      "learning_rate": 0.00015245283018867927,
+      "loss": 0.6334,
+      "step": 255
+    },
+    {
+      "epoch": 1.190725396133159,
+      "grad_norm": 0.36439597606658936,
+      "learning_rate": 0.00015220125786163523,
+      "loss": 0.6121,
+      "step": 256
+    },
+    {
+      "epoch": 1.195377235063236,
+      "grad_norm": 0.3767489790916443,
+      "learning_rate": 0.00015194968553459122,
+      "loss": 0.646,
+      "step": 257
+    },
+    {
+      "epoch": 1.200029073993313,
+      "grad_norm": 0.3449689745903015,
+      "learning_rate": 0.00015169811320754717,
+      "loss": 0.6624,
+      "step": 258
+    },
+    {
+      "epoch": 1.20468091292339,
+      "grad_norm": 0.34724000096321106,
+      "learning_rate": 0.00015144654088050316,
+      "loss": 0.6141,
+      "step": 259
+    },
+    {
+      "epoch": 1.2093327518534671,
+      "grad_norm": 0.33465149998664856,
+      "learning_rate": 0.00015119496855345914,
+      "loss": 0.6644,
+      "step": 260
+    },
+    {
+      "epoch": 1.213984590783544,
+      "grad_norm": 0.345456600189209,
+      "learning_rate": 0.0001509433962264151,
+      "loss": 0.6535,
+      "step": 261
+    },
+    {
+      "epoch": 1.2186364297136212,
+      "grad_norm": 0.34758734703063965,
+      "learning_rate": 0.0001506918238993711,
+      "loss": 0.6327,
+      "step": 262
+    },
+    {
+      "epoch": 1.223288268643698,
+      "grad_norm": 0.3376777172088623,
+      "learning_rate": 0.00015044025157232707,
+      "loss": 0.6218,
+      "step": 263
+    },
+    {
+      "epoch": 1.2279401075737753,
+      "grad_norm": 0.38596582412719727,
+      "learning_rate": 0.00015018867924528303,
+      "loss": 0.6186,
+      "step": 264
+    },
+    {
+      "epoch": 1.2325919465038524,
+      "grad_norm": 0.32499659061431885,
+      "learning_rate": 0.000149937106918239,
+      "loss": 0.6438,
+      "step": 265
+    },
+    {
+      "epoch": 1.2372437854339293,
+      "grad_norm": 0.31884467601776123,
+      "learning_rate": 0.00014968553459119498,
+      "loss": 0.6204,
+      "step": 266
+    },
+    {
+      "epoch": 1.2418956243640065,
+      "grad_norm": 0.36743050813674927,
+      "learning_rate": 0.00014943396226415093,
+      "loss": 0.6284,
+      "step": 267
+    },
+    {
+      "epoch": 1.2465474632940834,
+      "grad_norm": 0.3131709098815918,
+      "learning_rate": 0.00014918238993710692,
+      "loss": 0.6086,
+      "step": 268
+    },
+    {
+      "epoch": 1.2511993022241605,
+      "grad_norm": 0.3433280289173126,
+      "learning_rate": 0.00014893081761006288,
+      "loss": 0.6472,
+      "step": 269
+    },
+    {
+      "epoch": 1.2558511411542375,
+      "grad_norm": 0.3455664813518524,
+      "learning_rate": 0.00014867924528301886,
+      "loss": 0.6213,
+      "step": 270
+    },
+    {
+      "epoch": 1.2605029800843146,
+      "grad_norm": 0.39432403445243835,
+      "learning_rate": 0.00014842767295597485,
+      "loss": 0.6557,
+      "step": 271
+    },
+    {
+      "epoch": 1.2651548190143918,
+      "grad_norm": 0.3267870247364044,
+      "learning_rate": 0.0001481761006289308,
+      "loss": 0.6232,
+      "step": 272
+    },
+    {
+      "epoch": 1.2698066579444687,
+      "grad_norm": 0.33328256011009216,
+      "learning_rate": 0.0001479245283018868,
+      "loss": 0.6525,
+      "step": 273
+    },
+    {
+      "epoch": 1.2744584968745456,
+      "grad_norm": 0.39522597193717957,
+      "learning_rate": 0.00014767295597484278,
+      "loss": 0.6516,
+      "step": 274
+    },
+    {
+      "epoch": 1.2791103358046227,
+      "grad_norm": 0.35576415061950684,
+      "learning_rate": 0.00014742138364779873,
+      "loss": 0.6497,
+      "step": 275
+    },
+    {
+      "epoch": 1.2837621747347,
+      "grad_norm": 0.3637307584285736,
+      "learning_rate": 0.00014716981132075472,
+      "loss": 0.6558,
+      "step": 276
+    },
+    {
+      "epoch": 1.2884140136647768,
+      "grad_norm": 0.32312607765197754,
+      "learning_rate": 0.00014691823899371068,
+      "loss": 0.5779,
+      "step": 277
+    },
+    {
+      "epoch": 1.293065852594854,
+      "grad_norm": 0.3506655991077423,
+      "learning_rate": 0.00014666666666666666,
+      "loss": 0.6267,
+      "step": 278
+    },
+    {
+      "epoch": 1.2977176915249309,
+      "grad_norm": 0.3418485224246979,
+      "learning_rate": 0.00014641509433962265,
+      "loss": 0.6405,
+      "step": 279
+    },
+    {
+      "epoch": 1.302369530455008,
+      "grad_norm": 0.316551148891449,
+      "learning_rate": 0.0001461635220125786,
+      "loss": 0.5953,
+      "step": 280
+    },
+    {
+      "epoch": 1.307021369385085,
+      "grad_norm": 0.30916550755500793,
+      "learning_rate": 0.0001459119496855346,
+      "loss": 0.598,
+      "step": 281
+    },
+    {
+      "epoch": 1.311673208315162,
+      "grad_norm": 0.3628326654434204,
+      "learning_rate": 0.00014566037735849055,
+      "loss": 0.6402,
+      "step": 282
+    },
+    {
+      "epoch": 1.3163250472452392,
+      "grad_norm": 0.32959505915641785,
+      "learning_rate": 0.00014540880503144653,
+      "loss": 0.6555,
+      "step": 283
+    },
+    {
+      "epoch": 1.3209768861753162,
+      "grad_norm": 0.3360130786895752,
+      "learning_rate": 0.00014515723270440252,
+      "loss": 0.6519,
+      "step": 284
+    },
+    {
+      "epoch": 1.325628725105393,
+      "grad_norm": 0.3171837627887726,
+      "learning_rate": 0.00014490566037735848,
+      "loss": 0.6144,
+      "step": 285
+    },
+    {
+      "epoch": 1.3302805640354702,
+      "grad_norm": 0.3459252417087555,
+      "learning_rate": 0.00014465408805031446,
+      "loss": 0.6175,
+      "step": 286
+    },
+    {
+      "epoch": 1.3349324029655474,
+      "grad_norm": 0.3131142556667328,
+      "learning_rate": 0.00014440251572327045,
+      "loss": 0.6159,
+      "step": 287
+    },
+    {
+      "epoch": 1.3395842418956243,
+      "grad_norm": 0.34040936827659607,
+      "learning_rate": 0.0001441509433962264,
+      "loss": 0.6307,
+      "step": 288
+    },
+    {
+      "epoch": 1.3442360808257015,
+      "grad_norm": 0.3348907232284546,
+      "learning_rate": 0.0001438993710691824,
+      "loss": 0.603,
+      "step": 289
+    },
+    {
+      "epoch": 1.3488879197557784,
+      "grad_norm": 0.304889053106308,
+      "learning_rate": 0.00014364779874213835,
+      "loss": 0.6162,
+      "step": 290
+    },
+    {
+      "epoch": 1.3535397586858555,
+      "grad_norm": 0.31203022599220276,
+      "learning_rate": 0.00014339622641509434,
+      "loss": 0.6529,
+      "step": 291
+    },
+    {
+      "epoch": 1.3581915976159324,
+      "grad_norm": 0.34511953592300415,
+      "learning_rate": 0.00014314465408805032,
+      "loss": 0.6115,
+      "step": 292
+    },
+    {
+      "epoch": 1.3628434365460096,
+      "grad_norm": 0.32442706823349,
+      "learning_rate": 0.00014289308176100628,
+      "loss": 0.5989,
+      "step": 293
+    },
+    {
+      "epoch": 1.3674952754760867,
+      "grad_norm": 0.3309711217880249,
+      "learning_rate": 0.00014264150943396226,
+      "loss": 0.6408,
+      "step": 294
+    },
+    {
+      "epoch": 1.3721471144061637,
+      "grad_norm": 0.3313584327697754,
+      "learning_rate": 0.00014238993710691825,
+      "loss": 0.6189,
+      "step": 295
+    },
+    {
+      "epoch": 1.3767989533362408,
+      "grad_norm": 0.3500613570213318,
+      "learning_rate": 0.0001421383647798742,
+      "loss": 0.6361,
+      "step": 296
+    },
+    {
+      "epoch": 1.3814507922663177,
+      "grad_norm": 0.31808218359947205,
+      "learning_rate": 0.0001418867924528302,
+      "loss": 0.6074,
+      "step": 297
+    },
+    {
+      "epoch": 1.3861026311963949,
+      "grad_norm": 0.32396867871284485,
+      "learning_rate": 0.00014163522012578615,
+      "loss": 0.6148,
+      "step": 298
+    },
+    {
+      "epoch": 1.3907544701264718,
+      "grad_norm": 0.32348358631134033,
+      "learning_rate": 0.00014138364779874214,
+      "loss": 0.6207,
+      "step": 299
+    },
+    {
+      "epoch": 1.395406309056549,
+      "grad_norm": 0.3096139430999756,
+      "learning_rate": 0.00014113207547169812,
+      "loss": 0.6074,
+      "step": 300
+    },
+    {
+      "epoch": 1.400058147986626,
+      "grad_norm": 0.3543049693107605,
+      "learning_rate": 0.00014088050314465408,
+      "loss": 0.6285,
+      "step": 301
+    },
+    {
+      "epoch": 1.404709986916703,
+      "grad_norm": 0.32986271381378174,
+      "learning_rate": 0.00014062893081761007,
+      "loss": 0.6012,
+      "step": 302
+    },
+    {
+      "epoch": 1.40936182584678,
+      "grad_norm": 0.36291420459747314,
+      "learning_rate": 0.00014037735849056602,
+      "loss": 0.6352,
+      "step": 303
+    },
+    {
+      "epoch": 1.414013664776857,
+      "grad_norm": 0.3590340316295624,
+      "learning_rate": 0.000140125786163522,
+      "loss": 0.6397,
+      "step": 304
+    },
+    {
+      "epoch": 1.4186655037069342,
+      "grad_norm": 0.3462887406349182,
+      "learning_rate": 0.000139874213836478,
+      "loss": 0.638,
+      "step": 305
+    },
+    {
+      "epoch": 1.4233173426370112,
+      "grad_norm": 0.35419946908950806,
+      "learning_rate": 0.00013962264150943395,
+      "loss": 0.6325,
+      "step": 306
+    },
+    {
+      "epoch": 1.4279691815670883,
+      "grad_norm": 0.3525190055370331,
+      "learning_rate": 0.00013937106918238994,
+      "loss": 0.6419,
+      "step": 307
+    },
+    {
+      "epoch": 1.4326210204971652,
+      "grad_norm": 0.35608404874801636,
+      "learning_rate": 0.00013911949685534592,
+      "loss": 0.6363,
+      "step": 308
+    },
+    {
+      "epoch": 1.4372728594272424,
+      "grad_norm": 0.3118886947631836,
+      "learning_rate": 0.00013886792452830188,
+      "loss": 0.6523,
+      "step": 309
+    },
+    {
+      "epoch": 1.4419246983573193,
+      "grad_norm": 0.37694457173347473,
+      "learning_rate": 0.00013861635220125787,
+      "loss": 0.5873,
+      "step": 310
+    },
+    {
+      "epoch": 1.4465765372873964,
+      "grad_norm": 0.340800404548645,
+      "learning_rate": 0.00013836477987421382,
+      "loss": 0.6288,
+      "step": 311
+    },
+    {
+      "epoch": 1.4512283762174736,
+      "grad_norm": 0.32622620463371277,
+      "learning_rate": 0.0001381132075471698,
+      "loss": 0.6009,
+      "step": 312
+    },
+    {
+      "epoch": 1.4558802151475505,
+      "grad_norm": 0.3755263388156891,
+      "learning_rate": 0.0001378616352201258,
+      "loss": 0.6062,
+      "step": 313
+    },
+    {
+      "epoch": 1.4605320540776274,
+      "grad_norm": 0.3234919309616089,
+      "learning_rate": 0.00013761006289308175,
+      "loss": 0.629,
+      "step": 314
+    },
+    {
+      "epoch": 1.4651838930077046,
+      "grad_norm": 0.33275604248046875,
+      "learning_rate": 0.00013735849056603774,
+      "loss": 0.5932,
+      "step": 315
+    },
+    {
+      "epoch": 1.4698357319377817,
+      "grad_norm": 0.3457011580467224,
+      "learning_rate": 0.0001371069182389937,
+      "loss": 0.6183,
+      "step": 316
+    },
+    {
+      "epoch": 1.4744875708678586,
+      "grad_norm": 0.33747875690460205,
+      "learning_rate": 0.00013685534591194968,
+      "loss": 0.6437,
+      "step": 317
+    },
+    {
+      "epoch": 1.4791394097979358,
+      "grad_norm": 0.3290356993675232,
+      "learning_rate": 0.00013660377358490567,
+      "loss": 0.5893,
+      "step": 318
+    },
+    {
+      "epoch": 1.4837912487280127,
+      "grad_norm": 0.32800349593162537,
+      "learning_rate": 0.00013635220125786162,
+      "loss": 0.5838,
+      "step": 319
+    },
+    {
+      "epoch": 1.4884430876580899,
+      "grad_norm": 0.3453249931335449,
+      "learning_rate": 0.0001361006289308176,
+      "loss": 0.6311,
+      "step": 320
+    },
+    {
+      "epoch": 1.4930949265881668,
+      "grad_norm": 0.31954941153526306,
+      "learning_rate": 0.0001358490566037736,
+      "loss": 0.617,
+      "step": 321
+    },
+    {
+      "epoch": 1.497746765518244,
+      "grad_norm": 0.3611024022102356,
+      "learning_rate": 0.00013559748427672955,
+      "loss": 0.5991,
+      "step": 322
+    },
+    {
+      "epoch": 1.502398604448321,
+      "grad_norm": 0.3487178683280945,
+      "learning_rate": 0.00013534591194968554,
+      "loss": 0.6347,
+      "step": 323
+    },
+    {
+      "epoch": 1.507050443378398,
+      "grad_norm": 0.32538601756095886,
+      "learning_rate": 0.0001350943396226415,
+      "loss": 0.6149,
+      "step": 324
+    },
+    {
+      "epoch": 1.511702282308475,
+      "grad_norm": 0.34200358390808105,
+      "learning_rate": 0.00013484276729559748,
+      "loss": 0.5899,
+      "step": 325
+    },
+    {
+      "epoch": 1.516354121238552,
+      "grad_norm": 0.3375025987625122,
+      "learning_rate": 0.00013459119496855347,
+      "loss": 0.6198,
+      "step": 326
+    },
+    {
+      "epoch": 1.5210059601686292,
+      "grad_norm": 0.32477498054504395,
+      "learning_rate": 0.00013433962264150943,
+      "loss": 0.6115,
+      "step": 327
+    },
+    {
+      "epoch": 1.5256577990987061,
+      "grad_norm": 0.3544205129146576,
+      "learning_rate": 0.0001340880503144654,
+      "loss": 0.5999,
+      "step": 328
+    },
+    {
+      "epoch": 1.5303096380287833,
+      "grad_norm": 0.3432596027851105,
+      "learning_rate": 0.00013383647798742137,
+      "loss": 0.6344,
+      "step": 329
+    },
+    {
+      "epoch": 1.5349614769588604,
+      "grad_norm": 0.33119988441467285,
+      "learning_rate": 0.00013358490566037735,
+      "loss": 0.6159,
+      "step": 330
+    },
+    {
+      "epoch": 1.5396133158889374,
+      "grad_norm": 0.3123464584350586,
+      "learning_rate": 0.00013333333333333334,
+      "loss": 0.5981,
+      "step": 331
+    },
+    {
+      "epoch": 1.5442651548190143,
+      "grad_norm": 0.3196665942668915,
+      "learning_rate": 0.0001330817610062893,
+      "loss": 0.6077,
+      "step": 332
+    },
+    {
+      "epoch": 1.5489169937490914,
+      "grad_norm": 0.31501784920692444,
+      "learning_rate": 0.00013283018867924528,
+      "loss": 0.5931,
+      "step": 333
+    },
+    {
+      "epoch": 1.5535688326791686,
+      "grad_norm": 0.3310091197490692,
+      "learning_rate": 0.00013257861635220127,
+      "loss": 0.5829,
+      "step": 334
+    },
+    {
+      "epoch": 1.5582206716092455,
+      "grad_norm": 0.31366297602653503,
+      "learning_rate": 0.00013232704402515723,
+      "loss": 0.6086,
+      "step": 335
+    },
+    {
+      "epoch": 1.5628725105393224,
+      "grad_norm": 0.3098692297935486,
+      "learning_rate": 0.0001320754716981132,
+      "loss": 0.6171,
+      "step": 336
+    },
+    {
+      "epoch": 1.5675243494693998,
+      "grad_norm": 0.37782078981399536,
+      "learning_rate": 0.00013182389937106917,
+      "loss": 0.6029,
+      "step": 337
+    },
+    {
+      "epoch": 1.5721761883994767,
+      "grad_norm": 0.32571208477020264,
+      "learning_rate": 0.00013157232704402516,
+      "loss": 0.5917,
+      "step": 338
+    },
+    {
+      "epoch": 1.5768280273295536,
+      "grad_norm": 0.34452372789382935,
+      "learning_rate": 0.00013132075471698114,
+      "loss": 0.6284,
+      "step": 339
+    },
+    {
+      "epoch": 1.5814798662596308,
+      "grad_norm": 0.32797664403915405,
+      "learning_rate": 0.0001310691823899371,
+      "loss": 0.6013,
+      "step": 340
+    },
+    {
+      "epoch": 1.586131705189708,
+      "grad_norm": 0.3250451385974884,
+      "learning_rate": 0.00013081761006289308,
+      "loss": 0.5937,
+      "step": 341
+    },
+    {
+      "epoch": 1.5907835441197848,
+      "grad_norm": 0.32028743624687195,
+      "learning_rate": 0.00013056603773584907,
+      "loss": 0.6022,
+      "step": 342
+    },
+    {
+      "epoch": 1.5954353830498618,
+      "grad_norm": 0.32761526107788086,
+      "learning_rate": 0.00013031446540880503,
+      "loss": 0.5698,
+      "step": 343
+    },
+    {
+      "epoch": 1.600087221979939,
+      "grad_norm": 0.35005152225494385,
+      "learning_rate": 0.000130062893081761,
+      "loss": 0.5754,
+      "step": 344
+    },
+    {
+      "epoch": 1.604739060910016,
+      "grad_norm": 0.3206526041030884,
+      "learning_rate": 0.00012981132075471697,
+      "loss": 0.6198,
+      "step": 345
+    },
+    {
+      "epoch": 1.609390899840093,
+      "grad_norm": 0.34356677532196045,
+      "learning_rate": 0.00012955974842767296,
+      "loss": 0.6207,
+      "step": 346
+    },
+    {
+      "epoch": 1.6140427387701701,
+      "grad_norm": 0.35704153776168823,
+      "learning_rate": 0.00012930817610062894,
+      "loss": 0.5972,
+      "step": 347
+    },
+    {
+      "epoch": 1.6186945777002473,
+      "grad_norm": 0.3307339549064636,
+      "learning_rate": 0.0001290566037735849,
+      "loss": 0.6255,
+      "step": 348
+    },
+    {
+      "epoch": 1.6233464166303242,
+      "grad_norm": 0.34370359778404236,
+      "learning_rate": 0.00012880503144654088,
+      "loss": 0.6034,
+      "step": 349
+    },
+    {
+      "epoch": 1.6279982555604011,
+      "grad_norm": 0.3570363521575928,
+      "learning_rate": 0.00012855345911949684,
+      "loss": 0.6212,
+      "step": 350
+    },
+    {
+      "epoch": 1.6326500944904783,
+      "grad_norm": 0.3335217833518982,
+      "learning_rate": 0.00012830188679245283,
+      "loss": 0.5909,
+      "step": 351
+    },
+    {
+      "epoch": 1.6373019334205554,
+      "grad_norm": 0.3052317202091217,
+      "learning_rate": 0.0001280503144654088,
+      "loss": 0.6204,
+      "step": 352
+    },
+    {
+      "epoch": 1.6419537723506323,
+      "grad_norm": 0.38466358184814453,
+      "learning_rate": 0.00012779874213836477,
+      "loss": 0.6044,
+      "step": 353
+    },
+    {
+      "epoch": 1.6466056112807093,
+      "grad_norm": 0.33801618218421936,
+      "learning_rate": 0.00012754716981132076,
+      "loss": 0.6241,
+      "step": 354
+    },
+    {
+      "epoch": 1.6512574502107864,
+      "grad_norm": 0.32812151312828064,
+      "learning_rate": 0.00012729559748427674,
+      "loss": 0.6162,
+      "step": 355
+    },
+    {
+      "epoch": 1.6559092891408635,
+      "grad_norm": 0.33312782645225525,
+      "learning_rate": 0.0001270440251572327,
+      "loss": 0.5978,
+      "step": 356
+    },
+    {
+      "epoch": 1.6605611280709405,
+      "grad_norm": 0.35066506266593933,
+      "learning_rate": 0.00012679245283018869,
+      "loss": 0.5978,
+      "step": 357
+    },
+    {
+      "epoch": 1.6652129670010176,
+      "grad_norm": 0.34376099705696106,
+      "learning_rate": 0.00012654088050314464,
+      "loss": 0.6299,
+      "step": 358
+    },
+    {
+      "epoch": 1.6698648059310948,
+      "grad_norm": 0.33675849437713623,
+      "learning_rate": 0.00012628930817610063,
+      "loss": 0.6199,
+      "step": 359
+    },
+    {
+      "epoch": 1.6745166448611717,
+      "grad_norm": 0.35658469796180725,
+      "learning_rate": 0.00012603773584905661,
+      "loss": 0.6255,
+      "step": 360
+    },
+    {
+      "epoch": 1.6791684837912486,
+      "grad_norm": 0.34058240056037903,
+      "learning_rate": 0.00012578616352201257,
+      "loss": 0.6324,
+      "step": 361
+    },
+    {
+      "epoch": 1.6838203227213258,
+      "grad_norm": 0.3173544108867645,
+      "learning_rate": 0.00012553459119496856,
+      "loss": 0.6038,
+      "step": 362
+    },
+    {
+      "epoch": 1.688472161651403,
+      "grad_norm": 0.34285029768943787,
+      "learning_rate": 0.00012528301886792452,
+      "loss": 0.5952,
+      "step": 363
+    },
+    {
+      "epoch": 1.6931240005814798,
+      "grad_norm": 0.3301316499710083,
+      "learning_rate": 0.0001250314465408805,
+      "loss": 0.6196,
+      "step": 364
+    },
+    {
+      "epoch": 1.6977758395115568,
+      "grad_norm": 0.3121924102306366,
+      "learning_rate": 0.00012477987421383649,
+      "loss": 0.5874,
+      "step": 365
+    },
+    {
+      "epoch": 1.7024276784416341,
+      "grad_norm": 0.3126750886440277,
+      "learning_rate": 0.00012452830188679244,
+      "loss": 0.5918,
+      "step": 366
+    },
+    {
+      "epoch": 1.707079517371711,
+      "grad_norm": 0.3309522569179535,
+      "learning_rate": 0.00012427672955974843,
+      "loss": 0.6172,
+      "step": 367
+    },
+    {
+      "epoch": 1.711731356301788,
+      "grad_norm": 0.32107484340667725,
+      "learning_rate": 0.00012402515723270442,
+      "loss": 0.5869,
+      "step": 368
+    },
+    {
+      "epoch": 1.716383195231865,
+      "grad_norm": 0.32367855310440063,
+      "learning_rate": 0.00012377358490566037,
+      "loss": 0.5977,
+      "step": 369
+    },
+    {
+      "epoch": 1.7210350341619423,
+      "grad_norm": 0.329386442899704,
+      "learning_rate": 0.00012352201257861636,
+      "loss": 0.6305,
+      "step": 370
+    },
+    {
+      "epoch": 1.7256868730920192,
+      "grad_norm": 0.3309147357940674,
+      "learning_rate": 0.00012327044025157232,
+      "loss": 0.639,
+      "step": 371
+    },
+    {
+      "epoch": 1.730338712022096,
+      "grad_norm": 0.30057141184806824,
+      "learning_rate": 0.0001230188679245283,
+      "loss": 0.613,
+      "step": 372
+    },
+    {
+      "epoch": 1.7349905509521732,
+      "grad_norm": 0.3404291272163391,
+      "learning_rate": 0.0001227672955974843,
+      "loss": 0.5942,
+      "step": 373
+    },
+    {
+      "epoch": 1.7396423898822504,
+      "grad_norm": 0.3339247405529022,
+      "learning_rate": 0.00012251572327044025,
+      "loss": 0.6349,
+      "step": 374
+    },
+    {
+      "epoch": 1.7442942288123273,
+      "grad_norm": 0.3592160940170288,
+      "learning_rate": 0.00012226415094339623,
+      "loss": 0.5744,
+      "step": 375
+    },
+    {
+      "epoch": 1.7489460677424045,
+      "grad_norm": 0.32319962978363037,
+      "learning_rate": 0.0001220125786163522,
+      "loss": 0.595,
+      "step": 376
+    },
+    {
+      "epoch": 1.7535979066724816,
+      "grad_norm": 0.3749141991138458,
+      "learning_rate": 0.00012176100628930817,
+      "loss": 0.6068,
+      "step": 377
+    },
+    {
+      "epoch": 1.7582497456025585,
+      "grad_norm": 0.3308943808078766,
+      "learning_rate": 0.00012150943396226415,
+      "loss": 0.6085,
+      "step": 378
+    },
+    {
+      "epoch": 1.7629015845326355,
+      "grad_norm": 0.3301700949668884,
+      "learning_rate": 0.00012125786163522013,
+      "loss": 0.604,
+      "step": 379
+    },
+    {
+      "epoch": 1.7675534234627126,
+      "grad_norm": 0.32405662536621094,
+      "learning_rate": 0.0001210062893081761,
+      "loss": 0.5597,
+      "step": 380
+    },
+    {
+      "epoch": 1.7722052623927897,
+      "grad_norm": 0.3239361345767975,
+      "learning_rate": 0.00012075471698113207,
+      "loss": 0.5604,
+      "step": 381
+    },
+    {
+      "epoch": 1.7768571013228667,
+      "grad_norm": 0.39090093970298767,
+      "learning_rate": 0.00012050314465408805,
+      "loss": 0.5884,
+      "step": 382
+    },
+    {
+      "epoch": 1.7815089402529436,
+      "grad_norm": 0.30942320823669434,
+      "learning_rate": 0.00012025157232704403,
+      "loss": 0.5928,
+      "step": 383
+    },
+    {
+      "epoch": 1.7861607791830207,
+      "grad_norm": 0.3391380310058594,
+      "learning_rate": 0.00012,
+      "loss": 0.5934,
+      "step": 384
+    },
+    {
+      "epoch": 1.7908126181130979,
+      "grad_norm": 0.2991212010383606,
+      "learning_rate": 0.00011974842767295597,
+      "loss": 0.5694,
+      "step": 385
+    },
+    {
+      "epoch": 1.7954644570431748,
+      "grad_norm": 0.30802592635154724,
+      "learning_rate": 0.00011949685534591195,
+      "loss": 0.6238,
+      "step": 386
+    },
+    {
+      "epoch": 1.800116295973252,
+      "grad_norm": 0.30259111523628235,
+      "learning_rate": 0.00011924528301886793,
+      "loss": 0.5919,
+      "step": 387
+    },
+    {
+      "epoch": 1.804768134903329,
+      "grad_norm": 0.3143281936645508,
+      "learning_rate": 0.0001189937106918239,
+      "loss": 0.6095,
+      "step": 388
+    },
+    {
+      "epoch": 1.809419973833406,
+      "grad_norm": 0.30476394295692444,
+      "learning_rate": 0.00011874213836477988,
+      "loss": 0.5943,
+      "step": 389
+    },
+    {
+      "epoch": 1.814071812763483,
+      "grad_norm": 0.3428845703601837,
+      "learning_rate": 0.00011849056603773585,
+      "loss": 0.6175,
+      "step": 390
+    },
+    {
+      "epoch": 1.81872365169356,
+      "grad_norm": 0.34092244505882263,
+      "learning_rate": 0.00011823899371069183,
+      "loss": 0.5998,
+      "step": 391
+    },
+    {
+      "epoch": 1.8233754906236372,
+      "grad_norm": 0.31098848581314087,
+      "learning_rate": 0.0001179874213836478,
+      "loss": 0.6101,
+      "step": 392
+    },
+    {
+      "epoch": 1.8280273295537142,
+      "grad_norm": 0.2941473722457886,
+      "learning_rate": 0.00011773584905660378,
+      "loss": 0.5759,
+      "step": 393
+    },
+    {
+      "epoch": 1.832679168483791,
+      "grad_norm": 0.33450087904930115,
+      "learning_rate": 0.00011748427672955975,
+      "loss": 0.5997,
+      "step": 394
+    },
+    {
+      "epoch": 1.8373310074138685,
+      "grad_norm": 0.3609963059425354,
+      "learning_rate": 0.00011723270440251572,
+      "loss": 0.5758,
+      "step": 395
+    },
+    {
+      "epoch": 1.8419828463439454,
+      "grad_norm": 0.31881803274154663,
+      "learning_rate": 0.0001169811320754717,
+      "loss": 0.5656,
+      "step": 396
+    },
+    {
+      "epoch": 1.8466346852740223,
+      "grad_norm": 0.3416830599308014,
+      "learning_rate": 0.00011672955974842768,
+      "loss": 0.5422,
+      "step": 397
+    },
+    {
+      "epoch": 1.8512865242040994,
+      "grad_norm": 0.3478597104549408,
+      "learning_rate": 0.00011647798742138365,
+      "loss": 0.5993,
+      "step": 398
+    },
+    {
+      "epoch": 1.8559383631341766,
+      "grad_norm": 0.3396543264389038,
+      "learning_rate": 0.00011622641509433962,
+      "loss": 0.597,
+      "step": 399
+    },
+    {
+      "epoch": 1.8605902020642535,
+      "grad_norm": 0.37878015637397766,
+      "learning_rate": 0.0001159748427672956,
+      "loss": 0.6112,
+      "step": 400
+    },
+    {
+      "epoch": 1.8652420409943304,
+      "grad_norm": 0.33623307943344116,
+      "learning_rate": 0.00011572327044025158,
+      "loss": 0.6076,
+      "step": 401
+    },
+    {
+      "epoch": 1.8698938799244076,
+      "grad_norm": 0.35553428530693054,
+      "learning_rate": 0.00011547169811320755,
+      "loss": 0.5959,
+      "step": 402
+    },
+    {
+      "epoch": 1.8745457188544847,
+      "grad_norm": 0.32533591985702515,
+      "learning_rate": 0.00011522012578616352,
+      "loss": 0.5598,
+      "step": 403
+    },
+    {
+      "epoch": 1.8791975577845617,
+      "grad_norm": 0.3280135989189148,
+      "learning_rate": 0.0001149685534591195,
+      "loss": 0.5792,
+      "step": 404
+    },
+    {
+      "epoch": 1.8838493967146388,
+      "grad_norm": 0.3364737927913666,
+      "learning_rate": 0.00011471698113207548,
+      "loss": 0.6273,
+      "step": 405
+    },
+    {
+      "epoch": 1.888501235644716,
+      "grad_norm": 0.35875532031059265,
+      "learning_rate": 0.00011446540880503145,
+      "loss": 0.5933,
+      "step": 406
+    },
+    {
+      "epoch": 1.8931530745747929,
+      "grad_norm": 0.3286273181438446,
+      "learning_rate": 0.00011421383647798742,
+      "loss": 0.5963,
+      "step": 407
+    },
+    {
+      "epoch": 1.8978049135048698,
+      "grad_norm": 0.3897830843925476,
+      "learning_rate": 0.00011396226415094339,
+      "loss": 0.6051,
+      "step": 408
+    },
+    {
+      "epoch": 1.902456752434947,
+      "grad_norm": 0.33309322595596313,
+      "learning_rate": 0.00011371069182389938,
+      "loss": 0.5732,
+      "step": 409
+    },
+    {
+      "epoch": 1.907108591365024,
+      "grad_norm": 0.338821679353714,
+      "learning_rate": 0.00011345911949685535,
+      "loss": 0.5996,
+      "step": 410
+    },
+    {
+      "epoch": 1.911760430295101,
+      "grad_norm": 0.31202319264411926,
+      "learning_rate": 0.00011320754716981132,
+      "loss": 0.6289,
+      "step": 411
+    },
+    {
+      "epoch": 1.916412269225178,
+      "grad_norm": 0.36233726143836975,
+      "learning_rate": 0.00011295597484276729,
+      "loss": 0.6103,
+      "step": 412
+    },
+    {
+      "epoch": 1.921064108155255,
+      "grad_norm": 0.3464321792125702,
+      "learning_rate": 0.00011270440251572328,
+      "loss": 0.6323,
+      "step": 413
+    },
+    {
+      "epoch": 1.9257159470853322,
+      "grad_norm": 0.31249722838401794,
+      "learning_rate": 0.00011245283018867925,
+      "loss": 0.5887,
+      "step": 414
+    },
+    {
+      "epoch": 1.9303677860154091,
+      "grad_norm": 0.36603844165802,
+      "learning_rate": 0.00011220125786163522,
+      "loss": 0.628,
+      "step": 415
+    },
+    {
+      "epoch": 1.9350196249454863,
+      "grad_norm": 0.35427892208099365,
+      "learning_rate": 0.00011194968553459119,
+      "loss": 0.5998,
+      "step": 416
+    },
+    {
+      "epoch": 1.9396714638755634,
+      "grad_norm": 0.3365907669067383,
+      "learning_rate": 0.00011169811320754718,
+      "loss": 0.6052,
+      "step": 417
+    },
+    {
+      "epoch": 1.9443233028056404,
+      "grad_norm": 0.3800869584083557,
+      "learning_rate": 0.00011144654088050315,
+      "loss": 0.5754,
+      "step": 418
+    },
+    {
+      "epoch": 1.9489751417357173,
+      "grad_norm": 0.326170951128006,
+      "learning_rate": 0.00011119496855345912,
+      "loss": 0.6184,
+      "step": 419
+    },
+    {
+      "epoch": 1.9536269806657944,
+      "grad_norm": 0.37027937173843384,
+      "learning_rate": 0.00011094339622641509,
+      "loss": 0.5978,
+      "step": 420
+    },
+    {
+      "epoch": 1.9582788195958716,
+      "grad_norm": 0.3236027956008911,
+      "learning_rate": 0.00011069182389937108,
+      "loss": 0.5879,
+      "step": 421
+    },
+    {
+      "epoch": 1.9629306585259485,
+      "grad_norm": 0.3489690124988556,
+      "learning_rate": 0.00011044025157232705,
+      "loss": 0.5773,
+      "step": 422
+    },
+    {
+      "epoch": 1.9675824974560254,
+      "grad_norm": 0.39736804366111755,
+      "learning_rate": 0.00011018867924528302,
+      "loss": 0.625,
+      "step": 423
+    },
+    {
+      "epoch": 1.9722343363861028,
+      "grad_norm": 0.31319695711135864,
+      "learning_rate": 0.000109937106918239,
+      "loss": 0.5818,
+      "step": 424
+    },
+    {
+      "epoch": 1.9768861753161797,
+      "grad_norm": 0.3418712019920349,
+      "learning_rate": 0.00010968553459119497,
+      "loss": 0.5939,
+      "step": 425
+    },
+    {
+      "epoch": 1.9815380142462566,
+      "grad_norm": 0.33332154154777527,
+      "learning_rate": 0.00010943396226415095,
+      "loss": 0.5789,
+      "step": 426
+    },
+    {
+      "epoch": 1.9861898531763338,
+      "grad_norm": 0.31600314378738403,
+      "learning_rate": 0.00010918238993710692,
+      "loss": 0.5958,
+      "step": 427
+    },
+    {
+      "epoch": 1.990841692106411,
+      "grad_norm": 0.3286213278770447,
+      "learning_rate": 0.0001089308176100629,
+      "loss": 0.5958,
+      "step": 428
+    },
+    {
+      "epoch": 1.9954935310364879,
+      "grad_norm": 0.36490023136138916,
+      "learning_rate": 0.00010867924528301887,
+      "loss": 0.5537,
+      "step": 429
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.3325427174568176,
+      "learning_rate": 0.00010842767295597485,
+      "loss": 0.5761,
+      "step": 430
+    },
+    {
+      "epoch": 2.004651838930077,
+      "grad_norm": 0.3556664288043976,
+      "learning_rate": 0.00010817610062893082,
+      "loss": 0.5541,
+      "step": 431
+    },
+    {
+      "epoch": 2.0093036778601543,
+      "grad_norm": 0.3427799642086029,
+      "learning_rate": 0.0001079245283018868,
+      "loss": 0.5488,
+      "step": 432
+    },
+    {
+      "epoch": 2.013955516790231,
+      "grad_norm": 0.32484185695648193,
+      "learning_rate": 0.00010767295597484277,
+      "loss": 0.5474,
+      "step": 433
+    },
+    {
+      "epoch": 2.018607355720308,
+      "grad_norm": 0.37591978907585144,
+      "learning_rate": 0.00010742138364779875,
+      "loss": 0.6009,
+      "step": 434
+    },
+    {
+      "epoch": 2.023259194650385,
+      "grad_norm": 0.3013262450695038,
+      "learning_rate": 0.00010716981132075472,
+      "loss": 0.5342,
+      "step": 435
+    },
+    {
+      "epoch": 2.0279110335804624,
+      "grad_norm": 0.33676877617836,
+      "learning_rate": 0.0001069182389937107,
+      "loss": 0.5416,
+      "step": 436
+    },
+    {
+      "epoch": 2.0325628725105394,
+      "grad_norm": 0.3570246696472168,
+      "learning_rate": 0.00010666666666666667,
+      "loss": 0.5956,
+      "step": 437
+    },
+    {
+      "epoch": 2.0372147114406163,
+      "grad_norm": 0.31898951530456543,
+      "learning_rate": 0.00010641509433962265,
+      "loss": 0.5662,
+      "step": 438
+    },
+    {
+      "epoch": 2.041866550370693,
+      "grad_norm": 0.3282899558544159,
+      "learning_rate": 0.00010616352201257862,
+      "loss": 0.5815,
+      "step": 439
+    },
+    {
+      "epoch": 2.0465183893007706,
+      "grad_norm": 0.35512569546699524,
+      "learning_rate": 0.0001059119496855346,
+      "loss": 0.5766,
+      "step": 440
+    },
+    {
+      "epoch": 2.0511702282308475,
+      "grad_norm": 0.34569963812828064,
+      "learning_rate": 0.00010566037735849057,
+      "loss": 0.5842,
+      "step": 441
+    },
+    {
+      "epoch": 2.0558220671609244,
+      "grad_norm": 0.3220233619213104,
+      "learning_rate": 0.00010540880503144654,
+      "loss": 0.5761,
+      "step": 442
+    },
+    {
+      "epoch": 2.060473906091002,
+      "grad_norm": 0.35429680347442627,
+      "learning_rate": 0.00010515723270440252,
+      "loss": 0.5607,
+      "step": 443
+    },
+    {
+      "epoch": 2.0651257450210787,
+      "grad_norm": 0.3453116714954376,
+      "learning_rate": 0.0001049056603773585,
+      "loss": 0.5488,
+      "step": 444
+    },
+    {
+      "epoch": 2.0697775839511556,
+      "grad_norm": 0.3153882622718811,
+      "learning_rate": 0.00010465408805031447,
+      "loss": 0.5501,
+      "step": 445
+    },
+    {
+      "epoch": 2.0744294228812326,
+      "grad_norm": 0.31950825452804565,
+      "learning_rate": 0.00010440251572327044,
+      "loss": 0.5717,
+      "step": 446
+    },
+    {
+      "epoch": 2.07908126181131,
+      "grad_norm": 0.31009095907211304,
+      "learning_rate": 0.00010415094339622642,
+      "loss": 0.5662,
+      "step": 447
+    },
+    {
+      "epoch": 2.083733100741387,
+      "grad_norm": 0.3531000316143036,
+      "learning_rate": 0.0001038993710691824,
+      "loss": 0.5694,
+      "step": 448
+    },
+    {
+      "epoch": 2.0883849396714638,
+      "grad_norm": 0.3471253514289856,
+      "learning_rate": 0.00010364779874213837,
+      "loss": 0.6,
+      "step": 449
+    },
+    {
+      "epoch": 2.093036778601541,
+      "grad_norm": 0.3371501564979553,
+      "learning_rate": 0.00010339622641509434,
+      "loss": 0.5805,
+      "step": 450
+    },
+    {
+      "epoch": 2.097688617531618,
+      "grad_norm": 0.3082928955554962,
+      "learning_rate": 0.00010314465408805032,
+      "loss": 0.5673,
+      "step": 451
+    },
+    {
+      "epoch": 2.102340456461695,
+      "grad_norm": 0.33554595708847046,
+      "learning_rate": 0.0001028930817610063,
+      "loss": 0.555,
+      "step": 452
+    },
+    {
+      "epoch": 2.106992295391772,
+      "grad_norm": 0.34016352891921997,
+      "learning_rate": 0.00010264150943396227,
+      "loss": 0.5878,
+      "step": 453
+    },
+    {
+      "epoch": 2.1116441343218493,
+      "grad_norm": 0.34198808670043945,
+      "learning_rate": 0.00010238993710691824,
+      "loss": 0.5649,
+      "step": 454
+    },
+    {
+      "epoch": 2.116295973251926,
+      "grad_norm": 0.3680737018585205,
+      "learning_rate": 0.00010213836477987422,
+      "loss": 0.5804,
+      "step": 455
+    },
+    {
+      "epoch": 2.120947812182003,
+      "grad_norm": 0.35423514246940613,
+      "learning_rate": 0.0001018867924528302,
+      "loss": 0.5819,
+      "step": 456
+    },
+    {
+      "epoch": 2.12559965111208,
+      "grad_norm": 0.3589932918548584,
+      "learning_rate": 0.00010163522012578617,
+      "loss": 0.5886,
+      "step": 457
+    },
+    {
+      "epoch": 2.1302514900421574,
+      "grad_norm": 0.36457332968711853,
+      "learning_rate": 0.00010138364779874214,
+      "loss": 0.5786,
+      "step": 458
+    },
+    {
+      "epoch": 2.1349033289722343,
+      "grad_norm": 0.3131260275840759,
+      "learning_rate": 0.00010113207547169811,
+      "loss": 0.4942,
+      "step": 459
+    },
+    {
+      "epoch": 2.1395551679023113,
+      "grad_norm": 0.36935341358184814,
+      "learning_rate": 0.0001008805031446541,
+      "loss": 0.5654,
+      "step": 460
+    },
+    {
+      "epoch": 2.1442070068323886,
+      "grad_norm": 0.39576059579849243,
+      "learning_rate": 0.00010062893081761007,
+      "loss": 0.5421,
+      "step": 461
+    },
+    {
+      "epoch": 2.1488588457624656,
+      "grad_norm": 0.3363158106803894,
+      "learning_rate": 0.00010037735849056604,
+      "loss": 0.5623,
+      "step": 462
+    },
+    {
+      "epoch": 2.1535106846925425,
+      "grad_norm": 0.361555278301239,
+      "learning_rate": 0.00010012578616352201,
+      "loss": 0.5979,
+      "step": 463
+    },
+    {
+      "epoch": 2.1581625236226194,
+      "grad_norm": 0.38301882147789,
+      "learning_rate": 9.9874213836478e-05,
+      "loss": 0.594,
+      "step": 464
+    },
+    {
+      "epoch": 2.1628143625526968,
+      "grad_norm": 0.360763818025589,
+      "learning_rate": 9.962264150943397e-05,
+      "loss": 0.5591,
+      "step": 465
+    },
+    {
+      "epoch": 2.1674662014827737,
+      "grad_norm": 0.3606659173965454,
+      "learning_rate": 9.937106918238994e-05,
+      "loss": 0.5816,
+      "step": 466
+    },
+    {
+      "epoch": 2.1721180404128506,
+      "grad_norm": 0.3752624988555908,
+      "learning_rate": 9.911949685534591e-05,
+      "loss": 0.5643,
+      "step": 467
+    },
+    {
+      "epoch": 2.1767698793429275,
+      "grad_norm": 0.31429556012153625,
+      "learning_rate": 9.88679245283019e-05,
+      "loss": 0.5901,
+      "step": 468
+    },
+    {
+      "epoch": 2.181421718273005,
+      "grad_norm": 0.37386947870254517,
+      "learning_rate": 9.861635220125787e-05,
+      "loss": 0.554,
+      "step": 469
+    },
+    {
+      "epoch": 2.186073557203082,
+      "grad_norm": 0.3736591041088104,
+      "learning_rate": 9.836477987421384e-05,
+      "loss": 0.5644,
+      "step": 470
+    },
+    {
+      "epoch": 2.1907253961331588,
+      "grad_norm": 0.3222185969352722,
+      "learning_rate": 9.811320754716981e-05,
+      "loss": 0.5734,
+      "step": 471
+    },
+    {
+      "epoch": 2.195377235063236,
+      "grad_norm": 0.3444487750530243,
+      "learning_rate": 9.786163522012578e-05,
+      "loss": 0.5898,
+      "step": 472
+    },
+    {
+      "epoch": 2.200029073993313,
+      "grad_norm": 0.3368206322193146,
+      "learning_rate": 9.761006289308177e-05,
+      "loss": 0.5475,
+      "step": 473
+    },
+    {
+      "epoch": 2.20468091292339,
+      "grad_norm": 0.36262720823287964,
+      "learning_rate": 9.735849056603774e-05,
+      "loss": 0.5527,
+      "step": 474
+    },
+    {
+      "epoch": 2.209332751853467,
+      "grad_norm": 0.3376947045326233,
+      "learning_rate": 9.710691823899371e-05,
+      "loss": 0.577,
+      "step": 475
+    },
+    {
+      "epoch": 2.2139845907835443,
+      "grad_norm": 0.33253538608551025,
+      "learning_rate": 9.685534591194969e-05,
+      "loss": 0.5665,
+      "step": 476
+    },
+    {
+      "epoch": 2.218636429713621,
+      "grad_norm": 0.3008655607700348,
+      "learning_rate": 9.660377358490567e-05,
+      "loss": 0.5602,
+      "step": 477
+    },
+    {
+      "epoch": 2.223288268643698,
+      "grad_norm": 0.34846389293670654,
+      "learning_rate": 9.635220125786164e-05,
+      "loss": 0.5962,
+      "step": 478
+    },
+    {
+      "epoch": 2.227940107573775,
+      "grad_norm": 0.33933350443840027,
+      "learning_rate": 9.610062893081761e-05,
+      "loss": 0.5313,
+      "step": 479
+    },
+    {
+      "epoch": 2.2325919465038524,
+      "grad_norm": 0.33366158604621887,
+      "learning_rate": 9.584905660377359e-05,
+      "loss": 0.5431,
+      "step": 480
+    },
+    {
+      "epoch": 2.2372437854339293,
+      "grad_norm": 0.3173999488353729,
+      "learning_rate": 9.559748427672957e-05,
+      "loss": 0.546,
+      "step": 481
+    },
+    {
+      "epoch": 2.2418956243640062,
+      "grad_norm": 0.35807493329048157,
+      "learning_rate": 9.534591194968554e-05,
+      "loss": 0.5735,
+      "step": 482
+    },
+    {
+      "epoch": 2.2465474632940836,
+      "grad_norm": 0.3571740984916687,
+      "learning_rate": 9.509433962264151e-05,
+      "loss": 0.575,
+      "step": 483
+    },
+    {
+      "epoch": 2.2511993022241605,
+      "grad_norm": 0.3613911271095276,
+      "learning_rate": 9.484276729559749e-05,
+      "loss": 0.5734,
+      "step": 484
+    },
+    {
+      "epoch": 2.2558511411542375,
+      "grad_norm": 0.34501418471336365,
+      "learning_rate": 9.459119496855347e-05,
+      "loss": 0.5693,
+      "step": 485
+    },
+    {
+      "epoch": 2.260502980084315,
+      "grad_norm": 0.33172985911369324,
+      "learning_rate": 9.433962264150944e-05,
+      "loss": 0.5247,
+      "step": 486
+    },
+    {
+      "epoch": 2.2651548190143918,
+      "grad_norm": 0.3598814308643341,
+      "learning_rate": 9.408805031446541e-05,
+      "loss": 0.5689,
+      "step": 487
+    },
+    {
+      "epoch": 2.2698066579444687,
+      "grad_norm": 0.35693806409835815,
+      "learning_rate": 9.383647798742139e-05,
+      "loss": 0.5741,
+      "step": 488
+    },
+    {
+      "epoch": 2.2744584968745456,
+      "grad_norm": 0.3337371349334717,
+      "learning_rate": 9.358490566037736e-05,
+      "loss": 0.5505,
+      "step": 489
+    },
+    {
+      "epoch": 2.2791103358046225,
+      "grad_norm": 0.3407386839389801,
+      "learning_rate": 9.333333333333334e-05,
+      "loss": 0.6109,
+      "step": 490
+    },
+    {
+      "epoch": 2.2837621747347,
+      "grad_norm": 0.3525364100933075,
+      "learning_rate": 9.308176100628931e-05,
+      "loss": 0.5898,
+      "step": 491
+    },
+    {
+      "epoch": 2.288414013664777,
+      "grad_norm": 0.3354165554046631,
+      "learning_rate": 9.283018867924529e-05,
+      "loss": 0.6044,
+      "step": 492
+    },
+    {
+      "epoch": 2.2930658525948537,
+      "grad_norm": 0.3306461274623871,
+      "learning_rate": 9.257861635220126e-05,
+      "loss": 0.5948,
+      "step": 493
+    },
+    {
+      "epoch": 2.297717691524931,
+      "grad_norm": 0.3609619736671448,
+      "learning_rate": 9.232704402515724e-05,
+      "loss": 0.5795,
+      "step": 494
+    },
+    {
+      "epoch": 2.302369530455008,
+      "grad_norm": 0.34186404943466187,
+      "learning_rate": 9.207547169811322e-05,
+      "loss": 0.599,
+      "step": 495
+    },
+    {
+      "epoch": 2.307021369385085,
+      "grad_norm": 0.3355623781681061,
+      "learning_rate": 9.182389937106919e-05,
+      "loss": 0.5879,
+      "step": 496
+    },
+    {
+      "epoch": 2.3116732083151623,
+      "grad_norm": 0.34636420011520386,
+      "learning_rate": 9.157232704402516e-05,
+      "loss": 0.5377,
+      "step": 497
+    },
+    {
+      "epoch": 2.3163250472452392,
+      "grad_norm": 0.32569974660873413,
+      "learning_rate": 9.132075471698114e-05,
+      "loss": 0.5494,
+      "step": 498
+    },
+    {
+      "epoch": 2.320976886175316,
+      "grad_norm": 0.3366355299949646,
+      "learning_rate": 9.106918238993712e-05,
+      "loss": 0.563,
+      "step": 499
+    },
+    {
+      "epoch": 2.325628725105393,
+      "grad_norm": 0.31733641028404236,
+      "learning_rate": 9.081761006289309e-05,
+      "loss": 0.5832,
+      "step": 500
+    },
+    {
+      "epoch": 2.3302805640354705,
+      "grad_norm": 0.34694549441337585,
+      "learning_rate": 9.056603773584906e-05,
+      "loss": 0.5825,
+      "step": 501
+    },
+    {
+      "epoch": 2.3349324029655474,
+      "grad_norm": 0.3364946246147156,
+      "learning_rate": 9.031446540880504e-05,
+      "loss": 0.558,
+      "step": 502
+    },
+    {
+      "epoch": 2.3395842418956243,
+      "grad_norm": 0.37570393085479736,
+      "learning_rate": 9.006289308176102e-05,
+      "loss": 0.5492,
+      "step": 503
+    },
+    {
+      "epoch": 2.3442360808257012,
+      "grad_norm": 0.36170029640197754,
+      "learning_rate": 8.981132075471699e-05,
+      "loss": 0.562,
+      "step": 504
+    },
+    {
+      "epoch": 2.3488879197557786,
+      "grad_norm": 0.33349472284317017,
+      "learning_rate": 8.955974842767296e-05,
+      "loss": 0.5728,
+      "step": 505
+    },
+    {
+      "epoch": 2.3535397586858555,
+      "grad_norm": 0.35422810912132263,
+      "learning_rate": 8.930817610062893e-05,
+      "loss": 0.5884,
+      "step": 506
+    },
+    {
+      "epoch": 2.3581915976159324,
+      "grad_norm": 0.35913020372390747,
+      "learning_rate": 8.905660377358492e-05,
+      "loss": 0.5548,
+      "step": 507
+    },
+    {
+      "epoch": 2.36284343654601,
+      "grad_norm": 0.33548977971076965,
+      "learning_rate": 8.880503144654089e-05,
+      "loss": 0.5622,
+      "step": 508
+    },
+    {
+      "epoch": 2.3674952754760867,
+      "grad_norm": 0.35967713594436646,
+      "learning_rate": 8.855345911949686e-05,
+      "loss": 0.5655,
+      "step": 509
+    },
+    {
+      "epoch": 2.3721471144061637,
+      "grad_norm": 0.32967039942741394,
+      "learning_rate": 8.830188679245283e-05,
+      "loss": 0.5356,
+      "step": 510
+    },
+    {
+      "epoch": 2.3767989533362406,
+      "grad_norm": 0.3087659776210785,
+      "learning_rate": 8.805031446540882e-05,
+      "loss": 0.5647,
+      "step": 511
+    },
+    {
+      "epoch": 2.381450792266318,
+      "grad_norm": 0.35358235239982605,
+      "learning_rate": 8.779874213836479e-05,
+      "loss": 0.536,
+      "step": 512
+    },
+    {
+      "epoch": 2.386102631196395,
+      "grad_norm": 0.3334801495075226,
+      "learning_rate": 8.754716981132076e-05,
+      "loss": 0.5711,
+      "step": 513
+    },
+    {
+      "epoch": 2.390754470126472,
+      "grad_norm": 0.31288623809814453,
+      "learning_rate": 8.729559748427673e-05,
+      "loss": 0.5554,
+      "step": 514
+    },
+    {
+      "epoch": 2.3954063090565487,
+      "grad_norm": 0.3208356499671936,
+      "learning_rate": 8.704402515723272e-05,
+      "loss": 0.5546,
+      "step": 515
+    },
+    {
+      "epoch": 2.400058147986626,
+      "grad_norm": 0.33053919672966003,
+      "learning_rate": 8.679245283018869e-05,
+      "loss": 0.5525,
+      "step": 516
+    },
+    {
+      "epoch": 2.404709986916703,
+      "grad_norm": 0.3359244167804718,
+      "learning_rate": 8.654088050314466e-05,
+      "loss": 0.5776,
+      "step": 517
+    },
+    {
+      "epoch": 2.40936182584678,
+      "grad_norm": 0.33414962887763977,
+      "learning_rate": 8.628930817610063e-05,
+      "loss": 0.5597,
+      "step": 518
+    },
+    {
+      "epoch": 2.4140136647768573,
+      "grad_norm": 0.323444128036499,
+      "learning_rate": 8.603773584905662e-05,
+      "loss": 0.5751,
+      "step": 519
+    },
+    {
+      "epoch": 2.4186655037069342,
+      "grad_norm": 0.32793116569519043,
+      "learning_rate": 8.578616352201259e-05,
+      "loss": 0.5631,
+      "step": 520
+    },
+    {
+      "epoch": 2.423317342637011,
+      "grad_norm": 0.3437434136867523,
+      "learning_rate": 8.553459119496856e-05,
+      "loss": 0.5764,
+      "step": 521
+    },
+    {
+      "epoch": 2.427969181567088,
+      "grad_norm": 0.3580510914325714,
+      "learning_rate": 8.528301886792453e-05,
+      "loss": 0.5694,
+      "step": 522
+    },
+    {
+      "epoch": 2.4326210204971654,
+      "grad_norm": 0.3285934627056122,
+      "learning_rate": 8.50314465408805e-05,
+      "loss": 0.5554,
+      "step": 523
+    },
+    {
+      "epoch": 2.4372728594272424,
+      "grad_norm": 0.3104308843612671,
+      "learning_rate": 8.477987421383649e-05,
+      "loss": 0.586,
+      "step": 524
+    },
+    {
+      "epoch": 2.4419246983573193,
+      "grad_norm": 0.3188004791736603,
+      "learning_rate": 8.452830188679246e-05,
+      "loss": 0.564,
+      "step": 525
+    },
+    {
+      "epoch": 2.446576537287396,
+      "grad_norm": 0.35406947135925293,
+      "learning_rate": 8.427672955974843e-05,
+      "loss": 0.5822,
+      "step": 526
+    },
+    {
+      "epoch": 2.4512283762174736,
+      "grad_norm": 0.3209180235862732,
+      "learning_rate": 8.40251572327044e-05,
+      "loss": 0.56,
+      "step": 527
+    },
+    {
+      "epoch": 2.4558802151475505,
+      "grad_norm": 0.338275283575058,
+      "learning_rate": 8.377358490566039e-05,
+      "loss": 0.5539,
+      "step": 528
+    },
+    {
+      "epoch": 2.4605320540776274,
+      "grad_norm": 0.3473256826400757,
+      "learning_rate": 8.352201257861636e-05,
+      "loss": 0.5644,
+      "step": 529
+    },
+    {
+      "epoch": 2.465183893007705,
+      "grad_norm": 0.3183133006095886,
+      "learning_rate": 8.327044025157233e-05,
+      "loss": 0.6156,
+      "step": 530
+    },
+    {
+      "epoch": 2.4698357319377817,
+      "grad_norm": 0.3253916800022125,
+      "learning_rate": 8.30188679245283e-05,
+      "loss": 0.5778,
+      "step": 531
+    },
+    {
+      "epoch": 2.4744875708678586,
+      "grad_norm": 0.3112945556640625,
+      "learning_rate": 8.276729559748429e-05,
+      "loss": 0.5561,
+      "step": 532
+    },
+    {
+      "epoch": 2.4791394097979356,
+      "grad_norm": 0.3238106667995453,
+      "learning_rate": 8.251572327044026e-05,
+      "loss": 0.5905,
+      "step": 533
+    },
+    {
+      "epoch": 2.483791248728013,
+      "grad_norm": 0.30079522728919983,
+      "learning_rate": 8.226415094339623e-05,
+      "loss": 0.5518,
+      "step": 534
+    },
+    {
+      "epoch": 2.48844308765809,
+      "grad_norm": 0.32537469267845154,
+      "learning_rate": 8.20125786163522e-05,
+      "loss": 0.5662,
+      "step": 535
+    },
+    {
+      "epoch": 2.493094926588167,
+      "grad_norm": 0.33754420280456543,
+      "learning_rate": 8.176100628930818e-05,
+      "loss": 0.5727,
+      "step": 536
+    },
+    {
+      "epoch": 2.4977467655182437,
+      "grad_norm": 0.31966105103492737,
+      "learning_rate": 8.150943396226416e-05,
+      "loss": 0.5807,
+      "step": 537
+    },
+    {
+      "epoch": 2.502398604448321,
+      "grad_norm": 0.3507080674171448,
+      "learning_rate": 8.125786163522013e-05,
+      "loss": 0.5604,
+      "step": 538
+    },
+    {
+      "epoch": 2.507050443378398,
+      "grad_norm": 0.316303551197052,
+      "learning_rate": 8.10062893081761e-05,
+      "loss": 0.5817,
+      "step": 539
+    },
+    {
+      "epoch": 2.511702282308475,
+      "grad_norm": 0.3277745544910431,
+      "learning_rate": 8.075471698113208e-05,
+      "loss": 0.5583,
+      "step": 540
+    },
+    {
+      "epoch": 2.5163541212385523,
+      "grad_norm": 0.3472454249858856,
+      "learning_rate": 8.050314465408806e-05,
+      "loss": 0.572,
+      "step": 541
+    },
+    {
+      "epoch": 2.521005960168629,
+      "grad_norm": 0.32334104180336,
+      "learning_rate": 8.025157232704403e-05,
+      "loss": 0.5791,
+      "step": 542
+    },
+    {
+      "epoch": 2.525657799098706,
+      "grad_norm": 0.30027705430984497,
+      "learning_rate": 8e-05,
+      "loss": 0.5707,
+      "step": 543
+    },
+    {
+      "epoch": 2.5303096380287835,
+      "grad_norm": 0.32912296056747437,
+      "learning_rate": 7.974842767295598e-05,
+      "loss": 0.5626,
+      "step": 544
+    },
+    {
+      "epoch": 2.5349614769588604,
+      "grad_norm": 0.30900466442108154,
+      "learning_rate": 7.949685534591196e-05,
+      "loss": 0.5267,
+      "step": 545
+    },
+    {
+      "epoch": 2.5396133158889374,
+      "grad_norm": 0.3330588936805725,
+      "learning_rate": 7.924528301886794e-05,
+      "loss": 0.5767,
+      "step": 546
+    },
+    {
+      "epoch": 2.5442651548190143,
+      "grad_norm": 0.3172569274902344,
+      "learning_rate": 7.899371069182391e-05,
+      "loss": 0.5151,
+      "step": 547
+    },
+    {
+      "epoch": 2.548916993749091,
+      "grad_norm": 0.3521912097930908,
+      "learning_rate": 7.874213836477988e-05,
+      "loss": 0.5458,
+      "step": 548
+    },
+    {
+      "epoch": 2.5535688326791686,
+      "grad_norm": 0.3407733738422394,
+      "learning_rate": 7.849056603773586e-05,
+      "loss": 0.61,
+      "step": 549
+    },
+    {
+      "epoch": 2.5582206716092455,
+      "grad_norm": 0.32567381858825684,
+      "learning_rate": 7.823899371069184e-05,
+      "loss": 0.5683,
+      "step": 550
+    },
+    {
+      "epoch": 2.5628725105393224,
+      "grad_norm": 0.3341250717639923,
+      "learning_rate": 7.798742138364781e-05,
+      "loss": 0.5633,
+      "step": 551
+    },
+    {
+      "epoch": 2.5675243494694,
+      "grad_norm": 0.33645099401474,
+      "learning_rate": 7.773584905660378e-05,
+      "loss": 0.5856,
+      "step": 552
+    },
+    {
+      "epoch": 2.5721761883994767,
+      "grad_norm": 0.35284698009490967,
+      "learning_rate": 7.748427672955975e-05,
+      "loss": 0.5478,
+      "step": 553
+    },
+    {
+      "epoch": 2.5768280273295536,
+      "grad_norm": 0.32906201481819153,
+      "learning_rate": 7.723270440251574e-05,
+      "loss": 0.5544,
+      "step": 554
+    },
+    {
+      "epoch": 2.581479866259631,
+      "grad_norm": 0.32479235529899597,
+      "learning_rate": 7.698113207547171e-05,
+      "loss": 0.5437,
+      "step": 555
+    },
+    {
+      "epoch": 2.586131705189708,
+      "grad_norm": 0.31013309955596924,
+      "learning_rate": 7.672955974842768e-05,
+      "loss": 0.5749,
+      "step": 556
+    },
+    {
+      "epoch": 2.590783544119785,
+      "grad_norm": 0.3508475124835968,
+      "learning_rate": 7.647798742138365e-05,
+      "loss": 0.558,
+      "step": 557
+    },
+    {
+      "epoch": 2.5954353830498618,
+      "grad_norm": 0.31400588154792786,
+      "learning_rate": 7.622641509433964e-05,
+      "loss": 0.5372,
+      "step": 558
+    },
+    {
+      "epoch": 2.6000872219799387,
+      "grad_norm": 0.3100167214870453,
+      "learning_rate": 7.597484276729561e-05,
+      "loss": 0.5523,
+      "step": 559
+    },
+    {
+      "epoch": 2.604739060910016,
+      "grad_norm": 0.35659855604171753,
+      "learning_rate": 7.572327044025158e-05,
+      "loss": 0.5663,
+      "step": 560
+    },
+    {
+      "epoch": 2.609390899840093,
+      "grad_norm": 0.3730183243751526,
+      "learning_rate": 7.547169811320755e-05,
+      "loss": 0.5208,
+      "step": 561
+    },
+    {
+      "epoch": 2.61404273877017,
+      "grad_norm": 0.33730876445770264,
+      "learning_rate": 7.522012578616354e-05,
+      "loss": 0.5577,
+      "step": 562
+    },
+    {
+      "epoch": 2.6186945777002473,
+      "grad_norm": 0.3258856236934662,
+      "learning_rate": 7.49685534591195e-05,
+      "loss": 0.5731,
+      "step": 563
+    },
+    {
+      "epoch": 2.623346416630324,
+      "grad_norm": 0.3567054867744446,
+      "learning_rate": 7.471698113207547e-05,
+      "loss": 0.6177,
+      "step": 564
+    },
+    {
+      "epoch": 2.627998255560401,
+      "grad_norm": 0.3461082875728607,
+      "learning_rate": 7.446540880503144e-05,
+      "loss": 0.5534,
+      "step": 565
+    },
+    {
+      "epoch": 2.6326500944904785,
+      "grad_norm": 0.2919813096523285,
+      "learning_rate": 7.421383647798742e-05,
+      "loss": 0.5511,
+      "step": 566
+    },
+    {
+      "epoch": 2.6373019334205554,
+      "grad_norm": 0.3393804132938385,
+      "learning_rate": 7.39622641509434e-05,
+      "loss": 0.5434,
+      "step": 567
+    },
+    {
+      "epoch": 2.6419537723506323,
+      "grad_norm": 0.3376845121383667,
+      "learning_rate": 7.371069182389937e-05,
+      "loss": 0.57,
+      "step": 568
+    },
+    {
+      "epoch": 2.6466056112807093,
+      "grad_norm": 0.3444107472896576,
+      "learning_rate": 7.345911949685534e-05,
+      "loss": 0.5322,
+      "step": 569
+    },
+    {
+      "epoch": 2.651257450210786,
+      "grad_norm": 0.31978002190589905,
+      "learning_rate": 7.320754716981132e-05,
+      "loss": 0.5695,
+      "step": 570
+    },
+    {
+      "epoch": 2.6559092891408635,
+      "grad_norm": 0.3549588918685913,
+      "learning_rate": 7.29559748427673e-05,
+      "loss": 0.5852,
+      "step": 571
+    },
+    {
+      "epoch": 2.6605611280709405,
+      "grad_norm": 0.3475392758846283,
+      "learning_rate": 7.270440251572327e-05,
+      "loss": 0.5797,
+      "step": 572
+    },
+    {
+      "epoch": 2.6652129670010174,
+      "grad_norm": 0.3519194424152374,
+      "learning_rate": 7.245283018867924e-05,
+      "loss": 0.555,
+      "step": 573
+    },
+    {
+      "epoch": 2.6698648059310948,
+      "grad_norm": 0.34987887740135193,
+      "learning_rate": 7.220125786163522e-05,
+      "loss": 0.578,
+      "step": 574
+    },
+    {
+      "epoch": 2.6745166448611717,
+      "grad_norm": 0.3432933986186981,
+      "learning_rate": 7.19496855345912e-05,
+      "loss": 0.5605,
+      "step": 575
+    },
+    {
+      "epoch": 2.6791684837912486,
+      "grad_norm": 0.3519866168498993,
+      "learning_rate": 7.169811320754717e-05,
+      "loss": 0.5685,
+      "step": 576
+    },
+    {
+      "epoch": 2.683820322721326,
+      "grad_norm": 0.3178909718990326,
+      "learning_rate": 7.144654088050314e-05,
+      "loss": 0.5362,
+      "step": 577
+    },
+    {
+      "epoch": 2.688472161651403,
+      "grad_norm": 0.3298041820526123,
+      "learning_rate": 7.119496855345912e-05,
+      "loss": 0.5296,
+      "step": 578
+    },
+    {
+      "epoch": 2.69312400058148,
+      "grad_norm": 0.329574316740036,
+      "learning_rate": 7.09433962264151e-05,
+      "loss": 0.5692,
+      "step": 579
+    },
+    {
+      "epoch": 2.6977758395115568,
+      "grad_norm": 0.3557285666465759,
+      "learning_rate": 7.069182389937107e-05,
+      "loss": 0.5278,
+      "step": 580
+    },
+    {
+      "epoch": 2.702427678441634,
+      "grad_norm": 0.34053951501846313,
+      "learning_rate": 7.044025157232704e-05,
+      "loss": 0.5699,
+      "step": 581
+    },
+    {
+      "epoch": 2.707079517371711,
+      "grad_norm": 0.34094059467315674,
+      "learning_rate": 7.018867924528301e-05,
+      "loss": 0.5149,
+      "step": 582
+    },
+    {
+      "epoch": 2.711731356301788,
+      "grad_norm": 0.33889856934547424,
+      "learning_rate": 6.9937106918239e-05,
+      "loss": 0.5413,
+      "step": 583
+    },
+    {
+      "epoch": 2.716383195231865,
+      "grad_norm": 0.31016072630882263,
+      "learning_rate": 6.968553459119497e-05,
+      "loss": 0.5759,
+      "step": 584
+    },
+    {
+      "epoch": 2.7210350341619423,
+      "grad_norm": 0.34233781695365906,
+      "learning_rate": 6.943396226415094e-05,
+      "loss": 0.5772,
+      "step": 585
+    },
+    {
+      "epoch": 2.725686873092019,
+      "grad_norm": 0.3177289068698883,
+      "learning_rate": 6.918238993710691e-05,
+      "loss": 0.5569,
+      "step": 586
+    },
+    {
+      "epoch": 2.730338712022096,
+      "grad_norm": 0.32834571599960327,
+      "learning_rate": 6.89308176100629e-05,
+      "loss": 0.5824,
+      "step": 587
+    },
+    {
+      "epoch": 2.7349905509521735,
+      "grad_norm": 0.33808374404907227,
+      "learning_rate": 6.867924528301887e-05,
+      "loss": 0.5806,
+      "step": 588
+    },
+    {
+      "epoch": 2.7396423898822504,
+      "grad_norm": 0.3351944088935852,
+      "learning_rate": 6.842767295597484e-05,
+      "loss": 0.5773,
+      "step": 589
+    },
+    {
+      "epoch": 2.7442942288123273,
+      "grad_norm": 0.3587758541107178,
+      "learning_rate": 6.817610062893081e-05,
+      "loss": 0.5383,
+      "step": 590
+    },
+    {
+      "epoch": 2.7489460677424047,
+      "grad_norm": 0.3201444447040558,
+      "learning_rate": 6.79245283018868e-05,
+      "loss": 0.5616,
+      "step": 591
+    },
+    {
+      "epoch": 2.7535979066724816,
+      "grad_norm": 0.32416731119155884,
+      "learning_rate": 6.767295597484277e-05,
+      "loss": 0.5687,
+      "step": 592
+    },
+    {
+      "epoch": 2.7582497456025585,
+      "grad_norm": 0.31704238057136536,
+      "learning_rate": 6.742138364779874e-05,
+      "loss": 0.5405,
+      "step": 593
+    },
+    {
+      "epoch": 2.7629015845326355,
+      "grad_norm": 0.33414769172668457,
+      "learning_rate": 6.716981132075471e-05,
+      "loss": 0.5636,
+      "step": 594
+    },
+    {
+      "epoch": 2.7675534234627124,
+      "grad_norm": 0.33495667576789856,
+      "learning_rate": 6.691823899371068e-05,
+      "loss": 0.57,
+      "step": 595
+    },
+    {
+      "epoch": 2.7722052623927897,
+      "grad_norm": 0.3029557466506958,
+      "learning_rate": 6.666666666666667e-05,
+      "loss": 0.5501,
+      "step": 596
+    },
+    {
+      "epoch": 2.7768571013228667,
+      "grad_norm": 0.3150354027748108,
+      "learning_rate": 6.641509433962264e-05,
+      "loss": 0.563,
+      "step": 597
+    },
+    {
+      "epoch": 2.7815089402529436,
+      "grad_norm": 0.352458655834198,
+      "learning_rate": 6.616352201257861e-05,
+      "loss": 0.5392,
+      "step": 598
+    },
+    {
+      "epoch": 2.786160779183021,
+      "grad_norm": 0.3278923034667969,
+      "learning_rate": 6.591194968553458e-05,
+      "loss": 0.5517,
+      "step": 599
+    },
+    {
+      "epoch": 2.790812618113098,
+      "grad_norm": 0.3624614477157593,
+      "learning_rate": 6.566037735849057e-05,
+      "loss": 0.5607,
+      "step": 600
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 860,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 4,
+  "save_steps": 50,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 5.1645214199780966e+17,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}