{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 4.407227853680035,
  "eval_steps": 500,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.004407227853680035,
      "grad_norm": 8.981419563293457,
      "learning_rate": 1.8e-06,
      "loss": 0.7093,
      "step": 10
    },
    {
      "epoch": 0.00881445570736007,
      "grad_norm": 5.4393768310546875,
      "learning_rate": 3.8e-06,
      "loss": 0.6137,
      "step": 20
    },
    {
      "epoch": 0.013221683561040106,
      "grad_norm": 1.8733208179473877,
      "learning_rate": 5.8e-06,
      "loss": 0.3678,
      "step": 30
    },
    {
      "epoch": 0.01762891141472014,
      "grad_norm": 2.020907402038574,
      "learning_rate": 7.8e-06,
      "loss": 0.2417,
      "step": 40
    },
    {
      "epoch": 0.022036139268400177,
      "grad_norm": 1.422074317932129,
      "learning_rate": 9.800000000000001e-06,
      "loss": 0.218,
      "step": 50
    },
    {
      "epoch": 0.026443367122080213,
      "grad_norm": 1.8262766599655151,
      "learning_rate": 1.18e-05,
      "loss": 0.1871,
      "step": 60
    },
    {
      "epoch": 0.030850594975760245,
      "grad_norm": 1.3047466278076172,
      "learning_rate": 1.3800000000000002e-05,
      "loss": 0.14,
      "step": 70
    },
    {
      "epoch": 0.03525782282944028,
      "grad_norm": 2.0399861335754395,
      "learning_rate": 1.58e-05,
      "loss": 0.1485,
      "step": 80
    },
    {
      "epoch": 0.03966505068312032,
      "grad_norm": 1.3455944061279297,
      "learning_rate": 1.78e-05,
      "loss": 0.1179,
      "step": 90
    },
    {
      "epoch": 0.044072278536800354,
      "grad_norm": 0.7521724700927734,
      "learning_rate": 1.9800000000000004e-05,
      "loss": 0.127,
      "step": 100
    },
    {
      "epoch": 0.04847950639048039,
      "grad_norm": 1.2108312845230103,
      "learning_rate": 2.18e-05,
      "loss": 0.1332,
      "step": 110
    },
    {
      "epoch": 0.052886734244160426,
      "grad_norm": 1.6099722385406494,
      "learning_rate": 2.38e-05,
      "loss": 0.1393,
      "step": 120
    },
    {
      "epoch": 0.05729396209784046,
      "grad_norm": 1.0002557039260864,
      "learning_rate": 2.58e-05,
      "loss": 0.104,
      "step": 130
    },
    {
      "epoch": 0.06170118995152049,
      "grad_norm": 1.4535717964172363,
      "learning_rate": 2.7800000000000005e-05,
      "loss": 0.1093,
      "step": 140
    },
    {
      "epoch": 0.06610841780520053,
      "grad_norm": 1.6055448055267334,
      "learning_rate": 2.98e-05,
      "loss": 0.0997,
      "step": 150
    },
    {
      "epoch": 0.07051564565888056,
      "grad_norm": 1.1183305978775024,
      "learning_rate": 3.18e-05,
      "loss": 0.1009,
      "step": 160
    },
    {
      "epoch": 0.0749228735125606,
      "grad_norm": 1.49135160446167,
      "learning_rate": 3.38e-05,
      "loss": 0.0994,
      "step": 170
    },
    {
      "epoch": 0.07933010136624064,
      "grad_norm": 0.878442645072937,
      "learning_rate": 3.58e-05,
      "loss": 0.0908,
      "step": 180
    },
    {
      "epoch": 0.08373732921992066,
      "grad_norm": 1.150046944618225,
      "learning_rate": 3.7800000000000004e-05,
      "loss": 0.0934,
      "step": 190
    },
    {
      "epoch": 0.08814455707360071,
      "grad_norm": 1.331091284751892,
      "learning_rate": 3.9800000000000005e-05,
      "loss": 0.0947,
      "step": 200
    },
    {
      "epoch": 0.09255178492728074,
      "grad_norm": 0.9203300476074219,
      "learning_rate": 4.18e-05,
      "loss": 0.0934,
      "step": 210
    },
    {
      "epoch": 0.09695901278096078,
      "grad_norm": 1.698358178138733,
      "learning_rate": 4.38e-05,
      "loss": 0.0926,
      "step": 220
    },
    {
      "epoch": 0.10136624063464081,
      "grad_norm": 1.2027570009231567,
      "learning_rate": 4.58e-05,
      "loss": 0.0816,
      "step": 230
    },
    {
      "epoch": 0.10577346848832085,
      "grad_norm": 1.0524420738220215,
      "learning_rate": 4.78e-05,
      "loss": 0.0888,
      "step": 240
    },
    {
      "epoch": 0.11018069634200088,
      "grad_norm": 1.1894065141677856,
      "learning_rate": 4.9800000000000004e-05,
      "loss": 0.0765,
      "step": 250
    },
    {
      "epoch": 0.11458792419568092,
      "grad_norm": 0.8286505937576294,
      "learning_rate": 5.1800000000000005e-05,
      "loss": 0.0762,
      "step": 260
    },
    {
      "epoch": 0.11899515204936095,
      "grad_norm": 1.3551204204559326,
      "learning_rate": 5.380000000000001e-05,
      "loss": 0.0699,
      "step": 270
    },
    {
      "epoch": 0.12340237990304098,
      "grad_norm": 0.57041996717453,
      "learning_rate": 5.580000000000001e-05,
      "loss": 0.0849,
      "step": 280
    },
    {
      "epoch": 0.127809607756721,
      "grad_norm": 1.1933631896972656,
      "learning_rate": 5.7799999999999995e-05,
      "loss": 0.0591,
      "step": 290
    },
    {
      "epoch": 0.13221683561040107,
      "grad_norm": 1.1949830055236816,
      "learning_rate": 5.9800000000000003e-05,
      "loss": 0.0794,
      "step": 300
    },
    {
      "epoch": 0.1366240634640811,
      "grad_norm": 0.6119535565376282,
      "learning_rate": 6.18e-05,
      "loss": 0.0698,
      "step": 310
    },
    {
      "epoch": 0.14103129131776113,
      "grad_norm": 0.8757414817810059,
      "learning_rate": 6.38e-05,
      "loss": 0.0634,
      "step": 320
    },
    {
      "epoch": 0.14543851917144116,
      "grad_norm": 1.4264259338378906,
      "learning_rate": 6.58e-05,
      "loss": 0.0807,
      "step": 330
    },
    {
      "epoch": 0.1498457470251212,
      "grad_norm": 0.8428658246994019,
      "learning_rate": 6.780000000000001e-05,
      "loss": 0.0554,
      "step": 340
    },
    {
      "epoch": 0.15425297487880124,
      "grad_norm": 1.0794216394424438,
      "learning_rate": 6.98e-05,
      "loss": 0.0646,
      "step": 350
    },
    {
      "epoch": 0.15866020273248127,
      "grad_norm": 0.9395310282707214,
      "learning_rate": 7.18e-05,
      "loss": 0.0574,
      "step": 360
    },
    {
      "epoch": 0.1630674305861613,
      "grad_norm": 0.9479997158050537,
      "learning_rate": 7.38e-05,
      "loss": 0.0488,
      "step": 370
    },
    {
      "epoch": 0.16747465843984133,
      "grad_norm": 1.0815507173538208,
      "learning_rate": 7.58e-05,
      "loss": 0.0551,
      "step": 380
    },
    {
      "epoch": 0.17188188629352139,
      "grad_norm": 1.0078706741333008,
      "learning_rate": 7.780000000000001e-05,
      "loss": 0.0639,
      "step": 390
    },
    {
      "epoch": 0.17628911414720141,
      "grad_norm": 0.562309205532074,
      "learning_rate": 7.98e-05,
      "loss": 0.0544,
      "step": 400
    },
    {
      "epoch": 0.18069634200088144,
      "grad_norm": 1.353854775428772,
      "learning_rate": 8.18e-05,
      "loss": 0.0576,
      "step": 410
    },
    {
      "epoch": 0.18510356985456147,
      "grad_norm": 1.3798199892044067,
      "learning_rate": 8.38e-05,
      "loss": 0.0636,
      "step": 420
    },
    {
      "epoch": 0.18951079770824153,
      "grad_norm": 1.1835122108459473,
      "learning_rate": 8.58e-05,
      "loss": 0.0552,
      "step": 430
    },
    {
      "epoch": 0.19391802556192156,
      "grad_norm": 0.805357038974762,
      "learning_rate": 8.78e-05,
      "loss": 0.0669,
      "step": 440
    },
    {
      "epoch": 0.1983252534156016,
      "grad_norm": 1.0877079963684082,
      "learning_rate": 8.98e-05,
      "loss": 0.0601,
      "step": 450
    },
    {
      "epoch": 0.20273248126928162,
      "grad_norm": 0.7082098126411438,
      "learning_rate": 9.180000000000001e-05,
      "loss": 0.0553,
      "step": 460
    },
    {
      "epoch": 0.20713970912296165,
      "grad_norm": 0.7886426448822021,
      "learning_rate": 9.38e-05,
      "loss": 0.0588,
      "step": 470
    },
    {
      "epoch": 0.2115469369766417,
      "grad_norm": 0.9061591625213623,
      "learning_rate": 9.58e-05,
      "loss": 0.0532,
      "step": 480
    },
    {
      "epoch": 0.21595416483032173,
      "grad_norm": 0.9045799970626831,
      "learning_rate": 9.78e-05,
      "loss": 0.0576,
      "step": 490
    },
    {
      "epoch": 0.22036139268400176,
      "grad_norm": 1.5410330295562744,
      "learning_rate": 9.98e-05,
      "loss": 0.0482,
      "step": 500
    },
    {
      "epoch": 0.2247686205376818,
      "grad_norm": 1.0996057987213135,
      "learning_rate": 9.9999778549206e-05,
      "loss": 0.0567,
      "step": 510
    },
    {
      "epoch": 0.22917584839136185,
      "grad_norm": 0.9273228645324707,
      "learning_rate": 9.999901304280685e-05,
      "loss": 0.0474,
      "step": 520
    },
    {
      "epoch": 0.23358307624504188,
      "grad_norm": 0.8598151803016663,
      "learning_rate": 9.999770075521164e-05,
      "loss": 0.0549,
      "step": 530
    },
    {
      "epoch": 0.2379903040987219,
      "grad_norm": 0.9834341406822205,
      "learning_rate": 9.99958417007713e-05,
      "loss": 0.0671,
      "step": 540
    },
    {
      "epoch": 0.24239753195240193,
      "grad_norm": 1.224439263343811,
      "learning_rate": 9.999343589981615e-05,
      "loss": 0.05,
      "step": 550
    },
    {
      "epoch": 0.24680475980608196,
      "grad_norm": 0.6575846076011658,
      "learning_rate": 9.999048337865568e-05,
      "loss": 0.0546,
      "step": 560
    },
    {
      "epoch": 0.251211987659762,
      "grad_norm": 1.0721476078033447,
      "learning_rate": 9.998698416957815e-05,
      "loss": 0.0561,
      "step": 570
    },
    {
      "epoch": 0.255619215513442,
      "grad_norm": 0.6784175038337708,
      "learning_rate": 9.998293831085037e-05,
      "loss": 0.0673,
      "step": 580
    },
    {
      "epoch": 0.2600264433671221,
      "grad_norm": 1.1893671751022339,
      "learning_rate": 9.997834584671719e-05,
      "loss": 0.05,
      "step": 590
    },
    {
      "epoch": 0.26443367122080214,
      "grad_norm": 1.0835838317871094,
      "learning_rate": 9.997320682740107e-05,
      "loss": 0.0514,
      "step": 600
    },
    {
      "epoch": 0.26884089907448216,
      "grad_norm": 0.9374244809150696,
      "learning_rate": 9.996752130910149e-05,
      "loss": 0.0528,
      "step": 610
    },
    {
      "epoch": 0.2732481269281622,
      "grad_norm": 1.2545331716537476,
      "learning_rate": 9.99612893539944e-05,
      "loss": 0.062,
      "step": 620
    },
    {
      "epoch": 0.2776553547818422,
      "grad_norm": 0.7685592770576477,
      "learning_rate": 9.995451103023144e-05,
      "loss": 0.0562,
      "step": 630
    },
    {
      "epoch": 0.28206258263552225,
      "grad_norm": 0.8536335229873657,
      "learning_rate": 9.994718641193928e-05,
      "loss": 0.0465,
      "step": 640
    },
    {
      "epoch": 0.2864698104892023,
      "grad_norm": 1.3706673383712769,
      "learning_rate": 9.993931557921874e-05,
      "loss": 0.0464,
      "step": 650
    },
    {
      "epoch": 0.2908770383428823,
      "grad_norm": 0.7187947630882263,
      "learning_rate": 9.993089861814402e-05,
      "loss": 0.0526,
      "step": 660
    },
    {
      "epoch": 0.29528426619656234,
      "grad_norm": 0.617475688457489,
      "learning_rate": 9.992193562076166e-05,
      "loss": 0.0386,
      "step": 670
    },
    {
      "epoch": 0.2996914940502424,
      "grad_norm": 1.2751219272613525,
      "learning_rate": 9.991242668508954e-05,
      "loss": 0.057,
      "step": 680
    },
    {
      "epoch": 0.30409872190392245,
      "grad_norm": 0.8712632060050964,
      "learning_rate": 9.990237191511587e-05,
      "loss": 0.0435,
      "step": 690
    },
    {
      "epoch": 0.3085059497576025,
      "grad_norm": 0.6670446395874023,
      "learning_rate": 9.989177142079802e-05,
      "loss": 0.0442,
      "step": 700
    },
    {
      "epoch": 0.3129131776112825,
      "grad_norm": 1.0403811931610107,
      "learning_rate": 9.988062531806126e-05,
      "loss": 0.0491,
      "step": 710
    },
    {
      "epoch": 0.31732040546496254,
      "grad_norm": 0.7482120990753174,
      "learning_rate": 9.986893372879762e-05,
      "loss": 0.0398,
      "step": 720
    },
    {
      "epoch": 0.32172763331864257,
      "grad_norm": 0.6806433200836182,
      "learning_rate": 9.985669678086443e-05,
      "loss": 0.0517,
      "step": 730
    },
    {
      "epoch": 0.3261348611723226,
      "grad_norm": 0.9633274078369141,
      "learning_rate": 9.984391460808298e-05,
      "loss": 0.0505,
      "step": 740
    },
    {
      "epoch": 0.33054208902600263,
      "grad_norm": 0.8663274049758911,
      "learning_rate": 9.983058735023709e-05,
      "loss": 0.0468,
      "step": 750
    },
    {
      "epoch": 0.33494931687968266,
      "grad_norm": 0.6330543756484985,
      "learning_rate": 9.98167151530715e-05,
      "loss": 0.0495,
      "step": 760
    },
    {
      "epoch": 0.33935654473336274,
      "grad_norm": 0.851847767829895,
      "learning_rate": 9.980229816829034e-05,
      "loss": 0.0442,
      "step": 770
    },
    {
      "epoch": 0.34376377258704277,
      "grad_norm": 0.820501446723938,
      "learning_rate": 9.978733655355544e-05,
      "loss": 0.0484,
      "step": 780
    },
    {
      "epoch": 0.3481710004407228,
      "grad_norm": 0.7086553573608398,
      "learning_rate": 9.977183047248464e-05,
      "loss": 0.0359,
      "step": 790
    },
    {
      "epoch": 0.35257822829440283,
      "grad_norm": 0.4318119287490845,
      "learning_rate": 9.975578009464992e-05,
      "loss": 0.0513,
      "step": 800
    },
    {
      "epoch": 0.35698545614808286,
      "grad_norm": 0.3257259428501129,
      "learning_rate": 9.97391855955757e-05,
      "loss": 0.0393,
      "step": 810
    },
    {
      "epoch": 0.3613926840017629,
      "grad_norm": 0.7262775897979736,
      "learning_rate": 9.972204715673669e-05,
      "loss": 0.0574,
      "step": 820
    },
    {
      "epoch": 0.3657999118554429,
      "grad_norm": 0.6231600642204285,
      "learning_rate": 9.970436496555617e-05,
      "loss": 0.0475,
      "step": 830
    },
    {
      "epoch": 0.37020713970912295,
      "grad_norm": 0.6444229483604431,
      "learning_rate": 9.968613921540373e-05,
      "loss": 0.0422,
      "step": 840
    },
    {
      "epoch": 0.374614367562803,
      "grad_norm": 0.43685343861579895,
      "learning_rate": 9.966737010559326e-05,
      "loss": 0.04,
      "step": 850
    },
    {
      "epoch": 0.37902159541648306,
      "grad_norm": 0.7441055178642273,
      "learning_rate": 9.964805784138072e-05,
      "loss": 0.0478,
      "step": 860
    },
    {
      "epoch": 0.3834288232701631,
      "grad_norm": 0.6977639198303223,
      "learning_rate": 9.962820263396195e-05,
      "loss": 0.0437,
      "step": 870
    },
    {
      "epoch": 0.3878360511238431,
      "grad_norm": 0.7057613730430603,
      "learning_rate": 9.960780470047033e-05,
      "loss": 0.0426,
      "step": 880
    },
    {
      "epoch": 0.39224327897752315,
      "grad_norm": 1.0681592226028442,
      "learning_rate": 9.958686426397437e-05,
      "loss": 0.0417,
      "step": 890
    },
    {
      "epoch": 0.3966505068312032,
      "grad_norm": 0.7268592119216919,
      "learning_rate": 9.956538155347534e-05,
      "loss": 0.0487,
      "step": 900
    },
    {
      "epoch": 0.4010577346848832,
      "grad_norm": 0.7201013565063477,
      "learning_rate": 9.95433568039047e-05,
      "loss": 0.0452,
      "step": 910
    },
    {
      "epoch": 0.40546496253856323,
      "grad_norm": 0.9344309568405151,
      "learning_rate": 9.952079025612162e-05,
      "loss": 0.0478,
      "step": 920
    },
    {
      "epoch": 0.40987219039224326,
      "grad_norm": 0.7012525200843811,
      "learning_rate": 9.949768215691022e-05,
      "loss": 0.0615,
      "step": 930
    },
    {
      "epoch": 0.4142794182459233,
      "grad_norm": 0.9316278100013733,
      "learning_rate": 9.9474032758977e-05,
      "loss": 0.0386,
      "step": 940
    },
    {
      "epoch": 0.4186866460996034,
      "grad_norm": 0.5106643438339233,
      "learning_rate": 9.944984232094794e-05,
      "loss": 0.0389,
      "step": 950
    },
    {
      "epoch": 0.4230938739532834,
      "grad_norm": 1.0993642807006836,
      "learning_rate": 9.942511110736584e-05,
      "loss": 0.0551,
      "step": 960
    },
    {
      "epoch": 0.42750110180696343,
      "grad_norm": 0.8568926453590393,
      "learning_rate": 9.939983938868726e-05,
      "loss": 0.0451,
      "step": 970
    },
    {
      "epoch": 0.43190832966064346,
      "grad_norm": 0.4553903341293335,
      "learning_rate": 9.93740274412797e-05,
      "loss": 0.0447,
      "step": 980
    },
    {
      "epoch": 0.4363155575143235,
      "grad_norm": 0.6275448799133301,
      "learning_rate": 9.934767554741846e-05,
      "loss": 0.0479,
      "step": 990
    },
    {
      "epoch": 0.4407227853680035,
      "grad_norm": 0.424888551235199,
      "learning_rate": 9.932078399528361e-05,
      "loss": 0.046,
      "step": 1000
    },
    {
      "epoch": 0.44513001322168355,
      "grad_norm": 0.5453678369522095,
      "learning_rate": 9.929335307895689e-05,
      "loss": 0.0506,
      "step": 1010
    },
    {
      "epoch": 0.4495372410753636,
      "grad_norm": 0.7597624063491821,
      "learning_rate": 9.926538309841839e-05,
      "loss": 0.0364,
      "step": 1020
    },
    {
      "epoch": 0.4539444689290436,
      "grad_norm": 0.591235339641571,
      "learning_rate": 9.923687435954334e-05,
      "loss": 0.0426,
      "step": 1030
    },
    {
      "epoch": 0.4583516967827237,
      "grad_norm": 0.8528213500976562,
      "learning_rate": 9.920782717409873e-05,
      "loss": 0.0348,
      "step": 1040
    },
    {
      "epoch": 0.4627589246364037,
      "grad_norm": 0.7315279841423035,
      "learning_rate": 9.917824185973994e-05,
      "loss": 0.0415,
      "step": 1050
    },
    {
      "epoch": 0.46716615249008375,
      "grad_norm": 0.5451973676681519,
      "learning_rate": 9.914811874000723e-05,
      "loss": 0.0471,
      "step": 1060
    },
    {
      "epoch": 0.4715733803437638,
      "grad_norm": 0.6954655051231384,
      "learning_rate": 9.911745814432218e-05,
      "loss": 0.0379,
      "step": 1070
    },
    {
      "epoch": 0.4759806081974438,
      "grad_norm": 0.6766794919967651,
      "learning_rate": 9.90862604079842e-05,
      "loss": 0.0428,
      "step": 1080
    },
    {
      "epoch": 0.48038783605112384,
      "grad_norm": 0.5949427485466003,
      "learning_rate": 9.90545258721667e-05,
      "loss": 0.0371,
      "step": 1090
    },
    {
      "epoch": 0.48479506390480387,
      "grad_norm": 0.433505117893219,
      "learning_rate": 9.90222548839135e-05,
      "loss": 0.0371,
      "step": 1100
    },
    {
      "epoch": 0.4892022917584839,
      "grad_norm": 0.6289669275283813,
      "learning_rate": 9.898944779613495e-05,
      "loss": 0.0339,
      "step": 1110
    },
    {
      "epoch": 0.4936095196121639,
      "grad_norm": 0.5964171886444092,
      "learning_rate": 9.89561049676041e-05,
      "loss": 0.0347,
      "step": 1120
    },
    {
      "epoch": 0.498016747465844,
      "grad_norm": 0.49216127395629883,
      "learning_rate": 9.89222267629528e-05,
      "loss": 0.0397,
      "step": 1130
    },
    {
      "epoch": 0.502423975319524,
      "grad_norm": 1.0050740242004395,
      "learning_rate": 9.888781355266763e-05,
      "loss": 0.0332,
      "step": 1140
    },
    {
      "epoch": 0.5068312031732041,
      "grad_norm": 0.6160919070243835,
      "learning_rate": 9.885286571308598e-05,
      "loss": 0.0379,
      "step": 1150
    },
    {
      "epoch": 0.511238431026884,
      "grad_norm": 0.5290043354034424,
      "learning_rate": 9.881738362639182e-05,
      "loss": 0.0448,
      "step": 1160
    },
    {
      "epoch": 0.5156456588805641,
      "grad_norm": 0.6544097065925598,
      "learning_rate": 9.878136768061154e-05,
      "loss": 0.0407,
      "step": 1170
    },
    {
      "epoch": 0.5200528867342442,
      "grad_norm": 0.6088456511497498,
      "learning_rate": 9.874481826960979e-05,
      "loss": 0.0413,
      "step": 1180
    },
    {
      "epoch": 0.5244601145879242,
      "grad_norm": 0.7381882667541504,
      "learning_rate": 9.870773579308503e-05,
      "loss": 0.0408,
      "step": 1190
    },
    {
      "epoch": 0.5288673424416043,
      "grad_norm": 0.4340173900127411,
      "learning_rate": 9.867012065656533e-05,
      "loss": 0.048,
      "step": 1200
    },
    {
      "epoch": 0.5332745702952842,
      "grad_norm": 0.8161675333976746,
      "learning_rate": 9.863197327140376e-05,
      "loss": 0.0466,
      "step": 1210
    },
    {
      "epoch": 0.5376817981489643,
      "grad_norm": 0.5402148365974426,
      "learning_rate": 9.859329405477403e-05,
      "loss": 0.0418,
      "step": 1220
    },
    {
      "epoch": 0.5420890260026443,
      "grad_norm": 0.6980718970298767,
      "learning_rate": 9.855408342966585e-05,
      "loss": 0.0418,
      "step": 1230
    },
    {
      "epoch": 0.5464962538563244,
      "grad_norm": 0.5462673306465149,
      "learning_rate": 9.851434182488033e-05,
      "loss": 0.046,
      "step": 1240
    },
    {
      "epoch": 0.5509034817100044,
      "grad_norm": 0.7031043171882629,
      "learning_rate": 9.84740696750253e-05,
      "loss": 0.0375,
      "step": 1250
    },
    {
      "epoch": 0.5553107095636844,
      "grad_norm": 0.6925310492515564,
      "learning_rate": 9.843326742051055e-05,
      "loss": 0.0375,
      "step": 1260
    },
    {
      "epoch": 0.5597179374173645,
      "grad_norm": 0.6021462678909302,
      "learning_rate": 9.839193550754297e-05,
      "loss": 0.0338,
      "step": 1270
    },
    {
      "epoch": 0.5641251652710445,
      "grad_norm": 0.6178574562072754,
      "learning_rate": 9.835007438812177e-05,
      "loss": 0.043,
      "step": 1280
    },
    {
      "epoch": 0.5685323931247246,
      "grad_norm": 0.7377062439918518,
      "learning_rate": 9.830768452003341e-05,
      "loss": 0.0411,
      "step": 1290
    },
    {
      "epoch": 0.5729396209784046,
      "grad_norm": 0.6421175599098206,
      "learning_rate": 9.826476636684671e-05,
      "loss": 0.0451,
      "step": 1300
    },
    {
      "epoch": 0.5773468488320846,
      "grad_norm": 0.595180094242096,
      "learning_rate": 9.822132039790773e-05,
      "loss": 0.0357,
      "step": 1310
    },
    {
      "epoch": 0.5817540766857646,
      "grad_norm": 0.8394598960876465,
      "learning_rate": 9.817734708833461e-05,
      "loss": 0.0448,
      "step": 1320
    },
    {
      "epoch": 0.5861613045394447,
      "grad_norm": 0.8743864297866821,
      "learning_rate": 9.813284691901243e-05,
      "loss": 0.04,
      "step": 1330
    },
    {
      "epoch": 0.5905685323931247,
      "grad_norm": 0.9135868549346924,
      "learning_rate": 9.808782037658792e-05,
      "loss": 0.0535,
      "step": 1340
    },
    {
      "epoch": 0.5949757602468048,
      "grad_norm": 1.1122609376907349,
      "learning_rate": 9.804226795346411e-05,
      "loss": 0.0438,
      "step": 1350
    },
    {
      "epoch": 0.5993829881004848,
      "grad_norm": 0.7222380042076111,
      "learning_rate": 9.799619014779503e-05,
      "loss": 0.0474,
      "step": 1360
    },
    {
      "epoch": 0.6037902159541648,
      "grad_norm": 0.922113299369812,
      "learning_rate": 9.794958746348013e-05,
      "loss": 0.0405,
      "step": 1370
    },
    {
      "epoch": 0.6081974438078449,
      "grad_norm": 1.01888108253479,
      "learning_rate": 9.790246041015896e-05,
      "loss": 0.0419,
      "step": 1380
    },
    {
      "epoch": 0.6126046716615249,
      "grad_norm": 0.6181578636169434,
      "learning_rate": 9.785480950320538e-05,
      "loss": 0.0431,
      "step": 1390
    },
    {
      "epoch": 0.617011899515205,
      "grad_norm": 0.8436324000358582,
      "learning_rate": 9.78066352637221e-05,
      "loss": 0.0419,
      "step": 1400
    },
    {
      "epoch": 0.6214191273688849,
      "grad_norm": 0.9413007497787476,
      "learning_rate": 9.775793821853488e-05,
      "loss": 0.0488,
      "step": 1410
    },
    {
      "epoch": 0.625826355222565,
      "grad_norm": 0.4537977874279022,
      "learning_rate": 9.77087189001868e-05,
      "loss": 0.0435,
      "step": 1420
    },
    {
      "epoch": 0.630233583076245,
      "grad_norm": 0.7080145478248596,
      "learning_rate": 9.765897784693243e-05,
      "loss": 0.0484,
      "step": 1430
    },
    {
      "epoch": 0.6346408109299251,
      "grad_norm": 0.4663386940956116,
      "learning_rate": 9.760871560273197e-05,
      "loss": 0.0382,
      "step": 1440
    },
    {
      "epoch": 0.6390480387836052,
      "grad_norm": 0.43715405464172363,
      "learning_rate": 9.755793271724526e-05,
      "loss": 0.0416,
      "step": 1450
    },
    {
      "epoch": 0.6434552666372851,
      "grad_norm": 0.5236002802848816,
      "learning_rate": 9.750662974582584e-05,
      "loss": 0.0415,
      "step": 1460
    },
    {
      "epoch": 0.6478624944909652,
      "grad_norm": 0.8185550570487976,
      "learning_rate": 9.745480724951473e-05,
      "loss": 0.0347,
      "step": 1470
    },
    {
      "epoch": 0.6522697223446452,
      "grad_norm": 0.4215397238731384,
      "learning_rate": 9.740246579503447e-05,
      "loss": 0.039,
      "step": 1480
    },
    {
      "epoch": 0.6566769501983253,
      "grad_norm": 0.7232959270477295,
      "learning_rate": 9.734960595478284e-05,
      "loss": 0.0322,
      "step": 1490
    },
    {
      "epoch": 0.6610841780520053,
      "grad_norm": 0.5677363872528076,
      "learning_rate": 9.729622830682657e-05,
      "loss": 0.0395,
      "step": 1500
    },
    {
      "epoch": 0.6654914059056853,
      "grad_norm": 0.5613128542900085,
      "learning_rate": 9.724233343489504e-05,
      "loss": 0.0236,
      "step": 1510
    },
    {
      "epoch": 0.6698986337593653,
      "grad_norm": 0.43877318501472473,
      "learning_rate": 9.718792192837396e-05,
      "loss": 0.0355,
      "step": 1520
    },
    {
      "epoch": 0.6743058616130454,
      "grad_norm": 0.5547152757644653,
      "learning_rate": 9.713299438229886e-05,
      "loss": 0.0373,
      "step": 1530
    },
    {
      "epoch": 0.6787130894667255,
      "grad_norm": 0.6812986731529236,
      "learning_rate": 9.707755139734855e-05,
      "loss": 0.0378,
      "step": 1540
    },
    {
      "epoch": 0.6831203173204055,
      "grad_norm": 0.5369997024536133,
      "learning_rate": 9.702159357983866e-05,
      "loss": 0.0409,
      "step": 1550
    },
    {
      "epoch": 0.6875275451740855,
      "grad_norm": 0.6349169611930847,
      "learning_rate": 9.696512154171492e-05,
      "loss": 0.0329,
      "step": 1560
    },
    {
      "epoch": 0.6919347730277655,
      "grad_norm": 0.6953033208847046,
      "learning_rate": 9.690813590054645e-05,
      "loss": 0.0364,
      "step": 1570
    },
    {
      "epoch": 0.6963420008814456,
      "grad_norm": 0.6071102023124695,
      "learning_rate": 9.685063727951914e-05,
      "loss": 0.0359,
      "step": 1580
    },
    {
      "epoch": 0.7007492287351256,
      "grad_norm": 0.637032151222229,
      "learning_rate": 9.679262630742865e-05,
      "loss": 0.0419,
      "step": 1590
    },
    {
      "epoch": 0.7051564565888057,
      "grad_norm": 0.5836294889450073,
      "learning_rate": 9.673410361867373e-05,
      "loss": 0.0424,
      "step": 1600
    },
    {
      "epoch": 0.7095636844424856,
      "grad_norm": 0.5156326293945312,
      "learning_rate": 9.667506985324909e-05,
      "loss": 0.0411,
      "step": 1610
    },
    {
      "epoch": 0.7139709122961657,
      "grad_norm": 0.2799781262874603,
      "learning_rate": 9.661552565673855e-05,
      "loss": 0.0367,
      "step": 1620
    },
    {
      "epoch": 0.7183781401498458,
      "grad_norm": 0.7328316569328308,
      "learning_rate": 9.655547168030789e-05,
      "loss": 0.0415,
      "step": 1630
    },
    {
      "epoch": 0.7227853680035258,
      "grad_norm": 0.6962109804153442,
      "learning_rate": 9.649490858069777e-05,
      "loss": 0.0356,
      "step": 1640
    },
    {
      "epoch": 0.7271925958572059,
      "grad_norm": 0.4299297332763672,
      "learning_rate": 9.643383702021658e-05,
      "loss": 0.0332,
      "step": 1650
    },
    {
      "epoch": 0.7315998237108858,
      "grad_norm": 0.3541034758090973,
      "learning_rate": 9.637225766673307e-05,
      "loss": 0.0449,
      "step": 1660
    },
    {
      "epoch": 0.7360070515645659,
      "grad_norm": 0.5092020034790039,
      "learning_rate": 9.631017119366922e-05,
      "loss": 0.0357,
      "step": 1670
    },
    {
      "epoch": 0.7404142794182459,
      "grad_norm": 0.6205541491508484,
      "learning_rate": 9.624757827999273e-05,
      "loss": 0.0384,
      "step": 1680
    },
    {
      "epoch": 0.744821507271926,
      "grad_norm": 0.7075048685073853,
      "learning_rate": 9.618447961020971e-05,
      "loss": 0.0506,
      "step": 1690
    },
    {
      "epoch": 0.749228735125606,
      "grad_norm": 0.6099492311477661,
      "learning_rate": 9.612087587435707e-05,
      "loss": 0.0373,
      "step": 1700
    },
    {
      "epoch": 0.753635962979286,
      "grad_norm": 0.763870358467102,
      "learning_rate": 9.605676776799508e-05,
      "loss": 0.0334,
      "step": 1710
    },
    {
      "epoch": 0.7580431908329661,
      "grad_norm": 0.7031465768814087,
      "learning_rate": 9.599215599219973e-05,
      "loss": 0.0438,
      "step": 1720
    },
    {
      "epoch": 0.7624504186866461,
      "grad_norm": 0.5724653601646423,
      "learning_rate": 9.592704125355505e-05,
      "loss": 0.0365,
      "step": 1730
    },
    {
      "epoch": 0.7668576465403262,
      "grad_norm": 0.4059954285621643,
      "learning_rate": 9.586142426414538e-05,
      "loss": 0.0456,
      "step": 1740
    },
    {
      "epoch": 0.7712648743940062,
      "grad_norm": 0.625922441482544,
      "learning_rate": 9.57953057415476e-05,
      "loss": 0.0347,
      "step": 1750
    },
    {
      "epoch": 0.7756721022476862,
      "grad_norm": 0.6046133637428284,
      "learning_rate": 9.572868640882328e-05,
      "loss": 0.0374,
      "step": 1760
    },
    {
      "epoch": 0.7800793301013662,
      "grad_norm": 0.3539145290851593,
      "learning_rate": 9.56615669945108e-05,
      "loss": 0.0371,
      "step": 1770
    },
    {
      "epoch": 0.7844865579550463,
      "grad_norm": 0.30902236700057983,
      "learning_rate": 9.55939482326173e-05,
      "loss": 0.0337,
      "step": 1780
    },
    {
      "epoch": 0.7888937858087263,
      "grad_norm": 0.764219343662262,
      "learning_rate": 9.552583086261069e-05,
      "loss": 0.0385,
      "step": 1790
    },
    {
      "epoch": 0.7933010136624064,
      "grad_norm": 0.7797220945358276,
      "learning_rate": 9.545721562941168e-05,
      "loss": 0.0398,
      "step": 1800
    },
    {
      "epoch": 0.7977082415160864,
      "grad_norm": 0.5790883898735046,
      "learning_rate": 9.538810328338543e-05,
      "loss": 0.0383,
      "step": 1810
    },
    {
      "epoch": 0.8021154693697664,
      "grad_norm": 0.6082382798194885,
      "learning_rate": 9.531849458033349e-05,
      "loss": 0.0303,
      "step": 1820
    },
    {
      "epoch": 0.8065226972234465,
      "grad_norm": 0.46905484795570374,
      "learning_rate": 9.524839028148547e-05,
      "loss": 0.0395,
      "step": 1830
    },
    {
      "epoch": 0.8109299250771265,
      "grad_norm": 0.7650353908538818,
      "learning_rate": 9.517779115349077e-05,
      "loss": 0.0401,
      "step": 1840
    },
    {
      "epoch": 0.8153371529308066,
      "grad_norm": 0.39779260754585266,
      "learning_rate": 9.510669796841014e-05,
      "loss": 0.0327,
      "step": 1850
    },
    {
      "epoch": 0.8197443807844865,
      "grad_norm": 0.6924306154251099,
      "learning_rate": 9.503511150370727e-05,
      "loss": 0.0384,
      "step": 1860
    },
    {
      "epoch": 0.8241516086381666,
      "grad_norm": 0.8675529360771179,
      "learning_rate": 9.496303254224024e-05,
      "loss": 0.0336,
      "step": 1870
    },
    {
      "epoch": 0.8285588364918466,
      "grad_norm": 0.5233723521232605,
      "learning_rate": 9.489046187225306e-05,
      "loss": 0.035,
      "step": 1880
    },
    {
      "epoch": 0.8329660643455267,
      "grad_norm": 0.5208832621574402,
      "learning_rate": 9.481740028736692e-05,
      "loss": 0.0382,
      "step": 1890
    },
    {
      "epoch": 0.8373732921992068,
      "grad_norm": 0.4628748595714569,
      "learning_rate": 9.474384858657164e-05,
      "loss": 0.0435,
      "step": 1900
    },
    {
      "epoch": 0.8417805200528867,
      "grad_norm": 0.7588955163955688,
      "learning_rate": 9.466980757421679e-05,
      "loss": 0.034,
      "step": 1910
    },
    {
      "epoch": 0.8461877479065668,
      "grad_norm": 0.5694310069084167,
      "learning_rate": 9.459527806000305e-05,
      "loss": 0.0342,
      "step": 1920
    },
    {
      "epoch": 0.8505949757602468,
      "grad_norm": 0.6176103949546814,
      "learning_rate": 9.452026085897325e-05,
      "loss": 0.0341,
      "step": 1930
    },
    {
      "epoch": 0.8550022036139269,
      "grad_norm": 0.6290533542633057,
      "learning_rate": 9.444475679150348e-05,
      "loss": 0.0325,
      "step": 1940
    },
    {
      "epoch": 0.8594094314676068,
      "grad_norm": 0.5656405687332153,
      "learning_rate": 9.436876668329411e-05,
      "loss": 0.0384,
      "step": 1950
    },
    {
      "epoch": 0.8638166593212869,
      "grad_norm": 0.6280255913734436,
      "learning_rate": 9.429229136536079e-05,
      "loss": 0.0341,
      "step": 1960
    },
    {
      "epoch": 0.8682238871749669,
      "grad_norm": 0.6733032464981079,
      "learning_rate": 9.421533167402534e-05,
      "loss": 0.0405,
      "step": 1970
    },
    {
      "epoch": 0.872631115028647,
      "grad_norm": 0.8397959470748901,
      "learning_rate": 9.413788845090666e-05,
      "loss": 0.0453,
      "step": 1980
    },
    {
      "epoch": 0.8770383428823271,
      "grad_norm": 0.6037388443946838,
      "learning_rate": 9.405996254291136e-05,
      "loss": 0.0356,
      "step": 1990
    },
    {
      "epoch": 0.881445570736007,
      "grad_norm": 0.4469926059246063,
      "learning_rate": 9.398155480222474e-05,
      "loss": 0.0401,
      "step": 2000
    },
    {
      "epoch": 0.8858527985896871,
      "grad_norm": 0.7656072378158569,
      "learning_rate": 9.390266608630128e-05,
      "loss": 0.0345,
      "step": 2010
    },
    {
      "epoch": 0.8902600264433671,
      "grad_norm": 0.49108415842056274,
      "learning_rate": 9.38232972578553e-05,
      "loss": 0.0299,
      "step": 2020
    },
    {
      "epoch": 0.8946672542970472,
      "grad_norm": 0.5365116000175476,
      "learning_rate": 9.374344918485164e-05,
      "loss": 0.0454,
      "step": 2030
    },
    {
      "epoch": 0.8990744821507272,
      "grad_norm": 0.5892438292503357,
      "learning_rate": 9.366312274049602e-05,
      "loss": 0.04,
      "step": 2040
    },
    {
      "epoch": 0.9034817100044072,
      "grad_norm": 0.906396746635437,
      "learning_rate": 9.358231880322554e-05,
      "loss": 0.0394,
      "step": 2050
    },
    {
      "epoch": 0.9078889378580872,
      "grad_norm": 0.8403815627098083,
      "learning_rate": 9.350103825669916e-05,
      "loss": 0.0423,
      "step": 2060
    },
    {
      "epoch": 0.9122961657117673,
      "grad_norm": 0.6807194948196411,
      "learning_rate": 9.341928198978787e-05,
      "loss": 0.0405,
      "step": 2070
    },
    {
      "epoch": 0.9167033935654474,
      "grad_norm": 0.6489959955215454,
      "learning_rate": 9.333705089656512e-05,
      "loss": 0.0384,
      "step": 2080
    },
    {
      "epoch": 0.9211106214191274,
      "grad_norm": 0.7752659916877747,
      "learning_rate": 9.325434587629698e-05,
      "loss": 0.0324,
      "step": 2090
    },
    {
      "epoch": 0.9255178492728074,
      "grad_norm": 0.3514426052570343,
      "learning_rate": 9.31711678334323e-05,
      "loss": 0.0303,
      "step": 2100
    },
    {
      "epoch": 0.9299250771264874,
      "grad_norm": 0.598884105682373,
      "learning_rate": 9.308751767759282e-05,
      "loss": 0.0403,
      "step": 2110
    },
    {
      "epoch": 0.9343323049801675,
      "grad_norm": 0.6693408489227295,
      "learning_rate": 9.300339632356325e-05,
      "loss": 0.0395,
      "step": 2120
    },
    {
      "epoch": 0.9387395328338475,
      "grad_norm": 0.6352264881134033,
      "learning_rate": 9.291880469128124e-05,
      "loss": 0.0425,
      "step": 2130
    },
    {
      "epoch": 0.9431467606875276,
      "grad_norm": 0.6078753471374512,
      "learning_rate": 9.283374370582732e-05,
      "loss": 0.0333,
      "step": 2140
    },
    {
      "epoch": 0.9475539885412075,
      "grad_norm": 0.5801308155059814,
      "learning_rate": 9.274821429741482e-05,
      "loss": 0.0352,
      "step": 2150
    },
    {
      "epoch": 0.9519612163948876,
      "grad_norm": 0.8792654871940613,
      "learning_rate": 9.266221740137961e-05,
      "loss": 0.0446,
      "step": 2160
    },
    {
      "epoch": 0.9563684442485677,
      "grad_norm": 0.478303462266922,
      "learning_rate": 9.257575395817001e-05,
      "loss": 0.027,
      "step": 2170
    },
    {
      "epoch": 0.9607756721022477,
      "grad_norm": 0.4599003493785858,
      "learning_rate": 9.248882491333637e-05,
      "loss": 0.0245,
      "step": 2180
    },
    {
      "epoch": 0.9651828999559278,
      "grad_norm": 1.0239288806915283,
      "learning_rate": 9.240143121752076e-05,
      "loss": 0.0378,
      "step": 2190
    },
    {
      "epoch": 0.9695901278096077,
      "grad_norm": 0.432131826877594,
      "learning_rate": 9.23135738264467e-05,
      "loss": 0.0363,
      "step": 2200
    },
    {
      "epoch": 0.9739973556632878,
      "grad_norm": 0.4591776430606842,
      "learning_rate": 9.222525370090849e-05,
      "loss": 0.0394,
      "step": 2210
    },
    {
      "epoch": 0.9784045835169678,
      "grad_norm": 0.38873565196990967,
      "learning_rate": 9.213647180676088e-05,
      "loss": 0.0337,
      "step": 2220
    },
    {
      "epoch": 0.9828118113706479,
      "grad_norm": 0.5276907086372375,
      "learning_rate": 9.204722911490846e-05,
      "loss": 0.0362,
      "step": 2230
    },
    {
      "epoch": 0.9872190392243279,
      "grad_norm": 0.6380802392959595,
      "learning_rate": 9.1957526601295e-05,
      "loss": 0.0389,
      "step": 2240
    },
    {
      "epoch": 0.9916262670780079,
      "grad_norm": 0.5179222226142883,
      "learning_rate": 9.186736524689281e-05,
      "loss": 0.0383,
      "step": 2250
    },
    {
      "epoch": 0.996033494931688,
      "grad_norm": 0.5359767079353333,
      "learning_rate": 9.177674603769204e-05,
      "loss": 0.0538,
      "step": 2260
    },
    {
      "epoch": 1.000440722785368,
      "grad_norm": 0.6518495082855225,
      "learning_rate": 9.168566996468983e-05,
      "loss": 0.0412,
      "step": 2270
    },
    {
      "epoch": 1.004847950639048,
      "grad_norm": 0.8117097616195679,
      "learning_rate": 9.159413802387951e-05,
      "loss": 0.0356,
      "step": 2280
    },
    {
      "epoch": 1.0092551784927282,
      "grad_norm": 0.49208784103393555,
      "learning_rate": 9.150215121623974e-05,
      "loss": 0.0383,
      "step": 2290
    },
    {
      "epoch": 1.0136624063464081,
      "grad_norm": 0.5838181376457214,
      "learning_rate": 9.140971054772349e-05,
      "loss": 0.0414,
      "step": 2300
    },
    {
      "epoch": 1.0180696342000881,
      "grad_norm": 0.4147219955921173,
      "learning_rate": 9.131681702924713e-05,
      "loss": 0.0296,
      "step": 2310
    },
    {
      "epoch": 1.022476862053768,
      "grad_norm": 0.45996376872062683,
      "learning_rate": 9.122347167667926e-05,
      "loss": 0.0323,
      "step": 2320
    },
    {
      "epoch": 1.0268840899074483,
      "grad_norm": 0.49214431643486023,
      "learning_rate": 9.112967551082973e-05,
      "loss": 0.0339,
      "step": 2330
    },
    {
      "epoch": 1.0312913177611283,
      "grad_norm": 0.8789698481559753,
      "learning_rate": 9.103542955743835e-05,
      "loss": 0.0421,
      "step": 2340
    },
    {
      "epoch": 1.0356985456148082,
      "grad_norm": 0.8325353264808655,
      "learning_rate": 9.094073484716381e-05,
      "loss": 0.0362,
      "step": 2350
    },
    {
      "epoch": 1.0401057734684884,
      "grad_norm": 0.8465186357498169,
      "learning_rate": 9.084559241557226e-05,
      "loss": 0.0397,
      "step": 2360
    },
    {
      "epoch": 1.0445130013221684,
      "grad_norm": 0.6591396927833557,
      "learning_rate": 9.075000330312608e-05,
      "loss": 0.0361,
      "step": 2370
    },
    {
      "epoch": 1.0489202291758484,
      "grad_norm": 0.7339377999305725,
      "learning_rate": 9.065396855517253e-05,
      "loss": 0.0364,
      "step": 2380
    },
    {
      "epoch": 1.0533274570295283,
      "grad_norm": 0.2985307276248932,
      "learning_rate": 9.055748922193219e-05,
      "loss": 0.0381,
      "step": 2390
    },
    {
      "epoch": 1.0577346848832085,
      "grad_norm": 0.6799618005752563,
      "learning_rate": 9.046056635848761e-05,
      "loss": 0.0359,
      "step": 2400
    },
    {
      "epoch": 1.0621419127368885,
      "grad_norm": 0.7417803406715393,
      "learning_rate": 9.036320102477169e-05,
      "loss": 0.034,
      "step": 2410
    },
    {
      "epoch": 1.0665491405905685,
      "grad_norm": 0.6348427534103394,
      "learning_rate": 9.02653942855561e-05,
      "loss": 0.0357,
      "step": 2420
    },
    {
      "epoch": 1.0709563684442487,
      "grad_norm": 0.5620206594467163,
      "learning_rate": 9.016714721043971e-05,
      "loss": 0.0342,
      "step": 2430
    },
    {
      "epoch": 1.0753635962979287,
      "grad_norm": 0.7759370803833008,
      "learning_rate": 9.006846087383675e-05,
      "loss": 0.0328,
      "step": 2440
    },
    {
      "epoch": 1.0797708241516086,
      "grad_norm": 0.6473592519760132,
      "learning_rate": 8.996933635496523e-05,
      "loss": 0.0381,
      "step": 2450
    },
    {
      "epoch": 1.0841780520052886,
      "grad_norm": 0.4338105022907257,
      "learning_rate": 8.986977473783498e-05,
      "loss": 0.0374,
      "step": 2460
    },
    {
      "epoch": 1.0885852798589688,
      "grad_norm": 0.443402498960495,
      "learning_rate": 8.97697771112359e-05,
      "loss": 0.0271,
      "step": 2470
    },
    {
      "epoch": 1.0929925077126488,
      "grad_norm": 0.60744309425354,
      "learning_rate": 8.966934456872602e-05,
      "loss": 0.0338,
      "step": 2480
    },
    {
      "epoch": 1.0973997355663287,
      "grad_norm": 0.9126889705657959,
      "learning_rate": 8.95684782086195e-05,
      "loss": 0.036,
      "step": 2490
    },
    {
      "epoch": 1.1018069634200087,
      "grad_norm": 0.915850043296814,
      "learning_rate": 8.946717913397476e-05,
      "loss": 0.039,
      "step": 2500
    },
    {
      "epoch": 1.106214191273689,
      "grad_norm": 0.9223534464836121,
      "learning_rate": 8.93654484525822e-05,
      "loss": 0.0371,
      "step": 2510
    },
    {
      "epoch": 1.110621419127369,
      "grad_norm": 0.4820459485054016,
      "learning_rate": 8.926328727695226e-05,
      "loss": 0.0325,
      "step": 2520
    },
    {
      "epoch": 1.1150286469810489,
      "grad_norm": 0.30257299542427063,
      "learning_rate": 8.916069672430319e-05,
      "loss": 0.0317,
      "step": 2530
    },
    {
      "epoch": 1.1194358748347288,
      "grad_norm": 0.5366955399513245,
      "learning_rate": 8.905767791654884e-05,
      "loss": 0.0376,
      "step": 2540
    },
    {
      "epoch": 1.123843102688409,
      "grad_norm": 0.31874722242355347,
      "learning_rate": 8.895423198028638e-05,
      "loss": 0.0353,
      "step": 2550
    },
    {
      "epoch": 1.128250330542089,
      "grad_norm": 0.4869764447212219,
      "learning_rate": 8.885036004678402e-05,
      "loss": 0.0424,
      "step": 2560
    },
    {
      "epoch": 1.132657558395769,
      "grad_norm": 0.5513941049575806,
      "learning_rate": 8.874606325196857e-05,
      "loss": 0.0373,
      "step": 2570
    },
    {
      "epoch": 1.1370647862494492,
      "grad_norm": 0.9100433588027954,
      "learning_rate": 8.864134273641304e-05,
      "loss": 0.0418,
      "step": 2580
    },
    {
      "epoch": 1.1414720141031292,
      "grad_norm": 0.6294649839401245,
      "learning_rate": 8.853619964532427e-05,
      "loss": 0.0326,
      "step": 2590
    },
    {
      "epoch": 1.1458792419568091,
      "grad_norm": 0.4285895526409149,
      "learning_rate": 8.843063512853019e-05,
      "loss": 0.0406,
      "step": 2600
    },
    {
      "epoch": 1.150286469810489,
      "grad_norm": 0.6396806240081787,
      "learning_rate": 8.832465034046749e-05,
      "loss": 0.0318,
      "step": 2610
    },
    {
      "epoch": 1.1546936976641693,
      "grad_norm": 0.720501720905304,
      "learning_rate": 8.821824644016882e-05,
      "loss": 0.0315,
      "step": 2620
    },
    {
      "epoch": 1.1591009255178493,
      "grad_norm": 0.6347301602363586,
      "learning_rate": 8.811142459125019e-05,
      "loss": 0.032,
      "step": 2630
    },
    {
      "epoch": 1.1635081533715292,
      "grad_norm": 0.5643287897109985,
      "learning_rate": 8.800418596189822e-05,
      "loss": 0.0411,
      "step": 2640
    },
    {
      "epoch": 1.1679153812252094,
      "grad_norm": 0.4921813905239105,
      "learning_rate": 8.789653172485737e-05,
      "loss": 0.0355,
      "step": 2650
    },
    {
      "epoch": 1.1723226090788894,
      "grad_norm": 0.5636128187179565,
      "learning_rate": 8.778846305741715e-05,
      "loss": 0.0306,
      "step": 2660
    },
    {
      "epoch": 1.1767298369325694,
      "grad_norm": 0.6810116767883301,
      "learning_rate": 8.767998114139918e-05,
      "loss": 0.0361,
      "step": 2670
    },
    {
      "epoch": 1.1811370647862494,
      "grad_norm": 0.5802286863327026,
      "learning_rate": 8.757108716314429e-05,
      "loss": 0.0362,
      "step": 2680
    },
    {
      "epoch": 1.1855442926399296,
      "grad_norm": 0.41435301303863525,
      "learning_rate": 8.746178231349962e-05,
      "loss": 0.0312,
      "step": 2690
    },
    {
      "epoch": 1.1899515204936095,
      "grad_norm": 0.6326773166656494,
      "learning_rate": 8.735206778780549e-05,
      "loss": 0.0308,
      "step": 2700
    },
    {
      "epoch": 1.1943587483472895,
      "grad_norm": 0.6125770807266235,
      "learning_rate": 8.724194478588234e-05,
      "loss": 0.0289,
      "step": 2710
    },
    {
      "epoch": 1.1987659762009697,
      "grad_norm": 0.4746556878089905,
      "learning_rate": 8.713141451201772e-05,
      "loss": 0.0328,
      "step": 2720
    },
    {
      "epoch": 1.2031732040546497,
      "grad_norm": 0.6342913508415222,
      "learning_rate": 8.702047817495295e-05,
      "loss": 0.0284,
      "step": 2730
    },
    {
      "epoch": 1.2075804319083296,
      "grad_norm": 0.5063793659210205,
      "learning_rate": 8.69091369878701e-05,
      "loss": 0.0275,
      "step": 2740
    },
    {
      "epoch": 1.2119876597620096,
      "grad_norm": 0.6542287468910217,
      "learning_rate": 8.679739216837849e-05,
      "loss": 0.0328,
      "step": 2750
    },
    {
      "epoch": 1.2163948876156898,
      "grad_norm": 0.4223293960094452,
      "learning_rate": 8.66852449385016e-05,
      "loss": 0.0281,
      "step": 2760
    },
    {
      "epoch": 1.2208021154693698,
      "grad_norm": 0.4498244822025299,
      "learning_rate": 8.657269652466356e-05,
      "loss": 0.0279,
      "step": 2770
    },
    {
      "epoch": 1.2252093433230498,
      "grad_norm": 0.42610597610473633,
      "learning_rate": 8.645974815767577e-05,
      "loss": 0.0301,
      "step": 2780
    },
    {
      "epoch": 1.22961657117673,
      "grad_norm": 0.4645084738731384,
      "learning_rate": 8.634640107272351e-05,
      "loss": 0.0358,
      "step": 2790
    },
    {
      "epoch": 1.23402379903041,
      "grad_norm": 0.792888879776001,
      "learning_rate": 8.623265650935234e-05,
      "loss": 0.0353,
      "step": 2800
    },
    {
      "epoch": 1.23843102688409,
      "grad_norm": 0.563528835773468,
      "learning_rate": 8.611851571145456e-05,
      "loss": 0.028,
      "step": 2810
    },
    {
      "epoch": 1.2428382547377699,
      "grad_norm": 0.4328922629356384,
      "learning_rate": 8.600397992725566e-05,
      "loss": 0.0273,
      "step": 2820
    },
    {
      "epoch": 1.24724548259145,
      "grad_norm": 0.49706098437309265,
      "learning_rate": 8.588905040930061e-05,
      "loss": 0.0362,
      "step": 2830
    },
    {
      "epoch": 1.25165271044513,
      "grad_norm": 0.5285065770149231,
      "learning_rate": 8.577372841444022e-05,
      "loss": 0.0355,
      "step": 2840
    },
    {
      "epoch": 1.25605993829881,
      "grad_norm": 0.4812205135822296,
      "learning_rate": 8.565801520381736e-05,
      "loss": 0.0315,
      "step": 2850
    },
    {
      "epoch": 1.2604671661524902,
      "grad_norm": 0.4882432818412781,
      "learning_rate": 8.554191204285313e-05,
      "loss": 0.0289,
      "step": 2860
    },
    {
      "epoch": 1.2648743940061702,
      "grad_norm": 0.46949517726898193,
      "learning_rate": 8.542542020123315e-05,
      "loss": 0.0317,
      "step": 2870
    },
    {
      "epoch": 1.2692816218598502,
      "grad_norm": 0.3332348167896271,
      "learning_rate": 8.530854095289347e-05,
      "loss": 0.0264,
      "step": 2880
    },
    {
      "epoch": 1.2736888497135301,
      "grad_norm": 0.41327717900276184,
      "learning_rate": 8.519127557600688e-05,
      "loss": 0.0372,
      "step": 2890
    },
    {
      "epoch": 1.27809607756721,
      "grad_norm": 0.44787606596946716,
      "learning_rate": 8.507362535296871e-05,
      "loss": 0.0305,
      "step": 2900
    },
    {
      "epoch": 1.2825033054208903,
      "grad_norm": 0.5057454705238342,
      "learning_rate": 8.495559157038299e-05,
      "loss": 0.0329,
      "step": 2910
    },
    {
      "epoch": 1.2869105332745703,
      "grad_norm": 0.4290280342102051,
      "learning_rate": 8.483717551904823e-05,
      "loss": 0.0285,
      "step": 2920
    },
    {
      "epoch": 1.2913177611282503,
      "grad_norm": 0.6101882457733154,
      "learning_rate": 8.47183784939434e-05,
      "loss": 0.0271,
      "step": 2930
    },
    {
      "epoch": 1.2957249889819304,
      "grad_norm": 0.5664324164390564,
      "learning_rate": 8.459920179421374e-05,
      "loss": 0.0376,
      "step": 2940
    },
    {
      "epoch": 1.3001322168356104,
      "grad_norm": 0.7244129776954651,
      "learning_rate": 8.447964672315656e-05,
      "loss": 0.0353,
      "step": 2950
    },
    {
      "epoch": 1.3045394446892904,
      "grad_norm": 0.35245728492736816,
      "learning_rate": 8.435971458820692e-05,
      "loss": 0.0314,
      "step": 2960
    },
    {
      "epoch": 1.3089466725429704,
      "grad_norm": 0.5011683702468872,
      "learning_rate": 8.423940670092345e-05,
      "loss": 0.031,
      "step": 2970
    },
    {
      "epoch": 1.3133539003966506,
      "grad_norm": 0.5053640007972717,
      "learning_rate": 8.411872437697394e-05,
      "loss": 0.0307,
      "step": 2980
    },
    {
      "epoch": 1.3177611282503305,
      "grad_norm": 0.7924652099609375,
      "learning_rate": 8.399766893612096e-05,
      "loss": 0.0328,
      "step": 2990
    },
    {
      "epoch": 1.3221683561040105,
      "grad_norm": 0.4984727203845978,
      "learning_rate": 8.38762417022074e-05,
      "loss": 0.0373,
      "step": 3000
    },
    {
      "epoch": 1.3265755839576907,
      "grad_norm": 0.5146403908729553,
      "learning_rate": 8.375444400314204e-05,
      "loss": 0.0309,
      "step": 3010
    },
    {
      "epoch": 1.3309828118113707,
      "grad_norm": 0.7703981995582581,
      "learning_rate": 8.3632277170885e-05,
      "loss": 0.0368,
      "step": 3020
    },
    {
      "epoch": 1.3353900396650507,
      "grad_norm": 0.7175225615501404,
      "learning_rate": 8.350974254143318e-05,
      "loss": 0.0288,
      "step": 3030
    },
    {
      "epoch": 1.3397972675187306,
      "grad_norm": 0.4650418758392334,
      "learning_rate": 8.338684145480566e-05,
      "loss": 0.0352,
      "step": 3040
    },
    {
      "epoch": 1.3442044953724108,
      "grad_norm": 0.5747969150543213,
      "learning_rate": 8.326357525502904e-05,
      "loss": 0.0316,
      "step": 3050
    },
    {
      "epoch": 1.3486117232260908,
      "grad_norm": 0.6155479550361633,
      "learning_rate": 8.313994529012273e-05,
      "loss": 0.0294,
      "step": 3060
    },
    {
      "epoch": 1.3530189510797708,
      "grad_norm": 0.5643554329872131,
      "learning_rate": 8.301595291208422e-05,
      "loss": 0.0292,
      "step": 3070
    },
    {
      "epoch": 1.357426178933451,
      "grad_norm": 0.4302073121070862,
      "learning_rate": 8.289159947687427e-05,
      "loss": 0.038,
      "step": 3080
    },
    {
      "epoch": 1.361833406787131,
      "grad_norm": 0.6329596042633057,
      "learning_rate": 8.276688634440216e-05,
      "loss": 0.0357,
      "step": 3090
    },
    {
      "epoch": 1.366240634640811,
      "grad_norm": 0.5275095701217651,
      "learning_rate": 8.26418148785107e-05,
      "loss": 0.0344,
      "step": 3100
    },
    {
      "epoch": 1.3706478624944909,
      "grad_norm": 0.6163100600242615,
      "learning_rate": 8.251638644696141e-05,
      "loss": 0.0294,
      "step": 3110
    },
    {
      "epoch": 1.375055090348171,
      "grad_norm": 0.6012453436851501,
      "learning_rate": 8.23906024214195e-05,
      "loss": 0.0311,
      "step": 3120
    },
    {
      "epoch": 1.379462318201851,
      "grad_norm": 0.6177839040756226,
      "learning_rate": 8.226446417743897e-05,
      "loss": 0.0318,
      "step": 3130
    },
    {
      "epoch": 1.383869546055531,
      "grad_norm": 0.45512518286705017,
      "learning_rate": 8.213797309444742e-05,
      "loss": 0.0286,
      "step": 3140
    },
    {
      "epoch": 1.3882767739092112,
      "grad_norm": 0.6371186971664429,
      "learning_rate": 8.201113055573105e-05,
      "loss": 0.0359,
      "step": 3150
    },
    {
      "epoch": 1.3926840017628912,
      "grad_norm": 0.2811204493045807,
      "learning_rate": 8.188393794841958e-05,
      "loss": 0.03,
      "step": 3160
    },
    {
      "epoch": 1.3970912296165712,
      "grad_norm": 0.478295236825943,
      "learning_rate": 8.175639666347094e-05,
      "loss": 0.0319,
      "step": 3170
    },
    {
      "epoch": 1.4014984574702511,
      "grad_norm": 0.5943291783332825,
      "learning_rate": 8.162850809565623e-05,
      "loss": 0.0329,
      "step": 3180
    },
    {
      "epoch": 1.4059056853239311,
      "grad_norm": 0.7045369744300842,
      "learning_rate": 8.150027364354431e-05,
      "loss": 0.0292,
      "step": 3190
    },
    {
      "epoch": 1.4103129131776113,
      "grad_norm": 0.5297116041183472,
      "learning_rate": 8.137169470948662e-05,
      "loss": 0.0414,
      "step": 3200
    },
    {
      "epoch": 1.4147201410312913,
      "grad_norm": 0.3332461416721344,
      "learning_rate": 8.124277269960179e-05,
      "loss": 0.029,
      "step": 3210
    },
    {
      "epoch": 1.4191273688849715,
      "grad_norm": 0.23516874015331268,
      "learning_rate": 8.111350902376023e-05,
      "loss": 0.0298,
      "step": 3220
    },
    {
      "epoch": 1.4235345967386515,
      "grad_norm": 0.3020011782646179,
      "learning_rate": 8.098390509556883e-05,
      "loss": 0.0251,
      "step": 3230
    },
    {
      "epoch": 1.4279418245923314,
      "grad_norm": 0.5564290285110474,
      "learning_rate": 8.085396233235536e-05,
      "loss": 0.0293,
      "step": 3240
    },
    {
      "epoch": 1.4323490524460114,
      "grad_norm": 0.481466680765152,
      "learning_rate": 8.072368215515306e-05,
      "loss": 0.0347,
      "step": 3250
    },
    {
      "epoch": 1.4367562802996914,
      "grad_norm": 0.3064061999320984,
      "learning_rate": 8.059306598868506e-05,
      "loss": 0.0301,
      "step": 3260
    },
    {
      "epoch": 1.4411635081533716,
      "grad_norm": 0.4470309317111969,
      "learning_rate": 8.046211526134888e-05,
      "loss": 0.0339,
      "step": 3270
    },
    {
      "epoch": 1.4455707360070515,
      "grad_norm": 0.34815454483032227,
      "learning_rate": 8.033083140520065e-05,
      "loss": 0.0302,
      "step": 3280
    },
    {
      "epoch": 1.4499779638607315,
      "grad_norm": 0.4798082113265991,
      "learning_rate": 8.019921585593962e-05,
      "loss": 0.027,
      "step": 3290
    },
    {
      "epoch": 1.4543851917144117,
      "grad_norm": 0.7006734013557434,
      "learning_rate": 8.006727005289232e-05,
      "loss": 0.0423,
      "step": 3300
    },
    {
      "epoch": 1.4587924195680917,
      "grad_norm": 0.5315399765968323,
      "learning_rate": 7.993499543899692e-05,
      "loss": 0.0253,
      "step": 3310
    },
    {
      "epoch": 1.4631996474217717,
      "grad_norm": 0.524401843547821,
      "learning_rate": 7.980239346078742e-05,
      "loss": 0.0299,
      "step": 3320
    },
    {
      "epoch": 1.4676068752754516,
      "grad_norm": 0.43416568636894226,
      "learning_rate": 7.966946556837778e-05,
      "loss": 0.034,
      "step": 3330
    },
    {
      "epoch": 1.4720141031291318,
      "grad_norm": 0.4605284333229065,
      "learning_rate": 7.953621321544616e-05,
      "loss": 0.0317,
      "step": 3340
    },
    {
      "epoch": 1.4764213309828118,
      "grad_norm": 0.34317782521247864,
      "learning_rate": 7.940263785921896e-05,
      "loss": 0.0328,
      "step": 3350
    },
    {
      "epoch": 1.4808285588364918,
      "grad_norm": 0.40754324197769165,
      "learning_rate": 7.926874096045482e-05,
      "loss": 0.0363,
      "step": 3360
    },
    {
      "epoch": 1.485235786690172,
      "grad_norm": 0.37132081389427185,
      "learning_rate": 7.913452398342881e-05,
      "loss": 0.0377,
      "step": 3370
    },
    {
      "epoch": 1.489643014543852,
      "grad_norm": 0.3234936594963074,
      "learning_rate": 7.89999883959163e-05,
      "loss": 0.0279,
      "step": 3380
    },
    {
      "epoch": 1.494050242397532,
      "grad_norm": 0.5448654294013977,
      "learning_rate": 7.886513566917687e-05,
      "loss": 0.0308,
      "step": 3390
    },
    {
      "epoch": 1.498457470251212,
      "grad_norm": 0.4141952097415924,
      "learning_rate": 7.872996727793838e-05,
      "loss": 0.0333,
      "step": 3400
    },
    {
      "epoch": 1.5028646981048919,
      "grad_norm": 0.3153703808784485,
      "learning_rate": 7.859448470038069e-05,
      "loss": 0.0275,
      "step": 3410
    },
    {
      "epoch": 1.507271925958572,
      "grad_norm": 0.4379798173904419,
      "learning_rate": 7.845868941811956e-05,
      "loss": 0.0397,
      "step": 3420
    },
    {
      "epoch": 1.511679153812252,
      "grad_norm": 0.4414808452129364,
      "learning_rate": 7.832258291619043e-05,
      "loss": 0.0281,
      "step": 3430
    },
    {
      "epoch": 1.5160863816659322,
      "grad_norm": 0.4763164520263672,
      "learning_rate": 7.81861666830322e-05,
      "loss": 0.034,
      "step": 3440
    },
    {
      "epoch": 1.5204936095196122,
      "grad_norm": 0.5171635150909424,
      "learning_rate": 7.804944221047097e-05,
      "loss": 0.03,
      "step": 3450
    },
    {
      "epoch": 1.5249008373732922,
      "grad_norm": 0.47638601064682007,
      "learning_rate": 7.791241099370364e-05,
      "loss": 0.0345,
      "step": 3460
    },
    {
      "epoch": 1.5293080652269722,
      "grad_norm": 0.3092629313468933,
      "learning_rate": 7.777507453128163e-05,
      "loss": 0.0303,
      "step": 3470
    },
    {
      "epoch": 1.5337152930806521,
      "grad_norm": 0.605987012386322,
      "learning_rate": 7.763743432509451e-05,
      "loss": 0.0296,
      "step": 3480
    },
    {
      "epoch": 1.5381225209343323,
      "grad_norm": 0.5758124589920044,
      "learning_rate": 7.749949188035353e-05,
      "loss": 0.0345,
      "step": 3490
    },
    {
      "epoch": 1.5425297487880123,
      "grad_norm": 0.43425044417381287,
      "learning_rate": 7.736124870557516e-05,
      "loss": 0.0303,
      "step": 3500
    },
    {
      "epoch": 1.5469369766416925,
      "grad_norm": 0.4632278084754944,
      "learning_rate": 7.722270631256459e-05,
      "loss": 0.0314,
      "step": 3510
    },
    {
      "epoch": 1.5513442044953725,
      "grad_norm": 1.0620092153549194,
      "learning_rate": 7.708386621639925e-05,
      "loss": 0.0375,
      "step": 3520
    },
    {
      "epoch": 1.5557514323490524,
      "grad_norm": 0.3753572106361389,
      "learning_rate": 7.694472993541219e-05,
      "loss": 0.0262,
      "step": 3530
    },
    {
      "epoch": 1.5601586602027324,
      "grad_norm": 0.6743049025535583,
      "learning_rate": 7.680529899117547e-05,
      "loss": 0.0344,
      "step": 3540
    },
    {
      "epoch": 1.5645658880564124,
      "grad_norm": 0.361783504486084,
      "learning_rate": 7.666557490848358e-05,
      "loss": 0.0299,
      "step": 3550
    },
    {
      "epoch": 1.5689731159100926,
      "grad_norm": 0.24776755273342133,
      "learning_rate": 7.65255592153367e-05,
      "loss": 0.0362,
      "step": 3560
    },
    {
      "epoch": 1.5733803437637726,
      "grad_norm": 0.6891716122627258,
      "learning_rate": 7.638525344292402e-05,
      "loss": 0.0379,
      "step": 3570
    },
    {
      "epoch": 1.5777875716174528,
      "grad_norm": 0.5458760261535645,
      "learning_rate": 7.624465912560697e-05,
      "loss": 0.0342,
      "step": 3580
    },
    {
      "epoch": 1.5821947994711327,
      "grad_norm": 0.46650922298431396,
      "learning_rate": 7.610377780090249e-05,
      "loss": 0.0324,
      "step": 3590
    },
    {
      "epoch": 1.5866020273248127,
      "grad_norm": 0.5069313049316406,
      "learning_rate": 7.596261100946618e-05,
      "loss": 0.0332,
      "step": 3600
    },
    {
      "epoch": 1.5910092551784927,
      "grad_norm": 0.5984185934066772,
      "learning_rate": 7.582116029507542e-05,
      "loss": 0.0315,
      "step": 3610
    },
    {
      "epoch": 1.5954164830321726,
      "grad_norm": 0.5433853268623352,
      "learning_rate": 7.56794272046126e-05,
      "loss": 0.0321,
      "step": 3620
    },
    {
      "epoch": 1.5998237108858528,
      "grad_norm": 0.3341008126735687,
      "learning_rate": 7.55374132880481e-05,
      "loss": 0.0371,
      "step": 3630
    },
    {
      "epoch": 1.6042309387395328,
      "grad_norm": 0.6386590600013733,
      "learning_rate": 7.539512009842333e-05,
      "loss": 0.028,
      "step": 3640
    },
    {
      "epoch": 1.608638166593213,
      "grad_norm": 0.5230645537376404,
      "learning_rate": 7.525254919183382e-05,
      "loss": 0.0347,
      "step": 3650
    },
    {
      "epoch": 1.613045394446893,
      "grad_norm": 0.6428864002227783,
      "learning_rate": 7.510970212741215e-05,
      "loss": 0.0299,
      "step": 3660
    },
    {
      "epoch": 1.617452622300573,
      "grad_norm": 0.3960852026939392,
      "learning_rate": 7.496658046731096e-05,
      "loss": 0.0327,
      "step": 3670
    },
    {
      "epoch": 1.621859850154253,
      "grad_norm": 0.4268948435783386,
      "learning_rate": 7.482318577668578e-05,
      "loss": 0.0232,
      "step": 3680
    },
    {
      "epoch": 1.626267078007933,
      "grad_norm": 0.7047191858291626,
      "learning_rate": 7.467951962367796e-05,
      "loss": 0.0291,
      "step": 3690
    },
    {
      "epoch": 1.6306743058616129,
      "grad_norm": 0.4119497835636139,
      "learning_rate": 7.453558357939755e-05,
      "loss": 0.0265,
      "step": 3700
    },
    {
      "epoch": 1.635081533715293,
      "grad_norm": 0.39977771043777466,
      "learning_rate": 7.439137921790606e-05,
      "loss": 0.0328,
      "step": 3710
    },
    {
      "epoch": 1.6394887615689733,
      "grad_norm": 0.5855114459991455,
      "learning_rate": 7.42469081161993e-05,
      "loss": 0.0336,
      "step": 3720
    },
    {
      "epoch": 1.6438959894226532,
      "grad_norm": 0.6724711656570435,
      "learning_rate": 7.410217185419006e-05,
      "loss": 0.0298,
      "step": 3730
    },
    {
      "epoch": 1.6483032172763332,
      "grad_norm": 0.36661335825920105,
      "learning_rate": 7.395717201469095e-05,
      "loss": 0.0297,
      "step": 3740
    },
    {
      "epoch": 1.6527104451300132,
      "grad_norm": 0.5452461242675781,
      "learning_rate": 7.381191018339696e-05,
      "loss": 0.0337,
      "step": 3750
    },
    {
      "epoch": 1.6571176729836932,
      "grad_norm": 0.7773560285568237,
      "learning_rate": 7.36663879488682e-05,
      "loss": 0.0225,
      "step": 3760
    },
    {
      "epoch": 1.6615249008373731,
      "grad_norm": 0.6551713943481445,
      "learning_rate": 7.352060690251254e-05,
      "loss": 0.0253,
      "step": 3770
    },
    {
      "epoch": 1.6659321286910533,
      "grad_norm": 0.3749849796295166,
      "learning_rate": 7.337456863856811e-05,
      "loss": 0.0254,
      "step": 3780
    },
    {
      "epoch": 1.6703393565447333,
      "grad_norm": 0.7387410998344421,
      "learning_rate": 7.3228274754086e-05,
      "loss": 0.0314,
      "step": 3790
    },
    {
      "epoch": 1.6747465843984135,
      "grad_norm": 0.4090714454650879,
      "learning_rate": 7.308172684891267e-05,
      "loss": 0.0319,
      "step": 3800
    },
    {
      "epoch": 1.6791538122520935,
      "grad_norm": 0.34972548484802246,
      "learning_rate": 7.293492652567255e-05,
      "loss": 0.0295,
      "step": 3810
    },
    {
      "epoch": 1.6835610401057735,
      "grad_norm": 0.44757863879203796,
      "learning_rate": 7.278787538975043e-05,
      "loss": 0.0324,
      "step": 3820
    },
    {
      "epoch": 1.6879682679594534,
      "grad_norm": 0.35655754804611206,
      "learning_rate": 7.2640575049274e-05,
      "loss": 0.0271,
      "step": 3830
    },
    {
      "epoch": 1.6923754958131334,
      "grad_norm": 0.4466415345668793,
      "learning_rate": 7.249302711509616e-05,
      "loss": 0.0344,
      "step": 3840
    },
    {
      "epoch": 1.6967827236668136,
      "grad_norm": 0.31867215037345886,
      "learning_rate": 7.23452332007775e-05,
      "loss": 0.0256,
      "step": 3850
    },
    {
      "epoch": 1.7011899515204936,
      "grad_norm": 0.4352557063102722,
      "learning_rate": 7.219719492256858e-05,
      "loss": 0.0377,
      "step": 3860
    },
    {
      "epoch": 1.7055971793741738,
      "grad_norm": 0.574970006942749,
      "learning_rate": 7.20489138993923e-05,
      "loss": 0.0293,
      "step": 3870
    },
    {
      "epoch": 1.7100044072278537,
      "grad_norm": 0.5501832365989685,
      "learning_rate": 7.190039175282614e-05,
      "loss": 0.0322,
      "step": 3880
    },
    {
      "epoch": 1.7144116350815337,
      "grad_norm": 0.46910592913627625,
      "learning_rate": 7.175163010708455e-05,
      "loss": 0.0337,
      "step": 3890
    },
    {
      "epoch": 1.7188188629352137,
      "grad_norm": 0.9607611894607544,
      "learning_rate": 7.1602630589001e-05,
      "loss": 0.0325,
      "step": 3900
    },
    {
      "epoch": 1.7232260907888937,
      "grad_norm": 0.273421972990036,
      "learning_rate": 7.14533948280104e-05,
      "loss": 0.0242,
      "step": 3910
    },
    {
      "epoch": 1.7276333186425739,
      "grad_norm": 0.5107641816139221,
      "learning_rate": 7.130392445613109e-05,
      "loss": 0.0293,
      "step": 3920
    },
    {
      "epoch": 1.7320405464962538,
      "grad_norm": 0.4398612678050995,
      "learning_rate": 7.115422110794711e-05,
      "loss": 0.0338,
      "step": 3930
    },
    {
      "epoch": 1.736447774349934,
      "grad_norm": 0.6604986190795898,
      "learning_rate": 7.100428642059033e-05,
      "loss": 0.0331,
      "step": 3940
    },
    {
      "epoch": 1.740855002203614,
      "grad_norm": 0.6657695174217224,
      "learning_rate": 7.08541220337224e-05,
      "loss": 0.0327,
      "step": 3950
    },
    {
      "epoch": 1.745262230057294,
      "grad_norm": 0.4896192252635956,
      "learning_rate": 7.070372958951706e-05,
      "loss": 0.0339,
      "step": 3960
    },
    {
      "epoch": 1.749669457910974,
      "grad_norm": 0.49432122707366943,
      "learning_rate": 7.055311073264194e-05,
      "loss": 0.0311,
      "step": 3970
    },
    {
      "epoch": 1.754076685764654,
      "grad_norm": 0.6652064919471741,
      "learning_rate": 7.040226711024077e-05,
      "loss": 0.0288,
      "step": 3980
    },
    {
      "epoch": 1.7584839136183341,
      "grad_norm": 0.2407093495130539,
      "learning_rate": 7.02512003719152e-05,
      "loss": 0.027,
      "step": 3990
    },
    {
      "epoch": 1.762891141472014,
      "grad_norm": 0.49707257747650146,
      "learning_rate": 7.00999121697069e-05,
      "loss": 0.0236,
      "step": 4000
    },
    {
      "epoch": 1.7672983693256943,
      "grad_norm": 0.564982533454895,
      "learning_rate": 6.99484041580794e-05,
      "loss": 0.0265,
      "step": 4010
    },
    {
      "epoch": 1.7717055971793743,
      "grad_norm": 0.5058256387710571,
      "learning_rate": 6.979667799390004e-05,
      "loss": 0.0244,
      "step": 4020
    },
    {
      "epoch": 1.7761128250330542,
      "grad_norm": 0.47129127383232117,
      "learning_rate": 6.964473533642185e-05,
      "loss": 0.035,
      "step": 4030
    },
    {
      "epoch": 1.7805200528867342,
      "grad_norm": 0.5903072953224182,
      "learning_rate": 6.949257784726539e-05,
      "loss": 0.0281,
      "step": 4040
    },
    {
      "epoch": 1.7849272807404142,
      "grad_norm": 0.6064934134483337,
      "learning_rate": 6.934020719040056e-05,
      "loss": 0.0334,
      "step": 4050
    },
    {
      "epoch": 1.7893345085940942,
      "grad_norm": 0.7642019987106323,
      "learning_rate": 6.918762503212848e-05,
      "loss": 0.0336,
      "step": 4060
    },
    {
      "epoch": 1.7937417364477743,
      "grad_norm": 0.5818179249763489,
      "learning_rate": 6.903483304106319e-05,
      "loss": 0.0371,
      "step": 4070
    },
    {
      "epoch": 1.7981489643014545,
      "grad_norm": 0.4175790846347809,
      "learning_rate": 6.888183288811341e-05,
      "loss": 0.0289,
      "step": 4080
    },
    {
      "epoch": 1.8025561921551345,
      "grad_norm": 0.4765195846557617,
      "learning_rate": 6.87286262464643e-05,
      "loss": 0.0279,
      "step": 4090
    },
    {
      "epoch": 1.8069634200088145,
      "grad_norm": 0.4594866633415222,
      "learning_rate": 6.857521479155915e-05,
      "loss": 0.0291,
      "step": 4100
    },
    {
      "epoch": 1.8113706478624945,
      "grad_norm": 0.33618780970573425,
      "learning_rate": 6.842160020108104e-05,
      "loss": 0.0233,
      "step": 4110
    },
    {
      "epoch": 1.8157778757161744,
      "grad_norm": 0.3849072754383087,
      "learning_rate": 6.826778415493455e-05,
      "loss": 0.0262,
      "step": 4120
    },
    {
      "epoch": 1.8201851035698544,
      "grad_norm": 0.516969621181488,
      "learning_rate": 6.811376833522729e-05,
      "loss": 0.0226,
      "step": 4130
    },
    {
      "epoch": 1.8245923314235346,
      "grad_norm": 0.6272823214530945,
      "learning_rate": 6.795955442625159e-05,
      "loss": 0.0243,
      "step": 4140
    },
    {
      "epoch": 1.8289995592772146,
      "grad_norm": 0.605594277381897,
      "learning_rate": 6.780514411446608e-05,
      "loss": 0.0284,
      "step": 4150
    },
    {
      "epoch": 1.8334067871308948,
      "grad_norm": 0.28106987476348877,
      "learning_rate": 6.765053908847716e-05,
      "loss": 0.0232,
      "step": 4160
    },
    {
      "epoch": 1.8378140149845748,
      "grad_norm": 0.40030765533447266,
      "learning_rate": 6.749574103902064e-05,
      "loss": 0.0236,
      "step": 4170
    },
    {
      "epoch": 1.8422212428382547,
      "grad_norm": 0.3321658670902252,
      "learning_rate": 6.734075165894317e-05,
      "loss": 0.0238,
      "step": 4180
    },
    {
      "epoch": 1.8466284706919347,
      "grad_norm": 0.33565711975097656,
      "learning_rate": 6.71855726431838e-05,
      "loss": 0.0266,
      "step": 4190
    },
    {
      "epoch": 1.8510356985456147,
      "grad_norm": 0.6143445372581482,
      "learning_rate": 6.703020568875538e-05,
      "loss": 0.0252,
      "step": 4200
    },
    {
      "epoch": 1.8554429263992949,
      "grad_norm": 0.3084467351436615,
      "learning_rate": 6.687465249472603e-05,
      "loss": 0.0247,
      "step": 4210
    },
    {
      "epoch": 1.8598501542529748,
      "grad_norm": 0.3428739905357361,
      "learning_rate": 6.671891476220055e-05,
      "loss": 0.0234,
      "step": 4220
    },
    {
      "epoch": 1.864257382106655,
      "grad_norm": 0.37586015462875366,
      "learning_rate": 6.656299419430183e-05,
      "loss": 0.0259,
      "step": 4230
    },
    {
      "epoch": 1.868664609960335,
      "grad_norm": 0.401557981967926,
      "learning_rate": 6.640689249615223e-05,
      "loss": 0.0203,
      "step": 4240
    },
    {
      "epoch": 1.873071837814015,
      "grad_norm": 0.5719128251075745,
      "learning_rate": 6.625061137485491e-05,
      "loss": 0.0321,
      "step": 4250
    },
    {
      "epoch": 1.877479065667695,
      "grad_norm": 0.558519184589386,
      "learning_rate": 6.609415253947517e-05,
      "loss": 0.0258,
      "step": 4260
    },
    {
      "epoch": 1.881886293521375,
      "grad_norm": 0.37527602910995483,
      "learning_rate": 6.593751770102178e-05,
      "loss": 0.0283,
      "step": 4270
    },
    {
      "epoch": 1.8862935213750551,
      "grad_norm": 0.4086023271083832,
      "learning_rate": 6.578070857242823e-05,
      "loss": 0.0274,
      "step": 4280
    },
    {
      "epoch": 1.890700749228735,
      "grad_norm": 0.793245255947113,
      "learning_rate": 6.562372686853402e-05,
      "loss": 0.0415,
      "step": 4290
    },
    {
      "epoch": 1.8951079770824153,
      "grad_norm": 0.4924085736274719,
      "learning_rate": 6.546657430606593e-05,
      "loss": 0.0284,
      "step": 4300
    },
    {
      "epoch": 1.8995152049360953,
      "grad_norm": 0.6175187230110168,
      "learning_rate": 6.530925260361918e-05,
      "loss": 0.028,
      "step": 4310
    },
    {
      "epoch": 1.9039224327897752,
      "grad_norm": 0.5409990549087524,
      "learning_rate": 6.515176348163871e-05,
      "loss": 0.0289,
      "step": 4320
    },
    {
      "epoch": 1.9083296606434552,
      "grad_norm": 0.4849074184894562,
      "learning_rate": 6.499410866240032e-05,
      "loss": 0.0297,
      "step": 4330
    },
    {
      "epoch": 1.9127368884971352,
      "grad_norm": 0.6663230061531067,
      "learning_rate": 6.48362898699919e-05,
      "loss": 0.0241,
      "step": 4340
    },
    {
      "epoch": 1.9171441163508154,
      "grad_norm": 0.5513930916786194,
      "learning_rate": 6.467830883029443e-05,
      "loss": 0.0231,
      "step": 4350
    },
    {
      "epoch": 1.9215513442044954,
      "grad_norm": 0.4695577919483185,
      "learning_rate": 6.452016727096326e-05,
      "loss": 0.0296,
      "step": 4360
    },
    {
      "epoch": 1.9259585720581756,
      "grad_norm": 0.4701121747493744,
      "learning_rate": 6.436186692140916e-05,
      "loss": 0.0266,
      "step": 4370
    },
    {
      "epoch": 1.9303657999118555,
      "grad_norm": 0.327288419008255,
      "learning_rate": 6.420340951277938e-05,
      "loss": 0.0291,
      "step": 4380
    },
    {
      "epoch": 1.9347730277655355,
      "grad_norm": 0.5551244616508484,
      "learning_rate": 6.404479677793874e-05,
      "loss": 0.0331,
      "step": 4390
    },
    {
      "epoch": 1.9391802556192155,
      "grad_norm": 0.43837469816207886,
      "learning_rate": 6.388603045145075e-05,
      "loss": 0.0253,
      "step": 4400
    },
    {
      "epoch": 1.9435874834728954,
      "grad_norm": 0.701884925365448,
      "learning_rate": 6.372711226955843e-05,
      "loss": 0.0268,
      "step": 4410
    },
    {
      "epoch": 1.9479947113265754,
      "grad_norm": 0.30174151062965393,
      "learning_rate": 6.356804397016564e-05,
      "loss": 0.0223,
      "step": 4420
    },
    {
      "epoch": 1.9524019391802556,
      "grad_norm": 0.5513481497764587,
      "learning_rate": 6.340882729281779e-05,
      "loss": 0.0268,
      "step": 4430
    },
    {
      "epoch": 1.9568091670339358,
      "grad_norm": 0.5355939269065857,
      "learning_rate": 6.324946397868294e-05,
      "loss": 0.0298,
      "step": 4440
    },
    {
      "epoch": 1.9612163948876158,
      "grad_norm": 0.4604257047176361,
      "learning_rate": 6.308995577053276e-05,
      "loss": 0.0321,
      "step": 4450
    },
    {
      "epoch": 1.9656236227412958,
      "grad_norm": 0.4191978871822357,
      "learning_rate": 6.293030441272347e-05,
      "loss": 0.0297,
      "step": 4460
    },
    {
      "epoch": 1.9700308505949757,
      "grad_norm": 0.6228031516075134,
      "learning_rate": 6.277051165117677e-05,
      "loss": 0.028,
      "step": 4470
    },
    {
      "epoch": 1.9744380784486557,
      "grad_norm": 0.4991246461868286,
      "learning_rate": 6.261057923336064e-05,
      "loss": 0.0275,
      "step": 4480
    },
    {
      "epoch": 1.9788453063023357,
      "grad_norm": 0.43567246198654175,
      "learning_rate": 6.245050890827042e-05,
      "loss": 0.0264,
      "step": 4490
    },
    {
      "epoch": 1.9832525341560159,
      "grad_norm": 0.3800245225429535,
      "learning_rate": 6.229030242640952e-05,
      "loss": 0.0346,
      "step": 4500
    },
    {
      "epoch": 1.9876597620096959,
      "grad_norm": 0.3371240794658661,
      "learning_rate": 6.212996153977037e-05,
      "loss": 0.0201,
      "step": 4510
    },
    {
      "epoch": 1.992066989863376,
      "grad_norm": 0.5228076577186584,
      "learning_rate": 6.196948800181523e-05,
      "loss": 0.0291,
      "step": 4520
    },
    {
      "epoch": 1.996474217717056,
      "grad_norm": 0.6688753366470337,
      "learning_rate": 6.180888356745695e-05,
      "loss": 0.0239,
      "step": 4530
    },
    {
      "epoch": 2.000881445570736,
      "grad_norm": 0.4080394208431244,
      "learning_rate": 6.164814999303995e-05,
      "loss": 0.0226,
      "step": 4540
    },
    {
      "epoch": 2.005288673424416,
      "grad_norm": 0.4964958727359772,
      "learning_rate": 6.148728903632081e-05,
      "loss": 0.0336,
      "step": 4550
    },
    {
      "epoch": 2.009695901278096,
      "grad_norm": 0.46922144293785095,
      "learning_rate": 6.132630245644921e-05,
      "loss": 0.0256,
      "step": 4560
    },
    {
      "epoch": 2.014103129131776,
      "grad_norm": 0.43690091371536255,
      "learning_rate": 6.116519201394857e-05,
      "loss": 0.0278,
      "step": 4570
    },
    {
      "epoch": 2.0185103569854563,
      "grad_norm": 0.3830350637435913,
      "learning_rate": 6.10039594706969e-05,
      "loss": 0.0246,
      "step": 4580
    },
    {
      "epoch": 2.0229175848391363,
      "grad_norm": 0.4406307637691498,
      "learning_rate": 6.084260658990744e-05,
      "loss": 0.0242,
      "step": 4590
    },
    {
      "epoch": 2.0273248126928163,
      "grad_norm": 0.6453173160552979,
      "learning_rate": 6.068113513610943e-05,
      "loss": 0.0205,
      "step": 4600
    },
    {
      "epoch": 2.0317320405464963,
      "grad_norm": 0.4437095522880554,
      "learning_rate": 6.0519546875128876e-05,
      "loss": 0.0273,
      "step": 4610
    },
    {
      "epoch": 2.0361392684001762,
      "grad_norm": 0.3416881859302521,
      "learning_rate": 6.035784357406906e-05,
      "loss": 0.0307,
      "step": 4620
    },
    {
      "epoch": 2.040546496253856,
      "grad_norm": 0.3453536033630371,
      "learning_rate": 6.01960270012914e-05,
      "loss": 0.0173,
      "step": 4630
    },
    {
      "epoch": 2.044953724107536,
      "grad_norm": 0.4587254524230957,
      "learning_rate": 6.003409892639599e-05,
      "loss": 0.0262,
      "step": 4640
    },
    {
      "epoch": 2.0493609519612166,
      "grad_norm": 0.481233686208725,
      "learning_rate": 5.9872061120202336e-05,
      "loss": 0.0237,
      "step": 4650
    },
    {
      "epoch": 2.0537681798148966,
      "grad_norm": 0.34292593598365784,
      "learning_rate": 5.9709915354729914e-05,
      "loss": 0.0334,
      "step": 4660
    },
    {
      "epoch": 2.0581754076685765,
      "grad_norm": 0.4022336006164551,
      "learning_rate": 5.9547663403178824e-05,
      "loss": 0.0315,
      "step": 4670
    },
    {
      "epoch": 2.0625826355222565,
      "grad_norm": 0.4429273009300232,
      "learning_rate": 5.9385307039910445e-05,
      "loss": 0.0219,
      "step": 4680
    },
    {
      "epoch": 2.0669898633759365,
      "grad_norm": 0.4632837772369385,
      "learning_rate": 5.922284804042792e-05,
      "loss": 0.0269,
      "step": 4690
    },
    {
      "epoch": 2.0713970912296165,
      "grad_norm": 0.48913586139678955,
      "learning_rate": 5.906028818135687e-05,
      "loss": 0.0256,
      "step": 4700
    },
    {
      "epoch": 2.0758043190832964,
      "grad_norm": 0.5592131614685059,
      "learning_rate": 5.889762924042585e-05,
      "loss": 0.0289,
      "step": 4710
    },
    {
      "epoch": 2.080211546936977,
      "grad_norm": 0.46295300126075745,
      "learning_rate": 5.873487299644699e-05,
      "loss": 0.0226,
      "step": 4720
    },
    {
      "epoch": 2.084618774790657,
      "grad_norm": 0.5707088708877563,
      "learning_rate": 5.857202122929649e-05,
      "loss": 0.0231,
      "step": 4730
    },
    {
      "epoch": 2.089026002644337,
      "grad_norm": 0.48545607924461365,
      "learning_rate": 5.840907571989518e-05,
      "loss": 0.0266,
      "step": 4740
    },
    {
      "epoch": 2.0934332304980168,
      "grad_norm": 0.281380295753479,
      "learning_rate": 5.824603825018904e-05,
      "loss": 0.0187,
      "step": 4750
    },
    {
      "epoch": 2.0978404583516967,
      "grad_norm": 0.48434731364250183,
      "learning_rate": 5.808291060312975e-05,
      "loss": 0.0255,
      "step": 4760
    },
    {
      "epoch": 2.1022476862053767,
      "grad_norm": 0.34401217103004456,
      "learning_rate": 5.7919694562655083e-05,
      "loss": 0.0254,
      "step": 4770
    },
    {
      "epoch": 2.1066549140590567,
      "grad_norm": 0.3194863200187683,
      "learning_rate": 5.775639191366954e-05,
      "loss": 0.0296,
      "step": 4780
    },
    {
      "epoch": 2.111062141912737,
      "grad_norm": 0.5448789596557617,
      "learning_rate": 5.75930044420247e-05,
      "loss": 0.0274,
      "step": 4790
    },
    {
      "epoch": 2.115469369766417,
      "grad_norm": 0.3996848464012146,
      "learning_rate": 5.74295339344998e-05,
      "loss": 0.0267,
      "step": 4800
    },
    {
      "epoch": 2.119876597620097,
      "grad_norm": 0.29342353343963623,
      "learning_rate": 5.726598217878211e-05,
      "loss": 0.0247,
      "step": 4810
    },
    {
      "epoch": 2.124283825473777,
      "grad_norm": 0.3241419196128845,
      "learning_rate": 5.71023509634474e-05,
      "loss": 0.0199,
      "step": 4820
    },
    {
      "epoch": 2.128691053327457,
      "grad_norm": 0.38731175661087036,
      "learning_rate": 5.693864207794049e-05,
      "loss": 0.0264,
      "step": 4830
    },
    {
      "epoch": 2.133098281181137,
      "grad_norm": 0.3758256435394287,
      "learning_rate": 5.677485731255545e-05,
      "loss": 0.0284,
      "step": 4840
    },
    {
      "epoch": 2.137505509034817,
      "grad_norm": 0.5041866302490234,
      "learning_rate": 5.6610998458416296e-05,
      "loss": 0.022,
      "step": 4850
    },
    {
      "epoch": 2.1419127368884974,
      "grad_norm": 0.5127224326133728,
      "learning_rate": 5.644706730745716e-05,
      "loss": 0.022,
      "step": 4860
    },
    {
      "epoch": 2.1463199647421773,
      "grad_norm": 0.6049852967262268,
      "learning_rate": 5.628306565240287e-05,
      "loss": 0.0345,
      "step": 4870
    },
    {
      "epoch": 2.1507271925958573,
      "grad_norm": 0.5794158577919006,
      "learning_rate": 5.611899528674923e-05,
      "loss": 0.0261,
      "step": 4880
    },
    {
      "epoch": 2.1551344204495373,
      "grad_norm": 0.5450758934020996,
      "learning_rate": 5.595485800474349e-05,
      "loss": 0.0275,
      "step": 4890
    },
    {
      "epoch": 2.1595416483032173,
      "grad_norm": 0.29294103384017944,
      "learning_rate": 5.579065560136467e-05,
      "loss": 0.024,
      "step": 4900
    },
    {
      "epoch": 2.1639488761568972,
      "grad_norm": 0.42734894156455994,
      "learning_rate": 5.562638987230392e-05,
      "loss": 0.0252,
      "step": 4910
    },
    {
      "epoch": 2.168356104010577,
      "grad_norm": 0.37511804699897766,
      "learning_rate": 5.546206261394498e-05,
      "loss": 0.027,
      "step": 4920
    },
    {
      "epoch": 2.172763331864257,
      "grad_norm": 0.33593010902404785,
      "learning_rate": 5.529767562334437e-05,
      "loss": 0.0233,
      "step": 4930
    },
    {
      "epoch": 2.1771705597179376,
      "grad_norm": 0.7282082438468933,
      "learning_rate": 5.5133230698211926e-05,
      "loss": 0.0307,
      "step": 4940
    },
    {
      "epoch": 2.1815777875716176,
      "grad_norm": 0.43649062514305115,
      "learning_rate": 5.496872963689096e-05,
      "loss": 0.0218,
      "step": 4950
    },
    {
      "epoch": 2.1859850154252976,
      "grad_norm": 0.4204055964946747,
      "learning_rate": 5.4804174238338756e-05,
      "loss": 0.0266,
      "step": 4960
    },
    {
      "epoch": 2.1903922432789775,
      "grad_norm": 0.4718764126300812,
      "learning_rate": 5.463956630210678e-05,
      "loss": 0.0253,
      "step": 4970
    },
    {
      "epoch": 2.1947994711326575,
      "grad_norm": 0.3668510317802429,
      "learning_rate": 5.4474907628321046e-05,
      "loss": 0.0229,
      "step": 4980
    },
    {
      "epoch": 2.1992066989863375,
      "grad_norm": 0.3522864282131195,
      "learning_rate": 5.431020001766244e-05,
      "loss": 0.0222,
      "step": 4990
    },
    {
      "epoch": 2.2036139268400174,
      "grad_norm": 0.5479294657707214,
      "learning_rate": 5.4145445271346986e-05,
      "loss": 0.0246,
      "step": 5000
    },
    {
      "epoch": 2.208021154693698,
      "grad_norm": 0.6383090615272522,
      "learning_rate": 5.398064519110622e-05,
      "loss": 0.0272,
      "step": 5010
    },
    {
      "epoch": 2.212428382547378,
      "grad_norm": 0.3878839910030365,
      "learning_rate": 5.3815801579167394e-05,
      "loss": 0.0277,
      "step": 5020
    },
    {
      "epoch": 2.216835610401058,
      "grad_norm": 0.33870816230773926,
      "learning_rate": 5.365091623823382e-05,
      "loss": 0.0226,
      "step": 5030
    },
    {
      "epoch": 2.221242838254738,
      "grad_norm": 0.38512441515922546,
      "learning_rate": 5.348599097146521e-05,
      "loss": 0.025,
      "step": 5040
    },
    {
      "epoch": 2.2256500661084178,
      "grad_norm": 0.5055773258209229,
      "learning_rate": 5.3321027582457836e-05,
      "loss": 0.0254,
      "step": 5050
    },
    {
      "epoch": 2.2300572939620977,
      "grad_norm": 0.5212316513061523,
      "learning_rate": 5.315602787522491e-05,
      "loss": 0.0193,
      "step": 5060
    },
    {
      "epoch": 2.2344645218157777,
      "grad_norm": 0.7157302498817444,
      "learning_rate": 5.299099365417678e-05,
      "loss": 0.027,
      "step": 5070
    },
    {
      "epoch": 2.2388717496694577,
      "grad_norm": 0.3394235670566559,
      "learning_rate": 5.2825926724101236e-05,
      "loss": 0.0215,
      "step": 5080
    },
    {
      "epoch": 2.243278977523138,
      "grad_norm": 0.805059015750885,
      "learning_rate": 5.26608288901438e-05,
      "loss": 0.0287,
      "step": 5090
    },
    {
      "epoch": 2.247686205376818,
      "grad_norm": 0.44006943702697754,
      "learning_rate": 5.24957019577879e-05,
      "loss": 0.0266,
      "step": 5100
    },
    {
      "epoch": 2.252093433230498,
      "grad_norm": 0.21787722408771515,
      "learning_rate": 5.2330547732835266e-05,
      "loss": 0.0237,
      "step": 5110
    },
    {
      "epoch": 2.256500661084178,
      "grad_norm": 0.47274816036224365,
      "learning_rate": 5.2165368021385996e-05,
      "loss": 0.027,
      "step": 5120
    },
    {
      "epoch": 2.260907888937858,
      "grad_norm": 0.5053203701972961,
      "learning_rate": 5.200016462981897e-05,
      "loss": 0.0244,
      "step": 5130
    },
    {
      "epoch": 2.265315116791538,
      "grad_norm": 0.3583391308784485,
      "learning_rate": 5.1834939364772015e-05,
      "loss": 0.0242,
      "step": 5140
    },
    {
      "epoch": 2.269722344645218,
      "grad_norm": 0.37365928292274475,
      "learning_rate": 5.166969403312214e-05,
      "loss": 0.0287,
      "step": 5150
    },
    {
      "epoch": 2.2741295724988984,
      "grad_norm": 0.27925682067871094,
      "learning_rate": 5.1504430441965844e-05,
      "loss": 0.0256,
      "step": 5160
    },
    {
      "epoch": 2.2785368003525783,
      "grad_norm": 0.3031574487686157,
      "learning_rate": 5.133915039859923e-05,
      "loss": 0.0261,
      "step": 5170
    },
    {
      "epoch": 2.2829440282062583,
      "grad_norm": 0.28002825379371643,
      "learning_rate": 5.1173855710498444e-05,
      "loss": 0.0273,
      "step": 5180
    },
    {
      "epoch": 2.2873512560599383,
      "grad_norm": 0.30599093437194824,
      "learning_rate": 5.100854818529967e-05,
      "loss": 0.0168,
      "step": 5190
    },
    {
      "epoch": 2.2917584839136182,
      "grad_norm": 0.3657265603542328,
      "learning_rate": 5.084322963077951e-05,
      "loss": 0.0246,
      "step": 5200
    },
    {
      "epoch": 2.2961657117672982,
      "grad_norm": 0.2867480218410492,
      "learning_rate": 5.067790185483522e-05,
      "loss": 0.0269,
      "step": 5210
    },
    {
      "epoch": 2.300572939620978,
      "grad_norm": 0.35093027353286743,
      "learning_rate": 5.0512566665464844e-05,
      "loss": 0.0173,
      "step": 5220
    },
    {
      "epoch": 2.3049801674746586,
      "grad_norm": 0.2638501524925232,
      "learning_rate": 5.034722587074755e-05,
      "loss": 0.0259,
      "step": 5230
    },
    {
      "epoch": 2.3093873953283386,
      "grad_norm": 0.27982184290885925,
      "learning_rate": 5.018188127882375e-05,
      "loss": 0.0256,
      "step": 5240
    },
    {
      "epoch": 2.3137946231820186,
      "grad_norm": 0.3896169662475586,
      "learning_rate": 5.0016534697875417e-05,
      "loss": 0.0182,
      "step": 5250
    },
    {
      "epoch": 2.3182018510356985,
      "grad_norm": 0.4068613350391388,
      "learning_rate": 4.9851187936106294e-05,
      "loss": 0.0235,
      "step": 5260
    },
    {
      "epoch": 2.3226090788893785,
      "grad_norm": 0.40756428241729736,
      "learning_rate": 4.968584280172206e-05,
      "loss": 0.0199,
      "step": 5270
    },
    {
      "epoch": 2.3270163067430585,
      "grad_norm": 0.3594902455806732,
      "learning_rate": 4.95205011029106e-05,
      "loss": 0.0154,
      "step": 5280
    },
    {
      "epoch": 2.3314235345967385,
      "grad_norm": 0.3173263370990753,
      "learning_rate": 4.935516464782227e-05,
      "loss": 0.0188,
      "step": 5290
    },
    {
      "epoch": 2.335830762450419,
      "grad_norm": 0.4052960276603699,
      "learning_rate": 4.918983524455003e-05,
      "loss": 0.0275,
      "step": 5300
    },
    {
      "epoch": 2.340237990304099,
      "grad_norm": 0.44275254011154175,
      "learning_rate": 4.9024514701109766e-05,
      "loss": 0.0381,
      "step": 5310
    },
    {
      "epoch": 2.344645218157779,
      "grad_norm": 0.3805629312992096,
      "learning_rate": 4.885920482542043e-05,
      "loss": 0.0278,
      "step": 5320
    },
    {
      "epoch": 2.349052446011459,
      "grad_norm": 0.3801402151584625,
      "learning_rate": 4.869390742528438e-05,
      "loss": 0.0256,
      "step": 5330
    },
    {
      "epoch": 2.3534596738651388,
      "grad_norm": 0.4606906771659851,
      "learning_rate": 4.852862430836744e-05,
      "loss": 0.021,
      "step": 5340
    },
    {
      "epoch": 2.3578669017188187,
      "grad_norm": 0.4528634250164032,
      "learning_rate": 4.836335728217933e-05,
      "loss": 0.0232,
      "step": 5350
    },
    {
      "epoch": 2.3622741295724987,
      "grad_norm": 0.46547994017601013,
      "learning_rate": 4.819810815405379e-05,
      "loss": 0.0295,
      "step": 5360
    },
    {
      "epoch": 2.366681357426179,
      "grad_norm": 0.3093867897987366,
      "learning_rate": 4.803287873112877e-05,
      "loss": 0.0223,
      "step": 5370
    },
    {
      "epoch": 2.371088585279859,
      "grad_norm": 0.4733358919620514,
      "learning_rate": 4.786767082032681e-05,
      "loss": 0.0271,
      "step": 5380
    },
    {
      "epoch": 2.375495813133539,
      "grad_norm": 0.33468925952911377,
      "learning_rate": 4.77024862283351e-05,
      "loss": 0.0271,
      "step": 5390
    },
    {
      "epoch": 2.379903040987219,
      "grad_norm": 0.6140813827514648,
      "learning_rate": 4.753732676158593e-05,
      "loss": 0.0239,
      "step": 5400
    },
    {
      "epoch": 2.384310268840899,
      "grad_norm": 0.7943770289421082,
      "learning_rate": 4.737219422623672e-05,
      "loss": 0.027,
      "step": 5410
    },
    {
      "epoch": 2.388717496694579,
      "grad_norm": 0.520611584186554,
      "learning_rate": 4.720709042815044e-05,
      "loss": 0.0281,
      "step": 5420
    },
    {
      "epoch": 2.393124724548259,
      "grad_norm": 0.3081592917442322,
      "learning_rate": 4.704201717287578e-05,
      "loss": 0.0208,
      "step": 5430
    },
    {
      "epoch": 2.3975319524019394,
      "grad_norm": 0.2500794529914856,
      "learning_rate": 4.6876976265627404e-05,
      "loss": 0.0244,
      "step": 5440
    },
    {
      "epoch": 2.4019391802556194,
      "grad_norm": 0.6113792061805725,
      "learning_rate": 4.671196951126626e-05,
      "loss": 0.0287,
      "step": 5450
    },
    {
      "epoch": 2.4063464081092993,
      "grad_norm": 0.418827086687088,
      "learning_rate": 4.654699871427971e-05,
      "loss": 0.0188,
      "step": 5460
    },
    {
      "epoch": 2.4107536359629793,
      "grad_norm": 0.5944854617118835,
      "learning_rate": 4.6382065678762034e-05,
      "loss": 0.0248,
      "step": 5470
    },
    {
      "epoch": 2.4151608638166593,
      "grad_norm": 0.3814733922481537,
      "learning_rate": 4.6217172208394424e-05,
      "loss": 0.0174,
      "step": 5480
    },
    {
      "epoch": 2.4195680916703393,
      "grad_norm": 0.47005438804626465,
      "learning_rate": 4.605232010642549e-05,
      "loss": 0.0195,
      "step": 5490
    },
    {
      "epoch": 2.4239753195240192,
      "grad_norm": 0.4051513671875,
      "learning_rate": 4.588751117565142e-05,
      "loss": 0.0175,
      "step": 5500
    },
    {
      "epoch": 2.4283825473776997,
      "grad_norm": 0.21102257072925568,
      "learning_rate": 4.5722747218396214e-05,
      "loss": 0.0212,
      "step": 5510
    },
    {
      "epoch": 2.4327897752313796,
      "grad_norm": 0.2332678735256195,
      "learning_rate": 4.5558030036492194e-05,
      "loss": 0.0205,
      "step": 5520
    },
    {
      "epoch": 2.4371970030850596,
      "grad_norm": 0.5153501033782959,
      "learning_rate": 4.539336143125999e-05,
      "loss": 0.0199,
      "step": 5530
    },
    {
      "epoch": 2.4416042309387396,
      "grad_norm": 0.35871756076812744,
      "learning_rate": 4.522874320348916e-05,
      "loss": 0.0185,
      "step": 5540
    },
    {
      "epoch": 2.4460114587924195,
      "grad_norm": 0.4586464762687683,
      "learning_rate": 4.506417715341821e-05,
      "loss": 0.0207,
      "step": 5550
    },
    {
      "epoch": 2.4504186866460995,
      "grad_norm": 0.4426180124282837,
      "learning_rate": 4.489966508071511e-05,
      "loss": 0.0237,
      "step": 5560
    },
    {
      "epoch": 2.4548259144997795,
      "grad_norm": 0.46985065937042236,
      "learning_rate": 4.4735208784457575e-05,
      "loss": 0.0222,
      "step": 5570
    },
    {
      "epoch": 2.45923314235346,
      "grad_norm": 0.25444307923316956,
      "learning_rate": 4.457081006311325e-05,
      "loss": 0.0183,
      "step": 5580
    },
    {
      "epoch": 2.46364037020714,
      "grad_norm": 0.3950861096382141,
      "learning_rate": 4.440647071452027e-05,
      "loss": 0.0209,
      "step": 5590
    },
    {
      "epoch": 2.46804759806082,
      "grad_norm": 0.3346171975135803,
      "learning_rate": 4.424219253586737e-05,
      "loss": 0.026,
      "step": 5600
    },
    {
      "epoch": 2.4724548259145,
      "grad_norm": 0.406186044216156,
      "learning_rate": 4.407797732367443e-05,
      "loss": 0.0239,
      "step": 5610
    },
    {
      "epoch": 2.47686205376818,
      "grad_norm": 0.5083613395690918,
      "learning_rate": 4.391382687377268e-05,
      "loss": 0.0225,
      "step": 5620
    },
    {
      "epoch": 2.48126928162186,
      "grad_norm": 0.39063209295272827,
      "learning_rate": 4.374974298128512e-05,
      "loss": 0.0158,
      "step": 5630
    },
    {
      "epoch": 2.4856765094755398,
      "grad_norm": 0.507192850112915,
      "learning_rate": 4.358572744060699e-05,
      "loss": 0.0247,
      "step": 5640
    },
    {
      "epoch": 2.49008373732922,
      "grad_norm": 0.28652122616767883,
      "learning_rate": 4.342178204538588e-05,
      "loss": 0.0229,
      "step": 5650
    },
    {
      "epoch": 2.4944909651829,
      "grad_norm": 0.4468642473220825,
      "learning_rate": 4.325790858850241e-05,
      "loss": 0.0293,
      "step": 5660
    },
    {
      "epoch": 2.49889819303658,
      "grad_norm": 0.30711686611175537,
      "learning_rate": 4.309410886205043e-05,
      "loss": 0.0147,
      "step": 5670
    },
    {
      "epoch": 2.50330542089026,
      "grad_norm": 0.6272444128990173,
      "learning_rate": 4.293038465731752e-05,
      "loss": 0.0262,
      "step": 5680
    },
    {
      "epoch": 2.50771264874394,
      "grad_norm": 0.4111109673976898,
      "learning_rate": 4.276673776476533e-05,
      "loss": 0.0176,
      "step": 5690
    },
    {
      "epoch": 2.51211987659762,
      "grad_norm": 0.3570762872695923,
      "learning_rate": 4.260316997401007e-05,
      "loss": 0.0237,
      "step": 5700
    },
    {
      "epoch": 2.5165271044513,
      "grad_norm": 0.5258123874664307,
      "learning_rate": 4.243968307380293e-05,
      "loss": 0.0225,
      "step": 5710
    },
    {
      "epoch": 2.5209343323049804,
      "grad_norm": 0.4450371563434601,
      "learning_rate": 4.22762788520104e-05,
      "loss": 0.0179,
      "step": 5720
    },
    {
      "epoch": 2.52534156015866,
      "grad_norm": 0.34086015820503235,
      "learning_rate": 4.211295909559491e-05,
      "loss": 0.0236,
      "step": 5730
    },
    {
      "epoch": 2.5297487880123404,
      "grad_norm": 0.32176846265792847,
      "learning_rate": 4.194972559059511e-05,
      "loss": 0.0215,
      "step": 5740
    },
    {
      "epoch": 2.5341560158660204,
      "grad_norm": 0.4261510670185089,
      "learning_rate": 4.178658012210651e-05,
      "loss": 0.0192,
      "step": 5750
    },
    {
      "epoch": 2.5385632437197003,
      "grad_norm": 0.3140770196914673,
      "learning_rate": 4.162352447426177e-05,
      "loss": 0.0182,
      "step": 5760
    },
    {
      "epoch": 2.5429704715733803,
      "grad_norm": 0.20170685648918152,
      "learning_rate": 4.146056043021135e-05,
      "loss": 0.0174,
      "step": 5770
    },
    {
      "epoch": 2.5473776994270603,
      "grad_norm": 0.4431476891040802,
      "learning_rate": 4.1297689772103944e-05,
      "loss": 0.0261,
      "step": 5780
    },
    {
      "epoch": 2.5517849272807407,
      "grad_norm": 0.35332852602005005,
      "learning_rate": 4.113491428106694e-05,
      "loss": 0.0198,
      "step": 5790
    },
    {
      "epoch": 2.55619215513442,
      "grad_norm": 0.31072738766670227,
      "learning_rate": 4.0972235737187055e-05,
      "loss": 0.0222,
      "step": 5800
    },
    {
      "epoch": 2.5605993829881006,
      "grad_norm": 0.36561596393585205,
      "learning_rate": 4.080965591949076e-05,
      "loss": 0.029,
      "step": 5810
    },
    {
      "epoch": 2.5650066108417806,
      "grad_norm": 0.6434592008590698,
      "learning_rate": 4.0647176605924924e-05,
      "loss": 0.0286,
      "step": 5820
    },
    {
      "epoch": 2.5694138386954606,
      "grad_norm": 0.3110179603099823,
      "learning_rate": 4.0484799573337255e-05,
      "loss": 0.021,
      "step": 5830
    },
    {
      "epoch": 2.5738210665491406,
      "grad_norm": 0.43782544136047363,
      "learning_rate": 4.032252659745699e-05,
      "loss": 0.0189,
      "step": 5840
    },
    {
      "epoch": 2.5782282944028205,
      "grad_norm": 0.2952023446559906,
      "learning_rate": 4.016035945287539e-05,
      "loss": 0.0218,
      "step": 5850
    },
    {
      "epoch": 2.5826355222565005,
      "grad_norm": 0.35426491498947144,
      "learning_rate": 3.999829991302635e-05,
      "loss": 0.017,
      "step": 5860
    },
    {
      "epoch": 2.5870427501101805,
      "grad_norm": 0.3844353258609772,
      "learning_rate": 3.983634975016707e-05,
      "loss": 0.0163,
      "step": 5870
    },
    {
      "epoch": 2.591449977963861,
      "grad_norm": 0.22439396381378174,
      "learning_rate": 3.967451073535854e-05,
      "loss": 0.0216,
      "step": 5880
    },
    {
      "epoch": 2.595857205817541,
      "grad_norm": 0.3587394952774048,
      "learning_rate": 3.951278463844633e-05,
      "loss": 0.0187,
      "step": 5890
    },
    {
      "epoch": 2.600264433671221,
      "grad_norm": 0.5323532819747925,
      "learning_rate": 3.935117322804111e-05,
      "loss": 0.0197,
      "step": 5900
    },
    {
      "epoch": 2.604671661524901,
      "grad_norm": 0.4443974792957306,
      "learning_rate": 3.918967827149938e-05,
      "loss": 0.0286,
      "step": 5910
    },
    {
      "epoch": 2.609078889378581,
      "grad_norm": 0.51837557554245,
      "learning_rate": 3.9028301534904094e-05,
      "loss": 0.0188,
      "step": 5920
    },
    {
      "epoch": 2.6134861172322608,
      "grad_norm": 0.5784846544265747,
      "learning_rate": 3.88670447830454e-05,
      "loss": 0.0253,
      "step": 5930
    },
    {
      "epoch": 2.6178933450859407,
      "grad_norm": 0.34923461079597473,
      "learning_rate": 3.870590977940132e-05,
      "loss": 0.0257,
      "step": 5940
    },
    {
      "epoch": 2.622300572939621,
      "grad_norm": 0.5016012787818909,
      "learning_rate": 3.8544898286118404e-05,
      "loss": 0.0186,
      "step": 5950
    },
    {
      "epoch": 2.626707800793301,
      "grad_norm": 0.46786078810691833,
      "learning_rate": 3.838401206399257e-05,
      "loss": 0.0223,
      "step": 5960
    },
    {
      "epoch": 2.631115028646981,
      "grad_norm": 0.32447516918182373,
      "learning_rate": 3.822325287244975e-05,
      "loss": 0.0223,
      "step": 5970
    },
    {
      "epoch": 2.635522256500661,
      "grad_norm": 0.23921582102775574,
      "learning_rate": 3.8062622469526725e-05,
      "loss": 0.0157,
      "step": 5980
    },
    {
      "epoch": 2.639929484354341,
      "grad_norm": 0.41544628143310547,
      "learning_rate": 3.790212261185183e-05,
      "loss": 0.0222,
      "step": 5990
    },
    {
      "epoch": 2.644336712208021,
      "grad_norm": 0.19495147466659546,
      "learning_rate": 3.7741755054625794e-05,
      "loss": 0.0255,
      "step": 6000
    },
    {
      "epoch": 2.648743940061701,
      "grad_norm": 0.26758095622062683,
      "learning_rate": 3.758152155160255e-05,
      "loss": 0.0194,
      "step": 6010
    },
    {
      "epoch": 2.6531511679153814,
      "grad_norm": 0.32311534881591797,
      "learning_rate": 3.742142385506999e-05,
      "loss": 0.0173,
      "step": 6020
    },
    {
      "epoch": 2.6575583957690614,
      "grad_norm": 0.36834850907325745,
      "learning_rate": 3.72614637158309e-05,
      "loss": 0.0253,
      "step": 6030
    },
    {
      "epoch": 2.6619656236227414,
      "grad_norm": 0.22646619379520416,
      "learning_rate": 3.710164288318371e-05,
      "loss": 0.0189,
      "step": 6040
    },
    {
      "epoch": 2.6663728514764213,
      "grad_norm": 0.4088250696659088,
      "learning_rate": 3.694196310490345e-05,
      "loss": 0.0223,
      "step": 6050
    },
    {
      "epoch": 2.6707800793301013,
      "grad_norm": 0.2535192668437958,
      "learning_rate": 3.678242612722259e-05,
      "loss": 0.0156,
      "step": 6060
    },
    {
      "epoch": 2.6751873071837813,
      "grad_norm": 0.25822219252586365,
      "learning_rate": 3.6623033694811953e-05,
      "loss": 0.0173,
      "step": 6070
    },
    {
      "epoch": 2.6795945350374613,
      "grad_norm": 0.27290865778923035,
      "learning_rate": 3.6463787550761665e-05,
      "loss": 0.0185,
      "step": 6080
    },
    {
      "epoch": 2.6840017628911417,
      "grad_norm": 0.478697270154953,
      "learning_rate": 3.630468943656202e-05,
      "loss": 0.0174,
      "step": 6090
    },
    {
      "epoch": 2.6884089907448216,
      "grad_norm": 0.5832062363624573,
      "learning_rate": 3.6145741092084523e-05,
      "loss": 0.0228,
      "step": 6100
    },
    {
      "epoch": 2.6928162185985016,
      "grad_norm": 0.4017068147659302,
      "learning_rate": 3.598694425556278e-05,
      "loss": 0.0238,
      "step": 6110
    },
    {
      "epoch": 2.6972234464521816,
      "grad_norm": 0.2960377335548401,
      "learning_rate": 3.58283006635736e-05,
      "loss": 0.0173,
      "step": 6120
    },
    {
      "epoch": 2.7016306743058616,
      "grad_norm": 0.42903682589530945,
      "learning_rate": 3.566981205101781e-05,
      "loss": 0.0225,
      "step": 6130
    },
    {
      "epoch": 2.7060379021595415,
      "grad_norm": 0.32902613282203674,
      "learning_rate": 3.5511480151101556e-05,
      "loss": 0.0205,
      "step": 6140
    },
    {
      "epoch": 2.7104451300132215,
      "grad_norm": 0.236186683177948,
      "learning_rate": 3.5353306695317104e-05,
      "loss": 0.0173,
      "step": 6150
    },
    {
      "epoch": 2.714852357866902,
      "grad_norm": 0.35425737500190735,
      "learning_rate": 3.519529341342402e-05,
      "loss": 0.0215,
      "step": 6160
    },
    {
      "epoch": 2.719259585720582,
      "grad_norm": 0.5131089091300964,
      "learning_rate": 3.503744203343026e-05,
      "loss": 0.0183,
      "step": 6170
    },
    {
      "epoch": 2.723666813574262,
      "grad_norm": 0.287217915058136,
      "learning_rate": 3.487975428157318e-05,
      "loss": 0.0257,
      "step": 6180
    },
    {
      "epoch": 2.728074041427942,
      "grad_norm": 0.3751159906387329,
      "learning_rate": 3.472223188230083e-05,
      "loss": 0.0175,
      "step": 6190
    },
    {
      "epoch": 2.732481269281622,
      "grad_norm": 0.4694988429546356,
      "learning_rate": 3.4564876558252866e-05,
      "loss": 0.0246,
      "step": 6200
    },
    {
      "epoch": 2.736888497135302,
      "grad_norm": 0.3909672498703003,
      "learning_rate": 3.440769003024195e-05,
      "loss": 0.0225,
      "step": 6210
    },
    {
      "epoch": 2.7412957249889818,
      "grad_norm": 0.3745891749858856,
      "learning_rate": 3.425067401723477e-05,
      "loss": 0.0293,
      "step": 6220
    },
    {
      "epoch": 2.745702952842662,
      "grad_norm": 0.45297297835350037,
      "learning_rate": 3.409383023633325e-05,
      "loss": 0.0185,
      "step": 6230
    },
    {
      "epoch": 2.750110180696342,
      "grad_norm": 0.4655900001525879,
      "learning_rate": 3.3937160402755894e-05,
      "loss": 0.0185,
      "step": 6240
    },
    {
      "epoch": 2.754517408550022,
      "grad_norm": 0.3401032090187073,
      "learning_rate": 3.378066622981885e-05,
      "loss": 0.0222,
      "step": 6250
    },
    {
      "epoch": 2.758924636403702,
      "grad_norm": 0.46499380469322205,
      "learning_rate": 3.362434942891738e-05,
      "loss": 0.0252,
      "step": 6260
    },
    {
      "epoch": 2.763331864257382,
      "grad_norm": 0.3436833620071411,
      "learning_rate": 3.346821170950693e-05,
      "loss": 0.0215,
      "step": 6270
    },
    {
      "epoch": 2.767739092111062,
      "grad_norm": 0.21363238990306854,
      "learning_rate": 3.3312254779084585e-05,
      "loss": 0.0218,
      "step": 6280
    },
    {
      "epoch": 2.772146319964742,
      "grad_norm": 0.5733113884925842,
      "learning_rate": 3.315648034317039e-05,
      "loss": 0.0242,
      "step": 6290
    },
    {
      "epoch": 2.7765535478184225,
      "grad_norm": 0.37033259868621826,
      "learning_rate": 3.3000890105288564e-05,
      "loss": 0.0194,
      "step": 6300
    },
    {
      "epoch": 2.780960775672102,
      "grad_norm": 0.45300716161727905,
      "learning_rate": 3.284548576694908e-05,
      "loss": 0.0167,
      "step": 6310
    },
    {
      "epoch": 2.7853680035257824,
      "grad_norm": 0.4063990116119385,
      "learning_rate": 3.2690269027628815e-05,
      "loss": 0.0199,
      "step": 6320
    },
    {
      "epoch": 2.7897752313794624,
      "grad_norm": 0.3002220094203949,
      "learning_rate": 3.253524158475324e-05,
      "loss": 0.0185,
      "step": 6330
    },
    {
      "epoch": 2.7941824592331423,
      "grad_norm": 0.4974387288093567,
      "learning_rate": 3.238040513367757e-05,
      "loss": 0.0255,
      "step": 6340
    },
    {
      "epoch": 2.7985896870868223,
      "grad_norm": 0.3639296889305115,
      "learning_rate": 3.222576136766843e-05,
      "loss": 0.0171,
      "step": 6350
    },
    {
      "epoch": 2.8029969149405023,
      "grad_norm": 0.292157918214798,
      "learning_rate": 3.2071311977885324e-05,
      "loss": 0.0176,
      "step": 6360
    },
    {
      "epoch": 2.8074041427941827,
      "grad_norm": 0.28334933519363403,
      "learning_rate": 3.191705865336197e-05,
      "loss": 0.0179,
      "step": 6370
    },
    {
      "epoch": 2.8118113706478622,
      "grad_norm": 0.3558843433856964,
      "learning_rate": 3.1763003080988075e-05,
      "loss": 0.0188,
      "step": 6380
    },
    {
      "epoch": 2.8162185985015427,
      "grad_norm": 0.5022380948066711,
      "learning_rate": 3.160914694549063e-05,
      "loss": 0.0291,
      "step": 6390
    },
    {
      "epoch": 2.8206258263552226,
      "grad_norm": 0.36502861976623535,
      "learning_rate": 3.145549192941573e-05,
      "loss": 0.024,
      "step": 6400
    },
    {
      "epoch": 2.8250330542089026,
      "grad_norm": 0.28113237023353577,
      "learning_rate": 3.130203971310999e-05,
      "loss": 0.0199,
      "step": 6410
    },
    {
      "epoch": 2.8294402820625826,
      "grad_norm": 0.28604063391685486,
      "learning_rate": 3.114879197470225e-05,
      "loss": 0.0177,
      "step": 6420
    },
    {
      "epoch": 2.8338475099162626,
      "grad_norm": 0.3796468675136566,
      "learning_rate": 3.0995750390085285e-05,
      "loss": 0.0212,
      "step": 6430
    },
    {
      "epoch": 2.838254737769943,
      "grad_norm": 0.4099070131778717,
      "learning_rate": 3.084291663289728e-05,
      "loss": 0.0156,
      "step": 6440
    },
    {
      "epoch": 2.8426619656236225,
      "grad_norm": 0.39487749338150024,
      "learning_rate": 3.069029237450375e-05,
      "loss": 0.0237,
      "step": 6450
    },
    {
      "epoch": 2.847069193477303,
      "grad_norm": 0.4001524746417999,
      "learning_rate": 3.053787928397911e-05,
      "loss": 0.016,
      "step": 6460
    },
    {
      "epoch": 2.851476421330983,
      "grad_norm": 0.4424799382686615,
      "learning_rate": 3.0385679028088526e-05,
      "loss": 0.0152,
      "step": 6470
    },
    {
      "epoch": 2.855883649184663,
      "grad_norm": 0.31638985872268677,
      "learning_rate": 3.023369327126959e-05,
      "loss": 0.015,
      "step": 6480
    },
    {
      "epoch": 2.860290877038343,
      "grad_norm": 0.41499805450439453,
      "learning_rate": 3.0081923675614198e-05,
      "loss": 0.0139,
      "step": 6490
    },
    {
      "epoch": 2.864698104892023,
      "grad_norm": 0.16572968661785126,
      "learning_rate": 2.993037190085034e-05,
      "loss": 0.0151,
      "step": 6500
    },
    {
      "epoch": 2.8691053327457032,
      "grad_norm": 0.38601207733154297,
      "learning_rate": 2.977903960432392e-05,
      "loss": 0.0264,
      "step": 6510
    },
    {
      "epoch": 2.8735125605993828,
      "grad_norm": 0.4049271047115326,
      "learning_rate": 2.9627928440980722e-05,
      "loss": 0.0263,
      "step": 6520
    },
    {
      "epoch": 2.877919788453063,
      "grad_norm": 0.35763201117515564,
      "learning_rate": 2.9477040063348183e-05,
      "loss": 0.0152,
      "step": 6530
    },
    {
      "epoch": 2.882327016306743,
      "grad_norm": 0.300096333026886,
      "learning_rate": 2.9326376121517456e-05,
      "loss": 0.0228,
      "step": 6540
    },
    {
      "epoch": 2.886734244160423,
      "grad_norm": 0.41060328483581543,
      "learning_rate": 2.9175938263125236e-05,
      "loss": 0.0213,
      "step": 6550
    },
    {
      "epoch": 2.891141472014103,
      "grad_norm": 0.443558007478714,
      "learning_rate": 2.9025728133335873e-05,
      "loss": 0.0149,
      "step": 6560
    },
    {
      "epoch": 2.895548699867783,
      "grad_norm": 0.4524803161621094,
      "learning_rate": 2.8875747374823288e-05,
      "loss": 0.0264,
      "step": 6570
    },
    {
      "epoch": 2.899955927721463,
      "grad_norm": 0.4135073721408844,
      "learning_rate": 2.872599762775298e-05,
      "loss": 0.0163,
      "step": 6580
    },
    {
      "epoch": 2.904363155575143,
      "grad_norm": 0.2794502079486847,
      "learning_rate": 2.857648052976425e-05,
      "loss": 0.0179,
      "step": 6590
    },
    {
      "epoch": 2.9087703834288234,
      "grad_norm": 0.5536352396011353,
      "learning_rate": 2.8427197715952047e-05,
      "loss": 0.0145,
      "step": 6600
    },
    {
      "epoch": 2.9131776112825034,
      "grad_norm": 0.3690505623817444,
      "learning_rate": 2.8278150818849393e-05,
      "loss": 0.0154,
      "step": 6610
    },
    {
      "epoch": 2.9175848391361834,
      "grad_norm": 0.285930871963501,
      "learning_rate": 2.812934146840922e-05,
      "loss": 0.0181,
      "step": 6620
    },
    {
      "epoch": 2.9219920669898634,
      "grad_norm": 0.3328244686126709,
      "learning_rate": 2.7980771291986764e-05,
      "loss": 0.0158,
      "step": 6630
    },
    {
      "epoch": 2.9263992948435433,
      "grad_norm": 0.27469179034233093,
      "learning_rate": 2.783244191432167e-05,
      "loss": 0.0197,
      "step": 6640
    },
    {
      "epoch": 2.9308065226972233,
      "grad_norm": 0.3617198169231415,
      "learning_rate": 2.768435495752022e-05,
      "loss": 0.0162,
      "step": 6650
    },
    {
      "epoch": 2.9352137505509033,
      "grad_norm": 0.2811722159385681,
      "learning_rate": 2.753651204103771e-05,
      "loss": 0.019,
      "step": 6660
    },
    {
      "epoch": 2.9396209784045837,
      "grad_norm": 0.3542955815792084,
      "learning_rate": 2.7388914781660523e-05,
      "loss": 0.0242,
      "step": 6670
    },
    {
      "epoch": 2.9440282062582637,
      "grad_norm": 0.31205010414123535,
      "learning_rate": 2.7241564793488693e-05,
      "loss": 0.0178,
      "step": 6680
    },
    {
      "epoch": 2.9484354341119436,
      "grad_norm": 0.3759402632713318,
      "learning_rate": 2.7094463687918037e-05,
      "loss": 0.0214,
      "step": 6690
    },
    {
      "epoch": 2.9528426619656236,
      "grad_norm": 0.4986971318721771,
      "learning_rate": 2.694761307362268e-05,
      "loss": 0.0229,
      "step": 6700
    },
    {
      "epoch": 2.9572498898193036,
      "grad_norm": 0.5741224884986877,
      "learning_rate": 2.6801014556537467e-05,
      "loss": 0.0219,
      "step": 6710
    },
    {
      "epoch": 2.9616571176729836,
      "grad_norm": 0.2744426727294922,
      "learning_rate": 2.6654669739840243e-05,
      "loss": 0.0176,
      "step": 6720
    },
    {
      "epoch": 2.9660643455266635,
      "grad_norm": 0.2514480650424957,
      "learning_rate": 2.650858022393451e-05,
      "loss": 0.0284,
      "step": 6730
    },
    {
      "epoch": 2.970471573380344,
      "grad_norm": 0.3333909511566162,
      "learning_rate": 2.6362747606431747e-05,
      "loss": 0.0221,
      "step": 6740
    },
    {
      "epoch": 2.974878801234024,
      "grad_norm": 0.4586556553840637,
      "learning_rate": 2.6217173482134172e-05,
      "loss": 0.0179,
      "step": 6750
    },
    {
      "epoch": 2.979286029087704,
      "grad_norm": 0.2954588532447815,
      "learning_rate": 2.6071859443017044e-05,
      "loss": 0.0243,
      "step": 6760
    },
    {
      "epoch": 2.983693256941384,
      "grad_norm": 0.34891384840011597,
      "learning_rate": 2.5926807078211414e-05,
      "loss": 0.02,
      "step": 6770
    },
    {
      "epoch": 2.988100484795064,
      "grad_norm": 0.45223283767700195,
      "learning_rate": 2.5782017973986728e-05,
      "loss": 0.0191,
      "step": 6780
    },
    {
      "epoch": 2.992507712648744,
      "grad_norm": 0.4650042951107025,
      "learning_rate": 2.5637493713733374e-05,
      "loss": 0.0172,
      "step": 6790
    },
    {
      "epoch": 2.996914940502424,
      "grad_norm": 0.23131594061851501,
      "learning_rate": 2.549323587794559e-05,
      "loss": 0.0281,
      "step": 6800
    },
    {
      "epoch": 3.001322168356104,
      "grad_norm": 0.43882203102111816,
      "learning_rate": 2.5349246044203895e-05,
      "loss": 0.0177,
      "step": 6810
    },
    {
      "epoch": 3.005729396209784,
      "grad_norm": 0.4513368308544159,
      "learning_rate": 2.520552578715808e-05,
      "loss": 0.0187,
      "step": 6820
    },
    {
      "epoch": 3.010136624063464,
      "grad_norm": 0.39059892296791077,
      "learning_rate": 2.506207667850981e-05,
      "loss": 0.0238,
      "step": 6830
    },
    {
      "epoch": 3.014543851917144,
      "grad_norm": 0.369302898645401,
      "learning_rate": 2.4918900286995555e-05,
      "loss": 0.0202,
      "step": 6840
    },
    {
      "epoch": 3.018951079770824,
      "grad_norm": 0.2748335003852844,
      "learning_rate": 2.4775998178369458e-05,
      "loss": 0.0155,
      "step": 6850
    },
    {
      "epoch": 3.023358307624504,
      "grad_norm": 0.5056502223014832,
      "learning_rate": 2.4633371915386017e-05,
      "loss": 0.0187,
      "step": 6860
    },
    {
      "epoch": 3.027765535478184,
      "grad_norm": 0.4222106337547302,
      "learning_rate": 2.4491023057783235e-05,
      "loss": 0.0161,
      "step": 6870
    },
    {
      "epoch": 3.0321727633318645,
      "grad_norm": 0.25616875290870667,
      "learning_rate": 2.4348953162265375e-05,
      "loss": 0.0165,
      "step": 6880
    },
    {
      "epoch": 3.0365799911855444,
      "grad_norm": 0.3021768033504486,
      "learning_rate": 2.420716378248607e-05,
      "loss": 0.0241,
      "step": 6890
    },
    {
      "epoch": 3.0409872190392244,
      "grad_norm": 0.3301573395729065,
      "learning_rate": 2.4065656469031266e-05,
      "loss": 0.0189,
      "step": 6900
    },
    {
      "epoch": 3.0453944468929044,
      "grad_norm": 0.34506645798683167,
      "learning_rate": 2.3924432769402268e-05,
      "loss": 0.0183,
      "step": 6910
    },
    {
      "epoch": 3.0498016747465844,
      "grad_norm": 0.21936343610286713,
      "learning_rate": 2.3783494227998844e-05,
      "loss": 0.0205,
      "step": 6920
    },
    {
      "epoch": 3.0542089026002643,
      "grad_norm": 0.3818731904029846,
      "learning_rate": 2.3642842386102264e-05,
      "loss": 0.0171,
      "step": 6930
    },
    {
      "epoch": 3.0586161304539443,
      "grad_norm": 0.32185959815979004,
      "learning_rate": 2.3502478781858567e-05,
      "loss": 0.0181,
      "step": 6940
    },
    {
      "epoch": 3.0630233583076247,
      "grad_norm": 0.26512759923934937,
      "learning_rate": 2.3362404950261628e-05,
      "loss": 0.0176,
      "step": 6950
    },
    {
      "epoch": 3.0674305861613047,
      "grad_norm": 0.41826969385147095,
      "learning_rate": 2.3222622423136458e-05,
      "loss": 0.0175,
      "step": 6960
    },
    {
      "epoch": 3.0718378140149847,
      "grad_norm": 0.2612951099872589,
      "learning_rate": 2.3083132729122332e-05,
      "loss": 0.0202,
      "step": 6970
    },
    {
      "epoch": 3.0762450418686647,
      "grad_norm": 0.29240697622299194,
      "learning_rate": 2.294393739365621e-05,
      "loss": 0.0178,
      "step": 6980
    },
    {
      "epoch": 3.0806522697223446,
      "grad_norm": 0.41177690029144287,
      "learning_rate": 2.2805037938956e-05,
      "loss": 0.0227,
      "step": 6990
    },
    {
      "epoch": 3.0850594975760246,
      "grad_norm": 0.3977733254432678,
      "learning_rate": 2.266643588400386e-05,
      "loss": 0.0207,
      "step": 7000
    },
    {
      "epoch": 3.0894667254297046,
      "grad_norm": 0.4900129437446594,
      "learning_rate": 2.252813274452969e-05,
      "loss": 0.022,
      "step": 7010
    },
    {
      "epoch": 3.0938739532833845,
      "grad_norm": 0.3002599775791168,
      "learning_rate": 2.2390130032994427e-05,
      "loss": 0.0174,
      "step": 7020
    },
    {
      "epoch": 3.098281181137065,
      "grad_norm": 0.3148794174194336,
      "learning_rate": 2.2252429258573633e-05,
      "loss": 0.0159,
      "step": 7030
    },
    {
      "epoch": 3.102688408990745,
      "grad_norm": 0.41567566990852356,
      "learning_rate": 2.2115031927140904e-05,
      "loss": 0.0194,
      "step": 7040
    },
    {
      "epoch": 3.107095636844425,
      "grad_norm": 0.40259844064712524,
      "learning_rate": 2.1977939541251463e-05,
      "loss": 0.0141,
      "step": 7050
    },
    {
      "epoch": 3.111502864698105,
      "grad_norm": 0.26434504985809326,
      "learning_rate": 2.1841153600125684e-05,
      "loss": 0.0159,
      "step": 7060
    },
    {
      "epoch": 3.115910092551785,
      "grad_norm": 0.6673845648765564,
      "learning_rate": 2.170467559963267e-05,
      "loss": 0.0256,
      "step": 7070
    },
    {
      "epoch": 3.120317320405465,
      "grad_norm": 0.45193445682525635,
      "learning_rate": 2.1568507032273982e-05,
      "loss": 0.0181,
      "step": 7080
    },
    {
      "epoch": 3.124724548259145,
      "grad_norm": 0.2907746732234955,
      "learning_rate": 2.1432649387167264e-05,
      "loss": 0.0165,
      "step": 7090
    },
    {
      "epoch": 3.1291317761128252,
      "grad_norm": 0.33764925599098206,
      "learning_rate": 2.1297104150029973e-05,
      "loss": 0.0217,
      "step": 7100
    },
    {
      "epoch": 3.133539003966505,
      "grad_norm": 0.5358210802078247,
      "learning_rate": 2.116187280316307e-05,
      "loss": 0.021,
      "step": 7110
    },
    {
      "epoch": 3.137946231820185,
      "grad_norm": 0.3613552451133728,
      "learning_rate": 2.1026956825434908e-05,
      "loss": 0.0148,
      "step": 7120
    },
    {
      "epoch": 3.142353459673865,
      "grad_norm": 0.40836426615715027,
      "learning_rate": 2.0892357692265017e-05,
      "loss": 0.0191,
      "step": 7130
    },
    {
      "epoch": 3.146760687527545,
      "grad_norm": 0.3762858808040619,
      "learning_rate": 2.0758076875607947e-05,
      "loss": 0.021,
      "step": 7140
    },
    {
      "epoch": 3.151167915381225,
      "grad_norm": 0.25755882263183594,
      "learning_rate": 2.0624115843937207e-05,
      "loss": 0.0174,
      "step": 7150
    },
    {
      "epoch": 3.155575143234905,
      "grad_norm": 0.32605254650115967,
      "learning_rate": 2.0490476062229157e-05,
      "loss": 0.023,
      "step": 7160
    },
    {
      "epoch": 3.1599823710885855,
      "grad_norm": 0.2508399784564972,
      "learning_rate": 2.035715899194704e-05,
      "loss": 0.0187,
      "step": 7170
    },
    {
      "epoch": 3.1643895989422655,
      "grad_norm": 0.4113372564315796,
      "learning_rate": 2.022416609102499e-05,
      "loss": 0.0187,
      "step": 7180
    },
    {
      "epoch": 3.1687968267959454,
      "grad_norm": 0.37528106570243835,
      "learning_rate": 2.009149881385205e-05,
      "loss": 0.0163,
      "step": 7190
    },
    {
      "epoch": 3.1732040546496254,
      "grad_norm": 0.38423988223075867,
      "learning_rate": 1.995915861125634e-05,
      "loss": 0.0198,
      "step": 7200
    },
    {
      "epoch": 3.1776112825033054,
      "grad_norm": 0.17892250418663025,
      "learning_rate": 1.9827146930489065e-05,
      "loss": 0.014,
      "step": 7210
    },
    {
      "epoch": 3.1820185103569854,
      "grad_norm": 0.19512003660202026,
      "learning_rate": 1.9695465215208848e-05,
      "loss": 0.0166,
      "step": 7220
    },
    {
      "epoch": 3.1864257382106653,
      "grad_norm": 0.24375677108764648,
      "learning_rate": 1.9564114905465813e-05,
      "loss": 0.0235,
      "step": 7230
    },
    {
      "epoch": 3.1908329660643453,
      "grad_norm": 0.61383056640625,
      "learning_rate": 1.9433097437685936e-05,
      "loss": 0.0156,
      "step": 7240
    },
    {
      "epoch": 3.1952401939180257,
      "grad_norm": 0.28985610604286194,
      "learning_rate": 1.930241424465521e-05,
      "loss": 0.0193,
      "step": 7250
    },
    {
      "epoch": 3.1996474217717057,
      "grad_norm": 0.2939668893814087,
      "learning_rate": 1.9172066755504115e-05,
      "loss": 0.0186,
      "step": 7260
    },
    {
      "epoch": 3.2040546496253857,
      "grad_norm": 0.28899717330932617,
      "learning_rate": 1.9042056395691914e-05,
      "loss": 0.0195,
      "step": 7270
    },
    {
      "epoch": 3.2084618774790656,
      "grad_norm": 0.1790951043367386,
      "learning_rate": 1.8912384586991066e-05,
      "loss": 0.0139,
      "step": 7280
    },
    {
      "epoch": 3.2128691053327456,
      "grad_norm": 0.3556305170059204,
      "learning_rate": 1.8783052747471717e-05,
      "loss": 0.0161,
      "step": 7290
    },
    {
      "epoch": 3.2172763331864256,
      "grad_norm": 0.40681588649749756,
      "learning_rate": 1.865406229148611e-05,
      "loss": 0.0181,
      "step": 7300
    },
    {
      "epoch": 3.2216835610401056,
      "grad_norm": 0.29141101241111755,
      "learning_rate": 1.8525414629653233e-05,
      "loss": 0.0134,
      "step": 7310
    },
    {
      "epoch": 3.226090788893786,
      "grad_norm": 0.28384751081466675,
      "learning_rate": 1.8397111168843255e-05,
      "loss": 0.0163,
      "step": 7320
    },
    {
      "epoch": 3.230498016747466,
      "grad_norm": 0.3549920320510864,
      "learning_rate": 1.8269153312162323e-05,
      "loss": 0.0163,
      "step": 7330
    },
    {
      "epoch": 3.234905244601146,
      "grad_norm": 0.22828508913516998,
      "learning_rate": 1.8141542458937054e-05,
      "loss": 0.0183,
      "step": 7340
    },
    {
      "epoch": 3.239312472454826,
      "grad_norm": 0.2825407385826111,
      "learning_rate": 1.8014280004699268e-05,
      "loss": 0.0205,
      "step": 7350
    },
    {
      "epoch": 3.243719700308506,
      "grad_norm": 0.4135924279689789,
      "learning_rate": 1.788736734117078e-05,
      "loss": 0.0202,
      "step": 7360
    },
    {
      "epoch": 3.248126928162186,
      "grad_norm": 0.4002763032913208,
      "learning_rate": 1.7760805856248152e-05,
      "loss": 0.0193,
      "step": 7370
    },
    {
      "epoch": 3.252534156015866,
      "grad_norm": 0.2891213595867157,
      "learning_rate": 1.7634596933987518e-05,
      "loss": 0.018,
      "step": 7380
    },
    {
      "epoch": 3.2569413838695462,
      "grad_norm": 0.27820175886154175,
      "learning_rate": 1.7508741954589404e-05,
      "loss": 0.0178,
      "step": 7390
    },
    {
      "epoch": 3.261348611723226,
      "grad_norm": 0.3232216238975525,
      "learning_rate": 1.7383242294383717e-05,
      "loss": 0.0163,
      "step": 7400
    },
    {
      "epoch": 3.265755839576906,
      "grad_norm": 0.28072431683540344,
      "learning_rate": 1.7258099325814632e-05,
      "loss": 0.0159,
      "step": 7410
    },
    {
      "epoch": 3.270163067430586,
      "grad_norm": 0.22099190950393677,
      "learning_rate": 1.7133314417425594e-05,
      "loss": 0.0263,
      "step": 7420
    },
    {
      "epoch": 3.274570295284266,
      "grad_norm": 0.20242473483085632,
      "learning_rate": 1.7008888933844408e-05,
      "loss": 0.0168,
      "step": 7430
    },
    {
      "epoch": 3.278977523137946,
      "grad_norm": 0.29331356287002563,
      "learning_rate": 1.6884824235768172e-05,
      "loss": 0.0134,
      "step": 7440
    },
    {
      "epoch": 3.283384750991626,
      "grad_norm": 0.3572656810283661,
      "learning_rate": 1.6761121679948592e-05,
      "loss": 0.018,
      "step": 7450
    },
    {
      "epoch": 3.2877919788453065,
      "grad_norm": 0.2896406948566437,
      "learning_rate": 1.663778261917695e-05,
      "loss": 0.0154,
      "step": 7460
    },
    {
      "epoch": 3.2921992066989865,
      "grad_norm": 0.40152695775032043,
      "learning_rate": 1.651480840226952e-05,
      "loss": 0.0191,
      "step": 7470
    },
    {
      "epoch": 3.2966064345526664,
      "grad_norm": 0.24419641494750977,
      "learning_rate": 1.639220037405258e-05,
      "loss": 0.0218,
      "step": 7480
    },
    {
      "epoch": 3.3010136624063464,
      "grad_norm": 0.519145131111145,
      "learning_rate": 1.6269959875347906e-05,
      "loss": 0.0172,
      "step": 7490
    },
    {
      "epoch": 3.3054208902600264,
      "grad_norm": 0.44710633158683777,
      "learning_rate": 1.614808824295802e-05,
      "loss": 0.016,
      "step": 7500
    },
    {
      "epoch": 3.3098281181137064,
      "grad_norm": 0.21060188114643097,
      "learning_rate": 1.602658680965152e-05,
      "loss": 0.0146,
      "step": 7510
    },
    {
      "epoch": 3.3142353459673863,
      "grad_norm": 0.3923156261444092,
      "learning_rate": 1.5905456904148686e-05,
      "loss": 0.0224,
      "step": 7520
    },
    {
      "epoch": 3.3186425738210668,
      "grad_norm": 0.4309709966182709,
      "learning_rate": 1.57846998511067e-05,
      "loss": 0.0189,
      "step": 7530
    },
    {
      "epoch": 3.3230498016747467,
      "grad_norm": 0.42418399453163147,
      "learning_rate": 1.566431697110538e-05,
      "loss": 0.016,
      "step": 7540
    },
    {
      "epoch": 3.3274570295284267,
      "grad_norm": 0.2486649602651596,
      "learning_rate": 1.554430958063259e-05,
      "loss": 0.019,
      "step": 7550
    },
    {
      "epoch": 3.3318642573821067,
      "grad_norm": 0.2620919644832611,
      "learning_rate": 1.5424678992069912e-05,
      "loss": 0.0154,
      "step": 7560
    },
    {
      "epoch": 3.3362714852357866,
      "grad_norm": 0.32689058780670166,
      "learning_rate": 1.5305426513678362e-05,
      "loss": 0.0132,
      "step": 7570
    },
    {
      "epoch": 3.3406787130894666,
      "grad_norm": 0.2152976244688034,
      "learning_rate": 1.518655344958388e-05,
      "loss": 0.0183,
      "step": 7580
    },
    {
      "epoch": 3.3450859409431466,
      "grad_norm": 0.2608199715614319,
      "learning_rate": 1.5068061099763275e-05,
      "loss": 0.0169,
      "step": 7590
    },
    {
      "epoch": 3.349493168796827,
      "grad_norm": 0.25575026869773865,
      "learning_rate": 1.494995076002988e-05,
      "loss": 0.0176,
      "step": 7600
    },
    {
      "epoch": 3.353900396650507,
      "grad_norm": 0.2881426513195038,
      "learning_rate": 1.4832223722019456e-05,
      "loss": 0.0144,
      "step": 7610
    },
    {
      "epoch": 3.358307624504187,
      "grad_norm": 0.21158578991889954,
      "learning_rate": 1.4714881273176035e-05,
      "loss": 0.0155,
      "step": 7620
    },
    {
      "epoch": 3.362714852357867,
      "grad_norm": 0.18288742005825043,
      "learning_rate": 1.4597924696737835e-05,
      "loss": 0.0229,
      "step": 7630
    },
    {
      "epoch": 3.367122080211547,
      "grad_norm": 0.2691928744316101,
      "learning_rate": 1.4481355271723252e-05,
      "loss": 0.0186,
      "step": 7640
    },
    {
      "epoch": 3.371529308065227,
      "grad_norm": 0.3401619791984558,
      "learning_rate": 1.4365174272916809e-05,
      "loss": 0.0151,
      "step": 7650
    },
    {
      "epoch": 3.375936535918907,
      "grad_norm": 0.2799758315086365,
      "learning_rate": 1.4249382970855319e-05,
      "loss": 0.0198,
      "step": 7660
    },
    {
      "epoch": 3.3803437637725873,
      "grad_norm": 0.5050471425056458,
      "learning_rate": 1.4133982631813903e-05,
      "loss": 0.0129,
      "step": 7670
    },
    {
      "epoch": 3.3847509916262672,
      "grad_norm": 0.3345412015914917,
      "learning_rate": 1.4018974517792194e-05,
      "loss": 0.0165,
      "step": 7680
    },
    {
      "epoch": 3.389158219479947,
      "grad_norm": 0.21150177717208862,
      "learning_rate": 1.390435988650048e-05,
      "loss": 0.0175,
      "step": 7690
    },
    {
      "epoch": 3.393565447333627,
      "grad_norm": 0.21793733537197113,
      "learning_rate": 1.3790139991346006e-05,
      "loss": 0.0115,
      "step": 7700
    },
    {
      "epoch": 3.397972675187307,
      "grad_norm": 0.2522967457771301,
      "learning_rate": 1.367631608141926e-05,
      "loss": 0.0153,
      "step": 7710
    },
    {
      "epoch": 3.402379903040987,
      "grad_norm": 0.2919367849826813,
      "learning_rate": 1.3562889401480278e-05,
      "loss": 0.0231,
      "step": 7720
    },
    {
      "epoch": 3.406787130894667,
      "grad_norm": 0.47937464714050293,
      "learning_rate": 1.3449861191945074e-05,
      "loss": 0.0185,
      "step": 7730
    },
    {
      "epoch": 3.4111943587483475,
      "grad_norm": 0.3898784816265106,
      "learning_rate": 1.3337232688872009e-05,
      "loss": 0.0126,
      "step": 7740
    },
    {
      "epoch": 3.4156015866020275,
      "grad_norm": 0.26319611072540283,
      "learning_rate": 1.3225005123948364e-05,
      "loss": 0.0192,
      "step": 7750
    },
    {
      "epoch": 3.4200088144557075,
      "grad_norm": 0.3580319285392761,
      "learning_rate": 1.311317972447681e-05,
      "loss": 0.0189,
      "step": 7760
    },
    {
      "epoch": 3.4244160423093875,
      "grad_norm": 0.3561249375343323,
      "learning_rate": 1.3001757713361996e-05,
      "loss": 0.0144,
      "step": 7770
    },
    {
      "epoch": 3.4288232701630674,
      "grad_norm": 0.40637272596359253,
      "learning_rate": 1.2890740309097204e-05,
      "loss": 0.0172,
      "step": 7780
    },
    {
      "epoch": 3.4332304980167474,
      "grad_norm": 0.419416606426239,
      "learning_rate": 1.2780128725750944e-05,
      "loss": 0.0207,
      "step": 7790
    },
    {
      "epoch": 3.4376377258704274,
      "grad_norm": 0.21601006388664246,
      "learning_rate": 1.266992417295379e-05,
      "loss": 0.0141,
      "step": 7800
    },
    {
      "epoch": 3.4420449537241073,
      "grad_norm": 0.21070635318756104,
      "learning_rate": 1.2560127855885073e-05,
      "loss": 0.016,
      "step": 7810
    },
    {
      "epoch": 3.4464521815777878,
      "grad_norm": 0.31750237941741943,
      "learning_rate": 1.2450740975259745e-05,
      "loss": 0.0128,
      "step": 7820
    },
    {
      "epoch": 3.4508594094314677,
      "grad_norm": 0.35502663254737854,
      "learning_rate": 1.234176472731517e-05,
      "loss": 0.013,
      "step": 7830
    },
    {
      "epoch": 3.4552666372851477,
      "grad_norm": 0.30502310395240784,
      "learning_rate": 1.2233200303798158e-05,
      "loss": 0.0155,
      "step": 7840
    },
    {
      "epoch": 3.4596738651388277,
      "grad_norm": 0.16081500053405762,
      "learning_rate": 1.2125048891951846e-05,
      "loss": 0.0159,
      "step": 7850
    },
    {
      "epoch": 3.4640810929925077,
      "grad_norm": 0.31936198472976685,
      "learning_rate": 1.2017311674502745e-05,
      "loss": 0.0171,
      "step": 7860
    },
    {
      "epoch": 3.4684883208461876,
      "grad_norm": 0.35154443979263306,
      "learning_rate": 1.1909989829647822e-05,
      "loss": 0.0208,
      "step": 7870
    },
    {
      "epoch": 3.4728955486998676,
      "grad_norm": 0.19019657373428345,
      "learning_rate": 1.1803084531041553e-05,
      "loss": 0.0154,
      "step": 7880
    },
    {
      "epoch": 3.4773027765535476,
      "grad_norm": 0.3997349143028259,
      "learning_rate": 1.1696596947783162e-05,
      "loss": 0.0222,
      "step": 7890
    },
    {
      "epoch": 3.481710004407228,
      "grad_norm": 0.39112651348114014,
      "learning_rate": 1.1590528244403803e-05,
      "loss": 0.0196,
      "step": 7900
    },
    {
      "epoch": 3.486117232260908,
      "grad_norm": 0.3202126920223236,
      "learning_rate": 1.148487958085382e-05,
      "loss": 0.0148,
      "step": 7910
    },
    {
      "epoch": 3.490524460114588,
      "grad_norm": 0.3292302191257477,
      "learning_rate": 1.1379652112490086e-05,
      "loss": 0.0164,
      "step": 7920
    },
    {
      "epoch": 3.494931687968268,
      "grad_norm": 0.3849090337753296,
      "learning_rate": 1.1274846990063315e-05,
      "loss": 0.0177,
      "step": 7930
    },
    {
      "epoch": 3.499338915821948,
      "grad_norm": 0.38877731561660767,
      "learning_rate": 1.117046535970554e-05,
      "loss": 0.02,
      "step": 7940
    },
    {
      "epoch": 3.503746143675628,
      "grad_norm": 0.19337135553359985,
      "learning_rate": 1.106650836291755e-05,
      "loss": 0.0125,
      "step": 7950
    },
    {
      "epoch": 3.508153371529308,
      "grad_norm": 0.18281033635139465,
      "learning_rate": 1.0962977136556418e-05,
      "loss": 0.012,
      "step": 7960
    },
    {
      "epoch": 3.5125605993829883,
      "grad_norm": 0.3720664381980896,
      "learning_rate": 1.0859872812823024e-05,
      "loss": 0.0199,
      "step": 7970
    },
    {
      "epoch": 3.5169678272366682,
      "grad_norm": 0.2982618808746338,
      "learning_rate": 1.0757196519249747e-05,
      "loss": 0.0144,
      "step": 7980
    },
    {
      "epoch": 3.521375055090348,
      "grad_norm": 0.3510649800300598,
      "learning_rate": 1.0654949378688077e-05,
      "loss": 0.0171,
      "step": 7990
    },
    {
      "epoch": 3.525782282944028,
      "grad_norm": 0.301533043384552,
      "learning_rate": 1.0553132509296376e-05,
      "loss": 0.0159,
      "step": 8000
    },
    {
      "epoch": 3.530189510797708,
      "grad_norm": 0.21104419231414795,
      "learning_rate": 1.0451747024527613e-05,
      "loss": 0.0161,
      "step": 8010
    },
    {
      "epoch": 3.534596738651388,
      "grad_norm": 0.3794727325439453,
      "learning_rate": 1.0350794033117189e-05,
      "loss": 0.014,
      "step": 8020
    },
    {
      "epoch": 3.539003966505068,
      "grad_norm": 0.27347055077552795,
      "learning_rate": 1.0250274639070856e-05,
      "loss": 0.016,
      "step": 8030
    },
    {
      "epoch": 3.5434111943587485,
      "grad_norm": 0.3846406936645508,
      "learning_rate": 1.0150189941652599e-05,
      "loss": 0.0156,
      "step": 8040
    },
    {
      "epoch": 3.5478184222124285,
      "grad_norm": 0.4758712947368622,
      "learning_rate": 1.0050541035372635e-05,
      "loss": 0.016,
      "step": 8050
    },
    {
      "epoch": 3.5522256500661085,
      "grad_norm": 0.42539891600608826,
      "learning_rate": 9.951329009975458e-06,
      "loss": 0.0185,
      "step": 8060
    },
    {
      "epoch": 3.5566328779197884,
      "grad_norm": 0.26795151829719543,
      "learning_rate": 9.852554950427845e-06,
      "loss": 0.0156,
      "step": 8070
    },
    {
      "epoch": 3.5610401057734684,
      "grad_norm": 0.3391098976135254,
      "learning_rate": 9.754219936907105e-06,
      "loss": 0.0225,
      "step": 8080
    },
    {
      "epoch": 3.5654473336271484,
      "grad_norm": 0.20026648044586182,
      "learning_rate": 9.656325044789194e-06,
      "loss": 0.0098,
      "step": 8090
    },
    {
      "epoch": 3.5698545614808284,
      "grad_norm": 0.32867705821990967,
      "learning_rate": 9.55887134463697e-06,
      "loss": 0.0175,
      "step": 8100
    },
    {
      "epoch": 3.5742617893345088,
      "grad_norm": 0.3004150390625,
      "learning_rate": 9.461859902188475e-06,
      "loss": 0.0177,
      "step": 8110
    },
    {
      "epoch": 3.5786690171881888,
      "grad_norm": 0.2241138070821762,
      "learning_rate": 9.365291778345303e-06,
      "loss": 0.0186,
      "step": 8120
    },
    {
      "epoch": 3.5830762450418687,
      "grad_norm": 0.21425291895866394,
      "learning_rate": 9.269168029160991e-06,
      "loss": 0.0148,
      "step": 8130
    },
    {
      "epoch": 3.5874834728955487,
      "grad_norm": 0.29779985547065735,
      "learning_rate": 9.173489705829447e-06,
      "loss": 0.02,
      "step": 8140
    },
    {
      "epoch": 3.5918907007492287,
      "grad_norm": 0.2450767606496811,
      "learning_rate": 9.078257854673516e-06,
      "loss": 0.0129,
      "step": 8150
    },
    {
      "epoch": 3.5962979286029086,
      "grad_norm": 0.256825715303421,
      "learning_rate": 8.983473517133429e-06,
      "loss": 0.0163,
      "step": 8160
    },
    {
      "epoch": 3.6007051564565886,
      "grad_norm": 0.2260303795337677,
      "learning_rate": 8.889137729755537e-06,
      "loss": 0.0174,
      "step": 8170
    },
    {
      "epoch": 3.605112384310269,
      "grad_norm": 0.28210824728012085,
      "learning_rate": 8.79525152418087e-06,
      "loss": 0.0119,
      "step": 8180
    },
    {
      "epoch": 3.609519612163949,
      "grad_norm": 0.4913744330406189,
      "learning_rate": 8.701815927133961e-06,
      "loss": 0.0177,
      "step": 8190
    },
    {
      "epoch": 3.613926840017629,
      "grad_norm": 0.3384150266647339,
      "learning_rate": 8.608831960411534e-06,
      "loss": 0.016,
      "step": 8200
    },
    {
      "epoch": 3.618334067871309,
      "grad_norm": 0.2618265151977539,
      "learning_rate": 8.516300640871321e-06,
      "loss": 0.0165,
      "step": 8210
    },
    {
      "epoch": 3.622741295724989,
      "grad_norm": 0.3150423765182495,
      "learning_rate": 8.424222980421038e-06,
      "loss": 0.0172,
      "step": 8220
    },
    {
      "epoch": 3.627148523578669,
      "grad_norm": 0.25074687600135803,
      "learning_rate": 8.332599986007184e-06,
      "loss": 0.0135,
      "step": 8230
    },
    {
      "epoch": 3.631555751432349,
      "grad_norm": 0.37179097533226013,
      "learning_rate": 8.241432659604203e-06,
      "loss": 0.0141,
      "step": 8240
    },
    {
      "epoch": 3.6359629792860293,
      "grad_norm": 0.339374840259552,
      "learning_rate": 8.150721998203331e-06,
      "loss": 0.0171,
      "step": 8250
    },
    {
      "epoch": 3.6403702071397093,
      "grad_norm": 0.18851503729820251,
      "learning_rate": 8.06046899380184e-06,
      "loss": 0.016,
      "step": 8260
    },
    {
      "epoch": 3.6447774349933892,
      "grad_norm": 0.22765935957431793,
      "learning_rate": 7.970674633392133e-06,
      "loss": 0.0155,
      "step": 8270
    },
    {
      "epoch": 3.649184662847069,
      "grad_norm": 0.346097856760025,
      "learning_rate": 7.881339898950924e-06,
      "loss": 0.0187,
      "step": 8280
    },
    {
      "epoch": 3.653591890700749,
      "grad_norm": 0.30000656843185425,
      "learning_rate": 7.792465767428597e-06,
      "loss": 0.02,
      "step": 8290
    },
    {
      "epoch": 3.657999118554429,
      "grad_norm": 0.19509530067443848,
      "learning_rate": 7.704053210738376e-06,
      "loss": 0.0094,
      "step": 8300
    },
    {
      "epoch": 3.662406346408109,
      "grad_norm": 0.38031843304634094,
      "learning_rate": 7.6161031957458494e-06,
      "loss": 0.0111,
      "step": 8310
    },
    {
      "epoch": 3.6668135742617896,
      "grad_norm": 0.26720842719078064,
      "learning_rate": 7.5286166842582605e-06,
      "loss": 0.0136,
      "step": 8320
    },
    {
      "epoch": 3.6712208021154695,
      "grad_norm": 0.36126887798309326,
      "learning_rate": 7.4415946330140814e-06,
      "loss": 0.0164,
      "step": 8330
    },
    {
      "epoch": 3.6756280299691495,
      "grad_norm": 0.21379412710666656,
      "learning_rate": 7.3550379936725644e-06,
      "loss": 0.0161,
      "step": 8340
    },
    {
      "epoch": 3.6800352578228295,
      "grad_norm": 0.2195032835006714,
      "learning_rate": 7.2689477128032035e-06,
      "loss": 0.018,
      "step": 8350
    },
    {
      "epoch": 3.6844424856765094,
      "grad_norm": 0.4028056263923645,
      "learning_rate": 7.183324731875551e-06,
      "loss": 0.018,
      "step": 8360
    },
    {
      "epoch": 3.6888497135301894,
      "grad_norm": 0.11799513548612595,
      "learning_rate": 7.098169987248782e-06,
      "loss": 0.0197,
      "step": 8370
    },
    {
      "epoch": 3.6932569413838694,
      "grad_norm": 0.2908453047275543,
      "learning_rate": 7.013484410161553e-06,
      "loss": 0.017,
      "step": 8380
    },
    {
      "epoch": 3.69766416923755,
      "grad_norm": 0.37175682187080383,
      "learning_rate": 6.92926892672176e-06,
      "loss": 0.0147,
      "step": 8390
    },
    {
      "epoch": 3.7020713970912293,
      "grad_norm": 0.3084324896335602,
      "learning_rate": 6.845524457896446e-06,
      "loss": 0.0145,
      "step": 8400
    },
    {
      "epoch": 3.7064786249449098,
      "grad_norm": 0.34085729718208313,
      "learning_rate": 6.7622519195017165e-06,
      "loss": 0.0144,
      "step": 8410
    },
    {
      "epoch": 3.7108858527985897,
      "grad_norm": 0.3780392110347748,
      "learning_rate": 6.679452222192684e-06,
      "loss": 0.0131,
      "step": 8420
    },
    {
      "epoch": 3.7152930806522697,
      "grad_norm": 0.4270978271961212,
      "learning_rate": 6.597126271453579e-06,
      "loss": 0.0145,
      "step": 8430
    },
    {
      "epoch": 3.7197003085059497,
      "grad_norm": 0.21092770993709564,
      "learning_rate": 6.51527496758782e-06,
      "loss": 0.0112,
      "step": 8440
    },
    {
      "epoch": 3.7241075363596297,
      "grad_norm": 0.2579943537712097,
      "learning_rate": 6.433899205708155e-06,
      "loss": 0.0154,
      "step": 8450
    },
    {
      "epoch": 3.72851476421331,
      "grad_norm": 0.3359133005142212,
      "learning_rate": 6.352999875726856e-06,
      "loss": 0.0131,
      "step": 8460
    },
    {
      "epoch": 3.7329219920669896,
      "grad_norm": 0.35096701979637146,
      "learning_rate": 6.272577862346052e-06,
      "loss": 0.0153,
      "step": 8470
    },
    {
      "epoch": 3.73732921992067,
      "grad_norm": 0.40706783533096313,
      "learning_rate": 6.192634045047996e-06,
      "loss": 0.0203,
      "step": 8480
    },
    {
      "epoch": 3.74173644777435,
      "grad_norm": 0.266539067029953,
      "learning_rate": 6.113169298085458e-06,
      "loss": 0.0172,
      "step": 8490
    },
    {
      "epoch": 3.74614367562803,
      "grad_norm": 0.20184855163097382,
      "learning_rate": 6.034184490472195e-06,
      "loss": 0.0131,
      "step": 8500
    },
    {
      "epoch": 3.75055090348171,
      "grad_norm": 0.23937053978443146,
      "learning_rate": 5.955680485973386e-06,
      "loss": 0.0242,
      "step": 8510
    },
    {
      "epoch": 3.75495813133539,
      "grad_norm": 0.23759597539901733,
      "learning_rate": 5.877658143096265e-06,
      "loss": 0.0137,
      "step": 8520
    },
    {
      "epoch": 3.7593653591890703,
      "grad_norm": 0.231092169880867,
      "learning_rate": 5.800118315080661e-06,
      "loss": 0.0113,
      "step": 8530
    },
    {
      "epoch": 3.76377258704275,
      "grad_norm": 0.25397223234176636,
      "learning_rate": 5.723061849889716e-06,
      "loss": 0.0147,
      "step": 8540
    },
    {
      "epoch": 3.7681798148964303,
      "grad_norm": 0.4719129502773285,
      "learning_rate": 5.646489590200604e-06,
      "loss": 0.015,
      "step": 8550
    },
    {
      "epoch": 3.7725870427501103,
      "grad_norm": 0.45554330945014954,
      "learning_rate": 5.570402373395256e-06,
      "loss": 0.0179,
      "step": 8560
    },
    {
      "epoch": 3.7769942706037902,
      "grad_norm": 0.23821285367012024,
      "learning_rate": 5.494801031551305e-06,
      "loss": 0.0127,
      "step": 8570
    },
    {
      "epoch": 3.78140149845747,
      "grad_norm": 0.28344210982322693,
      "learning_rate": 5.41968639143291e-06,
      "loss": 0.0159,
      "step": 8580
    },
    {
      "epoch": 3.78580872631115,
      "grad_norm": 0.28846192359924316,
      "learning_rate": 5.345059274481751e-06,
      "loss": 0.016,
      "step": 8590
    },
    {
      "epoch": 3.7902159541648306,
      "grad_norm": 0.1981518566608429,
      "learning_rate": 5.270920496808002e-06,
      "loss": 0.0164,
      "step": 8600
    },
    {
      "epoch": 3.79462318201851,
      "grad_norm": 0.1670496165752411,
      "learning_rate": 5.1972708691814695e-06,
      "loss": 0.0131,
      "step": 8610
    },
    {
      "epoch": 3.7990304098721905,
      "grad_norm": 0.20297828316688538,
      "learning_rate": 5.124111197022674e-06,
      "loss": 0.0145,
      "step": 8620
    },
    {
      "epoch": 3.8034376377258705,
      "grad_norm": 0.22566120326519012,
      "learning_rate": 5.051442280394081e-06,
      "loss": 0.0189,
      "step": 8630
    },
    {
      "epoch": 3.8078448655795505,
      "grad_norm": 0.3911871016025543,
      "learning_rate": 4.979264913991322e-06,
      "loss": 0.0101,
      "step": 8640
    },
    {
      "epoch": 3.8122520934332305,
      "grad_norm": 0.29937979578971863,
      "learning_rate": 4.907579887134489e-06,
      "loss": 0.0148,
      "step": 8650
    },
    {
      "epoch": 3.8166593212869104,
      "grad_norm": 0.18907302618026733,
      "learning_rate": 4.836387983759572e-06,
      "loss": 0.0109,
      "step": 8660
    },
    {
      "epoch": 3.8210665491405904,
      "grad_norm": 0.31525811553001404,
      "learning_rate": 4.765689982409816e-06,
      "loss": 0.0128,
      "step": 8670
    },
    {
      "epoch": 3.8254737769942704,
      "grad_norm": 0.4554682672023773,
      "learning_rate": 4.695486656227233e-06,
      "loss": 0.014,
      "step": 8680
    },
    {
      "epoch": 3.829881004847951,
      "grad_norm": 0.319570392370224,
      "learning_rate": 4.625778772944156e-06,
      "loss": 0.0172,
      "step": 8690
    },
    {
      "epoch": 3.8342882327016308,
      "grad_norm": 0.34560003876686096,
      "learning_rate": 4.556567094874825e-06,
      "loss": 0.0126,
      "step": 8700
    },
    {
      "epoch": 3.8386954605553107,
      "grad_norm": 0.4323245584964752,
      "learning_rate": 4.487852378907059e-06,
      "loss": 0.0139,
      "step": 8710
    },
    {
      "epoch": 3.8431026884089907,
      "grad_norm": 0.19029578566551208,
      "learning_rate": 4.419635376493986e-06,
      "loss": 0.0109,
      "step": 8720
    },
    {
      "epoch": 3.8475099162626707,
      "grad_norm": 0.25458836555480957,
      "learning_rate": 4.351916833645825e-06,
      "loss": 0.0174,
      "step": 8730
    },
    {
      "epoch": 3.8519171441163507,
      "grad_norm": 0.3662632405757904,
      "learning_rate": 4.284697490921691e-06,
      "loss": 0.0121,
      "step": 8740
    },
    {
      "epoch": 3.8563243719700306,
      "grad_norm": 0.27959346771240234,
      "learning_rate": 4.2179780834215585e-06,
      "loss": 0.0134,
      "step": 8750
    },
    {
      "epoch": 3.860731599823711,
      "grad_norm": 0.40293431282043457,
      "learning_rate": 4.151759340778178e-06,
      "loss": 0.0164,
      "step": 8760
    },
    {
      "epoch": 3.865138827677391,
      "grad_norm": 0.2881563603878021,
      "learning_rate": 4.086041987149109e-06,
      "loss": 0.0197,
      "step": 8770
    },
    {
      "epoch": 3.869546055531071,
      "grad_norm": 0.2514883279800415,
      "learning_rate": 4.020826741208811e-06,
      "loss": 0.014,
      "step": 8780
    },
    {
      "epoch": 3.873953283384751,
      "grad_norm": 0.2502042055130005,
      "learning_rate": 3.956114316140746e-06,
      "loss": 0.021,
      "step": 8790
    },
    {
      "epoch": 3.878360511238431,
      "grad_norm": 0.3575713336467743,
      "learning_rate": 3.891905419629643e-06,
      "loss": 0.0183,
      "step": 8800
    },
    {
      "epoch": 3.882767739092111,
      "grad_norm": 0.3218100666999817,
      "learning_rate": 3.8282007538536946e-06,
      "loss": 0.0232,
      "step": 8810
    },
    {
      "epoch": 3.887174966945791,
      "grad_norm": 0.4357834458351135,
      "learning_rate": 3.7650010154769265e-06,
      "loss": 0.0139,
      "step": 8820
    },
    {
      "epoch": 3.8915821947994713,
      "grad_norm": 0.2929894030094147,
      "learning_rate": 3.7023068956415608e-06,
      "loss": 0.0127,
      "step": 8830
    },
    {
      "epoch": 3.8959894226531513,
      "grad_norm": 0.28175050020217896,
      "learning_rate": 3.6401190799604303e-06,
      "loss": 0.0179,
      "step": 8840
    },
    {
      "epoch": 3.9003966505068313,
      "grad_norm": 0.31297770142555237,
      "learning_rate": 3.578438248509536e-06,
      "loss": 0.0151,
      "step": 8850
    },
    {
      "epoch": 3.9048038783605112,
      "grad_norm": 0.33773985505104065,
      "learning_rate": 3.5172650758205583e-06,
      "loss": 0.0145,
      "step": 8860
    },
    {
      "epoch": 3.909211106214191,
      "grad_norm": 0.36398860812187195,
      "learning_rate": 3.45660023087353e-06,
      "loss": 0.0159,
      "step": 8870
    },
    {
      "epoch": 3.913618334067871,
      "grad_norm": 0.2729854881763458,
      "learning_rate": 3.3964443770894528e-06,
      "loss": 0.0191,
      "step": 8880
    },
    {
      "epoch": 3.918025561921551,
      "grad_norm": 0.32151031494140625,
      "learning_rate": 3.3367981723231245e-06,
      "loss": 0.0157,
      "step": 8890
    },
    {
      "epoch": 3.9224327897752316,
      "grad_norm": 0.2357509881258011,
      "learning_rate": 3.2776622688558746e-06,
      "loss": 0.0122,
      "step": 8900
    },
    {
      "epoch": 3.9268400176289115,
      "grad_norm": 0.19904451072216034,
      "learning_rate": 3.2190373133884677e-06,
      "loss": 0.0185,
      "step": 8910
    },
    {
      "epoch": 3.9312472454825915,
      "grad_norm": 0.44772738218307495,
      "learning_rate": 3.1609239470340446e-06,
      "loss": 0.018,
      "step": 8920
    },
    {
      "epoch": 3.9356544733362715,
      "grad_norm": 0.25785723328590393,
      "learning_rate": 3.1033228053110373e-06,
      "loss": 0.0134,
      "step": 8930
    },
    {
      "epoch": 3.9400617011899515,
      "grad_norm": 0.24837853014469147,
      "learning_rate": 3.0462345181363314e-06,
      "loss": 0.014,
      "step": 8940
    },
    {
      "epoch": 3.9444689290436314,
      "grad_norm": 0.3545350730419159,
      "learning_rate": 2.9896597098182654e-06,
      "loss": 0.0189,
      "step": 8950
    },
    {
      "epoch": 3.9488761568973114,
      "grad_norm": 0.4729788601398468,
      "learning_rate": 2.933598999049891e-06,
      "loss": 0.018,
      "step": 8960
    },
    {
      "epoch": 3.953283384750992,
      "grad_norm": 0.248626708984375,
      "learning_rate": 2.8780529989021697e-06,
      "loss": 0.0127,
      "step": 8970
    },
    {
      "epoch": 3.957690612604672,
      "grad_norm": 0.2875106930732727,
      "learning_rate": 2.823022316817242e-06,
      "loss": 0.0168,
      "step": 8980
    },
    {
      "epoch": 3.962097840458352,
      "grad_norm": 0.3247435688972473,
      "learning_rate": 2.7685075546018456e-06,
      "loss": 0.0182,
      "step": 8990
    },
    {
      "epoch": 3.9665050683120318,
      "grad_norm": 0.31818291544914246,
      "learning_rate": 2.7145093084206598e-06,
      "loss": 0.0163,
      "step": 9000
    },
    {
      "epoch": 3.9709122961657117,
      "grad_norm": 0.3724207878112793,
      "learning_rate": 2.661028168789892e-06,
      "loss": 0.017,
      "step": 9010
    },
    {
      "epoch": 3.9753195240193917,
      "grad_norm": 0.24472211301326752,
      "learning_rate": 2.6080647205706855e-06,
      "loss": 0.0194,
      "step": 9020
    },
    {
      "epoch": 3.9797267518730717,
      "grad_norm": 0.29755479097366333,
      "learning_rate": 2.555619542962834e-06,
      "loss": 0.0142,
      "step": 9030
    },
    {
      "epoch": 3.984133979726752,
      "grad_norm": 0.23328396677970886,
      "learning_rate": 2.503693209498409e-06,
      "loss": 0.0152,
      "step": 9040
    },
    {
      "epoch": 3.988541207580432,
      "grad_norm": 0.45239877700805664,
      "learning_rate": 2.452286288035449e-06,
      "loss": 0.0155,
      "step": 9050
    },
    {
      "epoch": 3.992948435434112,
      "grad_norm": 0.43790799379348755,
      "learning_rate": 2.4013993407518363e-06,
      "loss": 0.0154,
      "step": 9060
    },
    {
      "epoch": 3.997355663287792,
      "grad_norm": 0.22011210024356842,
      "learning_rate": 2.351032924139063e-06,
      "loss": 0.0145,
      "step": 9070
    },
    {
      "epoch": 4.001762891141472,
      "grad_norm": 0.5300778746604919,
      "learning_rate": 2.30118758899619e-06,
      "loss": 0.0151,
      "step": 9080
    },
    {
      "epoch": 4.006170118995152,
      "grad_norm": 0.14596258103847504,
      "learning_rate": 2.2518638804238157e-06,
      "loss": 0.0142,
      "step": 9090
    },
    {
      "epoch": 4.010577346848832,
      "grad_norm": 0.20150664448738098,
      "learning_rate": 2.203062337818118e-06,
      "loss": 0.0136,
      "step": 9100
    },
    {
      "epoch": 4.014984574702512,
      "grad_norm": 0.15427398681640625,
      "learning_rate": 2.1547834948649483e-06,
      "loss": 0.0111,
      "step": 9110
    },
    {
      "epoch": 4.019391802556192,
      "grad_norm": 0.2912757694721222,
      "learning_rate": 2.1070278795340017e-06,
      "loss": 0.0163,
      "step": 9120
    },
    {
      "epoch": 4.023799030409872,
      "grad_norm": 0.18154922127723694,
      "learning_rate": 2.059796014073029e-06,
      "loss": 0.0142,
      "step": 9130
    },
    {
      "epoch": 4.028206258263552,
      "grad_norm": 0.2023886889219284,
      "learning_rate": 2.01308841500214e-06,
      "loss": 0.0147,
      "step": 9140
    },
    {
      "epoch": 4.032613486117232,
      "grad_norm": 0.329911470413208,
      "learning_rate": 1.9669055931081704e-06,
      "loss": 0.0101,
      "step": 9150
    },
    {
      "epoch": 4.037020713970913,
      "grad_norm": 0.26592153310775757,
      "learning_rate": 1.9212480534390507e-06,
      "loss": 0.0114,
      "step": 9160
    },
    {
      "epoch": 4.041427941824592,
      "grad_norm": 0.3038083016872406,
      "learning_rate": 1.8761162952983246e-06,
      "loss": 0.0169,
      "step": 9170
    },
    {
      "epoch": 4.045835169678273,
      "grad_norm": 0.31616950035095215,
      "learning_rate": 1.8315108122396618e-06,
      "loss": 0.0172,
      "step": 9180
    },
    {
      "epoch": 4.050242397531952,
      "grad_norm": 0.23818564414978027,
      "learning_rate": 1.787432092061475e-06,
      "loss": 0.016,
      "step": 9190
    },
    {
      "epoch": 4.054649625385633,
      "grad_norm": 0.2851960062980652,
      "learning_rate": 1.743880616801602e-06,
      "loss": 0.0106,
      "step": 9200
    },
    {
      "epoch": 4.059056853239312,
      "grad_norm": 0.4987255036830902,
      "learning_rate": 1.7008568627319865e-06,
      "loss": 0.0189,
      "step": 9210
    },
    {
      "epoch": 4.0634640810929925,
      "grad_norm": 0.4334008991718292,
      "learning_rate": 1.6583613003535226e-06,
      "loss": 0.0158,
      "step": 9220
    },
    {
      "epoch": 4.067871308946673,
      "grad_norm": 0.2709560990333557,
      "learning_rate": 1.6163943943908522e-06,
      "loss": 0.0141,
      "step": 9230
    },
    {
      "epoch": 4.0722785368003525,
      "grad_norm": 0.3422434628009796,
      "learning_rate": 1.5749566037873476e-06,
      "loss": 0.0185,
      "step": 9240
    },
    {
      "epoch": 4.076685764654033,
      "grad_norm": 0.3536073863506317,
      "learning_rate": 1.5340483817000428e-06,
      "loss": 0.0161,
      "step": 9250
    },
    {
      "epoch": 4.081092992507712,
      "grad_norm": 0.5355464220046997,
      "learning_rate": 1.4936701754947101e-06,
      "loss": 0.0134,
      "step": 9260
    },
    {
      "epoch": 4.085500220361393,
      "grad_norm": 0.17497968673706055,
      "learning_rate": 1.4538224267409361e-06,
      "loss": 0.0151,
      "step": 9270
    },
    {
      "epoch": 4.089907448215072,
      "grad_norm": 0.36362752318382263,
      "learning_rate": 1.414505571207314e-06,
      "loss": 0.0133,
      "step": 9280
    },
    {
      "epoch": 4.094314676068753,
      "grad_norm": 0.1893729567527771,
      "learning_rate": 1.3757200388566816e-06,
      "loss": 0.0178,
      "step": 9290
    },
    {
      "epoch": 4.098721903922433,
      "grad_norm": 0.4485119879245758,
      "learning_rate": 1.3374662538414074e-06,
      "loss": 0.0152,
      "step": 9300
    },
    {
      "epoch": 4.103129131776113,
      "grad_norm": 0.2299726903438568,
      "learning_rate": 1.2997446344987617e-06,
      "loss": 0.0151,
      "step": 9310
    },
    {
      "epoch": 4.107536359629793,
      "grad_norm": 0.30749672651290894,
      "learning_rate": 1.262555593346315e-06,
      "loss": 0.0288,
      "step": 9320
    },
    {
      "epoch": 4.111943587483473,
      "grad_norm": 0.3408014476299286,
      "learning_rate": 1.2258995370774685e-06,
      "loss": 0.0142,
      "step": 9330
    },
    {
      "epoch": 4.116350815337153,
      "grad_norm": 0.2513563632965088,
      "learning_rate": 1.1897768665569798e-06,
      "loss": 0.0157,
      "step": 9340
    },
    {
      "epoch": 4.120758043190833,
      "grad_norm": 0.1390959918498993,
      "learning_rate": 1.1541879768165954e-06,
      "loss": 0.0141,
      "step": 9350
    },
    {
      "epoch": 4.125165271044513,
      "grad_norm": 0.24601957201957703,
      "learning_rate": 1.1191332570507085e-06,
      "loss": 0.0091,
      "step": 9360
    },
    {
      "epoch": 4.1295724988981934,
      "grad_norm": 0.37747249007225037,
      "learning_rate": 1.0846130906121132e-06,
      "loss": 0.018,
      "step": 9370
    },
    {
      "epoch": 4.133979726751873,
      "grad_norm": 0.4120543301105499,
      "learning_rate": 1.0506278550078131e-06,
      "loss": 0.0095,
      "step": 9380
    },
    {
      "epoch": 4.138386954605553,
      "grad_norm": 0.23796918988227844,
      "learning_rate": 1.0171779218949185e-06,
      "loss": 0.0185,
      "step": 9390
    },
    {
      "epoch": 4.142794182459233,
      "grad_norm": 0.15384431183338165,
      "learning_rate": 9.842636570765174e-07,
      "loss": 0.0163,
      "step": 9400
    },
    {
      "epoch": 4.147201410312913,
      "grad_norm": 0.27213093638420105,
      "learning_rate": 9.518854204977612e-07,
      "loss": 0.0119,
      "step": 9410
    },
    {
      "epoch": 4.151608638166593,
      "grad_norm": 0.3461364805698395,
      "learning_rate": 9.200435662418349e-07,
      "loss": 0.0141,
      "step": 9420
    },
    {
      "epoch": 4.156015866020273,
      "grad_norm": 0.31474748253822327,
      "learning_rate": 8.887384425261658e-07,
      "loss": 0.0146,
      "step": 9430
    },
    {
      "epoch": 4.160423093873954,
      "grad_norm": 0.38170894980430603,
      "learning_rate": 8.579703916985648e-07,
      "loss": 0.0176,
      "step": 9440
    },
    {
      "epoch": 4.164830321727633,
      "grad_norm": 0.5210051536560059,
      "learning_rate": 8.277397502335194e-07,
      "loss": 0.0153,
      "step": 9450
    },
    {
      "epoch": 4.169237549581314,
      "grad_norm": 0.23647953569889069,
      "learning_rate": 7.980468487284675e-07,
      "loss": 0.016,
      "step": 9460
    },
    {
      "epoch": 4.173644777434993,
      "grad_norm": 0.23964443802833557,
      "learning_rate": 7.688920119002297e-07,
      "loss": 0.0128,
      "step": 9470
    },
    {
      "epoch": 4.178052005288674,
      "grad_norm": 0.17334182560443878,
      "learning_rate": 7.402755585814269e-07,
      "loss": 0.0091,
      "step": 9480
    },
    {
      "epoch": 4.182459233142353,
      "grad_norm": 0.3207050561904907,
      "learning_rate": 7.121978017170073e-07,
      "loss": 0.0166,
      "step": 9490
    },
    {
      "epoch": 4.1868664609960335,
      "grad_norm": 0.3219553828239441,
      "learning_rate": 6.846590483608306e-07,
      "loss": 0.011,
      "step": 9500
    },
    {
      "epoch": 4.191273688849714,
      "grad_norm": 0.21981462836265564,
      "learning_rate": 6.576595996722834e-07,
      "loss": 0.0189,
      "step": 9510
    },
    {
      "epoch": 4.1956809167033935,
      "grad_norm": 0.3288012444972992,
      "learning_rate": 6.311997509130141e-07,
      "loss": 0.0137,
      "step": 9520
    },
    {
      "epoch": 4.200088144557074,
      "grad_norm": 0.29040390253067017,
      "learning_rate": 6.052797914436803e-07,
      "loss": 0.0161,
      "step": 9530
    },
    {
      "epoch": 4.204495372410753,
      "grad_norm": 0.19576290249824524,
      "learning_rate": 5.799000047208181e-07,
      "loss": 0.0137,
      "step": 9540
    },
    {
      "epoch": 4.208902600264434,
      "grad_norm": 0.38006970286369324,
      "learning_rate": 5.550606682937054e-07,
      "loss": 0.0184,
      "step": 9550
    },
    {
      "epoch": 4.213309828118113,
      "grad_norm": 0.21636731922626495,
      "learning_rate": 5.307620538013481e-07,
      "loss": 0.0114,
      "step": 9560
    },
    {
      "epoch": 4.217717055971794,
      "grad_norm": 0.5447456240653992,
      "learning_rate": 5.070044269694874e-07,
      "loss": 0.0142,
      "step": 9570
    },
    {
      "epoch": 4.222124283825474,
      "grad_norm": 0.40838754177093506,
      "learning_rate": 4.837880476077417e-07,
      "loss": 0.0086,
      "step": 9580
    },
    {
      "epoch": 4.226531511679154,
      "grad_norm": 0.3780432939529419,
      "learning_rate": 4.6111316960670835e-07,
      "loss": 0.0154,
      "step": 9590
    },
    {
      "epoch": 4.230938739532834,
      "grad_norm": 0.3406049311161041,
      "learning_rate": 4.389800409352218e-07,
      "loss": 0.0183,
      "step": 9600
    },
    {
      "epoch": 4.235345967386514,
      "grad_norm": 0.13914929330348969,
      "learning_rate": 4.173889036376277e-07,
      "loss": 0.0118,
      "step": 9610
    },
    {
      "epoch": 4.239753195240194,
      "grad_norm": 0.18764470517635345,
      "learning_rate": 3.963399938311463e-07,
      "loss": 0.0154,
      "step": 9620
    },
    {
      "epoch": 4.244160423093874,
      "grad_norm": 0.2316211611032486,
      "learning_rate": 3.7583354170328545e-07,
      "loss": 0.0129,
      "step": 9630
    },
    {
      "epoch": 4.248567650947554,
      "grad_norm": 0.31830236315727234,
      "learning_rate": 3.558697715093207e-07,
      "loss": 0.0121,
      "step": 9640
    },
    {
      "epoch": 4.2529748788012345,
      "grad_norm": 0.1780199110507965,
      "learning_rate": 3.3644890156983576e-07,
      "loss": 0.0159,
      "step": 9650
    },
    {
      "epoch": 4.257382106654914,
      "grad_norm": 0.5262408256530762,
      "learning_rate": 3.175711442683638e-07,
      "loss": 0.0161,
      "step": 9660
    },
    {
      "epoch": 4.261789334508594,
      "grad_norm": 0.30066514015197754,
      "learning_rate": 2.9923670604902197e-07,
      "loss": 0.0207,
      "step": 9670
    },
    {
      "epoch": 4.266196562362274,
      "grad_norm": 0.18580789864063263,
      "learning_rate": 2.814457874143028e-07,
      "loss": 0.025,
      "step": 9680
    },
    {
      "epoch": 4.270603790215954,
      "grad_norm": 0.21357838809490204,
      "learning_rate": 2.641985829228366e-07,
      "loss": 0.0122,
      "step": 9690
    },
    {
      "epoch": 4.275011018069634,
      "grad_norm": 0.19489970803260803,
      "learning_rate": 2.474952811872877e-07,
      "loss": 0.0189,
      "step": 9700
    },
    {
      "epoch": 4.279418245923314,
      "grad_norm": 0.3230896592140198,
      "learning_rate": 2.3133606487228397e-07,
      "loss": 0.0139,
      "step": 9710
    },
    {
      "epoch": 4.283825473776995,
      "grad_norm": 0.15887413918972015,
      "learning_rate": 2.157211106924295e-07,
      "loss": 0.016,
      "step": 9720
    },
    {
      "epoch": 4.288232701630674,
      "grad_norm": 0.2264292687177658,
      "learning_rate": 2.006505894103672e-07,
      "loss": 0.0154,
      "step": 9730
    },
    {
      "epoch": 4.292639929484355,
      "grad_norm": 0.30880141258239746,
      "learning_rate": 1.8612466583489696e-07,
      "loss": 0.014,
      "step": 9740
    },
    {
      "epoch": 4.297047157338034,
      "grad_norm": 0.21884560585021973,
      "learning_rate": 1.7214349881918834e-07,
      "loss": 0.0145,
      "step": 9750
    },
    {
      "epoch": 4.301454385191715,
      "grad_norm": 0.18245521187782288,
      "learning_rate": 1.5870724125904845e-07,
      "loss": 0.0183,
      "step": 9760
    },
    {
      "epoch": 4.305861613045394,
      "grad_norm": 0.3281463384628296,
      "learning_rate": 1.4581604009124006e-07,
      "loss": 0.0136,
      "step": 9770
    },
    {
      "epoch": 4.310268840899075,
      "grad_norm": 0.4411262273788452,
      "learning_rate": 1.334700362918717e-07,
      "loss": 0.0157,
      "step": 9780
    },
    {
      "epoch": 4.314676068752754,
      "grad_norm": 0.24080494046211243,
      "learning_rate": 1.2166936487486015e-07,
      "loss": 0.018,
      "step": 9790
    },
    {
      "epoch": 4.3190832966064345,
      "grad_norm": 0.3871815502643585,
      "learning_rate": 1.1041415489045914e-07,
      "loss": 0.0135,
      "step": 9800
    },
    {
      "epoch": 4.323490524460115,
      "grad_norm": 0.3113195300102234,
      "learning_rate": 9.970452942384412e-08,
      "loss": 0.014,
      "step": 9810
    },
    {
      "epoch": 4.3278977523137945,
      "grad_norm": 0.1922612339258194,
      "learning_rate": 8.954060559375754e-08,
      "loss": 0.0149,
      "step": 9820
    },
    {
      "epoch": 4.332304980167475,
      "grad_norm": 0.39544400572776794,
      "learning_rate": 7.99224945512489e-08,
      "loss": 0.0139,
      "step": 9830
    },
    {
      "epoch": 4.336712208021154,
      "grad_norm": 0.20694600045681,
      "learning_rate": 7.085030147843675e-08,
      "loss": 0.0116,
      "step": 9840
    },
    {
      "epoch": 4.341119435874835,
      "grad_norm": 0.3235357999801636,
      "learning_rate": 6.232412558736523e-08,
      "loss": 0.0178,
      "step": 9850
    },
    {
      "epoch": 4.345526663728514,
      "grad_norm": 0.15688493847846985,
      "learning_rate": 5.434406011893822e-08,
      "loss": 0.0145,
      "step": 9860
    },
    {
      "epoch": 4.349933891582195,
      "grad_norm": 0.205477774143219,
      "learning_rate": 4.6910192341864664e-08,
      "loss": 0.0134,
      "step": 9870
    },
    {
      "epoch": 4.354341119435875,
      "grad_norm": 0.3022678792476654,
      "learning_rate": 4.0022603551737035e-08,
      "loss": 0.012,
      "step": 9880
    },
    {
      "epoch": 4.358748347289555,
      "grad_norm": 0.31290557980537415,
      "learning_rate": 3.3681369070120985e-08,
      "loss": 0.0179,
      "step": 9890
    },
    {
      "epoch": 4.363155575143235,
      "grad_norm": 0.44330283999443054,
      "learning_rate": 2.7886558243744866e-08,
      "loss": 0.0151,
      "step": 9900
    },
    {
      "epoch": 4.367562802996915,
      "grad_norm": 0.34386780858039856,
      "learning_rate": 2.2638234443722596e-08,
      "loss": 0.0166,
      "step": 9910
    },
    {
      "epoch": 4.371970030850595,
      "grad_norm": 0.27240562438964844,
      "learning_rate": 1.7936455064887504e-08,
      "loss": 0.0096,
      "step": 9920
    },
    {
      "epoch": 4.376377258704275,
      "grad_norm": 0.182706817984581,
      "learning_rate": 1.378127152514841e-08,
      "loss": 0.0167,
      "step": 9930
    },
    {
      "epoch": 4.380784486557955,
      "grad_norm": 0.24107477068901062,
      "learning_rate": 1.0172729264917857e-08,
      "loss": 0.0203,
      "step": 9940
    },
    {
      "epoch": 4.3851917144116355,
      "grad_norm": 0.113052137196064,
      "learning_rate": 7.1108677466458215e-09,
      "loss": 0.0157,
      "step": 9950
    },
    {
      "epoch": 4.389598942265315,
      "grad_norm": 0.33672088384628296,
      "learning_rate": 4.595720454353414e-09,
      "loss": 0.0204,
      "step": 9960
    },
    {
      "epoch": 4.394006170118995,
      "grad_norm": 0.2914753258228302,
      "learning_rate": 2.627314893294264e-09,
      "loss": 0.014,
      "step": 9970
    },
    {
      "epoch": 4.398413397972675,
      "grad_norm": 0.2843835651874542,
      "learning_rate": 1.2056725896270048e-09,
      "loss": 0.015,
      "step": 9980
    },
    {
      "epoch": 4.402820625826355,
      "grad_norm": 0.13735216856002808,
      "learning_rate": 3.308090902098826e-10,
      "loss": 0.0139,
      "step": 9990
    },
    {
      "epoch": 4.407227853680035,
      "grad_norm": 0.2054501622915268,
      "learning_rate": 2.7339624120159555e-12,
      "loss": 0.0163,
      "step": 10000
    },
    {
      "epoch": 4.407227853680035,
      "step": 10000,
      "total_flos": 0.0,
      "train_loss": 0.0015150180652737618,
      "train_runtime": 454.06,
      "train_samples_per_second": 352.376,
      "train_steps_per_second": 22.024
    }
  ],
  "logging_steps": 10,
  "max_steps": 10000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 1000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}