diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,4543 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 3219,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.004659832246039142,
+      "grad_norm": 12.350169326201001,
+      "learning_rate": 7.763975155279503e-07,
+      "loss": 1.4714,
+      "step": 5
+    },
+    {
+      "epoch": 0.009319664492078284,
+      "grad_norm": 12.346797651170705,
+      "learning_rate": 1.5527950310559006e-06,
+      "loss": 1.4407,
+      "step": 10
+    },
+    {
+      "epoch": 0.013979496738117428,
+      "grad_norm": 10.613939733228706,
+      "learning_rate": 2.329192546583851e-06,
+      "loss": 1.3684,
+      "step": 15
+    },
+    {
+      "epoch": 0.01863932898415657,
+      "grad_norm": 6.167457100740983,
+      "learning_rate": 3.1055900621118013e-06,
+      "loss": 1.2084,
+      "step": 20
+    },
+    {
+      "epoch": 0.023299161230195712,
+      "grad_norm": 2.305387878045333,
+      "learning_rate": 3.881987577639752e-06,
+      "loss": 1.0082,
+      "step": 25
+    },
+    {
+      "epoch": 0.027958993476234855,
+      "grad_norm": 1.0808248793823827,
+      "learning_rate": 4.658385093167702e-06,
+      "loss": 0.8963,
+      "step": 30
+    },
+    {
+      "epoch": 0.032618825722273995,
+      "grad_norm": 0.7619108984943135,
+      "learning_rate": 5.4347826086956525e-06,
+      "loss": 0.8111,
+      "step": 35
+    },
+    {
+      "epoch": 0.03727865796831314,
+      "grad_norm": 0.48868883744902675,
+      "learning_rate": 6.2111801242236025e-06,
+      "loss": 0.7828,
+      "step": 40
+    },
+    {
+      "epoch": 0.04193849021435228,
+      "grad_norm": 0.34317607297217406,
+      "learning_rate": 6.9875776397515525e-06,
+      "loss": 0.732,
+      "step": 45
+    },
+    {
+      "epoch": 0.046598322460391424,
+      "grad_norm": 0.27510329657234445,
+      "learning_rate": 7.763975155279503e-06,
+      "loss": 0.7134,
+      "step": 50
+    },
+    {
+      "epoch": 0.05125815470643057,
+      "grad_norm": 0.2593383235739799,
+      "learning_rate": 8.540372670807453e-06,
+      "loss": 0.6863,
+      "step": 55
+    },
+    {
+      "epoch": 0.05591798695246971,
+      "grad_norm": 0.20803490311680173,
+      "learning_rate": 9.316770186335403e-06,
+      "loss": 0.6527,
+      "step": 60
+    },
+    {
+      "epoch": 0.06057781919850885,
+      "grad_norm": 0.18707806008200195,
+      "learning_rate": 1.0093167701863353e-05,
+      "loss": 0.6396,
+      "step": 65
+    },
+    {
+      "epoch": 0.06523765144454799,
+      "grad_norm": 0.17361687344361165,
+      "learning_rate": 1.0869565217391305e-05,
+      "loss": 0.6376,
+      "step": 70
+    },
+    {
+      "epoch": 0.06989748369058714,
+      "grad_norm": 0.17740318808206565,
+      "learning_rate": 1.1645962732919255e-05,
+      "loss": 0.6197,
+      "step": 75
+    },
+    {
+      "epoch": 0.07455731593662628,
+      "grad_norm": 0.1830829354880775,
+      "learning_rate": 1.2422360248447205e-05,
+      "loss": 0.6132,
+      "step": 80
+    },
+    {
+      "epoch": 0.07921714818266543,
+      "grad_norm": 0.1721443847459758,
+      "learning_rate": 1.3198757763975155e-05,
+      "loss": 0.6035,
+      "step": 85
+    },
+    {
+      "epoch": 0.08387698042870456,
+      "grad_norm": 0.17132667339436553,
+      "learning_rate": 1.3975155279503105e-05,
+      "loss": 0.6131,
+      "step": 90
+    },
+    {
+      "epoch": 0.08853681267474371,
+      "grad_norm": 0.16965194551180893,
+      "learning_rate": 1.4751552795031057e-05,
+      "loss": 0.5942,
+      "step": 95
+    },
+    {
+      "epoch": 0.09319664492078285,
+      "grad_norm": 0.1763713891973871,
+      "learning_rate": 1.5527950310559007e-05,
+      "loss": 0.5795,
+      "step": 100
+    },
+    {
+      "epoch": 0.097856477166822,
+      "grad_norm": 0.15659150014943646,
+      "learning_rate": 1.630434782608696e-05,
+      "loss": 0.5749,
+      "step": 105
+    },
+    {
+      "epoch": 0.10251630941286113,
+      "grad_norm": 0.16376785841868574,
+      "learning_rate": 1.7080745341614907e-05,
+      "loss": 0.5765,
+      "step": 110
+    },
+    {
+      "epoch": 0.10717614165890028,
+      "grad_norm": 0.1609343354959525,
+      "learning_rate": 1.785714285714286e-05,
+      "loss": 0.5741,
+      "step": 115
+    },
+    {
+      "epoch": 0.11183597390493942,
+      "grad_norm": 0.15490460743142537,
+      "learning_rate": 1.8633540372670807e-05,
+      "loss": 0.5596,
+      "step": 120
+    },
+    {
+      "epoch": 0.11649580615097857,
+      "grad_norm": 0.15187248831053118,
+      "learning_rate": 1.940993788819876e-05,
+      "loss": 0.5561,
+      "step": 125
+    },
+    {
+      "epoch": 0.1211556383970177,
+      "grad_norm": 0.16231598905363784,
+      "learning_rate": 2.0186335403726707e-05,
+      "loss": 0.5609,
+      "step": 130
+    },
+    {
+      "epoch": 0.12581547064305684,
+      "grad_norm": 0.15986751085842757,
+      "learning_rate": 2.096273291925466e-05,
+      "loss": 0.5594,
+      "step": 135
+    },
+    {
+      "epoch": 0.13047530288909598,
+      "grad_norm": 0.16743294865258426,
+      "learning_rate": 2.173913043478261e-05,
+      "loss": 0.5432,
+      "step": 140
+    },
+    {
+      "epoch": 0.13513513513513514,
+      "grad_norm": 0.1484319141299328,
+      "learning_rate": 2.2515527950310562e-05,
+      "loss": 0.5712,
+      "step": 145
+    },
+    {
+      "epoch": 0.13979496738117428,
+      "grad_norm": 0.1576721338329723,
+      "learning_rate": 2.329192546583851e-05,
+      "loss": 0.5443,
+      "step": 150
+    },
+    {
+      "epoch": 0.14445479962721341,
+      "grad_norm": 0.15227239564649486,
+      "learning_rate": 2.4068322981366462e-05,
+      "loss": 0.5531,
+      "step": 155
+    },
+    {
+      "epoch": 0.14911463187325255,
+      "grad_norm": 0.15068906162669998,
+      "learning_rate": 2.484472049689441e-05,
+      "loss": 0.5326,
+      "step": 160
+    },
+    {
+      "epoch": 0.15377446411929171,
+      "grad_norm": 0.14913405493758147,
+      "learning_rate": 2.5621118012422362e-05,
+      "loss": 0.5447,
+      "step": 165
+    },
+    {
+      "epoch": 0.15843429636533085,
+      "grad_norm": 0.15044766007575655,
+      "learning_rate": 2.639751552795031e-05,
+      "loss": 0.5207,
+      "step": 170
+    },
+    {
+      "epoch": 0.16309412861137,
+      "grad_norm": 0.1610399582473269,
+      "learning_rate": 2.7173913043478262e-05,
+      "loss": 0.533,
+      "step": 175
+    },
+    {
+      "epoch": 0.16775396085740912,
+      "grad_norm": 0.16130174042514206,
+      "learning_rate": 2.795031055900621e-05,
+      "loss": 0.526,
+      "step": 180
+    },
+    {
+      "epoch": 0.1724137931034483,
+      "grad_norm": 0.14663650747441248,
+      "learning_rate": 2.8726708074534165e-05,
+      "loss": 0.5294,
+      "step": 185
+    },
+    {
+      "epoch": 0.17707362534948742,
+      "grad_norm": 0.1486001743429196,
+      "learning_rate": 2.9503105590062114e-05,
+      "loss": 0.5408,
+      "step": 190
+    },
+    {
+      "epoch": 0.18173345759552656,
+      "grad_norm": 0.17497333697410142,
+      "learning_rate": 3.0279503105590062e-05,
+      "loss": 0.5298,
+      "step": 195
+    },
+    {
+      "epoch": 0.1863932898415657,
+      "grad_norm": 0.1601437526619076,
+      "learning_rate": 3.1055900621118014e-05,
+      "loss": 0.5306,
+      "step": 200
+    },
+    {
+      "epoch": 0.19105312208760486,
+      "grad_norm": 0.1543179008220388,
+      "learning_rate": 3.183229813664597e-05,
+      "loss": 0.519,
+      "step": 205
+    },
+    {
+      "epoch": 0.195712954333644,
+      "grad_norm": 0.1651054178210597,
+      "learning_rate": 3.260869565217392e-05,
+      "loss": 0.5344,
+      "step": 210
+    },
+    {
+      "epoch": 0.20037278657968313,
+      "grad_norm": 0.14617187298201914,
+      "learning_rate": 3.3385093167701865e-05,
+      "loss": 0.5137,
+      "step": 215
+    },
+    {
+      "epoch": 0.20503261882572227,
+      "grad_norm": 0.16161866248880488,
+      "learning_rate": 3.4161490683229814e-05,
+      "loss": 0.519,
+      "step": 220
+    },
+    {
+      "epoch": 0.2096924510717614,
+      "grad_norm": 0.1662323815582868,
+      "learning_rate": 3.493788819875777e-05,
+      "loss": 0.5006,
+      "step": 225
+    },
+    {
+      "epoch": 0.21435228331780057,
+      "grad_norm": 0.14474895591349746,
+      "learning_rate": 3.571428571428572e-05,
+      "loss": 0.5161,
+      "step": 230
+    },
+    {
+      "epoch": 0.2190121155638397,
+      "grad_norm": 0.1357308477715445,
+      "learning_rate": 3.6490683229813665e-05,
+      "loss": 0.5128,
+      "step": 235
+    },
+    {
+      "epoch": 0.22367194780987884,
+      "grad_norm": 0.15933728534187916,
+      "learning_rate": 3.7267080745341614e-05,
+      "loss": 0.51,
+      "step": 240
+    },
+    {
+      "epoch": 0.22833178005591798,
+      "grad_norm": 0.16534519134066747,
+      "learning_rate": 3.804347826086957e-05,
+      "loss": 0.5157,
+      "step": 245
+    },
+    {
+      "epoch": 0.23299161230195714,
+      "grad_norm": 0.17244326329193677,
+      "learning_rate": 3.881987577639752e-05,
+      "loss": 0.5067,
+      "step": 250
+    },
+    {
+      "epoch": 0.23765144454799628,
+      "grad_norm": 0.1798597423839304,
+      "learning_rate": 3.9596273291925465e-05,
+      "loss": 0.5157,
+      "step": 255
+    },
+    {
+      "epoch": 0.2423112767940354,
+      "grad_norm": 0.1731038965183313,
+      "learning_rate": 4.0372670807453414e-05,
+      "loss": 0.5109,
+      "step": 260
+    },
+    {
+      "epoch": 0.24697110904007455,
+      "grad_norm": 0.1539562493485229,
+      "learning_rate": 4.114906832298137e-05,
+      "loss": 0.5082,
+      "step": 265
+    },
+    {
+      "epoch": 0.2516309412861137,
+      "grad_norm": 0.14556418549972353,
+      "learning_rate": 4.192546583850932e-05,
+      "loss": 0.506,
+      "step": 270
+    },
+    {
+      "epoch": 0.25629077353215285,
+      "grad_norm": 0.15400813143993836,
+      "learning_rate": 4.270186335403727e-05,
+      "loss": 0.5105,
+      "step": 275
+    },
+    {
+      "epoch": 0.26095060577819196,
+      "grad_norm": 0.1581556172786541,
+      "learning_rate": 4.347826086956522e-05,
+      "loss": 0.5046,
+      "step": 280
+    },
+    {
+      "epoch": 0.2656104380242311,
+      "grad_norm": 0.18296243968180886,
+      "learning_rate": 4.425465838509317e-05,
+      "loss": 0.5074,
+      "step": 285
+    },
+    {
+      "epoch": 0.2702702702702703,
+      "grad_norm": 0.16132323389914716,
+      "learning_rate": 4.5031055900621124e-05,
+      "loss": 0.5126,
+      "step": 290
+    },
+    {
+      "epoch": 0.2749301025163094,
+      "grad_norm": 0.1865293712611797,
+      "learning_rate": 4.580745341614907e-05,
+      "loss": 0.5158,
+      "step": 295
+    },
+    {
+      "epoch": 0.27958993476234856,
+      "grad_norm": 0.16782438536246205,
+      "learning_rate": 4.658385093167702e-05,
+      "loss": 0.5153,
+      "step": 300
+    },
+    {
+      "epoch": 0.2842497670083877,
+      "grad_norm": 0.15646398847700824,
+      "learning_rate": 4.736024844720497e-05,
+      "loss": 0.5042,
+      "step": 305
+    },
+    {
+      "epoch": 0.28890959925442683,
+      "grad_norm": 0.17365638860912874,
+      "learning_rate": 4.8136645962732924e-05,
+      "loss": 0.4964,
+      "step": 310
+    },
+    {
+      "epoch": 0.293569431500466,
+      "grad_norm": 0.1719285257394336,
+      "learning_rate": 4.891304347826087e-05,
+      "loss": 0.5037,
+      "step": 315
+    },
+    {
+      "epoch": 0.2982292637465051,
+      "grad_norm": 0.16543624419822456,
+      "learning_rate": 4.968944099378882e-05,
+      "loss": 0.4886,
+      "step": 320
+    },
+    {
+      "epoch": 0.30288909599254427,
+      "grad_norm": 0.17190813159052287,
+      "learning_rate": 4.994822229892993e-05,
+      "loss": 0.4859,
+      "step": 325
+    },
+    {
+      "epoch": 0.30754892823858343,
+      "grad_norm": 0.1444759626133901,
+      "learning_rate": 4.986192613047981e-05,
+      "loss": 0.4974,
+      "step": 330
+    },
+    {
+      "epoch": 0.31220876048462254,
+      "grad_norm": 0.16899405323995906,
+      "learning_rate": 4.977562996202969e-05,
+      "loss": 0.4949,
+      "step": 335
+    },
+    {
+      "epoch": 0.3168685927306617,
+      "grad_norm": 0.16541585685402416,
+      "learning_rate": 4.968933379357957e-05,
+      "loss": 0.5005,
+      "step": 340
+    },
+    {
+      "epoch": 0.32152842497670087,
+      "grad_norm": 0.16064447252389424,
+      "learning_rate": 4.9603037625129445e-05,
+      "loss": 0.4939,
+      "step": 345
+    },
+    {
+      "epoch": 0.32618825722274,
+      "grad_norm": 0.2030769676620295,
+      "learning_rate": 4.951674145667933e-05,
+      "loss": 0.508,
+      "step": 350
+    },
+    {
+      "epoch": 0.33084808946877914,
+      "grad_norm": 0.16222831537591864,
+      "learning_rate": 4.94304452882292e-05,
+      "loss": 0.495,
+      "step": 355
+    },
+    {
+      "epoch": 0.33550792171481825,
+      "grad_norm": 0.1492052661346749,
+      "learning_rate": 4.934414911977908e-05,
+      "loss": 0.4904,
+      "step": 360
+    },
+    {
+      "epoch": 0.3401677539608574,
+      "grad_norm": 0.15935338855159767,
+      "learning_rate": 4.9257852951328965e-05,
+      "loss": 0.4878,
+      "step": 365
+    },
+    {
+      "epoch": 0.3448275862068966,
+      "grad_norm": 0.15570525027405108,
+      "learning_rate": 4.917155678287884e-05,
+      "loss": 0.5007,
+      "step": 370
+    },
+    {
+      "epoch": 0.3494874184529357,
+      "grad_norm": 0.1567609642689268,
+      "learning_rate": 4.908526061442872e-05,
+      "loss": 0.4854,
+      "step": 375
+    },
+    {
+      "epoch": 0.35414725069897485,
+      "grad_norm": 0.16142419536236258,
+      "learning_rate": 4.89989644459786e-05,
+      "loss": 0.5015,
+      "step": 380
+    },
+    {
+      "epoch": 0.35880708294501396,
+      "grad_norm": 0.17291489492274068,
+      "learning_rate": 4.891266827752848e-05,
+      "loss": 0.5044,
+      "step": 385
+    },
+    {
+      "epoch": 0.3634669151910531,
+      "grad_norm": 0.18248668866361953,
+      "learning_rate": 4.882637210907836e-05,
+      "loss": 0.488,
+      "step": 390
+    },
+    {
+      "epoch": 0.3681267474370923,
+      "grad_norm": 0.17630594757341803,
+      "learning_rate": 4.874007594062824e-05,
+      "loss": 0.4954,
+      "step": 395
+    },
+    {
+      "epoch": 0.3727865796831314,
+      "grad_norm": 0.15330367362667366,
+      "learning_rate": 4.865377977217811e-05,
+      "loss": 0.4802,
+      "step": 400
+    },
+    {
+      "epoch": 0.37744641192917056,
+      "grad_norm": 0.15514197596294277,
+      "learning_rate": 4.8567483603728e-05,
+      "loss": 0.4989,
+      "step": 405
+    },
+    {
+      "epoch": 0.3821062441752097,
+      "grad_norm": 0.15209780228489939,
+      "learning_rate": 4.8481187435277875e-05,
+      "loss": 0.4941,
+      "step": 410
+    },
+    {
+      "epoch": 0.38676607642124883,
+      "grad_norm": 0.14305331492777879,
+      "learning_rate": 4.839489126682776e-05,
+      "loss": 0.4792,
+      "step": 415
+    },
+    {
+      "epoch": 0.391425908667288,
+      "grad_norm": 0.1481397853574671,
+      "learning_rate": 4.830859509837763e-05,
+      "loss": 0.4994,
+      "step": 420
+    },
+    {
+      "epoch": 0.3960857409133271,
+      "grad_norm": 0.14479208062471927,
+      "learning_rate": 4.822229892992751e-05,
+      "loss": 0.4977,
+      "step": 425
+    },
+    {
+      "epoch": 0.40074557315936626,
+      "grad_norm": 0.14967575520024354,
+      "learning_rate": 4.8136002761477395e-05,
+      "loss": 0.4832,
+      "step": 430
+    },
+    {
+      "epoch": 0.40540540540540543,
+      "grad_norm": 0.1525886217820535,
+      "learning_rate": 4.804970659302727e-05,
+      "loss": 0.4979,
+      "step": 435
+    },
+    {
+      "epoch": 0.41006523765144454,
+      "grad_norm": 0.1510539836108816,
+      "learning_rate": 4.796341042457715e-05,
+      "loss": 0.4932,
+      "step": 440
+    },
+    {
+      "epoch": 0.4147250698974837,
+      "grad_norm": 0.14255466931521418,
+      "learning_rate": 4.787711425612703e-05,
+      "loss": 0.4863,
+      "step": 445
+    },
+    {
+      "epoch": 0.4193849021435228,
+      "grad_norm": 0.14527150706011324,
+      "learning_rate": 4.779081808767691e-05,
+      "loss": 0.4776,
+      "step": 450
+    },
+    {
+      "epoch": 0.424044734389562,
+      "grad_norm": 0.13172030889656636,
+      "learning_rate": 4.770452191922679e-05,
+      "loss": 0.4841,
+      "step": 455
+    },
+    {
+      "epoch": 0.42870456663560114,
+      "grad_norm": 0.15405104318758273,
+      "learning_rate": 4.761822575077667e-05,
+      "loss": 0.4802,
+      "step": 460
+    },
+    {
+      "epoch": 0.43336439888164024,
+      "grad_norm": 0.14568312106835285,
+      "learning_rate": 4.753192958232654e-05,
+      "loss": 0.4762,
+      "step": 465
+    },
+    {
+      "epoch": 0.4380242311276794,
+      "grad_norm": 0.14175977685474392,
+      "learning_rate": 4.744563341387643e-05,
+      "loss": 0.4934,
+      "step": 470
+    },
+    {
+      "epoch": 0.4426840633737186,
+      "grad_norm": 0.14988327943801497,
+      "learning_rate": 4.7359337245426306e-05,
+      "loss": 0.4809,
+      "step": 475
+    },
+    {
+      "epoch": 0.4473438956197577,
+      "grad_norm": 0.14639420583818924,
+      "learning_rate": 4.7273041076976184e-05,
+      "loss": 0.4933,
+      "step": 480
+    },
+    {
+      "epoch": 0.45200372786579684,
+      "grad_norm": 0.16322206576886344,
+      "learning_rate": 4.718674490852606e-05,
+      "loss": 0.4849,
+      "step": 485
+    },
+    {
+      "epoch": 0.45666356011183595,
+      "grad_norm": 0.14137127430911123,
+      "learning_rate": 4.710044874007594e-05,
+      "loss": 0.4802,
+      "step": 490
+    },
+    {
+      "epoch": 0.4613233923578751,
+      "grad_norm": 0.16769224030133492,
+      "learning_rate": 4.7014152571625826e-05,
+      "loss": 0.4844,
+      "step": 495
+    },
+    {
+      "epoch": 0.4659832246039143,
+      "grad_norm": 0.13838134700309043,
+      "learning_rate": 4.6927856403175704e-05,
+      "loss": 0.4883,
+      "step": 500
+    },
+    {
+      "epoch": 0.4706430568499534,
+      "grad_norm": 0.13856705449643036,
+      "learning_rate": 4.684156023472558e-05,
+      "loss": 0.4741,
+      "step": 505
+    },
+    {
+      "epoch": 0.47530288909599255,
+      "grad_norm": 0.1423613301949002,
+      "learning_rate": 4.675526406627546e-05,
+      "loss": 0.4833,
+      "step": 510
+    },
+    {
+      "epoch": 0.47996272134203166,
+      "grad_norm": 0.15511304246710175,
+      "learning_rate": 4.666896789782534e-05,
+      "loss": 0.4813,
+      "step": 515
+    },
+    {
+      "epoch": 0.4846225535880708,
+      "grad_norm": 0.15335520203930875,
+      "learning_rate": 4.658267172937522e-05,
+      "loss": 0.4853,
+      "step": 520
+    },
+    {
+      "epoch": 0.48928238583411,
+      "grad_norm": 0.15779312069564705,
+      "learning_rate": 4.64963755609251e-05,
+      "loss": 0.4849,
+      "step": 525
+    },
+    {
+      "epoch": 0.4939422180801491,
+      "grad_norm": 0.16110647891199517,
+      "learning_rate": 4.641007939247497e-05,
+      "loss": 0.4813,
+      "step": 530
+    },
+    {
+      "epoch": 0.49860205032618826,
+      "grad_norm": 0.14254254783685247,
+      "learning_rate": 4.632378322402486e-05,
+      "loss": 0.4814,
+      "step": 535
+    },
+    {
+      "epoch": 0.5032618825722274,
+      "grad_norm": 0.16967666266664766,
+      "learning_rate": 4.6237487055574736e-05,
+      "loss": 0.4712,
+      "step": 540
+    },
+    {
+      "epoch": 0.5079217148182665,
+      "grad_norm": 0.15077882886548263,
+      "learning_rate": 4.6151190887124615e-05,
+      "loss": 0.477,
+      "step": 545
+    },
+    {
+      "epoch": 0.5125815470643057,
+      "grad_norm": 0.1721689867806337,
+      "learning_rate": 4.606489471867449e-05,
+      "loss": 0.479,
+      "step": 550
+    },
+    {
+      "epoch": 0.5172413793103449,
+      "grad_norm": 0.16402508300176416,
+      "learning_rate": 4.597859855022437e-05,
+      "loss": 0.4705,
+      "step": 555
+    },
+    {
+      "epoch": 0.5219012115563839,
+      "grad_norm": 0.15282835428057134,
+      "learning_rate": 4.589230238177425e-05,
+      "loss": 0.4954,
+      "step": 560
+    },
+    {
+      "epoch": 0.5265610438024231,
+      "grad_norm": 0.13305089920820062,
+      "learning_rate": 4.5806006213324134e-05,
+      "loss": 0.4703,
+      "step": 565
+    },
+    {
+      "epoch": 0.5312208760484622,
+      "grad_norm": 0.15502488989752575,
+      "learning_rate": 4.5719710044874006e-05,
+      "loss": 0.4717,
+      "step": 570
+    },
+    {
+      "epoch": 0.5358807082945014,
+      "grad_norm": 0.14215275099131902,
+      "learning_rate": 4.563341387642389e-05,
+      "loss": 0.4765,
+      "step": 575
+    },
+    {
+      "epoch": 0.5405405405405406,
+      "grad_norm": 0.13606318661955286,
+      "learning_rate": 4.554711770797377e-05,
+      "loss": 0.4773,
+      "step": 580
+    },
+    {
+      "epoch": 0.5452003727865797,
+      "grad_norm": 0.13803234590461655,
+      "learning_rate": 4.546082153952365e-05,
+      "loss": 0.4737,
+      "step": 585
+    },
+    {
+      "epoch": 0.5498602050326188,
+      "grad_norm": 0.1304023495075612,
+      "learning_rate": 4.5374525371073526e-05,
+      "loss": 0.47,
+      "step": 590
+    },
+    {
+      "epoch": 0.554520037278658,
+      "grad_norm": 0.13595610320559134,
+      "learning_rate": 4.5288229202623404e-05,
+      "loss": 0.4724,
+      "step": 595
+    },
+    {
+      "epoch": 0.5591798695246971,
+      "grad_norm": 0.14127960554484637,
+      "learning_rate": 4.520193303417328e-05,
+      "loss": 0.4723,
+      "step": 600
+    },
+    {
+      "epoch": 0.5638397017707363,
+      "grad_norm": 0.1411659700613266,
+      "learning_rate": 4.511563686572317e-05,
+      "loss": 0.4711,
+      "step": 605
+    },
+    {
+      "epoch": 0.5684995340167754,
+      "grad_norm": 0.15075307574214206,
+      "learning_rate": 4.5029340697273045e-05,
+      "loss": 0.4826,
+      "step": 610
+    },
+    {
+      "epoch": 0.5731593662628145,
+      "grad_norm": 0.13121271683698776,
+      "learning_rate": 4.4943044528822923e-05,
+      "loss": 0.4707,
+      "step": 615
+    },
+    {
+      "epoch": 0.5778191985088537,
+      "grad_norm": 0.14102440538804908,
+      "learning_rate": 4.48567483603728e-05,
+      "loss": 0.4657,
+      "step": 620
+    },
+    {
+      "epoch": 0.5824790307548928,
+      "grad_norm": 0.14071806374265922,
+      "learning_rate": 4.477045219192268e-05,
+      "loss": 0.4886,
+      "step": 625
+    },
+    {
+      "epoch": 0.587138863000932,
+      "grad_norm": 0.1426043588092053,
+      "learning_rate": 4.4684156023472565e-05,
+      "loss": 0.4804,
+      "step": 630
+    },
+    {
+      "epoch": 0.5917986952469712,
+      "grad_norm": 0.1573249110002429,
+      "learning_rate": 4.4597859855022436e-05,
+      "loss": 0.4833,
+      "step": 635
+    },
+    {
+      "epoch": 0.5964585274930102,
+      "grad_norm": 0.1381470261485967,
+      "learning_rate": 4.4511563686572315e-05,
+      "loss": 0.4748,
+      "step": 640
+    },
+    {
+      "epoch": 0.6011183597390494,
+      "grad_norm": 0.13857763778694906,
+      "learning_rate": 4.44252675181222e-05,
+      "loss": 0.4752,
+      "step": 645
+    },
+    {
+      "epoch": 0.6057781919850885,
+      "grad_norm": 0.13669044543311157,
+      "learning_rate": 4.433897134967208e-05,
+      "loss": 0.4833,
+      "step": 650
+    },
+    {
+      "epoch": 0.6104380242311277,
+      "grad_norm": 0.1373940584717527,
+      "learning_rate": 4.4252675181221956e-05,
+      "loss": 0.4807,
+      "step": 655
+    },
+    {
+      "epoch": 0.6150978564771669,
+      "grad_norm": 0.153124198142014,
+      "learning_rate": 4.4166379012771834e-05,
+      "loss": 0.4615,
+      "step": 660
+    },
+    {
+      "epoch": 0.6197576887232059,
+      "grad_norm": 0.13363428695342205,
+      "learning_rate": 4.408008284432171e-05,
+      "loss": 0.4627,
+      "step": 665
+    },
+    {
+      "epoch": 0.6244175209692451,
+      "grad_norm": 0.13741491227456037,
+      "learning_rate": 4.39937866758716e-05,
+      "loss": 0.462,
+      "step": 670
+    },
+    {
+      "epoch": 0.6290773532152842,
+      "grad_norm": 0.1315001113413596,
+      "learning_rate": 4.3907490507421476e-05,
+      "loss": 0.4703,
+      "step": 675
+    },
+    {
+      "epoch": 0.6337371854613234,
+      "grad_norm": 0.1412517524621701,
+      "learning_rate": 4.382119433897135e-05,
+      "loss": 0.473,
+      "step": 680
+    },
+    {
+      "epoch": 0.6383970177073626,
+      "grad_norm": 0.14302912715759702,
+      "learning_rate": 4.373489817052123e-05,
+      "loss": 0.4831,
+      "step": 685
+    },
+    {
+      "epoch": 0.6430568499534017,
+      "grad_norm": 0.15370048567957728,
+      "learning_rate": 4.364860200207111e-05,
+      "loss": 0.4641,
+      "step": 690
+    },
+    {
+      "epoch": 0.6477166821994408,
+      "grad_norm": 0.13975305956574144,
+      "learning_rate": 4.356230583362099e-05,
+      "loss": 0.4696,
+      "step": 695
+    },
+    {
+      "epoch": 0.65237651444548,
+      "grad_norm": 0.12797191030035393,
+      "learning_rate": 4.347600966517087e-05,
+      "loss": 0.4792,
+      "step": 700
+    },
+    {
+      "epoch": 0.6570363466915191,
+      "grad_norm": 0.1319162959342285,
+      "learning_rate": 4.3389713496720745e-05,
+      "loss": 0.4732,
+      "step": 705
+    },
+    {
+      "epoch": 0.6616961789375583,
+      "grad_norm": 0.15170714155287787,
+      "learning_rate": 4.330341732827063e-05,
+      "loss": 0.4744,
+      "step": 710
+    },
+    {
+      "epoch": 0.6663560111835974,
+      "grad_norm": 0.14121110766522413,
+      "learning_rate": 4.321712115982051e-05,
+      "loss": 0.4614,
+      "step": 715
+    },
+    {
+      "epoch": 0.6710158434296365,
+      "grad_norm": 0.1344125838779663,
+      "learning_rate": 4.3130824991370387e-05,
+      "loss": 0.4722,
+      "step": 720
+    },
+    {
+      "epoch": 0.6756756756756757,
+      "grad_norm": 0.1358154185321121,
+      "learning_rate": 4.3044528822920265e-05,
+      "loss": 0.4711,
+      "step": 725
+    },
+    {
+      "epoch": 0.6803355079217148,
+      "grad_norm": 0.14178213383027227,
+      "learning_rate": 4.295823265447014e-05,
+      "loss": 0.4654,
+      "step": 730
+    },
+    {
+      "epoch": 0.684995340167754,
+      "grad_norm": 0.13702822354309874,
+      "learning_rate": 4.287193648602002e-05,
+      "loss": 0.4707,
+      "step": 735
+    },
+    {
+      "epoch": 0.6896551724137931,
+      "grad_norm": 0.13473850778324256,
+      "learning_rate": 4.27856403175699e-05,
+      "loss": 0.4632,
+      "step": 740
+    },
+    {
+      "epoch": 0.6943150046598322,
+      "grad_norm": 0.14705160976632414,
+      "learning_rate": 4.269934414911978e-05,
+      "loss": 0.4585,
+      "step": 745
+    },
+    {
+      "epoch": 0.6989748369058714,
+      "grad_norm": 0.12494243179339545,
+      "learning_rate": 4.261304798066966e-05,
+      "loss": 0.463,
+      "step": 750
+    },
+    {
+      "epoch": 0.7036346691519105,
+      "grad_norm": 0.14760585327440534,
+      "learning_rate": 4.252675181221954e-05,
+      "loss": 0.4655,
+      "step": 755
+    },
+    {
+      "epoch": 0.7082945013979497,
+      "grad_norm": 0.13149872043153332,
+      "learning_rate": 4.244045564376942e-05,
+      "loss": 0.4704,
+      "step": 760
+    },
+    {
+      "epoch": 0.7129543336439889,
+      "grad_norm": 0.1346957538996617,
+      "learning_rate": 4.23541594753193e-05,
+      "loss": 0.4518,
+      "step": 765
+    },
+    {
+      "epoch": 0.7176141658900279,
+      "grad_norm": 0.1419479924810519,
+      "learning_rate": 4.2267863306869176e-05,
+      "loss": 0.4614,
+      "step": 770
+    },
+    {
+      "epoch": 0.7222739981360671,
+      "grad_norm": 0.13400137009346338,
+      "learning_rate": 4.2181567138419054e-05,
+      "loss": 0.4764,
+      "step": 775
+    },
+    {
+      "epoch": 0.7269338303821062,
+      "grad_norm": 0.12915763696515364,
+      "learning_rate": 4.209527096996894e-05,
+      "loss": 0.4602,
+      "step": 780
+    },
+    {
+      "epoch": 0.7315936626281454,
+      "grad_norm": 0.1408508827788197,
+      "learning_rate": 4.200897480151881e-05,
+      "loss": 0.4669,
+      "step": 785
+    },
+    {
+      "epoch": 0.7362534948741846,
+      "grad_norm": 0.13931248848561703,
+      "learning_rate": 4.1922678633068695e-05,
+      "loss": 0.4725,
+      "step": 790
+    },
+    {
+      "epoch": 0.7409133271202236,
+      "grad_norm": 0.13194611067278658,
+      "learning_rate": 4.1836382464618573e-05,
+      "loss": 0.4652,
+      "step": 795
+    },
+    {
+      "epoch": 0.7455731593662628,
+      "grad_norm": 0.13507850713769173,
+      "learning_rate": 4.175008629616845e-05,
+      "loss": 0.4631,
+      "step": 800
+    },
+    {
+      "epoch": 0.750232991612302,
+      "grad_norm": 0.13102336577361046,
+      "learning_rate": 4.166379012771833e-05,
+      "loss": 0.4624,
+      "step": 805
+    },
+    {
+      "epoch": 0.7548928238583411,
+      "grad_norm": 0.1503346188896977,
+      "learning_rate": 4.157749395926821e-05,
+      "loss": 0.4828,
+      "step": 810
+    },
+    {
+      "epoch": 0.7595526561043803,
+      "grad_norm": 0.13755304015787692,
+      "learning_rate": 4.1491197790818086e-05,
+      "loss": 0.4542,
+      "step": 815
+    },
+    {
+      "epoch": 0.7642124883504194,
+      "grad_norm": 0.14611114221324292,
+      "learning_rate": 4.140490162236797e-05,
+      "loss": 0.4717,
+      "step": 820
+    },
+    {
+      "epoch": 0.7688723205964585,
+      "grad_norm": 0.14547388184706703,
+      "learning_rate": 4.131860545391785e-05,
+      "loss": 0.4638,
+      "step": 825
+    },
+    {
+      "epoch": 0.7735321528424977,
+      "grad_norm": 0.1340279606082225,
+      "learning_rate": 4.123230928546773e-05,
+      "loss": 0.4576,
+      "step": 830
+    },
+    {
+      "epoch": 0.7781919850885368,
+      "grad_norm": 0.13111567567487978,
+      "learning_rate": 4.1146013117017606e-05,
+      "loss": 0.4636,
+      "step": 835
+    },
+    {
+      "epoch": 0.782851817334576,
+      "grad_norm": 0.14364924103023152,
+      "learning_rate": 4.1059716948567484e-05,
+      "loss": 0.4697,
+      "step": 840
+    },
+    {
+      "epoch": 0.7875116495806151,
+      "grad_norm": 0.14256722121465673,
+      "learning_rate": 4.097342078011737e-05,
+      "loss": 0.4784,
+      "step": 845
+    },
+    {
+      "epoch": 0.7921714818266542,
+      "grad_norm": 0.13862377678033566,
+      "learning_rate": 4.088712461166724e-05,
+      "loss": 0.4658,
+      "step": 850
+    },
+    {
+      "epoch": 0.7968313140726934,
+      "grad_norm": 0.13916453963168612,
+      "learning_rate": 4.080082844321712e-05,
+      "loss": 0.4681,
+      "step": 855
+    },
+    {
+      "epoch": 0.8014911463187325,
+      "grad_norm": 0.13127531193262942,
+      "learning_rate": 4.0714532274767004e-05,
+      "loss": 0.4769,
+      "step": 860
+    },
+    {
+      "epoch": 0.8061509785647717,
+      "grad_norm": 0.147359189100914,
+      "learning_rate": 4.062823610631688e-05,
+      "loss": 0.455,
+      "step": 865
+    },
+    {
+      "epoch": 0.8108108108108109,
+      "grad_norm": 0.14538857342581543,
+      "learning_rate": 4.054193993786676e-05,
+      "loss": 0.4718,
+      "step": 870
+    },
+    {
+      "epoch": 0.8154706430568499,
+      "grad_norm": 0.14157037473898393,
+      "learning_rate": 4.045564376941664e-05,
+      "loss": 0.4666,
+      "step": 875
+    },
+    {
+      "epoch": 0.8201304753028891,
+      "grad_norm": 0.13587421068081873,
+      "learning_rate": 4.036934760096652e-05,
+      "loss": 0.4565,
+      "step": 880
+    },
+    {
+      "epoch": 0.8247903075489282,
+      "grad_norm": 0.12626622675823906,
+      "learning_rate": 4.02830514325164e-05,
+      "loss": 0.4722,
+      "step": 885
+    },
+    {
+      "epoch": 0.8294501397949674,
+      "grad_norm": 0.1310221391424837,
+      "learning_rate": 4.019675526406628e-05,
+      "loss": 0.4584,
+      "step": 890
+    },
+    {
+      "epoch": 0.8341099720410066,
+      "grad_norm": 0.14031832208645464,
+      "learning_rate": 4.011045909561615e-05,
+      "loss": 0.4798,
+      "step": 895
+    },
+    {
+      "epoch": 0.8387698042870456,
+      "grad_norm": 0.12351330637273926,
+      "learning_rate": 4.0024162927166037e-05,
+      "loss": 0.4608,
+      "step": 900
+    },
+    {
+      "epoch": 0.8434296365330848,
+      "grad_norm": 0.12321052762320074,
+      "learning_rate": 3.9937866758715915e-05,
+      "loss": 0.4618,
+      "step": 905
+    },
+    {
+      "epoch": 0.848089468779124,
+      "grad_norm": 0.12779013441419323,
+      "learning_rate": 3.98515705902658e-05,
+      "loss": 0.4677,
+      "step": 910
+    },
+    {
+      "epoch": 0.8527493010251631,
+      "grad_norm": 0.13230130347945906,
+      "learning_rate": 3.976527442181567e-05,
+      "loss": 0.4606,
+      "step": 915
+    },
+    {
+      "epoch": 0.8574091332712023,
+      "grad_norm": 0.14276084370535033,
+      "learning_rate": 3.967897825336555e-05,
+      "loss": 0.4685,
+      "step": 920
+    },
+    {
+      "epoch": 0.8620689655172413,
+      "grad_norm": 0.140673576202374,
+      "learning_rate": 3.9592682084915434e-05,
+      "loss": 0.4543,
+      "step": 925
+    },
+    {
+      "epoch": 0.8667287977632805,
+      "grad_norm": 0.14600932835513855,
+      "learning_rate": 3.950638591646531e-05,
+      "loss": 0.4508,
+      "step": 930
+    },
+    {
+      "epoch": 0.8713886300093197,
+      "grad_norm": 0.13689713890744198,
+      "learning_rate": 3.942008974801519e-05,
+      "loss": 0.4643,
+      "step": 935
+    },
+    {
+      "epoch": 0.8760484622553588,
+      "grad_norm": 0.1377121416873878,
+      "learning_rate": 3.933379357956507e-05,
+      "loss": 0.4602,
+      "step": 940
+    },
+    {
+      "epoch": 0.880708294501398,
+      "grad_norm": 0.1299717200243323,
+      "learning_rate": 3.924749741111495e-05,
+      "loss": 0.4643,
+      "step": 945
+    },
+    {
+      "epoch": 0.8853681267474371,
+      "grad_norm": 0.1366198120846624,
+      "learning_rate": 3.916120124266483e-05,
+      "loss": 0.4725,
+      "step": 950
+    },
+    {
+      "epoch": 0.8900279589934762,
+      "grad_norm": 0.14916228680639607,
+      "learning_rate": 3.9074905074214704e-05,
+      "loss": 0.4706,
+      "step": 955
+    },
+    {
+      "epoch": 0.8946877912395154,
+      "grad_norm": 0.12395362629247658,
+      "learning_rate": 3.898860890576458e-05,
+      "loss": 0.4642,
+      "step": 960
+    },
+    {
+      "epoch": 0.8993476234855545,
+      "grad_norm": 0.1515261241641573,
+      "learning_rate": 3.890231273731447e-05,
+      "loss": 0.4565,
+      "step": 965
+    },
+    {
+      "epoch": 0.9040074557315937,
+      "grad_norm": 0.13077977431266727,
+      "learning_rate": 3.8816016568864345e-05,
+      "loss": 0.4534,
+      "step": 970
+    },
+    {
+      "epoch": 0.9086672879776329,
+      "grad_norm": 0.1389256848294849,
+      "learning_rate": 3.8729720400414224e-05,
+      "loss": 0.4747,
+      "step": 975
+    },
+    {
+      "epoch": 0.9133271202236719,
+      "grad_norm": 0.14787518515320647,
+      "learning_rate": 3.86434242319641e-05,
+      "loss": 0.4602,
+      "step": 980
+    },
+    {
+      "epoch": 0.9179869524697111,
+      "grad_norm": 0.1333404321295863,
+      "learning_rate": 3.855712806351398e-05,
+      "loss": 0.4617,
+      "step": 985
+    },
+    {
+      "epoch": 0.9226467847157502,
+      "grad_norm": 0.12814398155291287,
+      "learning_rate": 3.8470831895063865e-05,
+      "loss": 0.4591,
+      "step": 990
+    },
+    {
+      "epoch": 0.9273066169617894,
+      "grad_norm": 0.12909591290157696,
+      "learning_rate": 3.838453572661374e-05,
+      "loss": 0.4546,
+      "step": 995
+    },
+    {
+      "epoch": 0.9319664492078286,
+      "grad_norm": 0.1272598419716883,
+      "learning_rate": 3.8298239558163615e-05,
+      "loss": 0.4576,
+      "step": 1000
+    },
+    {
+      "epoch": 0.9366262814538676,
+      "grad_norm": 0.14650718436432028,
+      "learning_rate": 3.82119433897135e-05,
+      "loss": 0.4635,
+      "step": 1005
+    },
+    {
+      "epoch": 0.9412861136999068,
+      "grad_norm": 0.1410323274579289,
+      "learning_rate": 3.812564722126338e-05,
+      "loss": 0.4567,
+      "step": 1010
+    },
+    {
+      "epoch": 0.9459459459459459,
+      "grad_norm": 0.14510135338129224,
+      "learning_rate": 3.8039351052813256e-05,
+      "loss": 0.4658,
+      "step": 1015
+    },
+    {
+      "epoch": 0.9506057781919851,
+      "grad_norm": 0.1272462824199875,
+      "learning_rate": 3.7953054884363134e-05,
+      "loss": 0.4573,
+      "step": 1020
+    },
+    {
+      "epoch": 0.9552656104380243,
+      "grad_norm": 0.13611323589606997,
+      "learning_rate": 3.786675871591301e-05,
+      "loss": 0.4632,
+      "step": 1025
+    },
+    {
+      "epoch": 0.9599254426840633,
+      "grad_norm": 0.13327903041914013,
+      "learning_rate": 3.77804625474629e-05,
+      "loss": 0.4671,
+      "step": 1030
+    },
+    {
+      "epoch": 0.9645852749301025,
+      "grad_norm": 0.1290051375603818,
+      "learning_rate": 3.7694166379012776e-05,
+      "loss": 0.4594,
+      "step": 1035
+    },
+    {
+      "epoch": 0.9692451071761417,
+      "grad_norm": 0.11740331871486634,
+      "learning_rate": 3.7607870210562654e-05,
+      "loss": 0.4577,
+      "step": 1040
+    },
+    {
+      "epoch": 0.9739049394221808,
+      "grad_norm": 0.12067266089150469,
+      "learning_rate": 3.752157404211253e-05,
+      "loss": 0.4477,
+      "step": 1045
+    },
+    {
+      "epoch": 0.97856477166822,
+      "grad_norm": 0.12998833529275453,
+      "learning_rate": 3.743527787366241e-05,
+      "loss": 0.4811,
+      "step": 1050
+    },
+    {
+      "epoch": 0.983224603914259,
+      "grad_norm": 0.1373423483010429,
+      "learning_rate": 3.734898170521229e-05,
+      "loss": 0.4571,
+      "step": 1055
+    },
+    {
+      "epoch": 0.9878844361602982,
+      "grad_norm": 0.13051928524237108,
+      "learning_rate": 3.7262685536762174e-05,
+      "loss": 0.4555,
+      "step": 1060
+    },
+    {
+      "epoch": 0.9925442684063374,
+      "grad_norm": 0.13320028314655158,
+      "learning_rate": 3.7176389368312045e-05,
+      "loss": 0.4592,
+      "step": 1065
+    },
+    {
+      "epoch": 0.9972041006523765,
+      "grad_norm": 0.12664355828737522,
+      "learning_rate": 3.709009319986193e-05,
+      "loss": 0.4612,
+      "step": 1070
+    },
+    {
+      "epoch": 1.0018639328984156,
+      "grad_norm": 0.13316313381309125,
+      "learning_rate": 3.700379703141181e-05,
+      "loss": 0.4331,
+      "step": 1075
+    },
+    {
+      "epoch": 1.0065237651444547,
+      "grad_norm": 0.13036712991919813,
+      "learning_rate": 3.6917500862961687e-05,
+      "loss": 0.4294,
+      "step": 1080
+    },
+    {
+      "epoch": 1.011183597390494,
+      "grad_norm": 0.14200705698988264,
+      "learning_rate": 3.6831204694511565e-05,
+      "loss": 0.4347,
+      "step": 1085
+    },
+    {
+      "epoch": 1.015843429636533,
+      "grad_norm": 0.13227179905192146,
+      "learning_rate": 3.674490852606144e-05,
+      "loss": 0.4484,
+      "step": 1090
+    },
+    {
+      "epoch": 1.0205032618825722,
+      "grad_norm": 0.13701355595945078,
+      "learning_rate": 3.665861235761132e-05,
+      "loss": 0.4299,
+      "step": 1095
+    },
+    {
+      "epoch": 1.0251630941286114,
+      "grad_norm": 0.14751805432207601,
+      "learning_rate": 3.6572316189161206e-05,
+      "loss": 0.4187,
+      "step": 1100
+    },
+    {
+      "epoch": 1.0298229263746506,
+      "grad_norm": 0.13995469792085918,
+      "learning_rate": 3.6486020020711085e-05,
+      "loss": 0.4279,
+      "step": 1105
+    },
+    {
+      "epoch": 1.0344827586206897,
+      "grad_norm": 0.12802192835956913,
+      "learning_rate": 3.639972385226096e-05,
+      "loss": 0.4254,
+      "step": 1110
+    },
+    {
+      "epoch": 1.0391425908667289,
+      "grad_norm": 0.13611203034280261,
+      "learning_rate": 3.631342768381084e-05,
+      "loss": 0.4292,
+      "step": 1115
+    },
+    {
+      "epoch": 1.0438024231127678,
+      "grad_norm": 0.14957119627812435,
+      "learning_rate": 3.622713151536072e-05,
+      "loss": 0.4284,
+      "step": 1120
+    },
+    {
+      "epoch": 1.048462255358807,
+      "grad_norm": 0.12750160118155685,
+      "learning_rate": 3.6140835346910604e-05,
+      "loss": 0.4314,
+      "step": 1125
+    },
+    {
+      "epoch": 1.0531220876048462,
+      "grad_norm": 0.12688794669630618,
+      "learning_rate": 3.6054539178460476e-05,
+      "loss": 0.4299,
+      "step": 1130
+    },
+    {
+      "epoch": 1.0577819198508853,
+      "grad_norm": 0.12940262434679464,
+      "learning_rate": 3.5968243010010354e-05,
+      "loss": 0.4449,
+      "step": 1135
+    },
+    {
+      "epoch": 1.0624417520969245,
+      "grad_norm": 0.1344591647473135,
+      "learning_rate": 3.588194684156024e-05,
+      "loss": 0.4376,
+      "step": 1140
+    },
+    {
+      "epoch": 1.0671015843429636,
+      "grad_norm": 0.14018335523954348,
+      "learning_rate": 3.579565067311012e-05,
+      "loss": 0.4272,
+      "step": 1145
+    },
+    {
+      "epoch": 1.0717614165890028,
+      "grad_norm": 0.13197152436809545,
+      "learning_rate": 3.5709354504659995e-05,
+      "loss": 0.4238,
+      "step": 1150
+    },
+    {
+      "epoch": 1.076421248835042,
+      "grad_norm": 0.13997134504229464,
+      "learning_rate": 3.5623058336209874e-05,
+      "loss": 0.4325,
+      "step": 1155
+    },
+    {
+      "epoch": 1.0810810810810811,
+      "grad_norm": 0.1243946417467569,
+      "learning_rate": 3.553676216775975e-05,
+      "loss": 0.4266,
+      "step": 1160
+    },
+    {
+      "epoch": 1.0857409133271203,
+      "grad_norm": 0.13695982009093383,
+      "learning_rate": 3.545046599930964e-05,
+      "loss": 0.4365,
+      "step": 1165
+    },
+    {
+      "epoch": 1.0904007455731595,
+      "grad_norm": 0.13288687615455663,
+      "learning_rate": 3.536416983085951e-05,
+      "loss": 0.4434,
+      "step": 1170
+    },
+    {
+      "epoch": 1.0950605778191984,
+      "grad_norm": 0.13958380148520386,
+      "learning_rate": 3.5277873662409386e-05,
+      "loss": 0.4268,
+      "step": 1175
+    },
+    {
+      "epoch": 1.0997204100652376,
+      "grad_norm": 0.1325057320544698,
+      "learning_rate": 3.519157749395927e-05,
+      "loss": 0.4326,
+      "step": 1180
+    },
+    {
+      "epoch": 1.1043802423112767,
+      "grad_norm": 0.14287603414990682,
+      "learning_rate": 3.510528132550915e-05,
+      "loss": 0.4301,
+      "step": 1185
+    },
+    {
+      "epoch": 1.109040074557316,
+      "grad_norm": 0.1270711340175025,
+      "learning_rate": 3.501898515705903e-05,
+      "loss": 0.432,
+      "step": 1190
+    },
+    {
+      "epoch": 1.113699906803355,
+      "grad_norm": 0.12982857170178297,
+      "learning_rate": 3.4932688988608906e-05,
+      "loss": 0.4415,
+      "step": 1195
+    },
+    {
+      "epoch": 1.1183597390493942,
+      "grad_norm": 0.12844627972390424,
+      "learning_rate": 3.4846392820158784e-05,
+      "loss": 0.4306,
+      "step": 1200
+    },
+    {
+      "epoch": 1.1230195712954334,
+      "grad_norm": 0.12922294210960988,
+      "learning_rate": 3.476009665170867e-05,
+      "loss": 0.4294,
+      "step": 1205
+    },
+    {
+      "epoch": 1.1276794035414726,
+      "grad_norm": 0.1352975404490548,
+      "learning_rate": 3.467380048325855e-05,
+      "loss": 0.4261,
+      "step": 1210
+    },
+    {
+      "epoch": 1.1323392357875117,
+      "grad_norm": 0.13189191398268402,
+      "learning_rate": 3.458750431480842e-05,
+      "loss": 0.4357,
+      "step": 1215
+    },
+    {
+      "epoch": 1.1369990680335509,
+      "grad_norm": 0.13483385328571226,
+      "learning_rate": 3.4501208146358304e-05,
+      "loss": 0.4324,
+      "step": 1220
+    },
+    {
+      "epoch": 1.14165890027959,
+      "grad_norm": 0.12703977619657308,
+      "learning_rate": 3.441491197790818e-05,
+      "loss": 0.4308,
+      "step": 1225
+    },
+    {
+      "epoch": 1.146318732525629,
+      "grad_norm": 0.13021754329644358,
+      "learning_rate": 3.432861580945806e-05,
+      "loss": 0.4348,
+      "step": 1230
+    },
+    {
+      "epoch": 1.1509785647716682,
+      "grad_norm": 0.12630065096512738,
+      "learning_rate": 3.424231964100794e-05,
+      "loss": 0.4319,
+      "step": 1235
+    },
+    {
+      "epoch": 1.1556383970177073,
+      "grad_norm": 0.13231707856789027,
+      "learning_rate": 3.415602347255782e-05,
+      "loss": 0.4299,
+      "step": 1240
+    },
+    {
+      "epoch": 1.1602982292637465,
+      "grad_norm": 0.11994267531074697,
+      "learning_rate": 3.40697273041077e-05,
+      "loss": 0.4397,
+      "step": 1245
+    },
+    {
+      "epoch": 1.1649580615097856,
+      "grad_norm": 0.12151938834329092,
+      "learning_rate": 3.398343113565758e-05,
+      "loss": 0.4297,
+      "step": 1250
+    },
+    {
+      "epoch": 1.1696178937558248,
+      "grad_norm": 0.13529745087876494,
+      "learning_rate": 3.389713496720746e-05,
+      "loss": 0.4082,
+      "step": 1255
+    },
+    {
+      "epoch": 1.174277726001864,
+      "grad_norm": 0.13107210131065658,
+      "learning_rate": 3.381083879875734e-05,
+      "loss": 0.4281,
+      "step": 1260
+    },
+    {
+      "epoch": 1.1789375582479031,
+      "grad_norm": 0.12898473319316642,
+      "learning_rate": 3.3724542630307215e-05,
+      "loss": 0.4222,
+      "step": 1265
+    },
+    {
+      "epoch": 1.1835973904939423,
+      "grad_norm": 0.12054941758498935,
+      "learning_rate": 3.363824646185709e-05,
+      "loss": 0.4378,
+      "step": 1270
+    },
+    {
+      "epoch": 1.1882572227399812,
+      "grad_norm": 0.12459595748571121,
+      "learning_rate": 3.355195029340698e-05,
+      "loss": 0.4266,
+      "step": 1275
+    },
+    {
+      "epoch": 1.1929170549860204,
+      "grad_norm": 0.12885333056347273,
+      "learning_rate": 3.346565412495685e-05,
+      "loss": 0.4408,
+      "step": 1280
+    },
+    {
+      "epoch": 1.1975768872320596,
+      "grad_norm": 0.1345136282623542,
+      "learning_rate": 3.3379357956506735e-05,
+      "loss": 0.4397,
+      "step": 1285
+    },
+    {
+      "epoch": 1.2022367194780987,
+      "grad_norm": 0.12924598426792455,
+      "learning_rate": 3.329306178805661e-05,
+      "loss": 0.4229,
+      "step": 1290
+    },
+    {
+      "epoch": 1.206896551724138,
+      "grad_norm": 0.13303044283548981,
+      "learning_rate": 3.320676561960649e-05,
+      "loss": 0.4307,
+      "step": 1295
+    },
+    {
+      "epoch": 1.211556383970177,
+      "grad_norm": 0.12992512457217237,
+      "learning_rate": 3.312046945115637e-05,
+      "loss": 0.4371,
+      "step": 1300
+    },
+    {
+      "epoch": 1.2162162162162162,
+      "grad_norm": 0.13102164040507241,
+      "learning_rate": 3.303417328270625e-05,
+      "loss": 0.4345,
+      "step": 1305
+    },
+    {
+      "epoch": 1.2208760484622554,
+      "grad_norm": 0.13668744404898098,
+      "learning_rate": 3.2947877114256126e-05,
+      "loss": 0.4228,
+      "step": 1310
+    },
+    {
+      "epoch": 1.2255358807082946,
+      "grad_norm": 0.13766259918268445,
+      "learning_rate": 3.286158094580601e-05,
+      "loss": 0.4467,
+      "step": 1315
+    },
+    {
+      "epoch": 1.2301957129543337,
+      "grad_norm": 0.13811874768466145,
+      "learning_rate": 3.277528477735589e-05,
+      "loss": 0.4275,
+      "step": 1320
+    },
+    {
+      "epoch": 1.2348555452003729,
+      "grad_norm": 0.12526566487900087,
+      "learning_rate": 3.268898860890577e-05,
+      "loss": 0.4398,
+      "step": 1325
+    },
+    {
+      "epoch": 1.2395153774464118,
+      "grad_norm": 0.12659661692104035,
+      "learning_rate": 3.2602692440455645e-05,
+      "loss": 0.4281,
+      "step": 1330
+    },
+    {
+      "epoch": 1.244175209692451,
+      "grad_norm": 0.1284723173688922,
+      "learning_rate": 3.2516396272005524e-05,
+      "loss": 0.4259,
+      "step": 1335
+    },
+    {
+      "epoch": 1.2488350419384902,
+      "grad_norm": 0.1367455450805604,
+      "learning_rate": 3.243010010355541e-05,
+      "loss": 0.4329,
+      "step": 1340
+    },
+    {
+      "epoch": 1.2534948741845293,
+      "grad_norm": 0.12448257731756714,
+      "learning_rate": 3.234380393510528e-05,
+      "loss": 0.417,
+      "step": 1345
+    },
+    {
+      "epoch": 1.2581547064305685,
+      "grad_norm": 0.13076421796705795,
+      "learning_rate": 3.225750776665516e-05,
+      "loss": 0.4287,
+      "step": 1350
+    },
+    {
+      "epoch": 1.2628145386766076,
+      "grad_norm": 0.13344972221220325,
+      "learning_rate": 3.217121159820504e-05,
+      "loss": 0.4408,
+      "step": 1355
+    },
+    {
+      "epoch": 1.2674743709226468,
+      "grad_norm": 0.1311687720347414,
+      "learning_rate": 3.208491542975492e-05,
+      "loss": 0.4277,
+      "step": 1360
+    },
+    {
+      "epoch": 1.272134203168686,
+      "grad_norm": 0.1265111068231496,
+      "learning_rate": 3.19986192613048e-05,
+      "loss": 0.4282,
+      "step": 1365
+    },
+    {
+      "epoch": 1.2767940354147251,
+      "grad_norm": 0.13478264370046403,
+      "learning_rate": 3.191232309285468e-05,
+      "loss": 0.4338,
+      "step": 1370
+    },
+    {
+      "epoch": 1.281453867660764,
+      "grad_norm": 0.12087525381740438,
+      "learning_rate": 3.1826026924404556e-05,
+      "loss": 0.4204,
+      "step": 1375
+    },
+    {
+      "epoch": 1.2861136999068035,
+      "grad_norm": 0.13711344106582482,
+      "learning_rate": 3.173973075595444e-05,
+      "loss": 0.4361,
+      "step": 1380
+    },
+    {
+      "epoch": 1.2907735321528424,
+      "grad_norm": 0.13032304039963954,
+      "learning_rate": 3.165343458750431e-05,
+      "loss": 0.4383,
+      "step": 1385
+    },
+    {
+      "epoch": 1.2954333643988816,
+      "grad_norm": 0.133835937985698,
+      "learning_rate": 3.156713841905419e-05,
+      "loss": 0.4271,
+      "step": 1390
+    },
+    {
+      "epoch": 1.3000931966449207,
+      "grad_norm": 0.1285832322421832,
+      "learning_rate": 3.1480842250604076e-05,
+      "loss": 0.4329,
+      "step": 1395
+    },
+    {
+      "epoch": 1.30475302889096,
+      "grad_norm": 0.13025265319524737,
+      "learning_rate": 3.1394546082153954e-05,
+      "loss": 0.4263,
+      "step": 1400
+    },
+    {
+      "epoch": 1.309412861136999,
+      "grad_norm": 0.13755284902351525,
+      "learning_rate": 3.130824991370383e-05,
+      "loss": 0.4377,
+      "step": 1405
+    },
+    {
+      "epoch": 1.3140726933830382,
+      "grad_norm": 0.1268628990685633,
+      "learning_rate": 3.122195374525371e-05,
+      "loss": 0.4258,
+      "step": 1410
+    },
+    {
+      "epoch": 1.3187325256290774,
+      "grad_norm": 0.12395149080398458,
+      "learning_rate": 3.113565757680359e-05,
+      "loss": 0.4307,
+      "step": 1415
+    },
+    {
+      "epoch": 1.3233923578751166,
+      "grad_norm": 0.12137312622130847,
+      "learning_rate": 3.1049361408353474e-05,
+      "loss": 0.4157,
+      "step": 1420
+    },
+    {
+      "epoch": 1.3280521901211557,
+      "grad_norm": 0.12586583333199863,
+      "learning_rate": 3.096306523990335e-05,
+      "loss": 0.4274,
+      "step": 1425
+    },
+    {
+      "epoch": 1.3327120223671947,
+      "grad_norm": 0.13110006924668216,
+      "learning_rate": 3.0876769071453223e-05,
+      "loss": 0.4343,
+      "step": 1430
+    },
+    {
+      "epoch": 1.337371854613234,
+      "grad_norm": 0.12490509732559753,
+      "learning_rate": 3.079047290300311e-05,
+      "loss": 0.4375,
+      "step": 1435
+    },
+    {
+      "epoch": 1.342031686859273,
+      "grad_norm": 0.13725018271926348,
+      "learning_rate": 3.070417673455299e-05,
+      "loss": 0.4303,
+      "step": 1440
+    },
+    {
+      "epoch": 1.3466915191053122,
+      "grad_norm": 0.11519424778567773,
+      "learning_rate": 3.061788056610287e-05,
+      "loss": 0.4362,
+      "step": 1445
+    },
+    {
+      "epoch": 1.3513513513513513,
+      "grad_norm": 0.13793363658416938,
+      "learning_rate": 3.053158439765274e-05,
+      "loss": 0.4247,
+      "step": 1450
+    },
+    {
+      "epoch": 1.3560111835973905,
+      "grad_norm": 0.12378958839640934,
+      "learning_rate": 3.0445288229202625e-05,
+      "loss": 0.4314,
+      "step": 1455
+    },
+    {
+      "epoch": 1.3606710158434296,
+      "grad_norm": 0.12798948350845976,
+      "learning_rate": 3.0358992060752506e-05,
+      "loss": 0.4373,
+      "step": 1460
+    },
+    {
+      "epoch": 1.3653308480894688,
+      "grad_norm": 0.1388233268931078,
+      "learning_rate": 3.027269589230238e-05,
+      "loss": 0.4342,
+      "step": 1465
+    },
+    {
+      "epoch": 1.369990680335508,
+      "grad_norm": 0.12530110969734215,
+      "learning_rate": 3.018639972385226e-05,
+      "loss": 0.4277,
+      "step": 1470
+    },
+    {
+      "epoch": 1.3746505125815471,
+      "grad_norm": 0.1242292789970513,
+      "learning_rate": 3.010010355540214e-05,
+      "loss": 0.4261,
+      "step": 1475
+    },
+    {
+      "epoch": 1.3793103448275863,
+      "grad_norm": 0.11702700114038071,
+      "learning_rate": 3.001380738695202e-05,
+      "loss": 0.4118,
+      "step": 1480
+    },
+    {
+      "epoch": 1.3839701770736252,
+      "grad_norm": 0.1215598841563416,
+      "learning_rate": 2.99275112185019e-05,
+      "loss": 0.4294,
+      "step": 1485
+    },
+    {
+      "epoch": 1.3886300093196646,
+      "grad_norm": 0.1294068318266713,
+      "learning_rate": 2.984121505005178e-05,
+      "loss": 0.4355,
+      "step": 1490
+    },
+    {
+      "epoch": 1.3932898415657036,
+      "grad_norm": 0.12670304691268644,
+      "learning_rate": 2.9754918881601657e-05,
+      "loss": 0.4443,
+      "step": 1495
+    },
+    {
+      "epoch": 1.3979496738117427,
+      "grad_norm": 0.13938946380677994,
+      "learning_rate": 2.966862271315154e-05,
+      "loss": 0.4434,
+      "step": 1500
+    },
+    {
+      "epoch": 1.402609506057782,
+      "grad_norm": 0.1332196330307831,
+      "learning_rate": 2.9582326544701417e-05,
+      "loss": 0.4358,
+      "step": 1505
+    },
+    {
+      "epoch": 1.407269338303821,
+      "grad_norm": 0.12496498045573066,
+      "learning_rate": 2.9496030376251292e-05,
+      "loss": 0.4273,
+      "step": 1510
+    },
+    {
+      "epoch": 1.4119291705498602,
+      "grad_norm": 0.12738125354735483,
+      "learning_rate": 2.9409734207801177e-05,
+      "loss": 0.4314,
+      "step": 1515
+    },
+    {
+      "epoch": 1.4165890027958994,
+      "grad_norm": 0.13741372664747528,
+      "learning_rate": 2.9323438039351052e-05,
+      "loss": 0.4292,
+      "step": 1520
+    },
+    {
+      "epoch": 1.4212488350419386,
+      "grad_norm": 0.12674441072920845,
+      "learning_rate": 2.9237141870900937e-05,
+      "loss": 0.4283,
+      "step": 1525
+    },
+    {
+      "epoch": 1.4259086672879777,
+      "grad_norm": 0.12537380837723255,
+      "learning_rate": 2.9150845702450812e-05,
+      "loss": 0.4262,
+      "step": 1530
+    },
+    {
+      "epoch": 1.4305684995340169,
+      "grad_norm": 0.13009878858523846,
+      "learning_rate": 2.906454953400069e-05,
+      "loss": 0.4315,
+      "step": 1535
+    },
+    {
+      "epoch": 1.4352283317800558,
+      "grad_norm": 0.1347245621272462,
+      "learning_rate": 2.897825336555057e-05,
+      "loss": 0.4257,
+      "step": 1540
+    },
+    {
+      "epoch": 1.439888164026095,
+      "grad_norm": 0.12818624522340144,
+      "learning_rate": 2.889195719710045e-05,
+      "loss": 0.4275,
+      "step": 1545
+    },
+    {
+      "epoch": 1.4445479962721341,
+      "grad_norm": 0.1257189645154172,
+      "learning_rate": 2.8805661028650328e-05,
+      "loss": 0.4224,
+      "step": 1550
+    },
+    {
+      "epoch": 1.4492078285181733,
+      "grad_norm": 0.13324646992192718,
+      "learning_rate": 2.871936486020021e-05,
+      "loss": 0.4223,
+      "step": 1555
+    },
+    {
+      "epoch": 1.4538676607642125,
+      "grad_norm": 0.13046918868178478,
+      "learning_rate": 2.8633068691750088e-05,
+      "loss": 0.4226,
+      "step": 1560
+    },
+    {
+      "epoch": 1.4585274930102516,
+      "grad_norm": 0.1350884416121187,
+      "learning_rate": 2.854677252329997e-05,
+      "loss": 0.4251,
+      "step": 1565
+    },
+    {
+      "epoch": 1.4631873252562908,
+      "grad_norm": 0.12174153738821432,
+      "learning_rate": 2.8460476354849848e-05,
+      "loss": 0.4291,
+      "step": 1570
+    },
+    {
+      "epoch": 1.46784715750233,
+      "grad_norm": 0.11888798404722918,
+      "learning_rate": 2.8374180186399723e-05,
+      "loss": 0.4347,
+      "step": 1575
+    },
+    {
+      "epoch": 1.4725069897483691,
+      "grad_norm": 0.12286851119996506,
+      "learning_rate": 2.8287884017949608e-05,
+      "loss": 0.4319,
+      "step": 1580
+    },
+    {
+      "epoch": 1.477166821994408,
+      "grad_norm": 0.12722173293990185,
+      "learning_rate": 2.8201587849499482e-05,
+      "loss": 0.4293,
+      "step": 1585
+    },
+    {
+      "epoch": 1.4818266542404475,
+      "grad_norm": 0.12224161716127686,
+      "learning_rate": 2.811529168104936e-05,
+      "loss": 0.4392,
+      "step": 1590
+    },
+    {
+      "epoch": 1.4864864864864864,
+      "grad_norm": 0.12445271747167126,
+      "learning_rate": 2.8028995512599242e-05,
+      "loss": 0.4299,
+      "step": 1595
+    },
+    {
+      "epoch": 1.4911463187325256,
+      "grad_norm": 0.1253934708979439,
+      "learning_rate": 2.794269934414912e-05,
+      "loss": 0.4301,
+      "step": 1600
+    },
+    {
+      "epoch": 1.4958061509785647,
+      "grad_norm": 0.12839174521893781,
+      "learning_rate": 2.7856403175699002e-05,
+      "loss": 0.4374,
+      "step": 1605
+    },
+    {
+      "epoch": 1.500465983224604,
+      "grad_norm": 0.13294154965075383,
+      "learning_rate": 2.777010700724888e-05,
+      "loss": 0.4328,
+      "step": 1610
+    },
+    {
+      "epoch": 1.505125815470643,
+      "grad_norm": 0.12533773887658659,
+      "learning_rate": 2.768381083879876e-05,
+      "loss": 0.4266,
+      "step": 1615
+    },
+    {
+      "epoch": 1.5097856477166822,
+      "grad_norm": 0.12175895097429791,
+      "learning_rate": 2.759751467034864e-05,
+      "loss": 0.4272,
+      "step": 1620
+    },
+    {
+      "epoch": 1.5144454799627214,
+      "grad_norm": 0.11801816291060407,
+      "learning_rate": 2.751121850189852e-05,
+      "loss": 0.4155,
+      "step": 1625
+    },
+    {
+      "epoch": 1.5191053122087603,
+      "grad_norm": 0.12675847183671907,
+      "learning_rate": 2.7424922333448393e-05,
+      "loss": 0.4155,
+      "step": 1630
+    },
+    {
+      "epoch": 1.5237651444547997,
+      "grad_norm": 0.11624269893426281,
+      "learning_rate": 2.7338626164998278e-05,
+      "loss": 0.4188,
+      "step": 1635
+    },
+    {
+      "epoch": 1.5284249767008387,
+      "grad_norm": 0.13690694094251268,
+      "learning_rate": 2.7252329996548153e-05,
+      "loss": 0.4228,
+      "step": 1640
+    },
+    {
+      "epoch": 1.533084808946878,
+      "grad_norm": 0.13138759891161272,
+      "learning_rate": 2.7166033828098038e-05,
+      "loss": 0.4395,
+      "step": 1645
+    },
+    {
+      "epoch": 1.537744641192917,
+      "grad_norm": 0.13275814073432596,
+      "learning_rate": 2.7079737659647913e-05,
+      "loss": 0.4314,
+      "step": 1650
+    },
+    {
+      "epoch": 1.5424044734389561,
+      "grad_norm": 0.1226189725133784,
+      "learning_rate": 2.699344149119779e-05,
+      "loss": 0.4362,
+      "step": 1655
+    },
+    {
+      "epoch": 1.5470643056849953,
+      "grad_norm": 0.13765222101975425,
+      "learning_rate": 2.6907145322747673e-05,
+      "loss": 0.4296,
+      "step": 1660
+    },
+    {
+      "epoch": 1.5517241379310345,
+      "grad_norm": 0.12719184235417458,
+      "learning_rate": 2.682084915429755e-05,
+      "loss": 0.4177,
+      "step": 1665
+    },
+    {
+      "epoch": 1.5563839701770736,
+      "grad_norm": 0.12215606374571104,
+      "learning_rate": 2.673455298584743e-05,
+      "loss": 0.4245,
+      "step": 1670
+    },
+    {
+      "epoch": 1.5610438024231128,
+      "grad_norm": 0.13680462460977227,
+      "learning_rate": 2.664825681739731e-05,
+      "loss": 0.4347,
+      "step": 1675
+    },
+    {
+      "epoch": 1.565703634669152,
+      "grad_norm": 0.13053250544826855,
+      "learning_rate": 2.6561960648947186e-05,
+      "loss": 0.4225,
+      "step": 1680
+    },
+    {
+      "epoch": 1.570363466915191,
+      "grad_norm": 0.12220469725247401,
+      "learning_rate": 2.647566448049707e-05,
+      "loss": 0.4187,
+      "step": 1685
+    },
+    {
+      "epoch": 1.5750232991612303,
+      "grad_norm": 0.12413634796113582,
+      "learning_rate": 2.6389368312046945e-05,
+      "loss": 0.4336,
+      "step": 1690
+    },
+    {
+      "epoch": 1.5796831314072692,
+      "grad_norm": 0.12043064927800652,
+      "learning_rate": 2.6303072143596824e-05,
+      "loss": 0.4373,
+      "step": 1695
+    },
+    {
+      "epoch": 1.5843429636533086,
+      "grad_norm": 0.1263261695012958,
+      "learning_rate": 2.6216775975146705e-05,
+      "loss": 0.428,
+      "step": 1700
+    },
+    {
+      "epoch": 1.5890027958993476,
+      "grad_norm": 0.1227161558533969,
+      "learning_rate": 2.6130479806696584e-05,
+      "loss": 0.4295,
+      "step": 1705
+    },
+    {
+      "epoch": 1.5936626281453867,
+      "grad_norm": 0.1270643566660397,
+      "learning_rate": 2.6044183638246462e-05,
+      "loss": 0.4287,
+      "step": 1710
+    },
+    {
+      "epoch": 1.598322460391426,
+      "grad_norm": 0.12178949334565495,
+      "learning_rate": 2.5957887469796343e-05,
+      "loss": 0.4226,
+      "step": 1715
+    },
+    {
+      "epoch": 1.602982292637465,
+      "grad_norm": 0.12218928395909417,
+      "learning_rate": 2.587159130134622e-05,
+      "loss": 0.4293,
+      "step": 1720
+    },
+    {
+      "epoch": 1.6076421248835042,
+      "grad_norm": 0.11827488589570154,
+      "learning_rate": 2.5785295132896096e-05,
+      "loss": 0.4276,
+      "step": 1725
+    },
+    {
+      "epoch": 1.6123019571295434,
+      "grad_norm": 0.11597894089300773,
+      "learning_rate": 2.569899896444598e-05,
+      "loss": 0.4205,
+      "step": 1730
+    },
+    {
+      "epoch": 1.6169617893755825,
+      "grad_norm": 0.11829969858257218,
+      "learning_rate": 2.5612702795995856e-05,
+      "loss": 0.4301,
+      "step": 1735
+    },
+    {
+      "epoch": 1.6216216216216215,
+      "grad_norm": 0.12599256386564445,
+      "learning_rate": 2.552640662754574e-05,
+      "loss": 0.4262,
+      "step": 1740
+    },
+    {
+      "epoch": 1.6262814538676609,
+      "grad_norm": 0.12229509396504862,
+      "learning_rate": 2.5440110459095616e-05,
+      "loss": 0.4342,
+      "step": 1745
+    },
+    {
+      "epoch": 1.6309412861136998,
+      "grad_norm": 0.12932497872367343,
+      "learning_rate": 2.5353814290645494e-05,
+      "loss": 0.4301,
+      "step": 1750
+    },
+    {
+      "epoch": 1.6356011183597392,
+      "grad_norm": 0.12151548654285184,
+      "learning_rate": 2.5267518122195376e-05,
+      "loss": 0.4238,
+      "step": 1755
+    },
+    {
+      "epoch": 1.6402609506057781,
+      "grad_norm": 0.13275301961315697,
+      "learning_rate": 2.5181221953745254e-05,
+      "loss": 0.4321,
+      "step": 1760
+    },
+    {
+      "epoch": 1.6449207828518173,
+      "grad_norm": 0.1198726488552535,
+      "learning_rate": 2.5094925785295132e-05,
+      "loss": 0.4312,
+      "step": 1765
+    },
+    {
+      "epoch": 1.6495806150978565,
+      "grad_norm": 0.12759737182671188,
+      "learning_rate": 2.5008629616845014e-05,
+      "loss": 0.4235,
+      "step": 1770
+    },
+    {
+      "epoch": 1.6542404473438956,
+      "grad_norm": 0.12640626702085442,
+      "learning_rate": 2.4922333448394892e-05,
+      "loss": 0.4139,
+      "step": 1775
+    },
+    {
+      "epoch": 1.6589002795899348,
+      "grad_norm": 0.11793124068530246,
+      "learning_rate": 2.483603727994477e-05,
+      "loss": 0.4318,
+      "step": 1780
+    },
+    {
+      "epoch": 1.6635601118359737,
+      "grad_norm": 0.11638331589079352,
+      "learning_rate": 2.4749741111494652e-05,
+      "loss": 0.4263,
+      "step": 1785
+    },
+    {
+      "epoch": 1.6682199440820131,
+      "grad_norm": 0.11973553173133647,
+      "learning_rate": 2.466344494304453e-05,
+      "loss": 0.4265,
+      "step": 1790
+    },
+    {
+      "epoch": 1.672879776328052,
+      "grad_norm": 0.12303025961982857,
+      "learning_rate": 2.457714877459441e-05,
+      "loss": 0.4173,
+      "step": 1795
+    },
+    {
+      "epoch": 1.6775396085740915,
+      "grad_norm": 0.11944496103244483,
+      "learning_rate": 2.4490852606144287e-05,
+      "loss": 0.4183,
+      "step": 1800
+    },
+    {
+      "epoch": 1.6821994408201304,
+      "grad_norm": 0.12065646781676319,
+      "learning_rate": 2.440455643769417e-05,
+      "loss": 0.4204,
+      "step": 1805
+    },
+    {
+      "epoch": 1.6868592730661698,
+      "grad_norm": 0.12423973581566003,
+      "learning_rate": 2.4318260269244047e-05,
+      "loss": 0.4232,
+      "step": 1810
+    },
+    {
+      "epoch": 1.6915191053122087,
+      "grad_norm": 0.12361738007885231,
+      "learning_rate": 2.4231964100793925e-05,
+      "loss": 0.4121,
+      "step": 1815
+    },
+    {
+      "epoch": 1.696178937558248,
+      "grad_norm": 0.11581164871848279,
+      "learning_rate": 2.4145667932343803e-05,
+      "loss": 0.4206,
+      "step": 1820
+    },
+    {
+      "epoch": 1.700838769804287,
+      "grad_norm": 0.11920977376595605,
+      "learning_rate": 2.4059371763893685e-05,
+      "loss": 0.4306,
+      "step": 1825
+    },
+    {
+      "epoch": 1.7054986020503262,
+      "grad_norm": 0.12253816707092181,
+      "learning_rate": 2.3973075595443563e-05,
+      "loss": 0.4311,
+      "step": 1830
+    },
+    {
+      "epoch": 1.7101584342963654,
+      "grad_norm": 0.13310074425522717,
+      "learning_rate": 2.388677942699344e-05,
+      "loss": 0.4304,
+      "step": 1835
+    },
+    {
+      "epoch": 1.7148182665424043,
+      "grad_norm": 0.12604533496490997,
+      "learning_rate": 2.3800483258543323e-05,
+      "loss": 0.4234,
+      "step": 1840
+    },
+    {
+      "epoch": 1.7194780987884437,
+      "grad_norm": 0.1091858086630578,
+      "learning_rate": 2.37141870900932e-05,
+      "loss": 0.4215,
+      "step": 1845
+    },
+    {
+      "epoch": 1.7241379310344827,
+      "grad_norm": 0.11643139556699238,
+      "learning_rate": 2.3627890921643083e-05,
+      "loss": 0.4082,
+      "step": 1850
+    },
+    {
+      "epoch": 1.728797763280522,
+      "grad_norm": 0.1224738227288852,
+      "learning_rate": 2.3541594753192957e-05,
+      "loss": 0.4182,
+      "step": 1855
+    },
+    {
+      "epoch": 1.733457595526561,
+      "grad_norm": 0.12526731472504868,
+      "learning_rate": 2.345529858474284e-05,
+      "loss": 0.4214,
+      "step": 1860
+    },
+    {
+      "epoch": 1.7381174277726001,
+      "grad_norm": 0.12090065715127674,
+      "learning_rate": 2.3369002416292717e-05,
+      "loss": 0.4234,
+      "step": 1865
+    },
+    {
+      "epoch": 1.7427772600186393,
+      "grad_norm": 0.1187329361943555,
+      "learning_rate": 2.32827062478426e-05,
+      "loss": 0.4195,
+      "step": 1870
+    },
+    {
+      "epoch": 1.7474370922646785,
+      "grad_norm": 0.11657803046926513,
+      "learning_rate": 2.3196410079392474e-05,
+      "loss": 0.4289,
+      "step": 1875
+    },
+    {
+      "epoch": 1.7520969245107176,
+      "grad_norm": 0.12269935652485676,
+      "learning_rate": 2.3110113910942355e-05,
+      "loss": 0.4264,
+      "step": 1880
+    },
+    {
+      "epoch": 1.7567567567567568,
+      "grad_norm": 0.12895286053321006,
+      "learning_rate": 2.3023817742492234e-05,
+      "loss": 0.4261,
+      "step": 1885
+    },
+    {
+      "epoch": 1.761416589002796,
+      "grad_norm": 0.11673527851142344,
+      "learning_rate": 2.2937521574042115e-05,
+      "loss": 0.4221,
+      "step": 1890
+    },
+    {
+      "epoch": 1.766076421248835,
+      "grad_norm": 0.12166044566996549,
+      "learning_rate": 2.285122540559199e-05,
+      "loss": 0.4256,
+      "step": 1895
+    },
+    {
+      "epoch": 1.7707362534948743,
+      "grad_norm": 0.12454893915488945,
+      "learning_rate": 2.276492923714187e-05,
+      "loss": 0.4142,
+      "step": 1900
+    },
+    {
+      "epoch": 1.7753960857409132,
+      "grad_norm": 0.11530673814280247,
+      "learning_rate": 2.267863306869175e-05,
+      "loss": 0.4136,
+      "step": 1905
+    },
+    {
+      "epoch": 1.7800559179869526,
+      "grad_norm": 0.12822137957999336,
+      "learning_rate": 2.259233690024163e-05,
+      "loss": 0.4284,
+      "step": 1910
+    },
+    {
+      "epoch": 1.7847157502329916,
+      "grad_norm": 0.11710411627997598,
+      "learning_rate": 2.250604073179151e-05,
+      "loss": 0.4112,
+      "step": 1915
+    },
+    {
+      "epoch": 1.7893755824790307,
+      "grad_norm": 0.12915002596644523,
+      "learning_rate": 2.2419744563341388e-05,
+      "loss": 0.4304,
+      "step": 1920
+    },
+    {
+      "epoch": 1.7940354147250699,
+      "grad_norm": 0.12257330140914043,
+      "learning_rate": 2.233344839489127e-05,
+      "loss": 0.423,
+      "step": 1925
+    },
+    {
+      "epoch": 1.798695246971109,
+      "grad_norm": 0.1302222177961605,
+      "learning_rate": 2.2247152226441148e-05,
+      "loss": 0.4198,
+      "step": 1930
+    },
+    {
+      "epoch": 1.8033550792171482,
+      "grad_norm": 0.12419853476676282,
+      "learning_rate": 2.2160856057991026e-05,
+      "loss": 0.4251,
+      "step": 1935
+    },
+    {
+      "epoch": 1.8080149114631874,
+      "grad_norm": 0.11427266994118157,
+      "learning_rate": 2.2074559889540904e-05,
+      "loss": 0.4107,
+      "step": 1940
+    },
+    {
+      "epoch": 1.8126747437092265,
+      "grad_norm": 0.13078560899989555,
+      "learning_rate": 2.1988263721090786e-05,
+      "loss": 0.4347,
+      "step": 1945
+    },
+    {
+      "epoch": 1.8173345759552655,
+      "grad_norm": 0.13113649931101512,
+      "learning_rate": 2.1901967552640664e-05,
+      "loss": 0.4101,
+      "step": 1950
+    },
+    {
+      "epoch": 1.8219944082013049,
+      "grad_norm": 0.1225950944087108,
+      "learning_rate": 2.1815671384190542e-05,
+      "loss": 0.4173,
+      "step": 1955
+    },
+    {
+      "epoch": 1.8266542404473438,
+      "grad_norm": 0.11568029867325309,
+      "learning_rate": 2.172937521574042e-05,
+      "loss": 0.4208,
+      "step": 1960
+    },
+    {
+      "epoch": 1.8313140726933832,
+      "grad_norm": 0.12724487437140347,
+      "learning_rate": 2.1643079047290302e-05,
+      "loss": 0.4266,
+      "step": 1965
+    },
+    {
+      "epoch": 1.8359739049394221,
+      "grad_norm": 0.127778691747861,
+      "learning_rate": 2.155678287884018e-05,
+      "loss": 0.4252,
+      "step": 1970
+    },
+    {
+      "epoch": 1.8406337371854613,
+      "grad_norm": 0.12114870767961486,
+      "learning_rate": 2.147048671039006e-05,
+      "loss": 0.4342,
+      "step": 1975
+    },
+    {
+      "epoch": 1.8452935694315005,
+      "grad_norm": 0.1187496189210886,
+      "learning_rate": 2.1384190541939937e-05,
+      "loss": 0.4195,
+      "step": 1980
+    },
+    {
+      "epoch": 1.8499534016775396,
+      "grad_norm": 0.1164628648881079,
+      "learning_rate": 2.129789437348982e-05,
+      "loss": 0.415,
+      "step": 1985
+    },
+    {
+      "epoch": 1.8546132339235788,
+      "grad_norm": 0.12470690822345669,
+      "learning_rate": 2.1211598205039697e-05,
+      "loss": 0.4403,
+      "step": 1990
+    },
+    {
+      "epoch": 1.8592730661696177,
+      "grad_norm": 0.12445996081047758,
+      "learning_rate": 2.1125302036589575e-05,
+      "loss": 0.4203,
+      "step": 1995
+    },
+    {
+      "epoch": 1.8639328984156571,
+      "grad_norm": 0.13600096036836443,
+      "learning_rate": 2.1039005868139457e-05,
+      "loss": 0.4206,
+      "step": 2000
+    },
+    {
+      "epoch": 1.868592730661696,
+      "grad_norm": 0.12457644963404628,
+      "learning_rate": 2.0952709699689335e-05,
+      "loss": 0.4166,
+      "step": 2005
+    },
+    {
+      "epoch": 1.8732525629077355,
+      "grad_norm": 0.11967906954723008,
+      "learning_rate": 2.0866413531239216e-05,
+      "loss": 0.4309,
+      "step": 2010
+    },
+    {
+      "epoch": 1.8779123951537744,
+      "grad_norm": 0.11872193808374913,
+      "learning_rate": 2.078011736278909e-05,
+      "loss": 0.4164,
+      "step": 2015
+    },
+    {
+      "epoch": 1.8825722273998136,
+      "grad_norm": 0.13383323070130404,
+      "learning_rate": 2.0693821194338973e-05,
+      "loss": 0.4127,
+      "step": 2020
+    },
+    {
+      "epoch": 1.8872320596458527,
+      "grad_norm": 0.1279122954768431,
+      "learning_rate": 2.060752502588885e-05,
+      "loss": 0.4152,
+      "step": 2025
+    },
+    {
+      "epoch": 1.8918918918918919,
+      "grad_norm": 0.11805828962993543,
+      "learning_rate": 2.0521228857438733e-05,
+      "loss": 0.4236,
+      "step": 2030
+    },
+    {
+      "epoch": 1.896551724137931,
+      "grad_norm": 0.12084745609561812,
+      "learning_rate": 2.0434932688988608e-05,
+      "loss": 0.4179,
+      "step": 2035
+    },
+    {
+      "epoch": 1.9012115563839702,
+      "grad_norm": 0.11796964749427283,
+      "learning_rate": 2.034863652053849e-05,
+      "loss": 0.4203,
+      "step": 2040
+    },
+    {
+      "epoch": 1.9058713886300094,
+      "grad_norm": 0.12055130411080678,
+      "learning_rate": 2.0262340352088367e-05,
+      "loss": 0.4385,
+      "step": 2045
+    },
+    {
+      "epoch": 1.9105312208760483,
+      "grad_norm": 0.12175309233745629,
+      "learning_rate": 2.017604418363825e-05,
+      "loss": 0.4275,
+      "step": 2050
+    },
+    {
+      "epoch": 1.9151910531220877,
+      "grad_norm": 0.12315994569624988,
+      "learning_rate": 2.0089748015188127e-05,
+      "loss": 0.4203,
+      "step": 2055
+    },
+    {
+      "epoch": 1.9198508853681266,
+      "grad_norm": 0.1241591691754375,
+      "learning_rate": 2.0003451846738005e-05,
+      "loss": 0.4315,
+      "step": 2060
+    },
+    {
+      "epoch": 1.924510717614166,
+      "grad_norm": 0.12155940122395187,
+      "learning_rate": 1.9917155678287887e-05,
+      "loss": 0.4172,
+      "step": 2065
+    },
+    {
+      "epoch": 1.929170549860205,
+      "grad_norm": 0.11995818477216764,
+      "learning_rate": 1.9830859509837765e-05,
+      "loss": 0.424,
+      "step": 2070
+    },
+    {
+      "epoch": 1.9338303821062441,
+      "grad_norm": 0.12799192307196566,
+      "learning_rate": 1.9744563341387643e-05,
+      "loss": 0.4293,
+      "step": 2075
+    },
+    {
+      "epoch": 1.9384902143522833,
+      "grad_norm": 0.12582750538306092,
+      "learning_rate": 1.9658267172937522e-05,
+      "loss": 0.4233,
+      "step": 2080
+    },
+    {
+      "epoch": 1.9431500465983225,
+      "grad_norm": 0.12640657594715463,
+      "learning_rate": 1.9571971004487403e-05,
+      "loss": 0.4233,
+      "step": 2085
+    },
+    {
+      "epoch": 1.9478098788443616,
+      "grad_norm": 0.1207639597563511,
+      "learning_rate": 1.948567483603728e-05,
+      "loss": 0.4261,
+      "step": 2090
+    },
+    {
+      "epoch": 1.9524697110904008,
+      "grad_norm": 0.12211931670067579,
+      "learning_rate": 1.939937866758716e-05,
+      "loss": 0.4077,
+      "step": 2095
+    },
+    {
+      "epoch": 1.95712954333644,
+      "grad_norm": 0.1288552605369309,
+      "learning_rate": 1.9313082499137038e-05,
+      "loss": 0.4297,
+      "step": 2100
+    },
+    {
+      "epoch": 1.961789375582479,
+      "grad_norm": 0.12210521745064834,
+      "learning_rate": 1.922678633068692e-05,
+      "loss": 0.4216,
+      "step": 2105
+    },
+    {
+      "epoch": 1.9664492078285183,
+      "grad_norm": 0.12007842604067011,
+      "learning_rate": 1.9140490162236798e-05,
+      "loss": 0.4288,
+      "step": 2110
+    },
+    {
+      "epoch": 1.9711090400745572,
+      "grad_norm": 0.1200014373172409,
+      "learning_rate": 1.9054193993786676e-05,
+      "loss": 0.4337,
+      "step": 2115
+    },
+    {
+      "epoch": 1.9757688723205966,
+      "grad_norm": 0.12048963812079494,
+      "learning_rate": 1.8967897825336554e-05,
+      "loss": 0.4199,
+      "step": 2120
+    },
+    {
+      "epoch": 1.9804287045666356,
+      "grad_norm": 0.1149812695004315,
+      "learning_rate": 1.8881601656886436e-05,
+      "loss": 0.4197,
+      "step": 2125
+    },
+    {
+      "epoch": 1.9850885368126747,
+      "grad_norm": 0.12276686067910879,
+      "learning_rate": 1.8795305488436314e-05,
+      "loss": 0.4256,
+      "step": 2130
+    },
+    {
+      "epoch": 1.9897483690587139,
+      "grad_norm": 0.12328838371074524,
+      "learning_rate": 1.8709009319986192e-05,
+      "loss": 0.4091,
+      "step": 2135
+    },
+    {
+      "epoch": 1.994408201304753,
+      "grad_norm": 0.11641664305189577,
+      "learning_rate": 1.8622713151536074e-05,
+      "loss": 0.4196,
+      "step": 2140
+    },
+    {
+      "epoch": 1.9990680335507922,
+      "grad_norm": 0.12930630468630874,
+      "learning_rate": 1.8536416983085952e-05,
+      "loss": 0.4241,
+      "step": 2145
+    },
+    {
+      "epoch": 2.003727865796831,
+      "grad_norm": 0.10953015651877655,
+      "learning_rate": 1.8450120814635834e-05,
+      "loss": 0.3856,
+      "step": 2150
+    },
+    {
+      "epoch": 2.0083876980428705,
+      "grad_norm": 0.1312378903134541,
+      "learning_rate": 1.836382464618571e-05,
+      "loss": 0.4066,
+      "step": 2155
+    },
+    {
+      "epoch": 2.0130475302889095,
+      "grad_norm": 0.1238321092559146,
+      "learning_rate": 1.827752847773559e-05,
+      "loss": 0.3993,
+      "step": 2160
+    },
+    {
+      "epoch": 2.017707362534949,
+      "grad_norm": 0.1295497534836468,
+      "learning_rate": 1.819123230928547e-05,
+      "loss": 0.3989,
+      "step": 2165
+    },
+    {
+      "epoch": 2.022367194780988,
+      "grad_norm": 0.11789987259106473,
+      "learning_rate": 1.810493614083535e-05,
+      "loss": 0.4,
+      "step": 2170
+    },
+    {
+      "epoch": 2.027027027027027,
+      "grad_norm": 0.1172636163155523,
+      "learning_rate": 1.8018639972385225e-05,
+      "loss": 0.3921,
+      "step": 2175
+    },
+    {
+      "epoch": 2.031686859273066,
+      "grad_norm": 0.11854988304414193,
+      "learning_rate": 1.7932343803935107e-05,
+      "loss": 0.4029,
+      "step": 2180
+    },
+    {
+      "epoch": 2.0363466915191055,
+      "grad_norm": 0.11630378582387761,
+      "learning_rate": 1.7846047635484985e-05,
+      "loss": 0.395,
+      "step": 2185
+    },
+    {
+      "epoch": 2.0410065237651445,
+      "grad_norm": 0.11974443564714121,
+      "learning_rate": 1.7759751467034866e-05,
+      "loss": 0.3965,
+      "step": 2190
+    },
+    {
+      "epoch": 2.0456663560111834,
+      "grad_norm": 0.12306199256663979,
+      "learning_rate": 1.767345529858474e-05,
+      "loss": 0.4128,
+      "step": 2195
+    },
+    {
+      "epoch": 2.050326188257223,
+      "grad_norm": 0.11476203852138765,
+      "learning_rate": 1.7587159130134623e-05,
+      "loss": 0.4094,
+      "step": 2200
+    },
+    {
+      "epoch": 2.0549860205032617,
+      "grad_norm": 0.11478633224728738,
+      "learning_rate": 1.75008629616845e-05,
+      "loss": 0.3949,
+      "step": 2205
+    },
+    {
+      "epoch": 2.059645852749301,
+      "grad_norm": 0.11827360581744069,
+      "learning_rate": 1.7414566793234383e-05,
+      "loss": 0.3965,
+      "step": 2210
+    },
+    {
+      "epoch": 2.06430568499534,
+      "grad_norm": 0.12896180024375098,
+      "learning_rate": 1.732827062478426e-05,
+      "loss": 0.3972,
+      "step": 2215
+    },
+    {
+      "epoch": 2.0689655172413794,
+      "grad_norm": 0.11877652963168868,
+      "learning_rate": 1.724197445633414e-05,
+      "loss": 0.3881,
+      "step": 2220
+    },
+    {
+      "epoch": 2.0736253494874184,
+      "grad_norm": 0.11410062313843118,
+      "learning_rate": 1.715567828788402e-05,
+      "loss": 0.3864,
+      "step": 2225
+    },
+    {
+      "epoch": 2.0782851817334578,
+      "grad_norm": 0.11884676784919541,
+      "learning_rate": 1.70693821194339e-05,
+      "loss": 0.3892,
+      "step": 2230
+    },
+    {
+      "epoch": 2.0829450139794967,
+      "grad_norm": 0.11329026695681223,
+      "learning_rate": 1.6983085950983777e-05,
+      "loss": 0.3909,
+      "step": 2235
+    },
+    {
+      "epoch": 2.0876048462255357,
+      "grad_norm": 0.1149751820301733,
+      "learning_rate": 1.6896789782533655e-05,
+      "loss": 0.4079,
+      "step": 2240
+    },
+    {
+      "epoch": 2.092264678471575,
+      "grad_norm": 0.12241924570401962,
+      "learning_rate": 1.6810493614083537e-05,
+      "loss": 0.3973,
+      "step": 2245
+    },
+    {
+      "epoch": 2.096924510717614,
+      "grad_norm": 0.12349976311476825,
+      "learning_rate": 1.6724197445633415e-05,
+      "loss": 0.4018,
+      "step": 2250
+    },
+    {
+      "epoch": 2.1015843429636534,
+      "grad_norm": 0.12156351285203568,
+      "learning_rate": 1.6637901277183294e-05,
+      "loss": 0.4063,
+      "step": 2255
+    },
+    {
+      "epoch": 2.1062441752096923,
+      "grad_norm": 0.12268254335355998,
+      "learning_rate": 1.6551605108733172e-05,
+      "loss": 0.4007,
+      "step": 2260
+    },
+    {
+      "epoch": 2.1109040074557317,
+      "grad_norm": 0.1173789864096937,
+      "learning_rate": 1.6465308940283053e-05,
+      "loss": 0.3967,
+      "step": 2265
+    },
+    {
+      "epoch": 2.1155638397017706,
+      "grad_norm": 0.1332521790139701,
+      "learning_rate": 1.637901277183293e-05,
+      "loss": 0.4068,
+      "step": 2270
+    },
+    {
+      "epoch": 2.12022367194781,
+      "grad_norm": 0.11800696565580679,
+      "learning_rate": 1.629271660338281e-05,
+      "loss": 0.3978,
+      "step": 2275
+    },
+    {
+      "epoch": 2.124883504193849,
+      "grad_norm": 0.1317850150824812,
+      "learning_rate": 1.620642043493269e-05,
+      "loss": 0.4028,
+      "step": 2280
+    },
+    {
+      "epoch": 2.1295433364398884,
+      "grad_norm": 0.11781633353379023,
+      "learning_rate": 1.612012426648257e-05,
+      "loss": 0.3884,
+      "step": 2285
+    },
+    {
+      "epoch": 2.1342031686859273,
+      "grad_norm": 0.11487475918531667,
+      "learning_rate": 1.6033828098032448e-05,
+      "loss": 0.3978,
+      "step": 2290
+    },
+    {
+      "epoch": 2.1388630009319662,
+      "grad_norm": 0.12136094358999924,
+      "learning_rate": 1.5947531929582326e-05,
+      "loss": 0.3996,
+      "step": 2295
+    },
+    {
+      "epoch": 2.1435228331780056,
+      "grad_norm": 0.12032050267123132,
+      "learning_rate": 1.5861235761132208e-05,
+      "loss": 0.4007,
+      "step": 2300
+    },
+    {
+      "epoch": 2.1481826654240446,
+      "grad_norm": 0.1198008774929863,
+      "learning_rate": 1.5774939592682086e-05,
+      "loss": 0.3948,
+      "step": 2305
+    },
+    {
+      "epoch": 2.152842497670084,
+      "grad_norm": 0.11897806671606173,
+      "learning_rate": 1.5688643424231964e-05,
+      "loss": 0.3965,
+      "step": 2310
+    },
+    {
+      "epoch": 2.157502329916123,
+      "grad_norm": 0.13488070133741764,
+      "learning_rate": 1.5602347255781842e-05,
+      "loss": 0.4062,
+      "step": 2315
+    },
+    {
+      "epoch": 2.1621621621621623,
+      "grad_norm": 0.11496625417243356,
+      "learning_rate": 1.5516051087331724e-05,
+      "loss": 0.3866,
+      "step": 2320
+    },
+    {
+      "epoch": 2.1668219944082012,
+      "grad_norm": 0.12373802287948729,
+      "learning_rate": 1.5429754918881602e-05,
+      "loss": 0.4072,
+      "step": 2325
+    },
+    {
+      "epoch": 2.1714818266542406,
+      "grad_norm": 0.1253195505836586,
+      "learning_rate": 1.534345875043148e-05,
+      "loss": 0.3997,
+      "step": 2330
+    },
+    {
+      "epoch": 2.1761416589002796,
+      "grad_norm": 0.11709446735449353,
+      "learning_rate": 1.525716258198136e-05,
+      "loss": 0.3922,
+      "step": 2335
+    },
+    {
+      "epoch": 2.180801491146319,
+      "grad_norm": 0.11808560696242013,
+      "learning_rate": 1.517086641353124e-05,
+      "loss": 0.3924,
+      "step": 2340
+    },
+    {
+      "epoch": 2.185461323392358,
+      "grad_norm": 0.12505547109800066,
+      "learning_rate": 1.508457024508112e-05,
+      "loss": 0.4111,
+      "step": 2345
+    },
+    {
+      "epoch": 2.190121155638397,
+      "grad_norm": 0.1201661175945142,
+      "learning_rate": 1.4998274076630997e-05,
+      "loss": 0.3806,
+      "step": 2350
+    },
+    {
+      "epoch": 2.194780987884436,
+      "grad_norm": 0.12209252746949248,
+      "learning_rate": 1.4911977908180877e-05,
+      "loss": 0.407,
+      "step": 2355
+    },
+    {
+      "epoch": 2.199440820130475,
+      "grad_norm": 0.1301766081048796,
+      "learning_rate": 1.4825681739730757e-05,
+      "loss": 0.4033,
+      "step": 2360
+    },
+    {
+      "epoch": 2.2041006523765145,
+      "grad_norm": 0.12361015796162281,
+      "learning_rate": 1.4739385571280637e-05,
+      "loss": 0.386,
+      "step": 2365
+    },
+    {
+      "epoch": 2.2087604846225535,
+      "grad_norm": 0.12569297053787679,
+      "learning_rate": 1.4653089402830513e-05,
+      "loss": 0.3852,
+      "step": 2370
+    },
+    {
+      "epoch": 2.213420316868593,
+      "grad_norm": 0.12020144097813847,
+      "learning_rate": 1.4566793234380393e-05,
+      "loss": 0.3914,
+      "step": 2375
+    },
+    {
+      "epoch": 2.218080149114632,
+      "grad_norm": 0.11707191489060537,
+      "learning_rate": 1.4480497065930273e-05,
+      "loss": 0.4009,
+      "step": 2380
+    },
+    {
+      "epoch": 2.222739981360671,
+      "grad_norm": 0.11388019109100149,
+      "learning_rate": 1.4394200897480153e-05,
+      "loss": 0.3849,
+      "step": 2385
+    },
+    {
+      "epoch": 2.22739981360671,
+      "grad_norm": 0.12143573509003629,
+      "learning_rate": 1.4307904729030031e-05,
+      "loss": 0.3957,
+      "step": 2390
+    },
+    {
+      "epoch": 2.2320596458527495,
+      "grad_norm": 0.11360838925947578,
+      "learning_rate": 1.4221608560579911e-05,
+      "loss": 0.406,
+      "step": 2395
+    },
+    {
+      "epoch": 2.2367194780987885,
+      "grad_norm": 0.11738739553961004,
+      "learning_rate": 1.4135312392129791e-05,
+      "loss": 0.3926,
+      "step": 2400
+    },
+    {
+      "epoch": 2.2413793103448274,
+      "grad_norm": 0.11694310012607675,
+      "learning_rate": 1.404901622367967e-05,
+      "loss": 0.379,
+      "step": 2405
+    },
+    {
+      "epoch": 2.246039142590867,
+      "grad_norm": 0.11549493762191822,
+      "learning_rate": 1.3962720055229547e-05,
+      "loss": 0.4035,
+      "step": 2410
+    },
+    {
+      "epoch": 2.2506989748369057,
+      "grad_norm": 0.1185596055080423,
+      "learning_rate": 1.3876423886779427e-05,
+      "loss": 0.3981,
+      "step": 2415
+    },
+    {
+      "epoch": 2.255358807082945,
+      "grad_norm": 0.11396564713694664,
+      "learning_rate": 1.3790127718329307e-05,
+      "loss": 0.3982,
+      "step": 2420
+    },
+    {
+      "epoch": 2.260018639328984,
+      "grad_norm": 0.1261947393458037,
+      "learning_rate": 1.3703831549879187e-05,
+      "loss": 0.3983,
+      "step": 2425
+    },
+    {
+      "epoch": 2.2646784715750234,
+      "grad_norm": 0.1187193536373494,
+      "learning_rate": 1.3617535381429064e-05,
+      "loss": 0.3936,
+      "step": 2430
+    },
+    {
+      "epoch": 2.2693383038210624,
+      "grad_norm": 0.1184433393852463,
+      "learning_rate": 1.3531239212978944e-05,
+      "loss": 0.3985,
+      "step": 2435
+    },
+    {
+      "epoch": 2.2739981360671018,
+      "grad_norm": 0.11965320929840521,
+      "learning_rate": 1.3444943044528824e-05,
+      "loss": 0.394,
+      "step": 2440
+    },
+    {
+      "epoch": 2.2786579683131407,
+      "grad_norm": 0.12400378459157593,
+      "learning_rate": 1.3358646876078703e-05,
+      "loss": 0.4001,
+      "step": 2445
+    },
+    {
+      "epoch": 2.28331780055918,
+      "grad_norm": 0.12253491982520912,
+      "learning_rate": 1.327235070762858e-05,
+      "loss": 0.3965,
+      "step": 2450
+    },
+    {
+      "epoch": 2.287977632805219,
+      "grad_norm": 0.11653670138180043,
+      "learning_rate": 1.318605453917846e-05,
+      "loss": 0.4008,
+      "step": 2455
+    },
+    {
+      "epoch": 2.292637465051258,
+      "grad_norm": 0.12543467885972914,
+      "learning_rate": 1.309975837072834e-05,
+      "loss": 0.3986,
+      "step": 2460
+    },
+    {
+      "epoch": 2.2972972972972974,
+      "grad_norm": 0.12262709260177206,
+      "learning_rate": 1.301346220227822e-05,
+      "loss": 0.3923,
+      "step": 2465
+    },
+    {
+      "epoch": 2.3019571295433363,
+      "grad_norm": 0.1421271140321946,
+      "learning_rate": 1.2927166033828098e-05,
+      "loss": 0.3957,
+      "step": 2470
+    },
+    {
+      "epoch": 2.3066169617893757,
+      "grad_norm": 0.11406152806013625,
+      "learning_rate": 1.2840869865377978e-05,
+      "loss": 0.3993,
+      "step": 2475
+    },
+    {
+      "epoch": 2.3112767940354146,
+      "grad_norm": 0.11563527223268577,
+      "learning_rate": 1.2754573696927858e-05,
+      "loss": 0.3889,
+      "step": 2480
+    },
+    {
+      "epoch": 2.315936626281454,
+      "grad_norm": 0.11406004210681313,
+      "learning_rate": 1.2668277528477738e-05,
+      "loss": 0.3983,
+      "step": 2485
+    },
+    {
+      "epoch": 2.320596458527493,
+      "grad_norm": 0.1170532637727948,
+      "learning_rate": 1.2581981360027614e-05,
+      "loss": 0.3903,
+      "step": 2490
+    },
+    {
+      "epoch": 2.325256290773532,
+      "grad_norm": 0.11742691455202914,
+      "learning_rate": 1.2495685191577494e-05,
+      "loss": 0.4114,
+      "step": 2495
+    },
+    {
+      "epoch": 2.3299161230195713,
+      "grad_norm": 0.12105903642639991,
+      "learning_rate": 1.2409389023127374e-05,
+      "loss": 0.3946,
+      "step": 2500
+    },
+    {
+      "epoch": 2.3345759552656107,
+      "grad_norm": 0.12536217909992417,
+      "learning_rate": 1.2323092854677252e-05,
+      "loss": 0.3985,
+      "step": 2505
+    },
+    {
+      "epoch": 2.3392357875116496,
+      "grad_norm": 0.12165174655774741,
+      "learning_rate": 1.2236796686227132e-05,
+      "loss": 0.4105,
+      "step": 2510
+    },
+    {
+      "epoch": 2.3438956197576886,
+      "grad_norm": 0.12304120417719089,
+      "learning_rate": 1.215050051777701e-05,
+      "loss": 0.403,
+      "step": 2515
+    },
+    {
+      "epoch": 2.348555452003728,
+      "grad_norm": 0.11577921208055114,
+      "learning_rate": 1.206420434932689e-05,
+      "loss": 0.3997,
+      "step": 2520
+    },
+    {
+      "epoch": 2.353215284249767,
+      "grad_norm": 0.119210033514637,
+      "learning_rate": 1.1977908180876769e-05,
+      "loss": 0.3961,
+      "step": 2525
+    },
+    {
+      "epoch": 2.3578751164958063,
+      "grad_norm": 0.11890081230916477,
+      "learning_rate": 1.1891612012426649e-05,
+      "loss": 0.4024,
+      "step": 2530
+    },
+    {
+      "epoch": 2.362534948741845,
+      "grad_norm": 0.12138067213187034,
+      "learning_rate": 1.1805315843976528e-05,
+      "loss": 0.3987,
+      "step": 2535
+    },
+    {
+      "epoch": 2.3671947809878846,
+      "grad_norm": 0.11737327739321275,
+      "learning_rate": 1.1719019675526408e-05,
+      "loss": 0.3863,
+      "step": 2540
+    },
+    {
+      "epoch": 2.3718546132339235,
+      "grad_norm": 0.11325716607024172,
+      "learning_rate": 1.1632723507076287e-05,
+      "loss": 0.4048,
+      "step": 2545
+    },
+    {
+      "epoch": 2.3765144454799625,
+      "grad_norm": 0.11976758933505972,
+      "learning_rate": 1.1546427338626167e-05,
+      "loss": 0.3874,
+      "step": 2550
+    },
+    {
+      "epoch": 2.381174277726002,
+      "grad_norm": 0.11725995404113715,
+      "learning_rate": 1.1460131170176045e-05,
+      "loss": 0.3876,
+      "step": 2555
+    },
+    {
+      "epoch": 2.385834109972041,
+      "grad_norm": 0.11442972457056923,
+      "learning_rate": 1.1373835001725925e-05,
+      "loss": 0.3941,
+      "step": 2560
+    },
+    {
+      "epoch": 2.39049394221808,
+      "grad_norm": 0.11881533778110595,
+      "learning_rate": 1.1287538833275803e-05,
+      "loss": 0.4101,
+      "step": 2565
+    },
+    {
+      "epoch": 2.395153774464119,
+      "grad_norm": 0.11776130121473811,
+      "learning_rate": 1.1201242664825683e-05,
+      "loss": 0.401,
+      "step": 2570
+    },
+    {
+      "epoch": 2.3998136067101585,
+      "grad_norm": 0.11241758775061532,
+      "learning_rate": 1.1114946496375561e-05,
+      "loss": 0.3935,
+      "step": 2575
+    },
+    {
+      "epoch": 2.4044734389561975,
+      "grad_norm": 0.114639703038874,
+      "learning_rate": 1.1028650327925441e-05,
+      "loss": 0.3946,
+      "step": 2580
+    },
+    {
+      "epoch": 2.409133271202237,
+      "grad_norm": 0.11549066105187085,
+      "learning_rate": 1.094235415947532e-05,
+      "loss": 0.4164,
+      "step": 2585
+    },
+    {
+      "epoch": 2.413793103448276,
+      "grad_norm": 0.11341130941153825,
+      "learning_rate": 1.0856057991025199e-05,
+      "loss": 0.3902,
+      "step": 2590
+    },
+    {
+      "epoch": 2.418452935694315,
+      "grad_norm": 0.11693981529245728,
+      "learning_rate": 1.0769761822575077e-05,
+      "loss": 0.3981,
+      "step": 2595
+    },
+    {
+      "epoch": 2.423112767940354,
+      "grad_norm": 0.11445718156037789,
+      "learning_rate": 1.0683465654124957e-05,
+      "loss": 0.4,
+      "step": 2600
+    },
+    {
+      "epoch": 2.427772600186393,
+      "grad_norm": 0.11760337159363873,
+      "learning_rate": 1.0597169485674835e-05,
+      "loss": 0.4043,
+      "step": 2605
+    },
+    {
+      "epoch": 2.4324324324324325,
+      "grad_norm": 0.11202197847953917,
+      "learning_rate": 1.0510873317224715e-05,
+      "loss": 0.3938,
+      "step": 2610
+    },
+    {
+      "epoch": 2.4370922646784714,
+      "grad_norm": 0.11445956969198921,
+      "learning_rate": 1.0424577148774595e-05,
+      "loss": 0.3929,
+      "step": 2615
+    },
+    {
+      "epoch": 2.441752096924511,
+      "grad_norm": 0.12511703582866368,
+      "learning_rate": 1.0338280980324475e-05,
+      "loss": 0.394,
+      "step": 2620
+    },
+    {
+      "epoch": 2.4464119291705497,
+      "grad_norm": 0.11880110447016685,
+      "learning_rate": 1.0251984811874353e-05,
+      "loss": 0.3935,
+      "step": 2625
+    },
+    {
+      "epoch": 2.451071761416589,
+      "grad_norm": 0.11225204973027134,
+      "learning_rate": 1.0165688643424233e-05,
+      "loss": 0.3924,
+      "step": 2630
+    },
+    {
+      "epoch": 2.455731593662628,
+      "grad_norm": 0.11582628937927099,
+      "learning_rate": 1.0079392474974112e-05,
+      "loss": 0.3874,
+      "step": 2635
+    },
+    {
+      "epoch": 2.4603914259086674,
+      "grad_norm": 0.11602435972224132,
+      "learning_rate": 9.993096306523992e-06,
+      "loss": 0.3905,
+      "step": 2640
+    },
+    {
+      "epoch": 2.4650512581547064,
+      "grad_norm": 0.10934888235224466,
+      "learning_rate": 9.90680013807387e-06,
+      "loss": 0.3854,
+      "step": 2645
+    },
+    {
+      "epoch": 2.4697110904007458,
+      "grad_norm": 0.11803474684024798,
+      "learning_rate": 9.82050396962375e-06,
+      "loss": 0.4011,
+      "step": 2650
+    },
+    {
+      "epoch": 2.4743709226467847,
+      "grad_norm": 0.12146425611154846,
+      "learning_rate": 9.734207801173628e-06,
+      "loss": 0.3963,
+      "step": 2655
+    },
+    {
+      "epoch": 2.4790307548928237,
+      "grad_norm": 0.1135891275955787,
+      "learning_rate": 9.647911632723508e-06,
+      "loss": 0.3894,
+      "step": 2660
+    },
+    {
+      "epoch": 2.483690587138863,
+      "grad_norm": 0.11364709484218559,
+      "learning_rate": 9.561615464273386e-06,
+      "loss": 0.3971,
+      "step": 2665
+    },
+    {
+      "epoch": 2.488350419384902,
+      "grad_norm": 0.1156960179424466,
+      "learning_rate": 9.475319295823266e-06,
+      "loss": 0.3942,
+      "step": 2670
+    },
+    {
+      "epoch": 2.4930102516309414,
+      "grad_norm": 0.11421026915198766,
+      "learning_rate": 9.389023127373144e-06,
+      "loss": 0.4049,
+      "step": 2675
+    },
+    {
+      "epoch": 2.4976700838769803,
+      "grad_norm": 0.12630419601963916,
+      "learning_rate": 9.302726958923024e-06,
+      "loss": 0.402,
+      "step": 2680
+    },
+    {
+      "epoch": 2.5023299161230197,
+      "grad_norm": 0.12381720089412004,
+      "learning_rate": 9.216430790472904e-06,
+      "loss": 0.395,
+      "step": 2685
+    },
+    {
+      "epoch": 2.5069897483690586,
+      "grad_norm": 0.11590520822783415,
+      "learning_rate": 9.130134622022784e-06,
+      "loss": 0.4054,
+      "step": 2690
+    },
+    {
+      "epoch": 2.511649580615098,
+      "grad_norm": 0.11325007091884573,
+      "learning_rate": 9.043838453572662e-06,
+      "loss": 0.3938,
+      "step": 2695
+    },
+    {
+      "epoch": 2.516309412861137,
+      "grad_norm": 0.11668015282004271,
+      "learning_rate": 8.957542285122542e-06,
+      "loss": 0.3929,
+      "step": 2700
+    },
+    {
+      "epoch": 2.5209692451071763,
+      "grad_norm": 0.12134253905412665,
+      "learning_rate": 8.87124611667242e-06,
+      "loss": 0.3972,
+      "step": 2705
+    },
+    {
+      "epoch": 2.5256290773532153,
+      "grad_norm": 0.11147986068784514,
+      "learning_rate": 8.7849499482223e-06,
+      "loss": 0.3952,
+      "step": 2710
+    },
+    {
+      "epoch": 2.5302889095992542,
+      "grad_norm": 0.1175070018096563,
+      "learning_rate": 8.698653779772179e-06,
+      "loss": 0.3966,
+      "step": 2715
+    },
+    {
+      "epoch": 2.5349487418452936,
+      "grad_norm": 0.11180900390774859,
+      "learning_rate": 8.612357611322058e-06,
+      "loss": 0.4019,
+      "step": 2720
+    },
+    {
+      "epoch": 2.5396085740913326,
+      "grad_norm": 0.11858240406805994,
+      "learning_rate": 8.526061442871937e-06,
+      "loss": 0.3834,
+      "step": 2725
+    },
+    {
+      "epoch": 2.544268406337372,
+      "grad_norm": 0.11048731464663403,
+      "learning_rate": 8.439765274421817e-06,
+      "loss": 0.3873,
+      "step": 2730
+    },
+    {
+      "epoch": 2.548928238583411,
+      "grad_norm": 0.11583203624412546,
+      "learning_rate": 8.353469105971695e-06,
+      "loss": 0.3972,
+      "step": 2735
+    },
+    {
+      "epoch": 2.5535880708294503,
+      "grad_norm": 0.11481442666462767,
+      "learning_rate": 8.267172937521575e-06,
+      "loss": 0.3877,
+      "step": 2740
+    },
+    {
+      "epoch": 2.558247903075489,
+      "grad_norm": 0.1161649710465431,
+      "learning_rate": 8.180876769071453e-06,
+      "loss": 0.3928,
+      "step": 2745
+    },
+    {
+      "epoch": 2.562907735321528,
+      "grad_norm": 0.11339480078650765,
+      "learning_rate": 8.094580600621333e-06,
+      "loss": 0.3954,
+      "step": 2750
+    },
+    {
+      "epoch": 2.5675675675675675,
+      "grad_norm": 0.11642208056067498,
+      "learning_rate": 8.008284432171211e-06,
+      "loss": 0.3952,
+      "step": 2755
+    },
+    {
+      "epoch": 2.572227399813607,
+      "grad_norm": 0.1240778001140626,
+      "learning_rate": 7.921988263721091e-06,
+      "loss": 0.3991,
+      "step": 2760
+    },
+    {
+      "epoch": 2.576887232059646,
+      "grad_norm": 0.1257736322663224,
+      "learning_rate": 7.835692095270971e-06,
+      "loss": 0.4027,
+      "step": 2765
+    },
+    {
+      "epoch": 2.581547064305685,
+      "grad_norm": 0.1086334375913493,
+      "learning_rate": 7.749395926820851e-06,
+      "loss": 0.3853,
+      "step": 2770
+    },
+    {
+      "epoch": 2.586206896551724,
+      "grad_norm": 0.11912421788092825,
+      "learning_rate": 7.663099758370729e-06,
+      "loss": 0.3909,
+      "step": 2775
+    },
+    {
+      "epoch": 2.590866728797763,
+      "grad_norm": 0.11326469460055125,
+      "learning_rate": 7.576803589920608e-06,
+      "loss": 0.3968,
+      "step": 2780
+    },
+    {
+      "epoch": 2.5955265610438025,
+      "grad_norm": 0.11011449384331268,
+      "learning_rate": 7.490507421470487e-06,
+      "loss": 0.3931,
+      "step": 2785
+    },
+    {
+      "epoch": 2.6001863932898415,
+      "grad_norm": 0.10840392156022874,
+      "learning_rate": 7.4042112530203655e-06,
+      "loss": 0.4031,
+      "step": 2790
+    },
+    {
+      "epoch": 2.604846225535881,
+      "grad_norm": 0.12005454329238541,
+      "learning_rate": 7.317915084570245e-06,
+      "loss": 0.4044,
+      "step": 2795
+    },
+    {
+      "epoch": 2.60950605778192,
+      "grad_norm": 0.11302963595923632,
+      "learning_rate": 7.231618916120124e-06,
+      "loss": 0.3921,
+      "step": 2800
+    },
+    {
+      "epoch": 2.6141658900279587,
+      "grad_norm": 0.11048573980819337,
+      "learning_rate": 7.1453227476700035e-06,
+      "loss": 0.4,
+      "step": 2805
+    },
+    {
+      "epoch": 2.618825722273998,
+      "grad_norm": 0.11149852915184379,
+      "learning_rate": 7.059026579219883e-06,
+      "loss": 0.39,
+      "step": 2810
+    },
+    {
+      "epoch": 2.6234855545200375,
+      "grad_norm": 0.10930848145957994,
+      "learning_rate": 6.9727304107697625e-06,
+      "loss": 0.396,
+      "step": 2815
+    },
+    {
+      "epoch": 2.6281453867660765,
+      "grad_norm": 0.11659987957115736,
+      "learning_rate": 6.886434242319641e-06,
+      "loss": 0.408,
+      "step": 2820
+    },
+    {
+      "epoch": 2.6328052190121154,
+      "grad_norm": 0.11459021413610607,
+      "learning_rate": 6.800138073869521e-06,
+      "loss": 0.3997,
+      "step": 2825
+    },
+    {
+      "epoch": 2.637465051258155,
+      "grad_norm": 0.11554336347352002,
+      "learning_rate": 6.713841905419399e-06,
+      "loss": 0.3957,
+      "step": 2830
+    },
+    {
+      "epoch": 2.6421248835041937,
+      "grad_norm": 0.11801567512662643,
+      "learning_rate": 6.627545736969279e-06,
+      "loss": 0.4018,
+      "step": 2835
+    },
+    {
+      "epoch": 2.646784715750233,
+      "grad_norm": 0.11332961592872548,
+      "learning_rate": 6.541249568519157e-06,
+      "loss": 0.3974,
+      "step": 2840
+    },
+    {
+      "epoch": 2.651444547996272,
+      "grad_norm": 0.112839891691877,
+      "learning_rate": 6.454953400069037e-06,
+      "loss": 0.3923,
+      "step": 2845
+    },
+    {
+      "epoch": 2.6561043802423114,
+      "grad_norm": 0.11864176716488593,
+      "learning_rate": 6.368657231618916e-06,
+      "loss": 0.4025,
+      "step": 2850
+    },
+    {
+      "epoch": 2.6607642124883504,
+      "grad_norm": 0.11607211068944612,
+      "learning_rate": 6.282361063168796e-06,
+      "loss": 0.3988,
+      "step": 2855
+    },
+    {
+      "epoch": 2.6654240447343893,
+      "grad_norm": 0.1257781061133214,
+      "learning_rate": 6.196064894718675e-06,
+      "loss": 0.4061,
+      "step": 2860
+    },
+    {
+      "epoch": 2.6700838769804287,
+      "grad_norm": 0.11322644710408915,
+      "learning_rate": 6.109768726268554e-06,
+      "loss": 0.4003,
+      "step": 2865
+    },
+    {
+      "epoch": 2.674743709226468,
+      "grad_norm": 0.11414461506291558,
+      "learning_rate": 6.023472557818433e-06,
+      "loss": 0.3906,
+      "step": 2870
+    },
+    {
+      "epoch": 2.679403541472507,
+      "grad_norm": 0.11816277777817852,
+      "learning_rate": 5.937176389368312e-06,
+      "loss": 0.415,
+      "step": 2875
+    },
+    {
+      "epoch": 2.684063373718546,
+      "grad_norm": 0.11051213389692817,
+      "learning_rate": 5.850880220918191e-06,
+      "loss": 0.4042,
+      "step": 2880
+    },
+    {
+      "epoch": 2.6887232059645854,
+      "grad_norm": 0.12244908283109682,
+      "learning_rate": 5.76458405246807e-06,
+      "loss": 0.3946,
+      "step": 2885
+    },
+    {
+      "epoch": 2.6933830382106243,
+      "grad_norm": 0.11294169860767862,
+      "learning_rate": 5.67828788401795e-06,
+      "loss": 0.3994,
+      "step": 2890
+    },
+    {
+      "epoch": 2.6980428704566637,
+      "grad_norm": 0.11434679459435167,
+      "learning_rate": 5.591991715567829e-06,
+      "loss": 0.4002,
+      "step": 2895
+    },
+    {
+      "epoch": 2.7027027027027026,
+      "grad_norm": 0.11620853180012555,
+      "learning_rate": 5.5056955471177085e-06,
+      "loss": 0.3895,
+      "step": 2900
+    },
+    {
+      "epoch": 2.707362534948742,
+      "grad_norm": 0.11491084489403146,
+      "learning_rate": 5.4193993786675876e-06,
+      "loss": 0.3929,
+      "step": 2905
+    },
+    {
+      "epoch": 2.712022367194781,
+      "grad_norm": 0.11607994911766722,
+      "learning_rate": 5.333103210217467e-06,
+      "loss": 0.4095,
+      "step": 2910
+    },
+    {
+      "epoch": 2.71668219944082,
+      "grad_norm": 0.1195690887906885,
+      "learning_rate": 5.246807041767346e-06,
+      "loss": 0.3961,
+      "step": 2915
+    },
+    {
+      "epoch": 2.7213420316868593,
+      "grad_norm": 0.11206718026401996,
+      "learning_rate": 5.160510873317225e-06,
+      "loss": 0.3915,
+      "step": 2920
+    },
+    {
+      "epoch": 2.7260018639328987,
+      "grad_norm": 0.11277017080298303,
+      "learning_rate": 5.074214704867105e-06,
+      "loss": 0.3907,
+      "step": 2925
+    },
+    {
+      "epoch": 2.7306616961789376,
+      "grad_norm": 0.11567785848339723,
+      "learning_rate": 4.987918536416984e-06,
+      "loss": 0.4027,
+      "step": 2930
+    },
+    {
+      "epoch": 2.7353215284249766,
+      "grad_norm": 0.11071994046092117,
+      "learning_rate": 4.901622367966863e-06,
+      "loss": 0.3925,
+      "step": 2935
+    },
+    {
+      "epoch": 2.739981360671016,
+      "grad_norm": 0.11413536613298304,
+      "learning_rate": 4.815326199516742e-06,
+      "loss": 0.4024,
+      "step": 2940
+    },
+    {
+      "epoch": 2.744641192917055,
+      "grad_norm": 0.11063106074559857,
+      "learning_rate": 4.729030031066621e-06,
+      "loss": 0.3952,
+      "step": 2945
+    },
+    {
+      "epoch": 2.7493010251630943,
+      "grad_norm": 0.11298752654192667,
+      "learning_rate": 4.6427338626165e-06,
+      "loss": 0.4051,
+      "step": 2950
+    },
+    {
+      "epoch": 2.753960857409133,
+      "grad_norm": 0.10842779275670957,
+      "learning_rate": 4.556437694166379e-06,
+      "loss": 0.3889,
+      "step": 2955
+    },
+    {
+      "epoch": 2.7586206896551726,
+      "grad_norm": 0.11389010252297402,
+      "learning_rate": 4.470141525716258e-06,
+      "loss": 0.3915,
+      "step": 2960
+    },
+    {
+      "epoch": 2.7632805219012115,
+      "grad_norm": 0.1121236713425753,
+      "learning_rate": 4.383845357266138e-06,
+      "loss": 0.3958,
+      "step": 2965
+    },
+    {
+      "epoch": 2.7679403541472505,
+      "grad_norm": 0.11286882820617526,
+      "learning_rate": 4.297549188816017e-06,
+      "loss": 0.4006,
+      "step": 2970
+    },
+    {
+      "epoch": 2.77260018639329,
+      "grad_norm": 0.11329231357821927,
+      "learning_rate": 4.211253020365896e-06,
+      "loss": 0.3985,
+      "step": 2975
+    },
+    {
+      "epoch": 2.7772600186393293,
+      "grad_norm": 0.11092813276847187,
+      "learning_rate": 4.124956851915775e-06,
+      "loss": 0.3924,
+      "step": 2980
+    },
+    {
+      "epoch": 2.781919850885368,
+      "grad_norm": 0.1118971253075082,
+      "learning_rate": 4.0386606834656544e-06,
+      "loss": 0.4032,
+      "step": 2985
+    },
+    {
+      "epoch": 2.786579683131407,
+      "grad_norm": 0.11176008258355923,
+      "learning_rate": 3.9523645150155335e-06,
+      "loss": 0.3975,
+      "step": 2990
+    },
+    {
+      "epoch": 2.7912395153774465,
+      "grad_norm": 0.10943035996887142,
+      "learning_rate": 3.8660683465654126e-06,
+      "loss": 0.3919,
+      "step": 2995
+    },
+    {
+      "epoch": 2.7958993476234855,
+      "grad_norm": 0.11483118883681243,
+      "learning_rate": 3.779772178115292e-06,
+      "loss": 0.3912,
+      "step": 3000
+    },
+    {
+      "epoch": 2.800559179869525,
+      "grad_norm": 0.1126488719999867,
+      "learning_rate": 3.693476009665171e-06,
+      "loss": 0.4027,
+      "step": 3005
+    },
+    {
+      "epoch": 2.805219012115564,
+      "grad_norm": 0.11286045873945728,
+      "learning_rate": 3.6071798412150506e-06,
+      "loss": 0.408,
+      "step": 3010
+    },
+    {
+      "epoch": 2.809878844361603,
+      "grad_norm": 0.10992331672038576,
+      "learning_rate": 3.5208836727649297e-06,
+      "loss": 0.387,
+      "step": 3015
+    },
+    {
+      "epoch": 2.814538676607642,
+      "grad_norm": 0.12352543400322044,
+      "learning_rate": 3.434587504314809e-06,
+      "loss": 0.3895,
+      "step": 3020
+    },
+    {
+      "epoch": 2.819198508853681,
+      "grad_norm": 0.11207025768331855,
+      "learning_rate": 3.348291335864688e-06,
+      "loss": 0.4045,
+      "step": 3025
+    },
+    {
+      "epoch": 2.8238583410997204,
+      "grad_norm": 0.10812227491647716,
+      "learning_rate": 3.2619951674145674e-06,
+      "loss": 0.41,
+      "step": 3030
+    },
+    {
+      "epoch": 2.8285181733457594,
+      "grad_norm": 0.11260215690311579,
+      "learning_rate": 3.1756989989644464e-06,
+      "loss": 0.3997,
+      "step": 3035
+    },
+    {
+      "epoch": 2.8331780055917988,
+      "grad_norm": 0.10737428204132128,
+      "learning_rate": 3.089402830514325e-06,
+      "loss": 0.3933,
+      "step": 3040
+    },
+    {
+      "epoch": 2.8378378378378377,
+      "grad_norm": 0.11227663694323252,
+      "learning_rate": 3.0031066620642046e-06,
+      "loss": 0.3913,
+      "step": 3045
+    },
+    {
+      "epoch": 2.842497670083877,
+      "grad_norm": 0.11546861018209845,
+      "learning_rate": 2.9168104936140837e-06,
+      "loss": 0.3989,
+      "step": 3050
+    },
+    {
+      "epoch": 2.847157502329916,
+      "grad_norm": 0.12043073090129779,
+      "learning_rate": 2.8305143251639627e-06,
+      "loss": 0.3955,
+      "step": 3055
+    },
+    {
+      "epoch": 2.8518173345759554,
+      "grad_norm": 0.10856123376161597,
+      "learning_rate": 2.7442181567138422e-06,
+      "loss": 0.3917,
+      "step": 3060
+    },
+    {
+      "epoch": 2.8564771668219944,
+      "grad_norm": 0.11152730615014406,
+      "learning_rate": 2.6579219882637213e-06,
+      "loss": 0.3927,
+      "step": 3065
+    },
+    {
+      "epoch": 2.8611369990680338,
+      "grad_norm": 0.11384123168547186,
+      "learning_rate": 2.5716258198136004e-06,
+      "loss": 0.3917,
+      "step": 3070
+    },
+    {
+      "epoch": 2.8657968313140727,
+      "grad_norm": 0.11257735330076836,
+      "learning_rate": 2.4853296513634795e-06,
+      "loss": 0.3968,
+      "step": 3075
+    },
+    {
+      "epoch": 2.8704566635601116,
+      "grad_norm": 0.11118072879806912,
+      "learning_rate": 2.399033482913359e-06,
+      "loss": 0.402,
+      "step": 3080
+    },
+    {
+      "epoch": 2.875116495806151,
+      "grad_norm": 0.10661830960311815,
+      "learning_rate": 2.312737314463238e-06,
+      "loss": 0.387,
+      "step": 3085
+    },
+    {
+      "epoch": 2.87977632805219,
+      "grad_norm": 0.11131116669341895,
+      "learning_rate": 2.226441146013117e-06,
+      "loss": 0.3925,
+      "step": 3090
+    },
+    {
+      "epoch": 2.8844361602982294,
+      "grad_norm": 0.10572688249561822,
+      "learning_rate": 2.140144977562996e-06,
+      "loss": 0.392,
+      "step": 3095
+    },
+    {
+      "epoch": 2.8890959925442683,
+      "grad_norm": 0.10952930062029194,
+      "learning_rate": 2.0538488091128757e-06,
+      "loss": 0.3984,
+      "step": 3100
+    },
+    {
+      "epoch": 2.8937558247903077,
+      "grad_norm": 0.11226013593138168,
+      "learning_rate": 1.9675526406627547e-06,
+      "loss": 0.386,
+      "step": 3105
+    },
+    {
+      "epoch": 2.8984156570363466,
+      "grad_norm": 0.11166941064892873,
+      "learning_rate": 1.8812564722126338e-06,
+      "loss": 0.4042,
+      "step": 3110
+    },
+    {
+      "epoch": 2.9030754892823856,
+      "grad_norm": 0.11100962355723291,
+      "learning_rate": 1.794960303762513e-06,
+      "loss": 0.3845,
+      "step": 3115
+    },
+    {
+      "epoch": 2.907735321528425,
+      "grad_norm": 0.11083283596472956,
+      "learning_rate": 1.7086641353123924e-06,
+      "loss": 0.3976,
+      "step": 3120
+    },
+    {
+      "epoch": 2.9123951537744643,
+      "grad_norm": 0.10761040599953665,
+      "learning_rate": 1.6223679668622715e-06,
+      "loss": 0.3869,
+      "step": 3125
+    },
+    {
+      "epoch": 2.9170549860205033,
+      "grad_norm": 0.11752701747700955,
+      "learning_rate": 1.5360717984121505e-06,
+      "loss": 0.3856,
+      "step": 3130
+    },
+    {
+      "epoch": 2.9217148182665422,
+      "grad_norm": 0.11262275125120517,
+      "learning_rate": 1.4497756299620296e-06,
+      "loss": 0.3958,
+      "step": 3135
+    },
+    {
+      "epoch": 2.9263746505125816,
+      "grad_norm": 0.11188745420795321,
+      "learning_rate": 1.363479461511909e-06,
+      "loss": 0.3858,
+      "step": 3140
+    },
+    {
+      "epoch": 2.9310344827586206,
+      "grad_norm": 0.10935869748960521,
+      "learning_rate": 1.277183293061788e-06,
+      "loss": 0.3979,
+      "step": 3145
+    },
+    {
+      "epoch": 2.93569431500466,
+      "grad_norm": 0.10792330843455245,
+      "learning_rate": 1.1908871246116673e-06,
+      "loss": 0.4015,
+      "step": 3150
+    },
+    {
+      "epoch": 2.940354147250699,
+      "grad_norm": 0.10989602088820866,
+      "learning_rate": 1.1045909561615463e-06,
+      "loss": 0.4,
+      "step": 3155
+    },
+    {
+      "epoch": 2.9450139794967383,
+      "grad_norm": 0.11439059237048368,
+      "learning_rate": 1.0182947877114256e-06,
+      "loss": 0.3966,
+      "step": 3160
+    },
+    {
+      "epoch": 2.949673811742777,
+      "grad_norm": 0.10987992776209123,
+      "learning_rate": 9.319986192613048e-07,
+      "loss": 0.3897,
+      "step": 3165
+    },
+    {
+      "epoch": 2.954333643988816,
+      "grad_norm": 0.10993119485555218,
+      "learning_rate": 8.45702450811184e-07,
+      "loss": 0.3976,
+      "step": 3170
+    },
+    {
+      "epoch": 2.9589934762348555,
+      "grad_norm": 0.10961246110438148,
+      "learning_rate": 7.594062823610632e-07,
+      "loss": 0.3986,
+      "step": 3175
+    },
+    {
+      "epoch": 2.963653308480895,
+      "grad_norm": 0.11191814527413249,
+      "learning_rate": 6.731101139109423e-07,
+      "loss": 0.399,
+      "step": 3180
+    },
+    {
+      "epoch": 2.968313140726934,
+      "grad_norm": 0.10780466688814475,
+      "learning_rate": 5.868139454608215e-07,
+      "loss": 0.3927,
+      "step": 3185
+    },
+    {
+      "epoch": 2.972972972972973,
+      "grad_norm": 0.1087111548107089,
+      "learning_rate": 5.005177770107007e-07,
+      "loss": 0.3905,
+      "step": 3190
+    },
+    {
+      "epoch": 2.977632805219012,
+      "grad_norm": 0.10853146248527311,
+      "learning_rate": 4.142216085605799e-07,
+      "loss": 0.3952,
+      "step": 3195
+    },
+    {
+      "epoch": 2.982292637465051,
+      "grad_norm": 0.1107565356873898,
+      "learning_rate": 3.279254401104591e-07,
+      "loss": 0.4026,
+      "step": 3200
+    },
+    {
+      "epoch": 2.9869524697110905,
+      "grad_norm": 0.11084938177916184,
+      "learning_rate": 2.416292716603383e-07,
+      "loss": 0.394,
+      "step": 3205
+    },
+    {
+      "epoch": 2.9916123019571295,
+      "grad_norm": 0.10980835190651844,
+      "learning_rate": 1.5533310321021747e-07,
+      "loss": 0.3963,
+      "step": 3210
+    },
+    {
+      "epoch": 2.996272134203169,
+      "grad_norm": 0.10906331971131258,
+      "learning_rate": 6.903693476009665e-08,
+      "loss": 0.4,
+      "step": 3215
+    },
+    {
+      "epoch": 3.0,
+      "step": 3219,
+      "total_flos": 1463203422797824.0,
+      "train_loss": 0.44736327451739705,
+      "train_runtime": 14518.7762,
+      "train_samples_per_second": 3.547,
+      "train_steps_per_second": 0.222
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 3219,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1463203422797824.0,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}