{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 780,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0038554216867469878,
      "grad_norm": 1.2289724349975586,
      "learning_rate": 0.0,
      "loss": 0.0363,
      "step": 1
    },
    {
      "epoch": 0.0077108433734939755,
      "grad_norm": 1.148133635520935,
      "learning_rate": 5.128205128205128e-07,
      "loss": 0.042,
      "step": 2
    },
    {
      "epoch": 0.011566265060240964,
      "grad_norm": 1.251704216003418,
      "learning_rate": 1.0256410256410257e-06,
      "loss": 0.0442,
      "step": 3
    },
    {
      "epoch": 0.015421686746987951,
      "grad_norm": 1.1709717512130737,
      "learning_rate": 1.5384615384615387e-06,
      "loss": 0.0363,
      "step": 4
    },
    {
      "epoch": 0.01927710843373494,
      "grad_norm": 1.1885852813720703,
      "learning_rate": 2.0512820512820513e-06,
      "loss": 0.0281,
      "step": 5
    },
    {
      "epoch": 0.02313253012048193,
      "grad_norm": 0.7513725161552429,
      "learning_rate": 2.564102564102564e-06,
      "loss": 0.0456,
      "step": 6
    },
    {
      "epoch": 0.026987951807228915,
      "grad_norm": 0.6604480147361755,
      "learning_rate": 3.0769230769230774e-06,
      "loss": 0.0336,
      "step": 7
    },
    {
      "epoch": 0.030843373493975902,
      "grad_norm": 0.5245765447616577,
      "learning_rate": 3.58974358974359e-06,
      "loss": 0.0351,
      "step": 8
    },
    {
      "epoch": 0.03469879518072289,
      "grad_norm": 0.35446059703826904,
      "learning_rate": 4.102564102564103e-06,
      "loss": 0.022,
      "step": 9
    },
    {
      "epoch": 0.03855421686746988,
      "grad_norm": 0.40557414293289185,
      "learning_rate": 4.615384615384616e-06,
      "loss": 0.0248,
      "step": 10
    },
    {
      "epoch": 0.042409638554216866,
      "grad_norm": 0.3844715356826782,
      "learning_rate": 5.128205128205128e-06,
      "loss": 0.0289,
      "step": 11
    },
    {
      "epoch": 0.04626506024096386,
      "grad_norm": 0.31521302461624146,
      "learning_rate": 5.641025641025641e-06,
      "loss": 0.0221,
      "step": 12
    },
    {
      "epoch": 0.05012048192771084,
      "grad_norm": 0.40441566705703735,
      "learning_rate": 6.153846153846155e-06,
      "loss": 0.0226,
      "step": 13
    },
    {
      "epoch": 0.05397590361445783,
      "grad_norm": 0.30345258116722107,
      "learning_rate": 6.666666666666667e-06,
      "loss": 0.0181,
      "step": 14
    },
    {
      "epoch": 0.05783132530120482,
      "grad_norm": 0.32419148087501526,
      "learning_rate": 7.17948717948718e-06,
      "loss": 0.0217,
      "step": 15
    },
    {
      "epoch": 0.061686746987951804,
      "grad_norm": 0.29662227630615234,
      "learning_rate": 7.692307692307694e-06,
      "loss": 0.0216,
      "step": 16
    },
    {
      "epoch": 0.0655421686746988,
      "grad_norm": 0.24562332034111023,
      "learning_rate": 8.205128205128205e-06,
      "loss": 0.0163,
      "step": 17
    },
    {
      "epoch": 0.06939759036144579,
      "grad_norm": 0.21251554787158966,
      "learning_rate": 8.717948717948719e-06,
      "loss": 0.0176,
      "step": 18
    },
    {
      "epoch": 0.07325301204819278,
      "grad_norm": 0.195358544588089,
      "learning_rate": 9.230769230769232e-06,
      "loss": 0.0121,
      "step": 19
    },
    {
      "epoch": 0.07710843373493977,
      "grad_norm": 0.1519210785627365,
      "learning_rate": 9.743589743589744e-06,
      "loss": 0.0103,
      "step": 20
    },
    {
      "epoch": 0.08096385542168674,
      "grad_norm": 0.29832565784454346,
      "learning_rate": 1.0256410256410256e-05,
      "loss": 0.0197,
      "step": 21
    },
    {
      "epoch": 0.08481927710843373,
      "grad_norm": 0.24802711606025696,
      "learning_rate": 1.076923076923077e-05,
      "loss": 0.0207,
      "step": 22
    },
    {
      "epoch": 0.08867469879518072,
      "grad_norm": 0.17878903448581696,
      "learning_rate": 1.1282051282051283e-05,
      "loss": 0.014,
      "step": 23
    },
    {
      "epoch": 0.09253012048192771,
      "grad_norm": 0.23687325417995453,
      "learning_rate": 1.1794871794871796e-05,
      "loss": 0.0153,
      "step": 24
    },
    {
      "epoch": 0.0963855421686747,
      "grad_norm": 0.29058969020843506,
      "learning_rate": 1.230769230769231e-05,
      "loss": 0.0149,
      "step": 25
    },
    {
      "epoch": 0.10024096385542168,
      "grad_norm": 0.1624310314655304,
      "learning_rate": 1.2820512820512823e-05,
      "loss": 0.0127,
      "step": 26
    },
    {
      "epoch": 0.10409638554216867,
      "grad_norm": 0.27256354689598083,
      "learning_rate": 1.3333333333333333e-05,
      "loss": 0.0173,
      "step": 27
    },
    {
      "epoch": 0.10795180722891566,
      "grad_norm": 0.24831782281398773,
      "learning_rate": 1.3846153846153847e-05,
      "loss": 0.0161,
      "step": 28
    },
    {
      "epoch": 0.11180722891566265,
      "grad_norm": 0.20417729020118713,
      "learning_rate": 1.435897435897436e-05,
      "loss": 0.0127,
      "step": 29
    },
    {
      "epoch": 0.11566265060240964,
      "grad_norm": 0.22253485023975372,
      "learning_rate": 1.4871794871794874e-05,
      "loss": 0.018,
      "step": 30
    },
    {
      "epoch": 0.11951807228915663,
      "grad_norm": 0.1634419709444046,
      "learning_rate": 1.5384615384615387e-05,
      "loss": 0.0092,
      "step": 31
    },
    {
      "epoch": 0.12337349397590361,
      "grad_norm": 0.16349917650222778,
      "learning_rate": 1.5897435897435897e-05,
      "loss": 0.0151,
      "step": 32
    },
    {
      "epoch": 0.1272289156626506,
      "grad_norm": 0.2271573394536972,
      "learning_rate": 1.641025641025641e-05,
      "loss": 0.0174,
      "step": 33
    },
    {
      "epoch": 0.1310843373493976,
      "grad_norm": 0.1916988492012024,
      "learning_rate": 1.6923076923076924e-05,
      "loss": 0.0126,
      "step": 34
    },
    {
      "epoch": 0.13493975903614458,
      "grad_norm": 0.23479051887989044,
      "learning_rate": 1.7435897435897438e-05,
      "loss": 0.0116,
      "step": 35
    },
    {
      "epoch": 0.13879518072289157,
      "grad_norm": 0.18296493589878082,
      "learning_rate": 1.794871794871795e-05,
      "loss": 0.0133,
      "step": 36
    },
    {
      "epoch": 0.14265060240963856,
      "grad_norm": 0.12307147681713104,
      "learning_rate": 1.8461538461538465e-05,
      "loss": 0.0093,
      "step": 37
    },
    {
      "epoch": 0.14650602409638555,
      "grad_norm": 0.1736001819372177,
      "learning_rate": 1.8974358974358975e-05,
      "loss": 0.0097,
      "step": 38
    },
    {
      "epoch": 0.15036144578313254,
      "grad_norm": 0.21048344671726227,
      "learning_rate": 1.9487179487179488e-05,
      "loss": 0.0121,
      "step": 39
    },
    {
      "epoch": 0.15421686746987953,
      "grad_norm": 0.18844228982925415,
      "learning_rate": 2e-05,
      "loss": 0.0145,
      "step": 40
    },
    {
      "epoch": 0.1580722891566265,
      "grad_norm": 0.24053065478801727,
      "learning_rate": 2.0512820512820512e-05,
      "loss": 0.0233,
      "step": 41
    },
    {
      "epoch": 0.16192771084337348,
      "grad_norm": 0.16952425241470337,
      "learning_rate": 2.102564102564103e-05,
      "loss": 0.024,
      "step": 42
    },
    {
      "epoch": 0.16578313253012048,
      "grad_norm": 0.1816851794719696,
      "learning_rate": 2.153846153846154e-05,
      "loss": 0.0165,
      "step": 43
    },
    {
      "epoch": 0.16963855421686747,
      "grad_norm": 0.17718464136123657,
      "learning_rate": 2.2051282051282056e-05,
      "loss": 0.0171,
      "step": 44
    },
    {
      "epoch": 0.17349397590361446,
      "grad_norm": 0.44719773530960083,
      "learning_rate": 2.2564102564102566e-05,
      "loss": 0.0357,
      "step": 45
    },
    {
      "epoch": 0.17734939759036145,
      "grad_norm": 0.1649208813905716,
      "learning_rate": 2.3076923076923076e-05,
      "loss": 0.016,
      "step": 46
    },
    {
      "epoch": 0.18120481927710844,
      "grad_norm": 0.17772121727466583,
      "learning_rate": 2.3589743589743593e-05,
      "loss": 0.0106,
      "step": 47
    },
    {
      "epoch": 0.18506024096385543,
      "grad_norm": 0.30155256390571594,
      "learning_rate": 2.4102564102564103e-05,
      "loss": 0.0132,
      "step": 48
    },
    {
      "epoch": 0.18891566265060242,
      "grad_norm": 0.1777803599834442,
      "learning_rate": 2.461538461538462e-05,
      "loss": 0.0089,
      "step": 49
    },
    {
      "epoch": 0.1927710843373494,
      "grad_norm": 0.23959968984127045,
      "learning_rate": 2.512820512820513e-05,
      "loss": 0.0137,
      "step": 50
    },
    {
      "epoch": 0.1966265060240964,
      "grad_norm": 0.2473219782114029,
      "learning_rate": 2.5641025641025646e-05,
      "loss": 0.0241,
      "step": 51
    },
    {
      "epoch": 0.20048192771084336,
      "grad_norm": 0.14216162264347076,
      "learning_rate": 2.6153846153846157e-05,
      "loss": 0.0089,
      "step": 52
    },
    {
      "epoch": 0.20433734939759035,
      "grad_norm": 0.16024824976921082,
      "learning_rate": 2.6666666666666667e-05,
      "loss": 0.0105,
      "step": 53
    },
    {
      "epoch": 0.20819277108433734,
      "grad_norm": 0.19614927470684052,
      "learning_rate": 2.7179487179487183e-05,
      "loss": 0.0102,
      "step": 54
    },
    {
      "epoch": 0.21204819277108433,
      "grad_norm": 0.18651551008224487,
      "learning_rate": 2.7692307692307694e-05,
      "loss": 0.0109,
      "step": 55
    },
    {
      "epoch": 0.21590361445783132,
      "grad_norm": 0.1358720362186432,
      "learning_rate": 2.820512820512821e-05,
      "loss": 0.0082,
      "step": 56
    },
    {
      "epoch": 0.2197590361445783,
      "grad_norm": 0.13998733460903168,
      "learning_rate": 2.871794871794872e-05,
      "loss": 0.0081,
      "step": 57
    },
    {
      "epoch": 0.2236144578313253,
      "grad_norm": 0.1299574077129364,
      "learning_rate": 2.923076923076923e-05,
      "loss": 0.0067,
      "step": 58
    },
    {
      "epoch": 0.2274698795180723,
      "grad_norm": 0.17477169632911682,
      "learning_rate": 2.9743589743589747e-05,
      "loss": 0.0091,
      "step": 59
    },
    {
      "epoch": 0.23132530120481928,
      "grad_norm": 0.27078455686569214,
      "learning_rate": 3.0256410256410257e-05,
      "loss": 0.0128,
      "step": 60
    },
    {
      "epoch": 0.23518072289156627,
      "grad_norm": 0.2184700220823288,
      "learning_rate": 3.0769230769230774e-05,
      "loss": 0.0236,
      "step": 61
    },
    {
      "epoch": 0.23903614457831326,
      "grad_norm": 0.21013228595256805,
      "learning_rate": 3.128205128205129e-05,
      "loss": 0.0114,
      "step": 62
    },
    {
      "epoch": 0.24289156626506025,
      "grad_norm": 0.4988560974597931,
      "learning_rate": 3.1794871794871795e-05,
      "loss": 0.0111,
      "step": 63
    },
    {
      "epoch": 0.24674698795180722,
      "grad_norm": 0.18484456837177277,
      "learning_rate": 3.230769230769231e-05,
      "loss": 0.0104,
      "step": 64
    },
    {
      "epoch": 0.25060240963855424,
      "grad_norm": 0.19871847331523895,
      "learning_rate": 3.282051282051282e-05,
      "loss": 0.0094,
      "step": 65
    },
    {
      "epoch": 0.2544578313253012,
      "grad_norm": 0.17952671647071838,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 0.0127,
      "step": 66
    },
    {
      "epoch": 0.2583132530120482,
      "grad_norm": 0.29055604338645935,
      "learning_rate": 3.384615384615385e-05,
      "loss": 0.023,
      "step": 67
    },
    {
      "epoch": 0.2621686746987952,
      "grad_norm": 0.2033795416355133,
      "learning_rate": 3.435897435897436e-05,
      "loss": 0.0091,
      "step": 68
    },
    {
      "epoch": 0.26602409638554214,
      "grad_norm": 0.2499230057001114,
      "learning_rate": 3.4871794871794875e-05,
      "loss": 0.0221,
      "step": 69
    },
    {
      "epoch": 0.26987951807228916,
      "grad_norm": 0.22038578987121582,
      "learning_rate": 3.538461538461539e-05,
      "loss": 0.0139,
      "step": 70
    },
    {
      "epoch": 0.2737349397590361,
      "grad_norm": 0.14657297730445862,
      "learning_rate": 3.58974358974359e-05,
      "loss": 0.0104,
      "step": 71
    },
    {
      "epoch": 0.27759036144578314,
      "grad_norm": 0.23861773312091827,
      "learning_rate": 3.6410256410256416e-05,
      "loss": 0.0126,
      "step": 72
    },
    {
      "epoch": 0.2814457831325301,
      "grad_norm": 0.21007320284843445,
      "learning_rate": 3.692307692307693e-05,
      "loss": 0.0133,
      "step": 73
    },
    {
      "epoch": 0.2853012048192771,
      "grad_norm": 0.20938654243946075,
      "learning_rate": 3.7435897435897436e-05,
      "loss": 0.0116,
      "step": 74
    },
    {
      "epoch": 0.2891566265060241,
      "grad_norm": 0.27394789457321167,
      "learning_rate": 3.794871794871795e-05,
      "loss": 0.0158,
      "step": 75
    },
    {
      "epoch": 0.2930120481927711,
      "grad_norm": 0.1749437153339386,
      "learning_rate": 3.846153846153846e-05,
      "loss": 0.0097,
      "step": 76
    },
    {
      "epoch": 0.29686746987951806,
      "grad_norm": 0.2208889275789261,
      "learning_rate": 3.8974358974358976e-05,
      "loss": 0.024,
      "step": 77
    },
    {
      "epoch": 0.3007228915662651,
      "grad_norm": 0.32802650332450867,
      "learning_rate": 3.948717948717949e-05,
      "loss": 0.015,
      "step": 78
    },
    {
      "epoch": 0.30457831325301205,
      "grad_norm": 0.21527938544750214,
      "learning_rate": 4e-05,
      "loss": 0.0137,
      "step": 79
    },
    {
      "epoch": 0.30843373493975906,
      "grad_norm": 0.2894026041030884,
      "learning_rate": 3.999979972589652e-05,
      "loss": 0.0177,
      "step": 80
    },
    {
      "epoch": 0.312289156626506,
      "grad_norm": 0.21148355305194855,
      "learning_rate": 3.9999198907597046e-05,
      "loss": 0.0146,
      "step": 81
    },
    {
      "epoch": 0.316144578313253,
      "grad_norm": 0.1950712502002716,
      "learning_rate": 3.999819755713442e-05,
      "loss": 0.0174,
      "step": 82
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.23408709466457367,
      "learning_rate": 3.9996795694563096e-05,
      "loss": 0.0142,
      "step": 83
    },
    {
      "epoch": 0.32385542168674697,
      "grad_norm": 0.4009031355381012,
      "learning_rate": 3.999499334795875e-05,
      "loss": 0.014,
      "step": 84
    },
    {
      "epoch": 0.327710843373494,
      "grad_norm": 0.2507084012031555,
      "learning_rate": 3.999279055341771e-05,
      "loss": 0.013,
      "step": 85
    },
    {
      "epoch": 0.33156626506024095,
      "grad_norm": 0.2922132909297943,
      "learning_rate": 3.999018735505626e-05,
      "loss": 0.013,
      "step": 86
    },
    {
      "epoch": 0.33542168674698797,
      "grad_norm": 0.2073613405227661,
      "learning_rate": 3.998718380500971e-05,
      "loss": 0.0107,
      "step": 87
    },
    {
      "epoch": 0.33927710843373493,
      "grad_norm": 0.3113904595375061,
      "learning_rate": 3.998377996343139e-05,
      "loss": 0.0337,
      "step": 88
    },
    {
      "epoch": 0.34313253012048195,
      "grad_norm": 0.18452070653438568,
      "learning_rate": 3.997997589849145e-05,
      "loss": 0.0137,
      "step": 89
    },
    {
      "epoch": 0.3469879518072289,
      "grad_norm": 0.24420331418514252,
      "learning_rate": 3.9975771686375434e-05,
      "loss": 0.0104,
      "step": 90
    },
    {
      "epoch": 0.35084337349397593,
      "grad_norm": 0.3163854479789734,
      "learning_rate": 3.9971167411282835e-05,
      "loss": 0.0133,
      "step": 91
    },
    {
      "epoch": 0.3546987951807229,
      "grad_norm": 0.11558058857917786,
      "learning_rate": 3.996616316542537e-05,
      "loss": 0.0069,
      "step": 92
    },
    {
      "epoch": 0.35855421686746985,
      "grad_norm": 2.0870206356048584,
      "learning_rate": 3.99607590490251e-05,
      "loss": 0.0227,
      "step": 93
    },
    {
      "epoch": 0.3624096385542169,
      "grad_norm": 0.23557044565677643,
      "learning_rate": 3.9954955170312504e-05,
      "loss": 0.0175,
      "step": 94
    },
    {
      "epoch": 0.36626506024096384,
      "grad_norm": 0.2593333423137665,
      "learning_rate": 3.9948751645524235e-05,
      "loss": 0.0184,
      "step": 95
    },
    {
      "epoch": 0.37012048192771085,
      "grad_norm": 0.5161668658256531,
      "learning_rate": 3.994214859890083e-05,
      "loss": 0.0265,
      "step": 96
    },
    {
      "epoch": 0.3739759036144578,
      "grad_norm": 0.5731181502342224,
      "learning_rate": 3.9935146162684206e-05,
      "loss": 0.0167,
      "step": 97
    },
    {
      "epoch": 0.37783132530120483,
      "grad_norm": 1.9357824325561523,
      "learning_rate": 3.992774447711503e-05,
      "loss": 0.0321,
      "step": 98
    },
    {
      "epoch": 0.3816867469879518,
      "grad_norm": 4.194522857666016,
      "learning_rate": 3.9919943690429906e-05,
      "loss": 0.086,
      "step": 99
    },
    {
      "epoch": 0.3855421686746988,
      "grad_norm": 0.3377005457878113,
      "learning_rate": 3.9911743958858376e-05,
      "loss": 0.0206,
      "step": 100
    },
    {
      "epoch": 0.3893975903614458,
      "grad_norm": 0.3599908649921417,
      "learning_rate": 3.9903145446619837e-05,
      "loss": 0.0207,
      "step": 101
    },
    {
      "epoch": 0.3932530120481928,
      "grad_norm": 0.3402753174304962,
      "learning_rate": 3.9894148325920226e-05,
      "loss": 0.0241,
      "step": 102
    },
    {
      "epoch": 0.39710843373493976,
      "grad_norm": 0.6041432023048401,
      "learning_rate": 3.9884752776948564e-05,
      "loss": 0.0138,
      "step": 103
    },
    {
      "epoch": 0.4009638554216867,
      "grad_norm": 4.082062721252441,
      "learning_rate": 3.987495898787336e-05,
      "loss": 0.1119,
      "step": 104
    },
    {
      "epoch": 0.40481927710843374,
      "grad_norm": 6.899234771728516,
      "learning_rate": 3.9864767154838864e-05,
      "loss": 0.1046,
      "step": 105
    },
    {
      "epoch": 0.4086746987951807,
      "grad_norm": 3.1316874027252197,
      "learning_rate": 3.985417748196108e-05,
      "loss": 0.0404,
      "step": 106
    },
    {
      "epoch": 0.4125301204819277,
      "grad_norm": 1.8661688566207886,
      "learning_rate": 3.9843190181323744e-05,
      "loss": 0.0348,
      "step": 107
    },
    {
      "epoch": 0.4163855421686747,
      "grad_norm": 0.5149962306022644,
      "learning_rate": 3.983180547297404e-05,
      "loss": 0.0284,
      "step": 108
    },
    {
      "epoch": 0.4202409638554217,
      "grad_norm": 0.3855750858783722,
      "learning_rate": 3.982002358491817e-05,
      "loss": 0.0242,
      "step": 109
    },
    {
      "epoch": 0.42409638554216866,
      "grad_norm": 0.35508987307548523,
      "learning_rate": 3.9807844753116864e-05,
      "loss": 0.0179,
      "step": 110
    },
    {
      "epoch": 0.4279518072289157,
      "grad_norm": 1.477452278137207,
      "learning_rate": 3.979526922148058e-05,
      "loss": 0.0436,
      "step": 111
    },
    {
      "epoch": 0.43180722891566264,
      "grad_norm": 0.41172751784324646,
      "learning_rate": 3.978229724186464e-05,
      "loss": 0.026,
      "step": 112
    },
    {
      "epoch": 0.43566265060240966,
      "grad_norm": 0.5889139771461487,
      "learning_rate": 3.9768929074064206e-05,
      "loss": 0.0249,
      "step": 113
    },
    {
      "epoch": 0.4395180722891566,
      "grad_norm": 0.38035663962364197,
      "learning_rate": 3.975516498580907e-05,
      "loss": 0.0257,
      "step": 114
    },
    {
      "epoch": 0.4433734939759036,
      "grad_norm": 0.35500895977020264,
      "learning_rate": 3.9741005252758255e-05,
      "loss": 0.026,
      "step": 115
    },
    {
      "epoch": 0.4472289156626506,
      "grad_norm": 0.6464815735816956,
      "learning_rate": 3.972645015849457e-05,
      "loss": 0.0499,
      "step": 116
    },
    {
      "epoch": 0.45108433734939757,
      "grad_norm": 0.20825283229351044,
      "learning_rate": 3.971149999451886e-05,
      "loss": 0.034,
      "step": 117
    },
    {
      "epoch": 0.4549397590361446,
      "grad_norm": 0.34638360142707825,
      "learning_rate": 3.9696155060244166e-05,
      "loss": 0.0226,
      "step": 118
    },
    {
      "epoch": 0.45879518072289155,
      "grad_norm": 0.27782803773880005,
      "learning_rate": 3.9680415662989806e-05,
      "loss": 0.0197,
      "step": 119
    },
    {
      "epoch": 0.46265060240963857,
      "grad_norm": 0.39609628915786743,
      "learning_rate": 3.966428211797514e-05,
      "loss": 0.0229,
      "step": 120
    },
    {
      "epoch": 0.46650602409638553,
      "grad_norm": 0.30235278606414795,
      "learning_rate": 3.9647754748313294e-05,
      "loss": 0.046,
      "step": 121
    },
    {
      "epoch": 0.47036144578313255,
      "grad_norm": 0.2774790823459625,
      "learning_rate": 3.963083388500469e-05,
      "loss": 0.0208,
      "step": 122
    },
    {
      "epoch": 0.4742168674698795,
      "grad_norm": 0.24587354063987732,
      "learning_rate": 3.96135198669304e-05,
      "loss": 0.026,
      "step": 123
    },
    {
      "epoch": 0.47807228915662653,
      "grad_norm": 0.2603200376033783,
      "learning_rate": 3.959581304084536e-05,
      "loss": 0.0189,
      "step": 124
    },
    {
      "epoch": 0.4819277108433735,
      "grad_norm": 0.33505868911743164,
      "learning_rate": 3.957771376137144e-05,
      "loss": 0.0227,
      "step": 125
    },
    {
      "epoch": 0.4857831325301205,
      "grad_norm": 0.32992950081825256,
      "learning_rate": 3.955922239099035e-05,
      "loss": 0.0207,
      "step": 126
    },
    {
      "epoch": 0.48963855421686747,
      "grad_norm": 0.2520611882209778,
      "learning_rate": 3.954033930003634e-05,
      "loss": 0.0308,
      "step": 127
    },
    {
      "epoch": 0.49349397590361443,
      "grad_norm": 0.3344120383262634,
      "learning_rate": 3.952106486668884e-05,
      "loss": 0.0187,
      "step": 128
    },
    {
      "epoch": 0.49734939759036145,
      "grad_norm": 0.46383559703826904,
      "learning_rate": 3.9501399476964806e-05,
      "loss": 0.0174,
      "step": 129
    },
    {
      "epoch": 0.5012048192771085,
      "grad_norm": 0.2674378752708435,
      "learning_rate": 3.9481343524711095e-05,
      "loss": 0.0164,
      "step": 130
    },
    {
      "epoch": 0.5050602409638554,
      "grad_norm": 0.21262381970882416,
      "learning_rate": 3.946089741159648e-05,
      "loss": 0.0088,
      "step": 131
    },
    {
      "epoch": 0.5089156626506024,
      "grad_norm": 0.2904111444950104,
      "learning_rate": 3.944006154710366e-05,
      "loss": 0.0195,
      "step": 132
    },
    {
      "epoch": 0.5127710843373494,
      "grad_norm": 0.42615893483161926,
      "learning_rate": 3.9418836348521045e-05,
      "loss": 0.0172,
      "step": 133
    },
    {
      "epoch": 0.5166265060240964,
      "grad_norm": 0.25124824047088623,
      "learning_rate": 3.939722224093439e-05,
      "loss": 0.0151,
      "step": 134
    },
    {
      "epoch": 0.5204819277108433,
      "grad_norm": 1.9261884689331055,
      "learning_rate": 3.937521965721831e-05,
      "loss": 0.0247,
      "step": 135
    },
    {
      "epoch": 0.5243373493975904,
      "grad_norm": 0.24449346959590912,
      "learning_rate": 3.9352829038027564e-05,
      "loss": 0.0113,
      "step": 136
    },
    {
      "epoch": 0.5281927710843374,
      "grad_norm": 2.4259743690490723,
      "learning_rate": 3.933005083178828e-05,
      "loss": 0.0533,
      "step": 137
    },
    {
      "epoch": 0.5320481927710843,
      "grad_norm": 0.3187374472618103,
      "learning_rate": 3.930688549468894e-05,
      "loss": 0.0179,
      "step": 138
    },
    {
      "epoch": 0.5359036144578313,
      "grad_norm": 2.9974002838134766,
      "learning_rate": 3.928333349067125e-05,
      "loss": 0.023,
      "step": 139
    },
    {
      "epoch": 0.5397590361445783,
      "grad_norm": 0.7724501490592957,
      "learning_rate": 3.9259395291420864e-05,
      "loss": 0.0179,
      "step": 140
    },
    {
      "epoch": 0.5436144578313253,
      "grad_norm": 0.2811576724052429,
      "learning_rate": 3.923507137635792e-05,
      "loss": 0.0118,
      "step": 141
    },
    {
      "epoch": 0.5474698795180722,
      "grad_norm": 0.301895409822464,
      "learning_rate": 3.921036223262745e-05,
      "loss": 0.0152,
      "step": 142
    },
    {
      "epoch": 0.5513253012048193,
      "grad_norm": 0.32171159982681274,
      "learning_rate": 3.9185268355089606e-05,
      "loss": 0.0297,
      "step": 143
    },
    {
      "epoch": 0.5551807228915663,
      "grad_norm": 0.16498999297618866,
      "learning_rate": 3.915979024630978e-05,
      "loss": 0.0134,
      "step": 144
    },
    {
      "epoch": 0.5590361445783133,
      "grad_norm": 0.27746957540512085,
      "learning_rate": 3.913392841654851e-05,
      "loss": 0.013,
      "step": 145
    },
    {
      "epoch": 0.5628915662650602,
      "grad_norm": 0.2298375517129898,
      "learning_rate": 3.9107683383751264e-05,
      "loss": 0.0106,
      "step": 146
    },
    {
      "epoch": 0.5667469879518072,
      "grad_norm": 0.2801443636417389,
      "learning_rate": 3.9081055673538093e-05,
      "loss": 0.0214,
      "step": 147
    },
    {
      "epoch": 0.5706024096385542,
      "grad_norm": 0.22764678299427032,
      "learning_rate": 3.9054045819193074e-05,
      "loss": 0.0148,
      "step": 148
    },
    {
      "epoch": 0.5744578313253013,
      "grad_norm": 0.1658417284488678,
      "learning_rate": 3.902665436165364e-05,
      "loss": 0.0333,
      "step": 149
    },
    {
      "epoch": 0.5783132530120482,
      "grad_norm": 0.1666492223739624,
      "learning_rate": 3.8998881849499756e-05,
      "loss": 0.0162,
      "step": 150
    },
    {
      "epoch": 0.5821686746987952,
      "grad_norm": 0.20655786991119385,
      "learning_rate": 3.897072883894291e-05,
      "loss": 0.0152,
      "step": 151
    },
    {
      "epoch": 0.5860240963855422,
      "grad_norm": 0.25162509083747864,
      "learning_rate": 3.894219589381501e-05,
      "loss": 0.0422,
      "step": 152
    },
    {
      "epoch": 0.5898795180722891,
      "grad_norm": 0.17071524262428284,
      "learning_rate": 3.8913283585557054e-05,
      "loss": 0.0082,
      "step": 153
    },
    {
      "epoch": 0.5937349397590361,
      "grad_norm": 0.20576122403144836,
      "learning_rate": 3.8883992493207696e-05,
      "loss": 0.0111,
      "step": 154
    },
    {
      "epoch": 0.5975903614457831,
      "grad_norm": 0.19477005302906036,
      "learning_rate": 3.885432320339167e-05,
      "loss": 0.0084,
      "step": 155
    },
    {
      "epoch": 0.6014457831325302,
      "grad_norm": 0.26111698150634766,
      "learning_rate": 3.8824276310308e-05,
      "loss": 0.0172,
      "step": 156
    },
    {
      "epoch": 0.6053012048192771,
      "grad_norm": 0.2800676226615906,
      "learning_rate": 3.879385241571817e-05,
      "loss": 0.0238,
      "step": 157
    },
    {
      "epoch": 0.6091566265060241,
      "grad_norm": 0.2781737148761749,
      "learning_rate": 3.876305212893399e-05,
      "loss": 0.0272,
      "step": 158
    },
    {
      "epoch": 0.6130120481927711,
      "grad_norm": 0.6798153519630432,
      "learning_rate": 3.873187606680543e-05,
      "loss": 0.0091,
      "step": 159
    },
    {
      "epoch": 0.6168674698795181,
      "grad_norm": 0.1639501005411148,
      "learning_rate": 3.8700324853708304e-05,
      "loss": 0.0233,
      "step": 160
    },
    {
      "epoch": 0.620722891566265,
      "grad_norm": 0.39862754940986633,
      "learning_rate": 3.866839912153168e-05,
      "loss": 0.016,
      "step": 161
    },
    {
      "epoch": 0.624578313253012,
      "grad_norm": 0.48113104701042175,
      "learning_rate": 3.863609950966531e-05,
      "loss": 0.016,
      "step": 162
    },
    {
      "epoch": 0.6284337349397591,
      "grad_norm": 0.17467358708381653,
      "learning_rate": 3.860342666498677e-05,
      "loss": 0.0103,
      "step": 163
    },
    {
      "epoch": 0.632289156626506,
      "grad_norm": 0.5942115783691406,
      "learning_rate": 3.8570381241848534e-05,
      "loss": 0.0158,
      "step": 164
    },
    {
      "epoch": 0.636144578313253,
      "grad_norm": 0.2018311470746994,
      "learning_rate": 3.853696390206484e-05,
      "loss": 0.0193,
      "step": 165
    },
    {
      "epoch": 0.64,
      "grad_norm": 0.5149483680725098,
      "learning_rate": 3.850317531489848e-05,
      "loss": 0.0268,
      "step": 166
    },
    {
      "epoch": 0.643855421686747,
      "grad_norm": 0.16745764017105103,
      "learning_rate": 3.846901615704734e-05,
      "loss": 0.0132,
      "step": 167
    },
    {
      "epoch": 0.6477108433734939,
      "grad_norm": 1.0696477890014648,
      "learning_rate": 3.843448711263089e-05,
      "loss": 0.0228,
      "step": 168
    },
    {
      "epoch": 0.651566265060241,
      "grad_norm": 0.27323034405708313,
      "learning_rate": 3.839958887317649e-05,
      "loss": 0.0309,
      "step": 169
    },
    {
      "epoch": 0.655421686746988,
      "grad_norm": 0.32417500019073486,
      "learning_rate": 3.8364322137605484e-05,
      "loss": 0.0282,
      "step": 170
    },
    {
      "epoch": 0.659277108433735,
      "grad_norm": 0.24416105449199677,
      "learning_rate": 3.832868761221926e-05,
      "loss": 0.0149,
      "step": 171
    },
    {
      "epoch": 0.6631325301204819,
      "grad_norm": 0.40103256702423096,
      "learning_rate": 3.829268601068509e-05,
      "loss": 0.0196,
      "step": 172
    },
    {
      "epoch": 0.6669879518072289,
      "grad_norm": 0.4967587888240814,
      "learning_rate": 3.825631805402182e-05,
      "loss": 0.027,
      "step": 173
    },
    {
      "epoch": 0.6708433734939759,
      "grad_norm": 0.32446494698524475,
      "learning_rate": 3.821958447058543e-05,
      "loss": 0.0186,
      "step": 174
    },
    {
      "epoch": 0.6746987951807228,
      "grad_norm": 0.17924204468727112,
      "learning_rate": 3.818248599605448e-05,
      "loss": 0.0158,
      "step": 175
    },
    {
      "epoch": 0.6785542168674699,
      "grad_norm": 0.32113662362098694,
      "learning_rate": 3.8145023373415326e-05,
      "loss": 0.0186,
      "step": 176
    },
    {
      "epoch": 0.6824096385542169,
      "grad_norm": 0.2955350875854492,
      "learning_rate": 3.810719735294731e-05,
      "loss": 0.0254,
      "step": 177
    },
    {
      "epoch": 0.6862650602409639,
      "grad_norm": 0.3955155611038208,
      "learning_rate": 3.806900869220765e-05,
      "loss": 0.0247,
      "step": 178
    },
    {
      "epoch": 0.6901204819277108,
      "grad_norm": 0.28007134795188904,
      "learning_rate": 3.8030458156016326e-05,
      "loss": 0.0159,
      "step": 179
    },
    {
      "epoch": 0.6939759036144578,
      "grad_norm": 0.3042643666267395,
      "learning_rate": 3.799154651644075e-05,
      "loss": 0.0163,
      "step": 180
    },
    {
      "epoch": 0.6978313253012048,
      "grad_norm": 0.18276739120483398,
      "learning_rate": 3.795227455278029e-05,
      "loss": 0.0088,
      "step": 181
    },
    {
      "epoch": 0.7016867469879519,
      "grad_norm": 0.19548135995864868,
      "learning_rate": 3.7912643051550674e-05,
      "loss": 0.0115,
      "step": 182
    },
    {
      "epoch": 0.7055421686746988,
      "grad_norm": 0.2164657711982727,
      "learning_rate": 3.787265280646825e-05,
      "loss": 0.0152,
      "step": 183
    },
    {
      "epoch": 0.7093975903614458,
      "grad_norm": 0.19452515244483948,
      "learning_rate": 3.783230461843406e-05,
      "loss": 0.008,
      "step": 184
    },
    {
      "epoch": 0.7132530120481928,
      "grad_norm": 0.27214348316192627,
      "learning_rate": 3.7791599295517825e-05,
      "loss": 0.0127,
      "step": 185
    },
    {
      "epoch": 0.7171084337349397,
      "grad_norm": 0.6383033990859985,
      "learning_rate": 3.775053765294175e-05,
      "loss": 0.0151,
      "step": 186
    },
    {
      "epoch": 0.7209638554216867,
      "grad_norm": 0.17882253229618073,
      "learning_rate": 3.7709120513064196e-05,
      "loss": 0.0092,
      "step": 187
    },
    {
      "epoch": 0.7248192771084337,
      "grad_norm": 0.2078862190246582,
      "learning_rate": 3.7667348705363227e-05,
      "loss": 0.0133,
      "step": 188
    },
    {
      "epoch": 0.7286746987951808,
      "grad_norm": 0.1957724541425705,
      "learning_rate": 3.762522306641998e-05,
      "loss": 0.0108,
      "step": 189
    },
    {
      "epoch": 0.7325301204819277,
      "grad_norm": 0.16418907046318054,
      "learning_rate": 3.75827444399019e-05,
      "loss": 0.0137,
      "step": 190
    },
    {
      "epoch": 0.7363855421686747,
      "grad_norm": 0.45467960834503174,
      "learning_rate": 3.7539913676545874e-05,
      "loss": 0.0142,
      "step": 191
    },
    {
      "epoch": 0.7402409638554217,
      "grad_norm": 1.2079054117202759,
      "learning_rate": 3.749673163414117e-05,
      "loss": 0.019,
      "step": 192
    },
    {
      "epoch": 0.7440963855421687,
      "grad_norm": 0.11845115572214127,
      "learning_rate": 3.745319917751229e-05,
      "loss": 0.0098,
      "step": 193
    },
    {
      "epoch": 0.7479518072289156,
      "grad_norm": 0.18299613893032074,
      "learning_rate": 3.740931717850159e-05,
      "loss": 0.0132,
      "step": 194
    },
    {
      "epoch": 0.7518072289156627,
      "grad_norm": 0.225267231464386,
      "learning_rate": 3.736508651595188e-05,
      "loss": 0.0148,
      "step": 195
    },
    {
      "epoch": 0.7556626506024097,
      "grad_norm": 0.49103838205337524,
      "learning_rate": 3.732050807568878e-05,
      "loss": 0.0132,
      "step": 196
    },
    {
      "epoch": 0.7595180722891566,
      "grad_norm": 0.19766205549240112,
      "learning_rate": 3.727558275050301e-05,
      "loss": 0.02,
      "step": 197
    },
    {
      "epoch": 0.7633734939759036,
      "grad_norm": 0.28864291310310364,
      "learning_rate": 3.7230311440132494e-05,
      "loss": 0.0222,
      "step": 198
    },
    {
      "epoch": 0.7672289156626506,
      "grad_norm": 0.28816428780555725,
      "learning_rate": 3.718469505124434e-05,
      "loss": 0.0256,
      "step": 199
    },
    {
      "epoch": 0.7710843373493976,
      "grad_norm": 0.20701606571674347,
      "learning_rate": 3.713873449741669e-05,
      "loss": 0.0114,
      "step": 200
    },
    {
      "epoch": 0.7749397590361445,
      "grad_norm": 0.5334108471870422,
      "learning_rate": 3.709243069912041e-05,
      "loss": 0.0119,
      "step": 201
    },
    {
      "epoch": 0.7787951807228916,
      "grad_norm": 0.7352932095527649,
      "learning_rate": 3.704578458370067e-05,
      "loss": 0.0148,
      "step": 202
    },
    {
      "epoch": 0.7826506024096386,
      "grad_norm": 0.22625291347503662,
      "learning_rate": 3.699879708535838e-05,
      "loss": 0.0164,
      "step": 203
    },
    {
      "epoch": 0.7865060240963856,
      "grad_norm": 0.2132428139448166,
      "learning_rate": 3.695146914513142e-05,
      "loss": 0.0083,
      "step": 204
    },
    {
      "epoch": 0.7903614457831325,
      "grad_norm": 0.15848131477832794,
      "learning_rate": 3.69038017108759e-05,
      "loss": 0.0098,
      "step": 205
    },
    {
      "epoch": 0.7942168674698795,
      "grad_norm": 1.2738759517669678,
      "learning_rate": 3.685579573724707e-05,
      "loss": 0.0137,
      "step": 206
    },
    {
      "epoch": 0.7980722891566265,
      "grad_norm": 0.2131299078464508,
      "learning_rate": 3.680745218568026e-05,
      "loss": 0.019,
      "step": 207
    },
    {
      "epoch": 0.8019277108433734,
      "grad_norm": 0.24905090034008026,
      "learning_rate": 3.6758772024371626e-05,
      "loss": 0.0242,
      "step": 208
    },
    {
      "epoch": 0.8057831325301205,
      "grad_norm": 0.36632227897644043,
      "learning_rate": 3.6709756228258735e-05,
      "loss": 0.021,
      "step": 209
    },
    {
      "epoch": 0.8096385542168675,
      "grad_norm": 0.19981122016906738,
      "learning_rate": 3.6660405779001036e-05,
      "loss": 0.0174,
      "step": 210
    },
    {
      "epoch": 0.8134939759036145,
      "grad_norm": 0.13923773169517517,
      "learning_rate": 3.6610721664960236e-05,
      "loss": 0.0088,
      "step": 211
    },
    {
      "epoch": 0.8173493975903614,
      "grad_norm": 0.27602291107177734,
      "learning_rate": 3.656070488118047e-05,
      "loss": 0.0391,
      "step": 212
    },
    {
      "epoch": 0.8212048192771084,
      "grad_norm": 1.3190431594848633,
      "learning_rate": 3.65103564293684e-05,
      "loss": 0.0255,
      "step": 213
    },
    {
      "epoch": 0.8250602409638554,
      "grad_norm": 0.15323355793952942,
      "learning_rate": 3.645967731787313e-05,
      "loss": 0.0164,
      "step": 214
    },
    {
      "epoch": 0.8289156626506025,
      "grad_norm": 0.15326736867427826,
      "learning_rate": 3.640866856166601e-05,
      "loss": 0.0109,
      "step": 215
    },
    {
      "epoch": 0.8327710843373494,
      "grad_norm": 0.2949405312538147,
      "learning_rate": 3.635733118232034e-05,
      "loss": 0.0117,
      "step": 216
    },
    {
      "epoch": 0.8366265060240964,
      "grad_norm": 0.243217334151268,
      "learning_rate": 3.6305666207990886e-05,
      "loss": 0.0191,
      "step": 217
    },
    {
      "epoch": 0.8404819277108434,
      "grad_norm": 0.43892231583595276,
      "learning_rate": 3.625367467339329e-05,
      "loss": 0.0174,
      "step": 218
    },
    {
      "epoch": 0.8443373493975903,
      "grad_norm": 0.36832818388938904,
      "learning_rate": 3.6201357619783336e-05,
      "loss": 0.0201,
      "step": 219
    },
    {
      "epoch": 0.8481927710843373,
      "grad_norm": 0.2650613486766815,
      "learning_rate": 3.614871609493614e-05,
      "loss": 0.0163,
      "step": 220
    },
    {
      "epoch": 0.8520481927710843,
      "grad_norm": 0.1648036539554596,
      "learning_rate": 3.609575115312511e-05,
      "loss": 0.0111,
      "step": 221
    },
    {
      "epoch": 0.8559036144578314,
      "grad_norm": 0.2738659083843231,
      "learning_rate": 3.604246385510088e-05,
      "loss": 0.0063,
      "step": 222
    },
    {
      "epoch": 0.8597590361445783,
      "grad_norm": 0.22977620363235474,
      "learning_rate": 3.598885526807003e-05,
      "loss": 0.015,
      "step": 223
    },
    {
      "epoch": 0.8636144578313253,
      "grad_norm": 0.535546064376831,
      "learning_rate": 3.593492646567372e-05,
      "loss": 0.0102,
      "step": 224
    },
    {
      "epoch": 0.8674698795180723,
      "grad_norm": 0.494342565536499,
      "learning_rate": 3.5880678527966224e-05,
      "loss": 0.0102,
      "step": 225
    },
    {
      "epoch": 0.8713253012048193,
      "grad_norm": 0.17625126242637634,
      "learning_rate": 3.582611254139324e-05,
      "loss": 0.0259,
      "step": 226
    },
    {
      "epoch": 0.8751807228915662,
      "grad_norm": 0.20568431913852692,
      "learning_rate": 3.577122959877017e-05,
      "loss": 0.0237,
      "step": 227
    },
    {
      "epoch": 0.8790361445783132,
      "grad_norm": 0.31444859504699707,
      "learning_rate": 3.571603079926024e-05,
      "loss": 0.0171,
      "step": 228
    },
    {
      "epoch": 0.8828915662650603,
      "grad_norm": 0.48810216784477234,
      "learning_rate": 3.566051724835245e-05,
      "loss": 0.01,
      "step": 229
    },
    {
      "epoch": 0.8867469879518072,
      "grad_norm": 0.4229196608066559,
      "learning_rate": 3.5604690057839456e-05,
      "loss": 0.0111,
      "step": 230
    },
    {
      "epoch": 0.8906024096385542,
      "grad_norm": 0.6600443124771118,
      "learning_rate": 3.554855034579532e-05,
      "loss": 0.0208,
      "step": 231
    },
    {
      "epoch": 0.8944578313253012,
      "grad_norm": 0.2891610860824585,
      "learning_rate": 3.549209923655309e-05,
      "loss": 0.0252,
      "step": 232
    },
    {
      "epoch": 0.8983132530120482,
      "grad_norm": 0.20314915478229523,
      "learning_rate": 3.5435337860682304e-05,
      "loss": 0.0164,
      "step": 233
    },
    {
      "epoch": 0.9021686746987951,
      "grad_norm": 0.19718259572982788,
      "learning_rate": 3.537826735496632e-05,
      "loss": 0.0125,
      "step": 234
    },
    {
      "epoch": 0.9060240963855422,
      "grad_norm": 1.074033498764038,
      "learning_rate": 3.532088886237956e-05,
      "loss": 0.015,
      "step": 235
    },
    {
      "epoch": 0.9098795180722892,
      "grad_norm": 0.7579637169837952,
      "learning_rate": 3.5263203532064674e-05,
      "loss": 0.0153,
      "step": 236
    },
    {
      "epoch": 0.9137349397590362,
      "grad_norm": 0.15700305998325348,
      "learning_rate": 3.520521251930941e-05,
      "loss": 0.0109,
      "step": 237
    },
    {
      "epoch": 0.9175903614457831,
      "grad_norm": 0.3394688069820404,
      "learning_rate": 3.5146916985523604e-05,
      "loss": 0.0156,
      "step": 238
    },
    {
      "epoch": 0.9214457831325301,
      "grad_norm": 1.3616353273391724,
      "learning_rate": 3.5088318098215805e-05,
      "loss": 0.0194,
      "step": 239
    },
    {
      "epoch": 0.9253012048192771,
      "grad_norm": 0.4036584198474884,
      "learning_rate": 3.5029417030970005e-05,
      "loss": 0.012,
      "step": 240
    },
    {
      "epoch": 0.929156626506024,
      "grad_norm": 0.18816007673740387,
      "learning_rate": 3.497021496342203e-05,
      "loss": 0.0146,
      "step": 241
    },
    {
      "epoch": 0.9330120481927711,
      "grad_norm": 0.17888401448726654,
      "learning_rate": 3.4910713081235975e-05,
      "loss": 0.0095,
      "step": 242
    },
    {
      "epoch": 0.9368674698795181,
      "grad_norm": 0.22190752625465393,
      "learning_rate": 3.485091257608047e-05,
      "loss": 0.018,
      "step": 243
    },
    {
      "epoch": 0.9407228915662651,
      "grad_norm": 0.48926958441734314,
      "learning_rate": 3.479081464560475e-05,
      "loss": 0.0407,
      "step": 244
    },
    {
      "epoch": 0.944578313253012,
      "grad_norm": 0.236537367105484,
      "learning_rate": 3.473042049341474e-05,
      "loss": 0.0152,
      "step": 245
    },
    {
      "epoch": 0.948433734939759,
      "grad_norm": 0.13449200987815857,
      "learning_rate": 3.466973132904892e-05,
      "loss": 0.0105,
      "step": 246
    },
    {
      "epoch": 0.952289156626506,
      "grad_norm": 0.23878124356269836,
      "learning_rate": 3.4608748367954064e-05,
      "loss": 0.0154,
      "step": 247
    },
    {
      "epoch": 0.9561445783132531,
      "grad_norm": 0.8225765824317932,
      "learning_rate": 3.4547472831460976e-05,
      "loss": 0.0258,
      "step": 248
    },
    {
      "epoch": 0.96,
      "grad_norm": 0.2288682758808136,
      "learning_rate": 3.4485905946759965e-05,
      "loss": 0.0137,
      "step": 249
    },
    {
      "epoch": 0.963855421686747,
      "grad_norm": 3.1449129581451416,
      "learning_rate": 3.442404894687629e-05,
      "loss": 0.009,
      "step": 250
    },
    {
      "epoch": 0.967710843373494,
      "grad_norm": 0.36979448795318604,
      "learning_rate": 3.4361903070645484e-05,
      "loss": 0.0135,
      "step": 251
    },
    {
      "epoch": 0.971566265060241,
      "grad_norm": 0.434368759393692,
      "learning_rate": 3.4299469562688494e-05,
      "loss": 0.0207,
      "step": 252
    },
    {
      "epoch": 0.9754216867469879,
      "grad_norm": 0.20272351801395416,
      "learning_rate": 3.423674967338681e-05,
      "loss": 0.0166,
      "step": 253
    },
    {
      "epoch": 0.9792771084337349,
      "grad_norm": 0.30525901913642883,
      "learning_rate": 3.4173744658857403e-05,
      "loss": 0.0088,
      "step": 254
    },
    {
      "epoch": 0.983132530120482,
      "grad_norm": 0.8094114661216736,
      "learning_rate": 3.411045578092754e-05,
      "loss": 0.0246,
      "step": 255
    },
    {
      "epoch": 0.9869879518072289,
      "grad_norm": 0.19253239035606384,
      "learning_rate": 3.404688430710955e-05,
      "loss": 0.0158,
      "step": 256
    },
    {
      "epoch": 0.9908433734939759,
      "grad_norm": 0.18241408467292786,
      "learning_rate": 3.398303151057543e-05,
      "loss": 0.0208,
      "step": 257
    },
    {
      "epoch": 0.9946987951807229,
      "grad_norm": 0.15632668137550354,
      "learning_rate": 3.391889867013134e-05,
      "loss": 0.014,
      "step": 258
    },
    {
      "epoch": 0.9985542168674699,
      "grad_norm": 0.15484513342380524,
      "learning_rate": 3.385448707019199e-05,
      "loss": 0.0098,
      "step": 259
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.44285017251968384,
      "learning_rate": 3.3789798000754924e-05,
      "loss": 0.0181,
      "step": 260
    },
    {
      "epoch": 1.003855421686747,
      "grad_norm": 0.22358648478984833,
      "learning_rate": 3.372483275737468e-05,
      "loss": 0.0123,
      "step": 261
    },
    {
      "epoch": 1.007710843373494,
      "grad_norm": 0.26415324211120605,
      "learning_rate": 3.365959264113683e-05,
      "loss": 0.0097,
      "step": 262
    },
    {
      "epoch": 1.011566265060241,
      "grad_norm": 0.16320234537124634,
      "learning_rate": 3.359407895863199e-05,
      "loss": 0.0168,
      "step": 263
    },
    {
      "epoch": 1.0154216867469879,
      "grad_norm": 0.11904699355363846,
      "learning_rate": 3.3528293021929523e-05,
      "loss": 0.011,
      "step": 264
    },
    {
      "epoch": 1.0192771084337349,
      "grad_norm": 0.19491620361804962,
      "learning_rate": 3.34622361485514e-05,
      "loss": 0.0203,
      "step": 265
    },
    {
      "epoch": 1.0231325301204819,
      "grad_norm": 0.11890948563814163,
      "learning_rate": 3.339590966144574e-05,
      "loss": 0.0039,
      "step": 266
    },
    {
      "epoch": 1.026987951807229,
      "grad_norm": 0.375266969203949,
      "learning_rate": 3.332931488896029e-05,
      "loss": 0.0108,
      "step": 267
    },
    {
      "epoch": 1.030843373493976,
      "grad_norm": 0.2763790488243103,
      "learning_rate": 3.326245316481591e-05,
      "loss": 0.0231,
      "step": 268
    },
    {
      "epoch": 1.034698795180723,
      "grad_norm": 0.13364073634147644,
      "learning_rate": 3.319532582807977e-05,
      "loss": 0.0096,
      "step": 269
    },
    {
      "epoch": 1.03855421686747,
      "grad_norm": 0.14633706212043762,
      "learning_rate": 3.31279342231386e-05,
      "loss": 0.0085,
      "step": 270
    },
    {
      "epoch": 1.0424096385542168,
      "grad_norm": 0.11920950561761856,
      "learning_rate": 3.30602796996717e-05,
      "loss": 0.0056,
      "step": 271
    },
    {
      "epoch": 1.0462650602409638,
      "grad_norm": 0.2369781881570816,
      "learning_rate": 3.299236361262401e-05,
      "loss": 0.0092,
      "step": 272
    },
    {
      "epoch": 1.0501204819277108,
      "grad_norm": 0.13494941592216492,
      "learning_rate": 3.2924187322178865e-05,
      "loss": 0.0084,
      "step": 273
    },
    {
      "epoch": 1.0539759036144578,
      "grad_norm": 0.11123005300760269,
      "learning_rate": 3.285575219373079e-05,
      "loss": 0.0058,
      "step": 274
    },
    {
      "epoch": 1.0578313253012048,
      "grad_norm": 0.18400225043296814,
      "learning_rate": 3.278705959785821e-05,
      "loss": 0.0113,
      "step": 275
    },
    {
      "epoch": 1.0616867469879518,
      "grad_norm": 0.19578677415847778,
      "learning_rate": 3.27181109102959e-05,
      "loss": 0.008,
      "step": 276
    },
    {
      "epoch": 1.0655421686746989,
      "grad_norm": 0.15978166460990906,
      "learning_rate": 3.2648907511907544e-05,
      "loss": 0.0074,
      "step": 277
    },
    {
      "epoch": 1.0693975903614459,
      "grad_norm": 0.22295606136322021,
      "learning_rate": 3.2579450788657997e-05,
      "loss": 0.033,
      "step": 278
    },
    {
      "epoch": 1.0732530120481927,
      "grad_norm": 0.11566543579101562,
      "learning_rate": 3.250974213158555e-05,
      "loss": 0.0097,
      "step": 279
    },
    {
      "epoch": 1.0771084337349397,
      "grad_norm": 0.14439509809017181,
      "learning_rate": 3.243978293677408e-05,
      "loss": 0.0099,
      "step": 280
    },
    {
      "epoch": 1.0809638554216867,
      "grad_norm": 0.21390187740325928,
      "learning_rate": 3.23695746053251e-05,
      "loss": 0.0115,
      "step": 281
    },
    {
      "epoch": 1.0848192771084337,
      "grad_norm": 0.1340283751487732,
      "learning_rate": 3.229911854332967e-05,
      "loss": 0.0112,
      "step": 282
    },
    {
      "epoch": 1.0886746987951808,
      "grad_norm": 0.08028711378574371,
      "learning_rate": 3.222841616184025e-05,
      "loss": 0.0044,
      "step": 283
    },
    {
      "epoch": 1.0925301204819278,
      "grad_norm": 0.0875745564699173,
      "learning_rate": 3.215746887684244e-05,
      "loss": 0.005,
      "step": 284
    },
    {
      "epoch": 1.0963855421686748,
      "grad_norm": 0.12632474303245544,
      "learning_rate": 3.208627810922665e-05,
      "loss": 0.0102,
      "step": 285
    },
    {
      "epoch": 1.1002409638554216,
      "grad_norm": 0.07411441951990128,
      "learning_rate": 3.201484528475958e-05,
      "loss": 0.0051,
      "step": 286
    },
    {
      "epoch": 1.1040963855421686,
      "grad_norm": 0.188252255320549,
      "learning_rate": 3.194317183405573e-05,
      "loss": 0.0088,
      "step": 287
    },
    {
      "epoch": 1.1079518072289156,
      "grad_norm": 1.1099765300750732,
      "learning_rate": 3.187125919254869e-05,
      "loss": 0.0182,
      "step": 288
    },
    {
      "epoch": 1.1118072289156626,
      "grad_norm": 0.09729217737913132,
      "learning_rate": 3.1799108800462466e-05,
      "loss": 0.0058,
      "step": 289
    },
    {
      "epoch": 1.1156626506024097,
      "grad_norm": 0.25208383798599243,
      "learning_rate": 3.172672210278254e-05,
      "loss": 0.0104,
      "step": 290
    },
    {
      "epoch": 1.1195180722891567,
      "grad_norm": 0.1512216329574585,
      "learning_rate": 3.1654100549227024e-05,
      "loss": 0.0058,
      "step": 291
    },
    {
      "epoch": 1.1233734939759037,
      "grad_norm": 0.17274995148181915,
      "learning_rate": 3.158124559421758e-05,
      "loss": 0.0059,
      "step": 292
    },
    {
      "epoch": 1.1272289156626507,
      "grad_norm": 0.32907241582870483,
      "learning_rate": 3.1508158696850275e-05,
      "loss": 0.0119,
      "step": 293
    },
    {
      "epoch": 1.1310843373493975,
      "grad_norm": 0.1462509036064148,
      "learning_rate": 3.14348413208664e-05,
      "loss": 0.0093,
      "step": 294
    },
    {
      "epoch": 1.1349397590361445,
      "grad_norm": 0.23990927636623383,
      "learning_rate": 3.136129493462312e-05,
      "loss": 0.0095,
      "step": 295
    },
    {
      "epoch": 1.1387951807228915,
      "grad_norm": 0.13981395959854126,
      "learning_rate": 3.12875210110641e-05,
      "loss": 0.006,
      "step": 296
    },
    {
      "epoch": 1.1426506024096386,
      "grad_norm": 0.10694962739944458,
      "learning_rate": 3.121352102768998e-05,
      "loss": 0.005,
      "step": 297
    },
    {
      "epoch": 1.1465060240963856,
      "grad_norm": 0.12599286437034607,
      "learning_rate": 3.113929646652879e-05,
      "loss": 0.0057,
      "step": 298
    },
    {
      "epoch": 1.1503614457831326,
      "grad_norm": 0.09407994896173477,
      "learning_rate": 3.106484881410628e-05,
      "loss": 0.0056,
      "step": 299
    },
    {
      "epoch": 1.1542168674698796,
      "grad_norm": 0.7781950831413269,
      "learning_rate": 3.0990179561416124e-05,
      "loss": 0.017,
      "step": 300
    },
    {
      "epoch": 1.1580722891566264,
      "grad_norm": 0.12835298478603363,
      "learning_rate": 3.091529020389009e-05,
      "loss": 0.0053,
      "step": 301
    },
    {
      "epoch": 1.1619277108433734,
      "grad_norm": 0.7488322854042053,
      "learning_rate": 3.0840182241368083e-05,
      "loss": 0.011,
      "step": 302
    },
    {
      "epoch": 1.1657831325301204,
      "grad_norm": 0.16700711846351624,
      "learning_rate": 3.076485717806808e-05,
      "loss": 0.0082,
      "step": 303
    },
    {
      "epoch": 1.1696385542168675,
      "grad_norm": 0.17361144721508026,
      "learning_rate": 3.0689316522556026e-05,
      "loss": 0.0097,
      "step": 304
    },
    {
      "epoch": 1.1734939759036145,
      "grad_norm": 0.08619613200426102,
      "learning_rate": 3.061356178771564e-05,
      "loss": 0.0082,
      "step": 305
    },
    {
      "epoch": 1.1773493975903615,
      "grad_norm": 0.15666592121124268,
      "learning_rate": 3.053759449071807e-05,
      "loss": 0.0106,
      "step": 306
    },
    {
      "epoch": 1.1812048192771085,
      "grad_norm": 0.2268570214509964,
      "learning_rate": 3.0461416152991555e-05,
      "loss": 0.0082,
      "step": 307
    },
    {
      "epoch": 1.1850602409638555,
      "grad_norm": 0.17297455668449402,
      "learning_rate": 3.038502830019092e-05,
      "loss": 0.0105,
      "step": 308
    },
    {
      "epoch": 1.1889156626506023,
      "grad_norm": 0.18929991126060486,
      "learning_rate": 3.0308432462167045e-05,
      "loss": 0.0076,
      "step": 309
    },
    {
      "epoch": 1.1927710843373494,
      "grad_norm": 0.25152456760406494,
      "learning_rate": 3.0231630172936205e-05,
      "loss": 0.0072,
      "step": 310
    },
    {
      "epoch": 1.1966265060240964,
      "grad_norm": 0.6995307207107544,
      "learning_rate": 3.015462297064936e-05,
      "loss": 0.0155,
      "step": 311
    },
    {
      "epoch": 1.2004819277108434,
      "grad_norm": 0.14346441626548767,
      "learning_rate": 3.0077412397561345e-05,
      "loss": 0.0194,
      "step": 312
    },
    {
      "epoch": 1.2043373493975904,
      "grad_norm": 0.2826595604419708,
      "learning_rate": 3.0000000000000004e-05,
      "loss": 0.0198,
      "step": 313
    },
    {
      "epoch": 1.2081927710843374,
      "grad_norm": 0.1361214816570282,
      "learning_rate": 2.992238732833518e-05,
      "loss": 0.007,
      "step": 314
    },
    {
      "epoch": 1.2120481927710842,
      "grad_norm": 0.13326340913772583,
      "learning_rate": 2.98445759369477e-05,
      "loss": 0.0116,
      "step": 315
    },
    {
      "epoch": 1.2159036144578312,
      "grad_norm": 0.09617264568805695,
      "learning_rate": 2.9766567384198228e-05,
      "loss": 0.0082,
      "step": 316
    },
    {
      "epoch": 1.2197590361445783,
      "grad_norm": 0.1366315484046936,
      "learning_rate": 2.9688363232396056e-05,
      "loss": 0.0075,
      "step": 317
    },
    {
      "epoch": 1.2236144578313253,
      "grad_norm": 0.696820080280304,
      "learning_rate": 2.960996504776783e-05,
      "loss": 0.0087,
      "step": 318
    },
    {
      "epoch": 1.2274698795180723,
      "grad_norm": 0.42925387620925903,
      "learning_rate": 2.9531374400426158e-05,
      "loss": 0.0277,
      "step": 319
    },
    {
      "epoch": 1.2313253012048193,
      "grad_norm": 0.16131658852100372,
      "learning_rate": 2.945259286433819e-05,
      "loss": 0.0097,
      "step": 320
    },
    {
      "epoch": 1.2351807228915663,
      "grad_norm": 0.1256534606218338,
      "learning_rate": 2.9373622017294075e-05,
      "loss": 0.0086,
      "step": 321
    },
    {
      "epoch": 1.2390361445783133,
      "grad_norm": 0.15791697800159454,
      "learning_rate": 2.9294463440875375e-05,
      "loss": 0.0119,
      "step": 322
    },
    {
      "epoch": 1.2428915662650604,
      "grad_norm": 0.12754394114017487,
      "learning_rate": 2.9215118720423375e-05,
      "loss": 0.0071,
      "step": 323
    },
    {
      "epoch": 1.2467469879518072,
      "grad_norm": 0.9191262125968933,
      "learning_rate": 2.913558944500736e-05,
      "loss": 0.0097,
      "step": 324
    },
    {
      "epoch": 1.2506024096385542,
      "grad_norm": 0.10129745304584503,
      "learning_rate": 2.9055877207392752e-05,
      "loss": 0.0162,
      "step": 325
    },
    {
      "epoch": 1.2544578313253012,
      "grad_norm": 0.12669669091701508,
      "learning_rate": 2.897598360400925e-05,
      "loss": 0.0051,
      "step": 326
    },
    {
      "epoch": 1.2583132530120482,
      "grad_norm": 0.1282287985086441,
      "learning_rate": 2.8895910234918828e-05,
      "loss": 0.0069,
      "step": 327
    },
    {
      "epoch": 1.2621686746987952,
      "grad_norm": 0.10596070438623428,
      "learning_rate": 2.8815658703783715e-05,
      "loss": 0.0044,
      "step": 328
    },
    {
      "epoch": 1.266024096385542,
      "grad_norm": 0.3934219777584076,
      "learning_rate": 2.873523061783426e-05,
      "loss": 0.0122,
      "step": 329
    },
    {
      "epoch": 1.269879518072289,
      "grad_norm": 0.13733914494514465,
      "learning_rate": 2.8654627587836724e-05,
      "loss": 0.0052,
      "step": 330
    },
    {
      "epoch": 1.273734939759036,
      "grad_norm": 0.21395160257816315,
      "learning_rate": 2.8573851228061084e-05,
      "loss": 0.0067,
      "step": 331
    },
    {
      "epoch": 1.277590361445783,
      "grad_norm": 0.23374490439891815,
      "learning_rate": 2.8492903156248642e-05,
      "loss": 0.0111,
      "step": 332
    },
    {
      "epoch": 1.28144578313253,
      "grad_norm": 0.14450624585151672,
      "learning_rate": 2.8411784993579633e-05,
      "loss": 0.0097,
      "step": 333
    },
    {
      "epoch": 1.2853012048192771,
      "grad_norm": 0.12868821620941162,
      "learning_rate": 2.8330498364640803e-05,
      "loss": 0.0088,
      "step": 334
    },
    {
      "epoch": 1.2891566265060241,
      "grad_norm": 0.2879028618335724,
      "learning_rate": 2.8249044897392814e-05,
      "loss": 0.0191,
      "step": 335
    },
    {
      "epoch": 1.2930120481927712,
      "grad_norm": 0.10924389213323593,
      "learning_rate": 2.8167426223137682e-05,
      "loss": 0.0063,
      "step": 336
    },
    {
      "epoch": 1.2968674698795182,
      "grad_norm": 0.14800770580768585,
      "learning_rate": 2.80856439764861e-05,
      "loss": 0.01,
      "step": 337
    },
    {
      "epoch": 1.3007228915662652,
      "grad_norm": 0.1235174834728241,
      "learning_rate": 2.8003699795324674e-05,
      "loss": 0.0044,
      "step": 338
    },
    {
      "epoch": 1.304578313253012,
      "grad_norm": 0.2660425305366516,
      "learning_rate": 2.792159532078314e-05,
      "loss": 0.0068,
      "step": 339
    },
    {
      "epoch": 1.308433734939759,
      "grad_norm": 0.10823825001716614,
      "learning_rate": 2.7839332197201505e-05,
      "loss": 0.0041,
      "step": 340
    },
    {
      "epoch": 1.312289156626506,
      "grad_norm": 0.9430158734321594,
      "learning_rate": 2.77569120720971e-05,
      "loss": 0.0141,
      "step": 341
    },
    {
      "epoch": 1.316144578313253,
      "grad_norm": 0.1147020012140274,
      "learning_rate": 2.7674336596131596e-05,
      "loss": 0.0075,
      "step": 342
    },
    {
      "epoch": 1.32,
      "grad_norm": 0.1446715146303177,
      "learning_rate": 2.7591607423077932e-05,
      "loss": 0.0094,
      "step": 343
    },
    {
      "epoch": 1.3238554216867469,
      "grad_norm": 0.08478965610265732,
      "learning_rate": 2.75087262097872e-05,
      "loss": 0.0034,
      "step": 344
    },
    {
      "epoch": 1.3277108433734939,
      "grad_norm": 0.08715037256479263,
      "learning_rate": 2.7425694616155474e-05,
      "loss": 0.0052,
      "step": 345
    },
    {
      "epoch": 1.331566265060241,
      "grad_norm": 2.43072772026062,
      "learning_rate": 2.7342514305090554e-05,
      "loss": 0.0223,
      "step": 346
    },
    {
      "epoch": 1.335421686746988,
      "grad_norm": 0.1595911830663681,
      "learning_rate": 2.7259186942478656e-05,
      "loss": 0.0042,
      "step": 347
    },
    {
      "epoch": 1.339277108433735,
      "grad_norm": 0.2817286550998688,
      "learning_rate": 2.717571419715107e-05,
      "loss": 0.027,
      "step": 348
    },
    {
      "epoch": 1.343132530120482,
      "grad_norm": 0.5103950500488281,
      "learning_rate": 2.7092097740850712e-05,
      "loss": 0.0085,
      "step": 349
    },
    {
      "epoch": 1.346987951807229,
      "grad_norm": 1.2923753261566162,
      "learning_rate": 2.700833924819868e-05,
      "loss": 0.01,
      "step": 350
    },
    {
      "epoch": 1.350843373493976,
      "grad_norm": 0.16906492412090302,
      "learning_rate": 2.692444039666066e-05,
      "loss": 0.0109,
      "step": 351
    },
    {
      "epoch": 1.354698795180723,
      "grad_norm": 0.37144047021865845,
      "learning_rate": 2.684040286651338e-05,
      "loss": 0.0113,
      "step": 352
    },
    {
      "epoch": 1.3585542168674698,
      "grad_norm": 0.3601152300834656,
      "learning_rate": 2.6756228340810946e-05,
      "loss": 0.0114,
      "step": 353
    },
    {
      "epoch": 1.3624096385542168,
      "grad_norm": 0.5444064140319824,
      "learning_rate": 2.6671918505351132e-05,
      "loss": 0.0086,
      "step": 354
    },
    {
      "epoch": 1.3662650602409638,
      "grad_norm": 0.14150206744670868,
      "learning_rate": 2.6587475048641596e-05,
      "loss": 0.0043,
      "step": 355
    },
    {
      "epoch": 1.3701204819277109,
      "grad_norm": 0.09511060267686844,
      "learning_rate": 2.6502899661866104e-05,
      "loss": 0.0062,
      "step": 356
    },
    {
      "epoch": 1.3739759036144579,
      "grad_norm": 0.16923631727695465,
      "learning_rate": 2.6418194038850634e-05,
      "loss": 0.0157,
      "step": 357
    },
    {
      "epoch": 1.377831325301205,
      "grad_norm": 0.13884925842285156,
      "learning_rate": 2.6333359876029455e-05,
      "loss": 0.0058,
      "step": 358
    },
    {
      "epoch": 1.3816867469879517,
      "grad_norm": 0.1783420443534851,
      "learning_rate": 2.624839887241115e-05,
      "loss": 0.0071,
      "step": 359
    },
    {
      "epoch": 1.3855421686746987,
      "grad_norm": 0.15586194396018982,
      "learning_rate": 2.6163312729544616e-05,
      "loss": 0.0079,
      "step": 360
    },
    {
      "epoch": 1.3893975903614457,
      "grad_norm": 0.22028474509716034,
      "learning_rate": 2.607810315148494e-05,
      "loss": 0.0078,
      "step": 361
    },
    {
      "epoch": 1.3932530120481927,
      "grad_norm": 0.14147606492042542,
      "learning_rate": 2.5992771844759315e-05,
      "loss": 0.0078,
      "step": 362
    },
    {
      "epoch": 1.3971084337349398,
      "grad_norm": 0.11384939402341843,
      "learning_rate": 2.5907320518332827e-05,
      "loss": 0.0067,
      "step": 363
    },
    {
      "epoch": 1.4009638554216868,
      "grad_norm": 0.6917237639427185,
      "learning_rate": 2.582175088357426e-05,
      "loss": 0.0097,
      "step": 364
    },
    {
      "epoch": 1.4048192771084338,
      "grad_norm": 0.26896387338638306,
      "learning_rate": 2.5736064654221808e-05,
      "loss": 0.0109,
      "step": 365
    },
    {
      "epoch": 1.4086746987951808,
      "grad_norm": 0.1673029661178589,
      "learning_rate": 2.5650263546348743e-05,
      "loss": 0.0082,
      "step": 366
    },
    {
      "epoch": 1.4125301204819278,
      "grad_norm": 0.1272229552268982,
      "learning_rate": 2.5564349278329056e-05,
      "loss": 0.0101,
      "step": 367
    },
    {
      "epoch": 1.4163855421686746,
      "grad_norm": 0.1690957248210907,
      "learning_rate": 2.547832357080305e-05,
      "loss": 0.0125,
      "step": 368
    },
    {
      "epoch": 1.4202409638554216,
      "grad_norm": 0.20461033284664154,
      "learning_rate": 2.539218814664288e-05,
      "loss": 0.0109,
      "step": 369
    },
    {
      "epoch": 1.4240963855421687,
      "grad_norm": 0.28373226523399353,
      "learning_rate": 2.5305944730918017e-05,
      "loss": 0.0152,
      "step": 370
    },
    {
      "epoch": 1.4279518072289157,
      "grad_norm": 0.2010459452867508,
      "learning_rate": 2.521959505086075e-05,
      "loss": 0.0067,
      "step": 371
    },
    {
      "epoch": 1.4318072289156627,
      "grad_norm": 0.28684207797050476,
      "learning_rate": 2.5133140835831537e-05,
      "loss": 0.0092,
      "step": 372
    },
    {
      "epoch": 1.4356626506024097,
      "grad_norm": 0.21436753869056702,
      "learning_rate": 2.5046583817284437e-05,
      "loss": 0.0095,
      "step": 373
    },
    {
      "epoch": 1.4395180722891565,
      "grad_norm": 0.12941661477088928,
      "learning_rate": 2.495992572873237e-05,
      "loss": 0.0063,
      "step": 374
    },
    {
      "epoch": 1.4433734939759035,
      "grad_norm": 0.15036840736865997,
      "learning_rate": 2.487316830571244e-05,
      "loss": 0.0125,
      "step": 375
    },
    {
      "epoch": 1.4472289156626506,
      "grad_norm": 0.1019950732588768,
      "learning_rate": 2.4786313285751158e-05,
      "loss": 0.004,
      "step": 376
    },
    {
      "epoch": 1.4510843373493976,
      "grad_norm": 0.12085995823144913,
      "learning_rate": 2.4699362408329646e-05,
      "loss": 0.0051,
      "step": 377
    },
    {
      "epoch": 1.4549397590361446,
      "grad_norm": 0.16224437952041626,
      "learning_rate": 2.4612317414848804e-05,
      "loss": 0.0057,
      "step": 378
    },
    {
      "epoch": 1.4587951807228916,
      "grad_norm": 0.49838605523109436,
      "learning_rate": 2.4525180048594452e-05,
      "loss": 0.0081,
      "step": 379
    },
    {
      "epoch": 1.4626506024096386,
      "grad_norm": 0.13203397393226624,
      "learning_rate": 2.4437952054702354e-05,
      "loss": 0.0084,
      "step": 380
    },
    {
      "epoch": 1.4665060240963856,
      "grad_norm": 0.2933284044265747,
      "learning_rate": 2.435063518012335e-05,
      "loss": 0.0167,
      "step": 381
    },
    {
      "epoch": 1.4703614457831327,
      "grad_norm": 0.16203272342681885,
      "learning_rate": 2.4263231173588323e-05,
      "loss": 0.0076,
      "step": 382
    },
    {
      "epoch": 1.4742168674698795,
      "grad_norm": 0.11736200004816055,
      "learning_rate": 2.4175741785573177e-05,
      "loss": 0.007,
      "step": 383
    },
    {
      "epoch": 1.4780722891566265,
      "grad_norm": 0.11185964941978455,
      "learning_rate": 2.4088168768263782e-05,
      "loss": 0.0079,
      "step": 384
    },
    {
      "epoch": 1.4819277108433735,
      "grad_norm": 0.1776389628648758,
      "learning_rate": 2.4000513875520892e-05,
      "loss": 0.0087,
      "step": 385
    },
    {
      "epoch": 1.4857831325301205,
      "grad_norm": 0.11572884768247604,
      "learning_rate": 2.3912778862845015e-05,
      "loss": 0.0069,
      "step": 386
    },
    {
      "epoch": 1.4896385542168675,
      "grad_norm": 0.12890145182609558,
      "learning_rate": 2.3824965487341247e-05,
      "loss": 0.0075,
      "step": 387
    },
    {
      "epoch": 1.4934939759036143,
      "grad_norm": 0.19975218176841736,
      "learning_rate": 2.3737075507684103e-05,
      "loss": 0.0101,
      "step": 388
    },
    {
      "epoch": 1.4973493975903613,
      "grad_norm": 0.32604387402534485,
      "learning_rate": 2.3649110684082258e-05,
      "loss": 0.0072,
      "step": 389
    },
    {
      "epoch": 1.5012048192771084,
      "grad_norm": 0.12934386730194092,
      "learning_rate": 2.3561072778243335e-05,
      "loss": 0.0078,
      "step": 390
    },
    {
      "epoch": 1.5050602409638554,
      "grad_norm": 0.11547217518091202,
      "learning_rate": 2.3472963553338614e-05,
      "loss": 0.0071,
      "step": 391
    },
    {
      "epoch": 1.5089156626506024,
      "grad_norm": 0.09941789507865906,
      "learning_rate": 2.3384784773967675e-05,
      "loss": 0.0123,
      "step": 392
    },
    {
      "epoch": 1.5127710843373494,
      "grad_norm": 0.7121322751045227,
      "learning_rate": 2.3296538206123134e-05,
      "loss": 0.0065,
      "step": 393
    },
    {
      "epoch": 1.5166265060240964,
      "grad_norm": 0.140323668718338,
      "learning_rate": 2.3208225617155206e-05,
      "loss": 0.0063,
      "step": 394
    },
    {
      "epoch": 1.5204819277108435,
      "grad_norm": 0.15863126516342163,
      "learning_rate": 2.311984877573636e-05,
      "loss": 0.0136,
      "step": 395
    },
    {
      "epoch": 1.5243373493975905,
      "grad_norm": 0.1590386927127838,
      "learning_rate": 2.3031409451825852e-05,
      "loss": 0.0146,
      "step": 396
    },
    {
      "epoch": 1.5281927710843375,
      "grad_norm": 0.12217574566602707,
      "learning_rate": 2.2942909416634326e-05,
      "loss": 0.0066,
      "step": 397
    },
    {
      "epoch": 1.5320481927710843,
      "grad_norm": 0.11038487404584885,
      "learning_rate": 2.285435044258829e-05,
      "loss": 0.0059,
      "step": 398
    },
    {
      "epoch": 1.5359036144578313,
      "grad_norm": 0.28101980686187744,
      "learning_rate": 2.2765734303294666e-05,
      "loss": 0.0091,
      "step": 399
    },
    {
      "epoch": 1.5397590361445783,
      "grad_norm": 0.136636421084404,
      "learning_rate": 2.2677062773505235e-05,
      "loss": 0.0045,
      "step": 400
    },
    {
      "epoch": 1.5436144578313253,
      "grad_norm": 0.09847535192966461,
      "learning_rate": 2.2588337629081107e-05,
      "loss": 0.0061,
      "step": 401
    },
    {
      "epoch": 1.5474698795180721,
      "grad_norm": 0.13071104884147644,
      "learning_rate": 2.249956064695717e-05,
      "loss": 0.0059,
      "step": 402
    },
    {
      "epoch": 1.5513253012048192,
      "grad_norm": 0.09452104568481445,
      "learning_rate": 2.2410733605106462e-05,
      "loss": 0.0049,
      "step": 403
    },
    {
      "epoch": 1.5551807228915662,
      "grad_norm": 0.12455491721630096,
      "learning_rate": 2.2321858282504606e-05,
      "loss": 0.0055,
      "step": 404
    },
    {
      "epoch": 1.5590361445783132,
      "grad_norm": 0.195816308259964,
      "learning_rate": 2.2232936459094158e-05,
      "loss": 0.0076,
      "step": 405
    },
    {
      "epoch": 1.5628915662650602,
      "grad_norm": 0.1624608188867569,
      "learning_rate": 2.214396991574896e-05,
      "loss": 0.0117,
      "step": 406
    },
    {
      "epoch": 1.5667469879518072,
      "grad_norm": 0.07928434759378433,
      "learning_rate": 2.205496043423849e-05,
      "loss": 0.0043,
      "step": 407
    },
    {
      "epoch": 1.5706024096385542,
      "grad_norm": 0.44694873690605164,
      "learning_rate": 2.1965909797192143e-05,
      "loss": 0.0082,
      "step": 408
    },
    {
      "epoch": 1.5744578313253013,
      "grad_norm": 0.1101914644241333,
      "learning_rate": 2.1876819788063586e-05,
      "loss": 0.0063,
      "step": 409
    },
    {
      "epoch": 1.5783132530120483,
      "grad_norm": 0.10395577549934387,
      "learning_rate": 2.1787692191094986e-05,
      "loss": 0.0058,
      "step": 410
    },
    {
      "epoch": 1.5821686746987953,
      "grad_norm": 0.14004680514335632,
      "learning_rate": 2.16985287912813e-05,
      "loss": 0.0062,
      "step": 411
    },
    {
      "epoch": 1.5860240963855423,
      "grad_norm": 0.15407921373844147,
      "learning_rate": 2.1609331374334523e-05,
      "loss": 0.007,
      "step": 412
    },
    {
      "epoch": 1.589879518072289,
      "grad_norm": 0.10258454084396362,
      "learning_rate": 2.1520101726647922e-05,
      "loss": 0.0115,
      "step": 413
    },
    {
      "epoch": 1.5937349397590361,
      "grad_norm": 0.13689640164375305,
      "learning_rate": 2.1430841635260274e-05,
      "loss": 0.025,
      "step": 414
    },
    {
      "epoch": 1.5975903614457831,
      "grad_norm": 0.10260256379842758,
      "learning_rate": 2.1341552887820048e-05,
      "loss": 0.0048,
      "step": 415
    },
    {
      "epoch": 1.6014457831325302,
      "grad_norm": 0.17679180204868317,
      "learning_rate": 2.125223727254964e-05,
      "loss": 0.0209,
      "step": 416
    },
    {
      "epoch": 1.605301204819277,
      "grad_norm": 0.16429021954536438,
      "learning_rate": 2.1162896578209517e-05,
      "loss": 0.018,
      "step": 417
    },
    {
      "epoch": 1.609156626506024,
      "grad_norm": 0.5342340469360352,
      "learning_rate": 2.1073532594062432e-05,
      "loss": 0.0063,
      "step": 418
    },
    {
      "epoch": 1.613012048192771,
      "grad_norm": 0.23090562224388123,
      "learning_rate": 2.0984147109837564e-05,
      "loss": 0.0043,
      "step": 419
    },
    {
      "epoch": 1.616867469879518,
      "grad_norm": 0.09459249675273895,
      "learning_rate": 2.0894741915694685e-05,
      "loss": 0.0051,
      "step": 420
    },
    {
      "epoch": 1.620722891566265,
      "grad_norm": 0.20056606829166412,
      "learning_rate": 2.0805318802188307e-05,
      "loss": 0.0108,
      "step": 421
    },
    {
      "epoch": 1.624578313253012,
      "grad_norm": 0.5049247145652771,
      "learning_rate": 2.071587956023181e-05,
      "loss": 0.0121,
      "step": 422
    },
    {
      "epoch": 1.628433734939759,
      "grad_norm": 0.10519257187843323,
      "learning_rate": 2.0626425981061608e-05,
      "loss": 0.0053,
      "step": 423
    },
    {
      "epoch": 1.632289156626506,
      "grad_norm": 0.1042436957359314,
      "learning_rate": 2.053695985620123e-05,
      "loss": 0.0082,
      "step": 424
    },
    {
      "epoch": 1.636144578313253,
      "grad_norm": 0.15873271226882935,
      "learning_rate": 2.0447482977425465e-05,
      "loss": 0.0088,
      "step": 425
    },
    {
      "epoch": 1.6400000000000001,
      "grad_norm": 0.09151361882686615,
      "learning_rate": 2.0357997136724486e-05,
      "loss": 0.0133,
      "step": 426
    },
    {
      "epoch": 1.6438554216867471,
      "grad_norm": 0.08998992294073105,
      "learning_rate": 2.0268504126267952e-05,
      "loss": 0.0049,
      "step": 427
    },
    {
      "epoch": 1.647710843373494,
      "grad_norm": 0.08661992847919464,
      "learning_rate": 2.0179005738369098e-05,
      "loss": 0.0051,
      "step": 428
    },
    {
      "epoch": 1.651566265060241,
      "grad_norm": 0.18930229544639587,
      "learning_rate": 2.008950376544887e-05,
      "loss": 0.0058,
      "step": 429
    },
    {
      "epoch": 1.655421686746988,
      "grad_norm": 0.10838332027196884,
      "learning_rate": 2e-05,
      "loss": 0.0099,
      "step": 430
    },
    {
      "epoch": 1.659277108433735,
      "grad_norm": 0.2523624300956726,
      "learning_rate": 1.9910496234551132e-05,
      "loss": 0.0149,
      "step": 431
    },
    {
      "epoch": 1.6631325301204818,
      "grad_norm": 0.10037925094366074,
      "learning_rate": 1.982099426163091e-05,
      "loss": 0.0061,
      "step": 432
    },
    {
      "epoch": 1.6669879518072288,
      "grad_norm": 0.0841156542301178,
      "learning_rate": 1.9731495873732055e-05,
      "loss": 0.0059,
      "step": 433
    },
    {
      "epoch": 1.6708433734939758,
      "grad_norm": 0.08965729922056198,
      "learning_rate": 1.9642002863275517e-05,
      "loss": 0.0051,
      "step": 434
    },
    {
      "epoch": 1.6746987951807228,
      "grad_norm": 0.10749141871929169,
      "learning_rate": 1.9552517022574542e-05,
      "loss": 0.0102,
      "step": 435
    },
    {
      "epoch": 1.6785542168674699,
      "grad_norm": 0.14290869235992432,
      "learning_rate": 1.946304014379878e-05,
      "loss": 0.008,
      "step": 436
    },
    {
      "epoch": 1.6824096385542169,
      "grad_norm": 0.19874858856201172,
      "learning_rate": 1.93735740189384e-05,
      "loss": 0.0089,
      "step": 437
    },
    {
      "epoch": 1.686265060240964,
      "grad_norm": 0.13204169273376465,
      "learning_rate": 1.9284120439768192e-05,
      "loss": 0.0142,
      "step": 438
    },
    {
      "epoch": 1.690120481927711,
      "grad_norm": 0.3346060812473297,
      "learning_rate": 1.9194681197811703e-05,
      "loss": 0.0047,
      "step": 439
    },
    {
      "epoch": 1.693975903614458,
      "grad_norm": 0.3231620788574219,
      "learning_rate": 1.910525808430532e-05,
      "loss": 0.0082,
      "step": 440
    },
    {
      "epoch": 1.697831325301205,
      "grad_norm": 0.17857253551483154,
      "learning_rate": 1.901585289016244e-05,
      "loss": 0.0035,
      "step": 441
    },
    {
      "epoch": 1.701686746987952,
      "grad_norm": 0.3094499707221985,
      "learning_rate": 1.8926467405937567e-05,
      "loss": 0.015,
      "step": 442
    },
    {
      "epoch": 1.7055421686746988,
      "grad_norm": 0.11924882233142853,
      "learning_rate": 1.8837103421790486e-05,
      "loss": 0.0198,
      "step": 443
    },
    {
      "epoch": 1.7093975903614458,
      "grad_norm": 0.15662431716918945,
      "learning_rate": 1.8747762727450363e-05,
      "loss": 0.0088,
      "step": 444
    },
    {
      "epoch": 1.7132530120481928,
      "grad_norm": 0.12857934832572937,
      "learning_rate": 1.8658447112179952e-05,
      "loss": 0.0177,
      "step": 445
    },
    {
      "epoch": 1.7171084337349396,
      "grad_norm": 0.0960603877902031,
      "learning_rate": 1.8569158364739736e-05,
      "loss": 0.0042,
      "step": 446
    },
    {
      "epoch": 1.7209638554216866,
      "grad_norm": 0.08039736747741699,
      "learning_rate": 1.8479898273352084e-05,
      "loss": 0.0043,
      "step": 447
    },
    {
      "epoch": 1.7248192771084336,
      "grad_norm": 0.0857943594455719,
      "learning_rate": 1.8390668625665483e-05,
      "loss": 0.0031,
      "step": 448
    },
    {
      "epoch": 1.7286746987951807,
      "grad_norm": 0.14046747982501984,
      "learning_rate": 1.83014712087187e-05,
      "loss": 0.0129,
      "step": 449
    },
    {
      "epoch": 1.7325301204819277,
      "grad_norm": 0.1024412289261818,
      "learning_rate": 1.821230780890502e-05,
      "loss": 0.0101,
      "step": 450
    },
    {
      "epoch": 1.7363855421686747,
      "grad_norm": 0.08862444758415222,
      "learning_rate": 1.8123180211936417e-05,
      "loss": 0.0069,
      "step": 451
    },
    {
      "epoch": 1.7402409638554217,
      "grad_norm": 0.06873127818107605,
      "learning_rate": 1.803409020280786e-05,
      "loss": 0.004,
      "step": 452
    },
    {
      "epoch": 1.7440963855421687,
      "grad_norm": 0.0650935024023056,
      "learning_rate": 1.794503956576152e-05,
      "loss": 0.0027,
      "step": 453
    },
    {
      "epoch": 1.7479518072289157,
      "grad_norm": 0.10328482836484909,
      "learning_rate": 1.7856030084251045e-05,
      "loss": 0.0047,
      "step": 454
    },
    {
      "epoch": 1.7518072289156628,
      "grad_norm": 0.14203296601772308,
      "learning_rate": 1.776706354090585e-05,
      "loss": 0.0081,
      "step": 455
    },
    {
      "epoch": 1.7556626506024098,
      "grad_norm": 0.08143935352563858,
      "learning_rate": 1.7678141717495394e-05,
      "loss": 0.0045,
      "step": 456
    },
    {
      "epoch": 1.7595180722891566,
      "grad_norm": 0.11309202760457993,
      "learning_rate": 1.758926639489354e-05,
      "loss": 0.0159,
      "step": 457
    },
    {
      "epoch": 1.7633734939759036,
      "grad_norm": 0.10518789291381836,
      "learning_rate": 1.7500439353042834e-05,
      "loss": 0.0126,
      "step": 458
    },
    {
      "epoch": 1.7672289156626506,
      "grad_norm": 0.09407763183116913,
      "learning_rate": 1.7411662370918893e-05,
      "loss": 0.0046,
      "step": 459
    },
    {
      "epoch": 1.7710843373493976,
      "grad_norm": 0.09223105013370514,
      "learning_rate": 1.7322937226494775e-05,
      "loss": 0.0056,
      "step": 460
    },
    {
      "epoch": 1.7749397590361444,
      "grad_norm": 0.10010652989149094,
      "learning_rate": 1.7234265696705344e-05,
      "loss": 0.0078,
      "step": 461
    },
    {
      "epoch": 1.7787951807228914,
      "grad_norm": 0.09317822009325027,
      "learning_rate": 1.7145649557411716e-05,
      "loss": 0.0043,
      "step": 462
    },
    {
      "epoch": 1.7826506024096385,
      "grad_norm": 0.14171656966209412,
      "learning_rate": 1.7057090583365678e-05,
      "loss": 0.0073,
      "step": 463
    },
    {
      "epoch": 1.7865060240963855,
      "grad_norm": 0.09350872039794922,
      "learning_rate": 1.6968590548174155e-05,
      "loss": 0.0052,
      "step": 464
    },
    {
      "epoch": 1.7903614457831325,
      "grad_norm": 0.13011880218982697,
      "learning_rate": 1.6880151224263646e-05,
      "loss": 0.0119,
      "step": 465
    },
    {
      "epoch": 1.7942168674698795,
      "grad_norm": 0.12144693732261658,
      "learning_rate": 1.6791774382844794e-05,
      "loss": 0.0108,
      "step": 466
    },
    {
      "epoch": 1.7980722891566265,
      "grad_norm": 0.5980355739593506,
      "learning_rate": 1.6703461793876876e-05,
      "loss": 0.0454,
      "step": 467
    },
    {
      "epoch": 1.8019277108433736,
      "grad_norm": 0.1022690162062645,
      "learning_rate": 1.6615215226032332e-05,
      "loss": 0.0046,
      "step": 468
    },
    {
      "epoch": 1.8057831325301206,
      "grad_norm": 0.09623578935861588,
      "learning_rate": 1.6527036446661396e-05,
      "loss": 0.0064,
      "step": 469
    },
    {
      "epoch": 1.8096385542168676,
      "grad_norm": 0.5450147390365601,
      "learning_rate": 1.643892722175667e-05,
      "loss": 0.0103,
      "step": 470
    },
    {
      "epoch": 1.8134939759036146,
      "grad_norm": 0.18981006741523743,
      "learning_rate": 1.635088931591775e-05,
      "loss": 0.0065,
      "step": 471
    },
    {
      "epoch": 1.8173493975903614,
      "grad_norm": 0.14748261868953705,
      "learning_rate": 1.6262924492315907e-05,
      "loss": 0.0198,
      "step": 472
    },
    {
      "epoch": 1.8212048192771084,
      "grad_norm": 0.11080063134431839,
      "learning_rate": 1.6175034512658753e-05,
      "loss": 0.0071,
      "step": 473
    },
    {
      "epoch": 1.8250602409638554,
      "grad_norm": 0.1321505457162857,
      "learning_rate": 1.6087221137154995e-05,
      "loss": 0.0115,
      "step": 474
    },
    {
      "epoch": 1.8289156626506025,
      "grad_norm": 0.1794983297586441,
      "learning_rate": 1.5999486124479115e-05,
      "loss": 0.0204,
      "step": 475
    },
    {
      "epoch": 1.8327710843373493,
      "grad_norm": 0.2090386003255844,
      "learning_rate": 1.591183123173622e-05,
      "loss": 0.0086,
      "step": 476
    },
    {
      "epoch": 1.8366265060240963,
      "grad_norm": 0.1864124834537506,
      "learning_rate": 1.5824258214426833e-05,
      "loss": 0.0037,
      "step": 477
    },
    {
      "epoch": 1.8404819277108433,
      "grad_norm": 0.0974712073802948,
      "learning_rate": 1.5736768826411683e-05,
      "loss": 0.0066,
      "step": 478
    },
    {
      "epoch": 1.8443373493975903,
      "grad_norm": 0.13470065593719482,
      "learning_rate": 1.5649364819876655e-05,
      "loss": 0.0082,
      "step": 479
    },
    {
      "epoch": 1.8481927710843373,
      "grad_norm": 0.15067611634731293,
      "learning_rate": 1.5562047945297652e-05,
      "loss": 0.0067,
      "step": 480
    },
    {
      "epoch": 1.8520481927710843,
      "grad_norm": 0.1268431693315506,
      "learning_rate": 1.547481995140556e-05,
      "loss": 0.0076,
      "step": 481
    },
    {
      "epoch": 1.8559036144578314,
      "grad_norm": 0.19485311210155487,
      "learning_rate": 1.53876825851512e-05,
      "loss": 0.0043,
      "step": 482
    },
    {
      "epoch": 1.8597590361445784,
      "grad_norm": 0.08730349689722061,
      "learning_rate": 1.5300637591670357e-05,
      "loss": 0.0081,
      "step": 483
    },
    {
      "epoch": 1.8636144578313254,
      "grad_norm": 0.10607759654521942,
      "learning_rate": 1.5213686714248852e-05,
      "loss": 0.0084,
      "step": 484
    },
    {
      "epoch": 1.8674698795180724,
      "grad_norm": 0.09020885825157166,
      "learning_rate": 1.5126831694287564e-05,
      "loss": 0.0061,
      "step": 485
    },
    {
      "epoch": 1.8713253012048194,
      "grad_norm": 0.16199110448360443,
      "learning_rate": 1.5040074271267633e-05,
      "loss": 0.0125,
      "step": 486
    },
    {
      "epoch": 1.8751807228915662,
      "grad_norm": 0.07592602074146271,
      "learning_rate": 1.4953416182715566e-05,
      "loss": 0.0041,
      "step": 487
    },
    {
      "epoch": 1.8790361445783132,
      "grad_norm": 0.3946191668510437,
      "learning_rate": 1.4866859164168466e-05,
      "loss": 0.0084,
      "step": 488
    },
    {
      "epoch": 1.8828915662650603,
      "grad_norm": 0.0946611613035202,
      "learning_rate": 1.478040494913926e-05,
      "loss": 0.0062,
      "step": 489
    },
    {
      "epoch": 1.886746987951807,
      "grad_norm": 0.11009645462036133,
      "learning_rate": 1.4694055269081987e-05,
      "loss": 0.0065,
      "step": 490
    },
    {
      "epoch": 1.890602409638554,
      "grad_norm": 0.18255947530269623,
      "learning_rate": 1.460781185335713e-05,
      "loss": 0.0065,
      "step": 491
    },
    {
      "epoch": 1.894457831325301,
      "grad_norm": 0.09501689672470093,
      "learning_rate": 1.452167642919695e-05,
      "loss": 0.0032,
      "step": 492
    },
    {
      "epoch": 1.8983132530120481,
      "grad_norm": 0.1402011215686798,
      "learning_rate": 1.443565072167095e-05,
      "loss": 0.0088,
      "step": 493
    },
    {
      "epoch": 1.9021686746987951,
      "grad_norm": 0.07192618399858475,
      "learning_rate": 1.4349736453651262e-05,
      "loss": 0.0041,
      "step": 494
    },
    {
      "epoch": 1.9060240963855422,
      "grad_norm": 0.1595878303050995,
      "learning_rate": 1.4263935345778202e-05,
      "loss": 0.0058,
      "step": 495
    },
    {
      "epoch": 1.9098795180722892,
      "grad_norm": 0.07809515297412872,
      "learning_rate": 1.4178249116425744e-05,
      "loss": 0.0037,
      "step": 496
    },
    {
      "epoch": 1.9137349397590362,
      "grad_norm": 0.09462330490350723,
      "learning_rate": 1.409267948166718e-05,
      "loss": 0.0051,
      "step": 497
    },
    {
      "epoch": 1.9175903614457832,
      "grad_norm": 0.21062414348125458,
      "learning_rate": 1.4007228155240696e-05,
      "loss": 0.0084,
      "step": 498
    },
    {
      "epoch": 1.9214457831325302,
      "grad_norm": 0.06321538984775543,
      "learning_rate": 1.3921896848515064e-05,
      "loss": 0.0073,
      "step": 499
    },
    {
      "epoch": 1.9253012048192772,
      "grad_norm": 0.1303102821111679,
      "learning_rate": 1.3836687270455388e-05,
      "loss": 0.0117,
      "step": 500
    },
    {
      "epoch": 1.929156626506024,
      "grad_norm": 0.09845706820487976,
      "learning_rate": 1.3751601127588849e-05,
      "loss": 0.0044,
      "step": 501
    },
    {
      "epoch": 1.933012048192771,
      "grad_norm": 0.07379204034805298,
      "learning_rate": 1.3666640123970556e-05,
      "loss": 0.0051,
      "step": 502
    },
    {
      "epoch": 1.936867469879518,
      "grad_norm": 0.07889371365308762,
      "learning_rate": 1.3581805961149371e-05,
      "loss": 0.008,
      "step": 503
    },
    {
      "epoch": 1.940722891566265,
      "grad_norm": 0.10923505574464798,
      "learning_rate": 1.3497100338133899e-05,
      "loss": 0.0136,
      "step": 504
    },
    {
      "epoch": 1.944578313253012,
      "grad_norm": 0.12238901853561401,
      "learning_rate": 1.341252495135841e-05,
      "loss": 0.006,
      "step": 505
    },
    {
      "epoch": 1.948433734939759,
      "grad_norm": 0.09399964660406113,
      "learning_rate": 1.3328081494648874e-05,
      "loss": 0.0076,
      "step": 506
    },
    {
      "epoch": 1.952289156626506,
      "grad_norm": 0.14163890480995178,
      "learning_rate": 1.324377165918906e-05,
      "loss": 0.006,
      "step": 507
    },
    {
      "epoch": 1.956144578313253,
      "grad_norm": 0.09812217950820923,
      "learning_rate": 1.3159597133486628e-05,
      "loss": 0.0058,
      "step": 508
    },
    {
      "epoch": 1.96,
      "grad_norm": 0.1172025129199028,
      "learning_rate": 1.3075559603339354e-05,
      "loss": 0.0085,
      "step": 509
    },
    {
      "epoch": 1.963855421686747,
      "grad_norm": 0.056579310446977615,
      "learning_rate": 1.299166075180133e-05,
      "loss": 0.0037,
      "step": 510
    },
    {
      "epoch": 1.967710843373494,
      "grad_norm": 0.11834102869033813,
      "learning_rate": 1.2907902259149287e-05,
      "loss": 0.0047,
      "step": 511
    },
    {
      "epoch": 1.971566265060241,
      "grad_norm": 0.4097451865673065,
      "learning_rate": 1.282428580284894e-05,
      "loss": 0.0158,
      "step": 512
    },
    {
      "epoch": 1.975421686746988,
      "grad_norm": 0.06863151490688324,
      "learning_rate": 1.274081305752135e-05,
      "loss": 0.0033,
      "step": 513
    },
    {
      "epoch": 1.979277108433735,
      "grad_norm": 0.09778847545385361,
      "learning_rate": 1.2657485694909451e-05,
      "loss": 0.0056,
      "step": 514
    },
    {
      "epoch": 1.983132530120482,
      "grad_norm": 0.10331977903842926,
      "learning_rate": 1.2574305383844528e-05,
      "loss": 0.0055,
      "step": 515
    },
    {
      "epoch": 1.9869879518072289,
      "grad_norm": 0.06463884562253952,
      "learning_rate": 1.2491273790212806e-05,
      "loss": 0.0039,
      "step": 516
    },
    {
      "epoch": 1.9908433734939759,
      "grad_norm": 0.12248596549034119,
      "learning_rate": 1.2408392576922075e-05,
      "loss": 0.0115,
      "step": 517
    },
    {
      "epoch": 1.994698795180723,
      "grad_norm": 0.15362606942653656,
      "learning_rate": 1.2325663403868406e-05,
      "loss": 0.0087,
      "step": 518
    },
    {
      "epoch": 1.99855421686747,
      "grad_norm": 0.06778371334075928,
      "learning_rate": 1.2243087927902905e-05,
      "loss": 0.0032,
      "step": 519
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.2598204016685486,
      "learning_rate": 1.2160667802798503e-05,
      "loss": 0.0092,
      "step": 520
    },
    {
      "epoch": 2.003855421686747,
      "grad_norm": 0.1877129226922989,
      "learning_rate": 1.2078404679216864e-05,
      "loss": 0.0076,
      "step": 521
    },
    {
      "epoch": 2.007710843373494,
      "grad_norm": 0.0740046426653862,
      "learning_rate": 1.199630020467533e-05,
      "loss": 0.0028,
      "step": 522
    },
    {
      "epoch": 2.011566265060241,
      "grad_norm": 0.14209777116775513,
      "learning_rate": 1.1914356023513904e-05,
      "loss": 0.008,
      "step": 523
    },
    {
      "epoch": 2.015421686746988,
      "grad_norm": 0.06664244830608368,
      "learning_rate": 1.1832573776862319e-05,
      "loss": 0.007,
      "step": 524
    },
    {
      "epoch": 2.019277108433735,
      "grad_norm": 0.0566011480987072,
      "learning_rate": 1.1750955102607193e-05,
      "loss": 0.0023,
      "step": 525
    },
    {
      "epoch": 2.023132530120482,
      "grad_norm": 0.059486355632543564,
      "learning_rate": 1.1669501635359212e-05,
      "loss": 0.0023,
      "step": 526
    },
    {
      "epoch": 2.026987951807229,
      "grad_norm": 0.048501964658498764,
      "learning_rate": 1.1588215006420374e-05,
      "loss": 0.0024,
      "step": 527
    },
    {
      "epoch": 2.0308433734939757,
      "grad_norm": 0.06546659767627716,
      "learning_rate": 1.1507096843751372e-05,
      "loss": 0.0038,
      "step": 528
    },
    {
      "epoch": 2.0346987951807227,
      "grad_norm": 0.055793650448322296,
      "learning_rate": 1.1426148771938915e-05,
      "loss": 0.0039,
      "step": 529
    },
    {
      "epoch": 2.0385542168674697,
      "grad_norm": 0.14630311727523804,
      "learning_rate": 1.1345372412163276e-05,
      "loss": 0.0028,
      "step": 530
    },
    {
      "epoch": 2.0424096385542168,
      "grad_norm": 0.052479639649391174,
      "learning_rate": 1.1264769382165748e-05,
      "loss": 0.0033,
      "step": 531
    },
    {
      "epoch": 2.0462650602409638,
      "grad_norm": 0.06099266931414604,
      "learning_rate": 1.1184341296216281e-05,
      "loss": 0.0023,
      "step": 532
    },
    {
      "epoch": 2.050120481927711,
      "grad_norm": 0.08203870803117752,
      "learning_rate": 1.110408976508118e-05,
      "loss": 0.0025,
      "step": 533
    },
    {
      "epoch": 2.053975903614458,
      "grad_norm": 0.031148897483944893,
      "learning_rate": 1.1024016395990758e-05,
      "loss": 0.0021,
      "step": 534
    },
    {
      "epoch": 2.057831325301205,
      "grad_norm": 0.10753555595874786,
      "learning_rate": 1.094412279260726e-05,
      "loss": 0.0084,
      "step": 535
    },
    {
      "epoch": 2.061686746987952,
      "grad_norm": 0.04244984686374664,
      "learning_rate": 1.0864410554992654e-05,
      "loss": 0.002,
      "step": 536
    },
    {
      "epoch": 2.065542168674699,
      "grad_norm": 0.03356681391596794,
      "learning_rate": 1.0784881279576635e-05,
      "loss": 0.0015,
      "step": 537
    },
    {
      "epoch": 2.069397590361446,
      "grad_norm": 0.0787050798535347,
      "learning_rate": 1.070553655912463e-05,
      "loss": 0.0028,
      "step": 538
    },
    {
      "epoch": 2.073253012048193,
      "grad_norm": 0.03804800286889076,
      "learning_rate": 1.0626377982705929e-05,
      "loss": 0.0021,
      "step": 539
    },
    {
      "epoch": 2.07710843373494,
      "grad_norm": 0.1359545737504959,
      "learning_rate": 1.0547407135661816e-05,
      "loss": 0.0112,
      "step": 540
    },
    {
      "epoch": 2.080963855421687,
      "grad_norm": 0.07904944568872452,
      "learning_rate": 1.0468625599573842e-05,
      "loss": 0.0029,
      "step": 541
    },
    {
      "epoch": 2.0848192771084335,
      "grad_norm": 0.0676993653178215,
      "learning_rate": 1.0390034952232178e-05,
      "loss": 0.0016,
      "step": 542
    },
    {
      "epoch": 2.0886746987951805,
      "grad_norm": 0.052145086228847504,
      "learning_rate": 1.0311636767603952e-05,
      "loss": 0.0024,
      "step": 543
    },
    {
      "epoch": 2.0925301204819275,
      "grad_norm": 0.1171824038028717,
      "learning_rate": 1.0233432615801786e-05,
      "loss": 0.009,
      "step": 544
    },
    {
      "epoch": 2.0963855421686746,
      "grad_norm": 0.7555789351463318,
      "learning_rate": 1.0155424063052306e-05,
      "loss": 0.0052,
      "step": 545
    },
    {
      "epoch": 2.1002409638554216,
      "grad_norm": 0.08210741728544235,
      "learning_rate": 1.0077612671664827e-05,
      "loss": 0.0054,
      "step": 546
    },
    {
      "epoch": 2.1040963855421686,
      "grad_norm": 0.11431938409805298,
      "learning_rate": 1.0000000000000006e-05,
      "loss": 0.0039,
      "step": 547
    },
    {
      "epoch": 2.1079518072289156,
      "grad_norm": 0.07425118237733841,
      "learning_rate": 9.922587602438657e-06,
      "loss": 0.0118,
      "step": 548
    },
    {
      "epoch": 2.1118072289156626,
      "grad_norm": 0.06563588976860046,
      "learning_rate": 9.84537702935065e-06,
      "loss": 0.0021,
      "step": 549
    },
    {
      "epoch": 2.1156626506024097,
      "grad_norm": 0.23397141695022583,
      "learning_rate": 9.768369827063803e-06,
      "loss": 0.0064,
      "step": 550
    },
    {
      "epoch": 2.1195180722891567,
      "grad_norm": 0.06807775050401688,
      "learning_rate": 9.691567537832964e-06,
      "loss": 0.0036,
      "step": 551
    },
    {
      "epoch": 2.1233734939759037,
      "grad_norm": 0.07760459929704666,
      "learning_rate": 9.614971699809081e-06,
      "loss": 0.005,
      "step": 552
    },
    {
      "epoch": 2.1272289156626507,
      "grad_norm": 0.08298410475254059,
      "learning_rate": 9.538583847008452e-06,
      "loss": 0.0072,
      "step": 553
    },
    {
      "epoch": 2.1310843373493977,
      "grad_norm": 0.053814083337783813,
      "learning_rate": 9.462405509281937e-06,
      "loss": 0.0021,
      "step": 554
    },
    {
      "epoch": 2.1349397590361447,
      "grad_norm": 0.05976598709821701,
      "learning_rate": 9.386438212284372e-06,
      "loss": 0.003,
      "step": 555
    },
    {
      "epoch": 2.1387951807228918,
      "grad_norm": 0.03989047929644585,
      "learning_rate": 9.310683477443979e-06,
      "loss": 0.002,
      "step": 556
    },
    {
      "epoch": 2.1426506024096383,
      "grad_norm": 0.2895124554634094,
      "learning_rate": 9.235142821931928e-06,
      "loss": 0.0134,
      "step": 557
    },
    {
      "epoch": 2.1465060240963854,
      "grad_norm": 0.07493002712726593,
      "learning_rate": 9.159817758631923e-06,
      "loss": 0.0028,
      "step": 558
    },
    {
      "epoch": 2.1503614457831324,
      "grad_norm": 0.037741731852293015,
      "learning_rate": 9.084709796109907e-06,
      "loss": 0.0012,
      "step": 559
    },
    {
      "epoch": 2.1542168674698794,
      "grad_norm": 0.06910698860883713,
      "learning_rate": 9.009820438583881e-06,
      "loss": 0.002,
      "step": 560
    },
    {
      "epoch": 2.1580722891566264,
      "grad_norm": 0.07993840426206589,
      "learning_rate": 8.93515118589373e-06,
      "loss": 0.0032,
      "step": 561
    },
    {
      "epoch": 2.1619277108433734,
      "grad_norm": 0.3232978880405426,
      "learning_rate": 8.860703533471218e-06,
      "loss": 0.0097,
      "step": 562
    },
    {
      "epoch": 2.1657831325301204,
      "grad_norm": 0.05569448694586754,
      "learning_rate": 8.786478972310023e-06,
      "loss": 0.0019,
      "step": 563
    },
    {
      "epoch": 2.1696385542168675,
      "grad_norm": 0.07345814257860184,
      "learning_rate": 8.712478988935904e-06,
      "loss": 0.0028,
      "step": 564
    },
    {
      "epoch": 2.1734939759036145,
      "grad_norm": 0.06032630801200867,
      "learning_rate": 8.638705065376887e-06,
      "loss": 0.0017,
      "step": 565
    },
    {
      "epoch": 2.1773493975903615,
      "grad_norm": 0.06941293925046921,
      "learning_rate": 8.565158679133605e-06,
      "loss": 0.0021,
      "step": 566
    },
    {
      "epoch": 2.1812048192771085,
      "grad_norm": 0.05227924510836601,
      "learning_rate": 8.491841303149728e-06,
      "loss": 0.0039,
      "step": 567
    },
    {
      "epoch": 2.1850602409638555,
      "grad_norm": 0.08419029414653778,
      "learning_rate": 8.418754405782423e-06,
      "loss": 0.0059,
      "step": 568
    },
    {
      "epoch": 2.1889156626506026,
      "grad_norm": 0.06260647624731064,
      "learning_rate": 8.345899450772975e-06,
      "loss": 0.0028,
      "step": 569
    },
    {
      "epoch": 2.1927710843373496,
      "grad_norm": 0.326419860124588,
      "learning_rate": 8.27327789721746e-06,
      "loss": 0.0083,
      "step": 570
    },
    {
      "epoch": 2.1966265060240966,
      "grad_norm": 0.05631480738520622,
      "learning_rate": 8.200891199537549e-06,
      "loss": 0.0014,
      "step": 571
    },
    {
      "epoch": 2.200481927710843,
      "grad_norm": 0.07872775942087173,
      "learning_rate": 8.128740807451311e-06,
      "loss": 0.0018,
      "step": 572
    },
    {
      "epoch": 2.20433734939759,
      "grad_norm": 0.13493163883686066,
      "learning_rate": 8.056828165944282e-06,
      "loss": 0.0085,
      "step": 573
    },
    {
      "epoch": 2.208192771084337,
      "grad_norm": 0.11313220113515854,
      "learning_rate": 7.98515471524042e-06,
      "loss": 0.0039,
      "step": 574
    },
    {
      "epoch": 2.212048192771084,
      "grad_norm": 0.06968442350625992,
      "learning_rate": 7.913721890773354e-06,
      "loss": 0.0023,
      "step": 575
    },
    {
      "epoch": 2.2159036144578312,
      "grad_norm": 0.6592087149620056,
      "learning_rate": 7.842531123157562e-06,
      "loss": 0.0027,
      "step": 576
    },
    {
      "epoch": 2.2197590361445783,
      "grad_norm": 0.08434339612722397,
      "learning_rate": 7.771583838159756e-06,
      "loss": 0.0036,
      "step": 577
    },
    {
      "epoch": 2.2236144578313253,
      "grad_norm": 0.05565265938639641,
      "learning_rate": 7.700881456670342e-06,
      "loss": 0.0022,
      "step": 578
    },
    {
      "epoch": 2.2274698795180723,
      "grad_norm": 0.06776664406061172,
      "learning_rate": 7.630425394674903e-06,
      "loss": 0.0055,
      "step": 579
    },
    {
      "epoch": 2.2313253012048193,
      "grad_norm": 0.0737040564417839,
      "learning_rate": 7.560217063225923e-06,
      "loss": 0.0045,
      "step": 580
    },
    {
      "epoch": 2.2351807228915663,
      "grad_norm": 0.0424475371837616,
      "learning_rate": 7.49025786841445e-06,
      "loss": 0.0016,
      "step": 581
    },
    {
      "epoch": 2.2390361445783133,
      "grad_norm": 0.05880943313241005,
      "learning_rate": 7.420549211342005e-06,
      "loss": 0.0036,
      "step": 582
    },
    {
      "epoch": 2.2428915662650604,
      "grad_norm": 0.07502184063196182,
      "learning_rate": 7.3510924880924575e-06,
      "loss": 0.0035,
      "step": 583
    },
    {
      "epoch": 2.2467469879518074,
      "grad_norm": 0.07997576147317886,
      "learning_rate": 7.2818890897041065e-06,
      "loss": 0.0034,
      "step": 584
    },
    {
      "epoch": 2.2506024096385544,
      "grad_norm": 0.05822072923183441,
      "learning_rate": 7.212940402141808e-06,
      "loss": 0.0024,
      "step": 585
    },
    {
      "epoch": 2.2544578313253014,
      "grad_norm": 0.05100448429584503,
      "learning_rate": 7.1442478062692135e-06,
      "loss": 0.006,
      "step": 586
    },
    {
      "epoch": 2.258313253012048,
      "grad_norm": 0.06905625760555267,
      "learning_rate": 7.075812677821145e-06,
      "loss": 0.0024,
      "step": 587
    },
    {
      "epoch": 2.262168674698795,
      "grad_norm": 0.055957138538360596,
      "learning_rate": 7.0076363873759865e-06,
      "loss": 0.0037,
      "step": 588
    },
    {
      "epoch": 2.266024096385542,
      "grad_norm": 0.03304177150130272,
      "learning_rate": 6.939720300328303e-06,
      "loss": 0.0018,
      "step": 589
    },
    {
      "epoch": 2.269879518072289,
      "grad_norm": 0.08473087102174759,
      "learning_rate": 6.8720657768614115e-06,
      "loss": 0.0058,
      "step": 590
    },
    {
      "epoch": 2.273734939759036,
      "grad_norm": 0.0625910758972168,
      "learning_rate": 6.8046741719202385e-06,
      "loss": 0.0049,
      "step": 591
    },
    {
      "epoch": 2.277590361445783,
      "grad_norm": 0.07574128359556198,
      "learning_rate": 6.737546835184101e-06,
      "loss": 0.0043,
      "step": 592
    },
    {
      "epoch": 2.28144578313253,
      "grad_norm": 0.07056653499603271,
      "learning_rate": 6.67068511103971e-06,
      "loss": 0.0119,
      "step": 593
    },
    {
      "epoch": 2.285301204819277,
      "grad_norm": 0.07596432417631149,
      "learning_rate": 6.604090338554267e-06,
      "loss": 0.0031,
      "step": 594
    },
    {
      "epoch": 2.289156626506024,
      "grad_norm": 0.0327884741127491,
      "learning_rate": 6.537763851448593e-06,
      "loss": 0.002,
      "step": 595
    },
    {
      "epoch": 2.293012048192771,
      "grad_norm": 0.03820427507162094,
      "learning_rate": 6.471706978070482e-06,
      "loss": 0.0017,
      "step": 596
    },
    {
      "epoch": 2.296867469879518,
      "grad_norm": 0.04365382716059685,
      "learning_rate": 6.4059210413680175e-06,
      "loss": 0.0017,
      "step": 597
    },
    {
      "epoch": 2.300722891566265,
      "grad_norm": 0.08569991588592529,
      "learning_rate": 6.340407358863167e-06,
      "loss": 0.0028,
      "step": 598
    },
    {
      "epoch": 2.304578313253012,
      "grad_norm": 0.34639790654182434,
      "learning_rate": 6.275167242625331e-06,
      "loss": 0.0121,
      "step": 599
    },
    {
      "epoch": 2.3084337349397592,
      "grad_norm": 0.0633152648806572,
      "learning_rate": 6.210201999245087e-06,
      "loss": 0.0027,
      "step": 600
    },
    {
      "epoch": 2.3122891566265062,
      "grad_norm": 0.06482216715812683,
      "learning_rate": 6.145512929808013e-06,
      "loss": 0.0028,
      "step": 601
    },
    {
      "epoch": 2.316144578313253,
      "grad_norm": 0.06370186805725098,
      "learning_rate": 6.081101329868664e-06,
      "loss": 0.0052,
      "step": 602
    },
    {
      "epoch": 2.32,
      "grad_norm": 0.05200124904513359,
      "learning_rate": 6.016968489424572e-06,
      "loss": 0.0022,
      "step": 603
    },
    {
      "epoch": 2.323855421686747,
      "grad_norm": 0.04066948965191841,
      "learning_rate": 5.95311569289045e-06,
      "loss": 0.0015,
      "step": 604
    },
    {
      "epoch": 2.327710843373494,
      "grad_norm": 0.04939762502908707,
      "learning_rate": 5.889544219072465e-06,
      "loss": 0.0018,
      "step": 605
    },
    {
      "epoch": 2.331566265060241,
      "grad_norm": 0.05605112388730049,
      "learning_rate": 5.8262553411426015e-06,
      "loss": 0.0023,
      "step": 606
    },
    {
      "epoch": 2.335421686746988,
      "grad_norm": 0.025311313569545746,
      "learning_rate": 5.7632503266131925e-06,
      "loss": 0.0013,
      "step": 607
    },
    {
      "epoch": 2.339277108433735,
      "grad_norm": 0.07716325670480728,
      "learning_rate": 5.700530437311509e-06,
      "loss": 0.0019,
      "step": 608
    },
    {
      "epoch": 2.343132530120482,
      "grad_norm": 0.035999130457639694,
      "learning_rate": 5.638096929354522e-06,
      "loss": 0.0011,
      "step": 609
    },
    {
      "epoch": 2.346987951807229,
      "grad_norm": 0.10247158259153366,
      "learning_rate": 5.5759510531237116e-06,
      "loss": 0.0077,
      "step": 610
    },
    {
      "epoch": 2.350843373493976,
      "grad_norm": 0.11552261561155319,
      "learning_rate": 5.514094053240035e-06,
      "loss": 0.0035,
      "step": 611
    },
    {
      "epoch": 2.354698795180723,
      "grad_norm": 0.09938646852970123,
      "learning_rate": 5.452527168539026e-06,
      "loss": 0.0018,
      "step": 612
    },
    {
      "epoch": 2.35855421686747,
      "grad_norm": 0.06877364218235016,
      "learning_rate": 5.39125163204594e-06,
      "loss": 0.0039,
      "step": 613
    },
    {
      "epoch": 2.362409638554217,
      "grad_norm": 0.03887651488184929,
      "learning_rate": 5.330268670951091e-06,
      "loss": 0.0013,
      "step": 614
    },
    {
      "epoch": 2.3662650602409636,
      "grad_norm": 0.08702801913022995,
      "learning_rate": 5.269579506585259e-06,
      "loss": 0.0084,
      "step": 615
    },
    {
      "epoch": 2.370120481927711,
      "grad_norm": 0.05485809966921806,
      "learning_rate": 5.209185354395261e-06,
      "loss": 0.0023,
      "step": 616
    },
    {
      "epoch": 2.3739759036144576,
      "grad_norm": 0.10920535027980804,
      "learning_rate": 5.149087423919541e-06,
      "loss": 0.0029,
      "step": 617
    },
    {
      "epoch": 2.3778313253012047,
      "grad_norm": 0.13908222317695618,
      "learning_rate": 5.089286918764031e-06,
      "loss": 0.0034,
      "step": 618
    },
    {
      "epoch": 2.3816867469879517,
      "grad_norm": 0.09357054531574249,
      "learning_rate": 5.029785036577976e-06,
      "loss": 0.0024,
      "step": 619
    },
    {
      "epoch": 2.3855421686746987,
      "grad_norm": 0.17332714796066284,
      "learning_rate": 4.97058296903e-06,
      "loss": 0.0064,
      "step": 620
    },
    {
      "epoch": 2.3893975903614457,
      "grad_norm": 0.04516534134745598,
      "learning_rate": 4.911681901784198e-06,
      "loss": 0.002,
      "step": 621
    },
    {
      "epoch": 2.3932530120481927,
      "grad_norm": 0.10836844146251678,
      "learning_rate": 4.853083014476403e-06,
      "loss": 0.0116,
      "step": 622
    },
    {
      "epoch": 2.3971084337349398,
      "grad_norm": 0.06490450352430344,
      "learning_rate": 4.794787480690597e-06,
      "loss": 0.0034,
      "step": 623
    },
    {
      "epoch": 2.4009638554216868,
      "grad_norm": 0.2077661007642746,
      "learning_rate": 4.736796467935336e-06,
      "loss": 0.0086,
      "step": 624
    },
    {
      "epoch": 2.404819277108434,
      "grad_norm": 0.050042327493429184,
      "learning_rate": 4.679111137620442e-06,
      "loss": 0.0018,
      "step": 625
    },
    {
      "epoch": 2.408674698795181,
      "grad_norm": 0.04626263305544853,
      "learning_rate": 4.621732645033687e-06,
      "loss": 0.0016,
      "step": 626
    },
    {
      "epoch": 2.412530120481928,
      "grad_norm": 0.03525455668568611,
      "learning_rate": 4.5646621393177e-06,
      "loss": 0.001,
      "step": 627
    },
    {
      "epoch": 2.416385542168675,
      "grad_norm": 0.06515587121248245,
      "learning_rate": 4.507900763446911e-06,
      "loss": 0.003,
      "step": 628
    },
    {
      "epoch": 2.420240963855422,
      "grad_norm": 0.05137157812714577,
      "learning_rate": 4.451449654204685e-06,
      "loss": 0.0016,
      "step": 629
    },
    {
      "epoch": 2.4240963855421684,
      "grad_norm": 0.09875486046075821,
      "learning_rate": 4.395309942160551e-06,
      "loss": 0.0019,
      "step": 630
    },
    {
      "epoch": 2.427951807228916,
      "grad_norm": 0.07232407480478287,
      "learning_rate": 4.339482751647557e-06,
      "loss": 0.0023,
      "step": 631
    },
    {
      "epoch": 2.4318072289156625,
      "grad_norm": 0.06509175151586533,
      "learning_rate": 4.283969200739764e-06,
      "loss": 0.0042,
      "step": 632
    },
    {
      "epoch": 2.4356626506024095,
      "grad_norm": 0.17352131009101868,
      "learning_rate": 4.228770401229824e-06,
      "loss": 0.0074,
      "step": 633
    },
    {
      "epoch": 2.4395180722891565,
      "grad_norm": 0.035422373563051224,
      "learning_rate": 4.173887458606767e-06,
      "loss": 0.0013,
      "step": 634
    },
    {
      "epoch": 2.4433734939759035,
      "grad_norm": 0.038832418620586395,
      "learning_rate": 4.119321472033779e-06,
      "loss": 0.0013,
      "step": 635
    },
    {
      "epoch": 2.4472289156626506,
      "grad_norm": 0.06765366345643997,
      "learning_rate": 4.065073534326282e-06,
      "loss": 0.0032,
      "step": 636
    },
    {
      "epoch": 2.4510843373493976,
      "grad_norm": 0.13772547245025635,
      "learning_rate": 4.011144731929981e-06,
      "loss": 0.0093,
      "step": 637
    },
    {
      "epoch": 2.4549397590361446,
      "grad_norm": 0.08776474744081497,
      "learning_rate": 3.957536144899123e-06,
      "loss": 0.0067,
      "step": 638
    },
    {
      "epoch": 2.4587951807228916,
      "grad_norm": 0.10551299154758453,
      "learning_rate": 3.904248846874894e-06,
      "loss": 0.0108,
      "step": 639
    },
    {
      "epoch": 2.4626506024096386,
      "grad_norm": 0.04805995151400566,
      "learning_rate": 3.851283905063863e-06,
      "loss": 0.0019,
      "step": 640
    },
    {
      "epoch": 2.4665060240963856,
      "grad_norm": 0.09927227348089218,
      "learning_rate": 3.7986423802166705e-06,
      "loss": 0.0067,
      "step": 641
    },
    {
      "epoch": 2.4703614457831327,
      "grad_norm": 0.11920815706253052,
      "learning_rate": 3.7463253266067147e-06,
      "loss": 0.0079,
      "step": 642
    },
    {
      "epoch": 2.4742168674698797,
      "grad_norm": 0.037468794733285904,
      "learning_rate": 3.694333792009115e-06,
      "loss": 0.0014,
      "step": 643
    },
    {
      "epoch": 2.4780722891566267,
      "grad_norm": 0.04393767938017845,
      "learning_rate": 3.64266881767966e-06,
      "loss": 0.001,
      "step": 644
    },
    {
      "epoch": 2.4819277108433733,
      "grad_norm": 0.045006439089775085,
      "learning_rate": 3.5913314383339937e-06,
      "loss": 0.0015,
      "step": 645
    },
    {
      "epoch": 2.4857831325301207,
      "grad_norm": 0.05892086401581764,
      "learning_rate": 3.5403226821268734e-06,
      "loss": 0.0013,
      "step": 646
    },
    {
      "epoch": 2.4896385542168673,
      "grad_norm": 0.09949485957622528,
      "learning_rate": 3.4896435706316e-06,
      "loss": 0.0037,
      "step": 647
    },
    {
      "epoch": 2.4934939759036143,
      "grad_norm": 0.08316222578287125,
      "learning_rate": 3.43929511881953e-06,
      "loss": 0.0061,
      "step": 648
    },
    {
      "epoch": 2.4973493975903613,
      "grad_norm": 0.11504265666007996,
      "learning_rate": 3.3892783350397675e-06,
      "loss": 0.0043,
      "step": 649
    },
    {
      "epoch": 2.5012048192771084,
      "grad_norm": 0.041751548647880554,
      "learning_rate": 3.3395942209989694e-06,
      "loss": 0.0014,
      "step": 650
    },
    {
      "epoch": 2.5050602409638554,
      "grad_norm": 0.12726637721061707,
      "learning_rate": 3.290243771741275e-06,
      "loss": 0.0062,
      "step": 651
    },
    {
      "epoch": 2.5089156626506024,
      "grad_norm": 0.10160968452692032,
      "learning_rate": 3.241227975628378e-06,
      "loss": 0.004,
      "step": 652
    },
    {
      "epoch": 2.5127710843373494,
      "grad_norm": 0.06926268339157104,
      "learning_rate": 3.1925478143197418e-06,
      "loss": 0.0028,
      "step": 653
    },
    {
      "epoch": 2.5166265060240964,
      "grad_norm": 0.07057193666696548,
      "learning_rate": 3.144204262752937e-06,
      "loss": 0.003,
      "step": 654
    },
    {
      "epoch": 2.5204819277108435,
      "grad_norm": 0.0741814598441124,
      "learning_rate": 3.0961982891241083e-06,
      "loss": 0.0022,
      "step": 655
    },
    {
      "epoch": 2.5243373493975905,
      "grad_norm": 0.03325735032558441,
      "learning_rate": 3.048530854868579e-06,
      "loss": 0.0011,
      "step": 656
    },
    {
      "epoch": 2.5281927710843375,
      "grad_norm": 0.07111026346683502,
      "learning_rate": 3.001202914641628e-06,
      "loss": 0.0021,
      "step": 657
    },
    {
      "epoch": 2.532048192771084,
      "grad_norm": 0.0835479348897934,
      "learning_rate": 2.954215416299331e-06,
      "loss": 0.0043,
      "step": 658
    },
    {
      "epoch": 2.5359036144578315,
      "grad_norm": 0.083519846200943,
      "learning_rate": 2.907569300879596e-06,
      "loss": 0.0036,
      "step": 659
    },
    {
      "epoch": 2.539759036144578,
      "grad_norm": 0.06607409566640854,
      "learning_rate": 2.8612655025833148e-06,
      "loss": 0.0013,
      "step": 660
    },
    {
      "epoch": 2.5436144578313256,
      "grad_norm": 0.06507234275341034,
      "learning_rate": 2.815304948755664e-06,
      "loss": 0.0037,
      "step": 661
    },
    {
      "epoch": 2.547469879518072,
      "grad_norm": 0.04953571408987045,
      "learning_rate": 2.769688559867514e-06,
      "loss": 0.0019,
      "step": 662
    },
    {
      "epoch": 2.551325301204819,
      "grad_norm": 0.24300339818000793,
      "learning_rate": 2.7244172494969978e-06,
      "loss": 0.0089,
      "step": 663
    },
    {
      "epoch": 2.555180722891566,
      "grad_norm": 0.03667943552136421,
      "learning_rate": 2.679491924311226e-06,
      "loss": 0.0013,
      "step": 664
    },
    {
      "epoch": 2.559036144578313,
      "grad_norm": 0.05591225251555443,
      "learning_rate": 2.6349134840481294e-06,
      "loss": 0.002,
      "step": 665
    },
    {
      "epoch": 2.56289156626506,
      "grad_norm": 0.08055668324232101,
      "learning_rate": 2.590682821498416e-06,
      "loss": 0.0023,
      "step": 666
    },
    {
      "epoch": 2.5667469879518072,
      "grad_norm": 0.0474967285990715,
      "learning_rate": 2.546800822487714e-06,
      "loss": 0.0025,
      "step": 667
    },
    {
      "epoch": 2.5706024096385542,
      "grad_norm": 0.061049655079841614,
      "learning_rate": 2.503268365858831e-06,
      "loss": 0.0023,
      "step": 668
    },
    {
      "epoch": 2.5744578313253013,
      "grad_norm": 0.07722913473844528,
      "learning_rate": 2.4600863234541338e-06,
      "loss": 0.0028,
      "step": 669
    },
    {
      "epoch": 2.5783132530120483,
      "grad_norm": 0.09591741114854813,
      "learning_rate": 2.417255560098111e-06,
      "loss": 0.0019,
      "step": 670
    },
    {
      "epoch": 2.5821686746987953,
      "grad_norm": 0.08037438243627548,
      "learning_rate": 2.374776933580025e-06,
      "loss": 0.0054,
      "step": 671
    },
    {
      "epoch": 2.5860240963855423,
      "grad_norm": 0.057184137403964996,
      "learning_rate": 2.3326512946367717e-06,
      "loss": 0.0022,
      "step": 672
    },
    {
      "epoch": 2.589879518072289,
      "grad_norm": 0.05303996801376343,
      "learning_rate": 2.2908794869358044e-06,
      "loss": 0.0013,
      "step": 673
    },
    {
      "epoch": 2.5937349397590364,
      "grad_norm": 0.04736126959323883,
      "learning_rate": 2.249462347058251e-06,
      "loss": 0.0024,
      "step": 674
    },
    {
      "epoch": 2.597590361445783,
      "grad_norm": 0.09622405469417572,
      "learning_rate": 2.2084007044821764e-06,
      "loss": 0.0057,
      "step": 675
    },
    {
      "epoch": 2.6014457831325304,
      "grad_norm": 0.052896302193403244,
      "learning_rate": 2.167695381565944e-06,
      "loss": 0.0022,
      "step": 676
    },
    {
      "epoch": 2.605301204819277,
      "grad_norm": 0.10612209141254425,
      "learning_rate": 2.127347193531757e-06,
      "loss": 0.0055,
      "step": 677
    },
    {
      "epoch": 2.609156626506024,
      "grad_norm": 0.08298195898532867,
      "learning_rate": 2.0873569484493305e-06,
      "loss": 0.0036,
      "step": 678
    },
    {
      "epoch": 2.613012048192771,
      "grad_norm": 0.06990071386098862,
      "learning_rate": 2.0477254472197237e-06,
      "loss": 0.0057,
      "step": 679
    },
    {
      "epoch": 2.616867469879518,
      "grad_norm": 0.06889458000659943,
      "learning_rate": 2.0084534835592584e-06,
      "loss": 0.0041,
      "step": 680
    },
    {
      "epoch": 2.620722891566265,
      "grad_norm": 0.04544433578848839,
      "learning_rate": 1.96954184398368e-06,
      "loss": 0.0018,
      "step": 681
    },
    {
      "epoch": 2.624578313253012,
      "grad_norm": 0.06777770817279816,
      "learning_rate": 1.930991307792358e-06,
      "loss": 0.0043,
      "step": 682
    },
    {
      "epoch": 2.628433734939759,
      "grad_norm": 0.054681550711393356,
      "learning_rate": 1.8928026470526917e-06,
      "loss": 0.0019,
      "step": 683
    },
    {
      "epoch": 2.632289156626506,
      "grad_norm": 0.10625132918357849,
      "learning_rate": 1.8549766265846725e-06,
      "loss": 0.0027,
      "step": 684
    },
    {
      "epoch": 2.636144578313253,
      "grad_norm": 0.081264428794384,
      "learning_rate": 1.817514003945524e-06,
      "loss": 0.0057,
      "step": 685
    },
    {
      "epoch": 2.64,
      "grad_norm": 0.04519634321331978,
      "learning_rate": 1.7804155294145765e-06,
      "loss": 0.0019,
      "step": 686
    },
    {
      "epoch": 2.643855421686747,
      "grad_norm": 0.02832118421792984,
      "learning_rate": 1.743681945978184e-06,
      "loss": 0.001,
      "step": 687
    },
    {
      "epoch": 2.6477108433734937,
      "grad_norm": 0.06859899312257767,
      "learning_rate": 1.7073139893149092e-06,
      "loss": 0.0037,
      "step": 688
    },
    {
      "epoch": 2.651566265060241,
      "grad_norm": 0.045318011194467545,
      "learning_rate": 1.6713123877807413e-06,
      "loss": 0.0019,
      "step": 689
    },
    {
      "epoch": 2.6554216867469878,
      "grad_norm": 0.09894543141126633,
      "learning_rate": 1.6356778623945223e-06,
      "loss": 0.0072,
      "step": 690
    },
    {
      "epoch": 2.659277108433735,
      "grad_norm": 0.06565819680690765,
      "learning_rate": 1.6004111268235156e-06,
      "loss": 0.0066,
      "step": 691
    },
    {
      "epoch": 2.663132530120482,
      "grad_norm": 0.043372225016355515,
      "learning_rate": 1.5655128873691138e-06,
      "loss": 0.0031,
      "step": 692
    },
    {
      "epoch": 2.666987951807229,
      "grad_norm": 0.042357612401247025,
      "learning_rate": 1.5309838429526714e-06,
      "loss": 0.0013,
      "step": 693
    },
    {
      "epoch": 2.670843373493976,
      "grad_norm": 0.1303417682647705,
      "learning_rate": 1.49682468510153e-06,
      "loss": 0.0114,
      "step": 694
    },
    {
      "epoch": 2.674698795180723,
      "grad_norm": 0.04758170619606972,
      "learning_rate": 1.4630360979351644e-06,
      "loss": 0.0014,
      "step": 695
    },
    {
      "epoch": 2.67855421686747,
      "grad_norm": 0.05521918088197708,
      "learning_rate": 1.4296187581514742e-06,
      "loss": 0.0042,
      "step": 696
    },
    {
      "epoch": 2.682409638554217,
      "grad_norm": 0.09340808540582657,
      "learning_rate": 1.396573335013236e-06,
      "loss": 0.0022,
      "step": 697
    },
    {
      "epoch": 2.686265060240964,
      "grad_norm": 0.04764275625348091,
      "learning_rate": 1.3639004903346954e-06,
      "loss": 0.0029,
      "step": 698
    },
    {
      "epoch": 2.690120481927711,
      "grad_norm": 1.6971749067306519,
      "learning_rate": 1.3316008784683265e-06,
      "loss": 0.0224,
      "step": 699
    },
    {
      "epoch": 2.693975903614458,
      "grad_norm": 0.048169564455747604,
      "learning_rate": 1.2996751462917057e-06,
      "loss": 0.0042,
      "step": 700
    },
    {
      "epoch": 2.697831325301205,
      "grad_norm": 0.07363072782754898,
      "learning_rate": 1.2681239331945695e-06,
      "loss": 0.0035,
      "step": 701
    },
    {
      "epoch": 2.701686746987952,
      "grad_norm": 0.07053504884243011,
      "learning_rate": 1.2369478710660166e-06,
      "loss": 0.0033,
      "step": 702
    },
    {
      "epoch": 2.7055421686746985,
      "grad_norm": 0.031916484236717224,
      "learning_rate": 1.2061475842818337e-06,
      "loss": 0.0011,
      "step": 703
    },
    {
      "epoch": 2.709397590361446,
      "grad_norm": 0.05330387130379677,
      "learning_rate": 1.175723689692001e-06,
      "loss": 0.0029,
      "step": 704
    },
    {
      "epoch": 2.7132530120481926,
      "grad_norm": 0.10069465637207031,
      "learning_rate": 1.1456767966083393e-06,
      "loss": 0.0101,
      "step": 705
    },
    {
      "epoch": 2.7171084337349396,
      "grad_norm": 0.06322697550058365,
      "learning_rate": 1.1160075067923092e-06,
      "loss": 0.0029,
      "step": 706
    },
    {
      "epoch": 2.7209638554216866,
      "grad_norm": 0.040642112493515015,
      "learning_rate": 1.086716414442952e-06,
      "loss": 0.0014,
      "step": 707
    },
    {
      "epoch": 2.7248192771084336,
      "grad_norm": 0.15921436250209808,
      "learning_rate": 1.057804106184992e-06,
      "loss": 0.0017,
      "step": 708
    },
    {
      "epoch": 2.7286746987951807,
      "grad_norm": 0.032873108983039856,
      "learning_rate": 1.0292711610570904e-06,
      "loss": 0.0014,
      "step": 709
    },
    {
      "epoch": 2.7325301204819277,
      "grad_norm": 0.0637291744351387,
      "learning_rate": 1.0011181505002488e-06,
      "loss": 0.0033,
      "step": 710
    },
    {
      "epoch": 2.7363855421686747,
      "grad_norm": 0.05275186896324158,
      "learning_rate": 9.733456383463658e-07,
      "loss": 0.0023,
      "step": 711
    },
    {
      "epoch": 2.7402409638554217,
      "grad_norm": 0.05407094582915306,
      "learning_rate": 9.459541808069295e-07,
      "loss": 0.0011,
      "step": 712
    },
    {
      "epoch": 2.7440963855421687,
      "grad_norm": 0.06292422115802765,
      "learning_rate": 9.189443264619102e-07,
      "loss": 0.0026,
      "step": 713
    },
    {
      "epoch": 2.7479518072289157,
      "grad_norm": 0.044170819222927094,
      "learning_rate": 8.923166162487407e-07,
      "loss": 0.0019,
      "step": 714
    },
    {
      "epoch": 2.7518072289156628,
      "grad_norm": 0.029595479369163513,
      "learning_rate": 8.660715834514977e-07,
      "loss": 0.0013,
      "step": 715
    },
    {
      "epoch": 2.75566265060241,
      "grad_norm": 0.15282854437828064,
      "learning_rate": 8.402097536902221e-07,
      "loss": 0.005,
      "step": 716
    },
    {
      "epoch": 2.759518072289157,
      "grad_norm": 0.04469212144613266,
      "learning_rate": 8.147316449103959e-07,
      "loss": 0.0018,
      "step": 717
    },
    {
      "epoch": 2.7633734939759034,
      "grad_norm": 0.06863706558942795,
      "learning_rate": 7.896377673725553e-07,
      "loss": 0.004,
      "step": 718
    },
    {
      "epoch": 2.767228915662651,
      "grad_norm": 0.11314541101455688,
      "learning_rate": 7.649286236420806e-07,
      "loss": 0.0047,
      "step": 719
    },
    {
      "epoch": 2.7710843373493974,
      "grad_norm": 0.03980047255754471,
      "learning_rate": 7.406047085791379e-07,
      "loss": 0.0012,
      "step": 720
    },
    {
      "epoch": 2.7749397590361444,
      "grad_norm": 0.05426950752735138,
      "learning_rate": 7.166665093287539e-07,
      "loss": 0.0017,
      "step": 721
    },
    {
      "epoch": 2.7787951807228914,
      "grad_norm": 0.05432049557566643,
      "learning_rate": 6.931145053110655e-07,
      "loss": 0.0017,
      "step": 722
    },
    {
      "epoch": 2.7826506024096385,
      "grad_norm": 0.09208190441131592,
      "learning_rate": 6.69949168211721e-07,
      "loss": 0.0061,
      "step": 723
    },
    {
      "epoch": 2.7865060240963855,
      "grad_norm": 0.04900306090712547,
      "learning_rate": 6.471709619724386e-07,
      "loss": 0.0043,
      "step": 724
    },
    {
      "epoch": 2.7903614457831325,
      "grad_norm": 0.06775905936956406,
      "learning_rate": 6.247803427816945e-07,
      "loss": 0.0037,
      "step": 725
    },
    {
      "epoch": 2.7942168674698795,
      "grad_norm": 0.060494791716337204,
      "learning_rate": 6.027777590656115e-07,
      "loss": 0.0059,
      "step": 726
    },
    {
      "epoch": 2.7980722891566265,
      "grad_norm": 0.04662139341235161,
      "learning_rate": 5.811636514789598e-07,
      "loss": 0.0016,
      "step": 727
    },
    {
      "epoch": 2.8019277108433736,
      "grad_norm": 0.05482866242527962,
      "learning_rate": 5.599384528963425e-07,
      "loss": 0.0031,
      "step": 728
    },
    {
      "epoch": 2.8057831325301206,
      "grad_norm": 0.03152560442686081,
      "learning_rate": 5.391025884035239e-07,
      "loss": 0.0009,
      "step": 729
    },
    {
      "epoch": 2.8096385542168676,
      "grad_norm": 0.32685551047325134,
      "learning_rate": 5.186564752889078e-07,
      "loss": 0.0047,
      "step": 730
    },
    {
      "epoch": 2.8134939759036146,
      "grad_norm": 0.11512124538421631,
      "learning_rate": 4.986005230351954e-07,
      "loss": 0.0019,
      "step": 731
    },
    {
      "epoch": 2.8173493975903616,
      "grad_norm": 0.062155649065971375,
      "learning_rate": 4.789351333111691e-07,
      "loss": 0.0023,
      "step": 732
    },
    {
      "epoch": 2.821204819277108,
      "grad_norm": 0.0548781119287014,
      "learning_rate": 4.5966069996365993e-07,
      "loss": 0.0017,
      "step": 733
    },
    {
      "epoch": 2.8250602409638557,
      "grad_norm": 0.09939337521791458,
      "learning_rate": 4.407776090096527e-07,
      "loss": 0.0075,
      "step": 734
    },
    {
      "epoch": 2.8289156626506022,
      "grad_norm": 0.18062038719654083,
      "learning_rate": 4.22286238628562e-07,
      "loss": 0.0072,
      "step": 735
    },
    {
      "epoch": 2.8327710843373493,
      "grad_norm": 0.06187034770846367,
      "learning_rate": 4.0418695915464657e-07,
      "loss": 0.0021,
      "step": 736
    },
    {
      "epoch": 2.8366265060240963,
      "grad_norm": 0.05907326936721802,
      "learning_rate": 3.8648013306960664e-07,
      "loss": 0.0044,
      "step": 737
    },
    {
      "epoch": 2.8404819277108433,
      "grad_norm": 0.03833287954330444,
      "learning_rate": 3.691661149953096e-07,
      "loss": 0.0019,
      "step": 738
    },
    {
      "epoch": 2.8443373493975903,
      "grad_norm": 0.06432440876960754,
      "learning_rate": 3.522452516867048e-07,
      "loss": 0.0036,
      "step": 739
    },
    {
      "epoch": 2.8481927710843373,
      "grad_norm": 0.09182235598564148,
      "learning_rate": 3.357178820248641e-07,
      "loss": 0.0032,
      "step": 740
    },
    {
      "epoch": 2.8520481927710843,
      "grad_norm": 0.07224789261817932,
      "learning_rate": 3.1958433701019697e-07,
      "loss": 0.0043,
      "step": 741
    },
    {
      "epoch": 2.8559036144578314,
      "grad_norm": 0.06781880557537079,
      "learning_rate": 3.038449397558396e-07,
      "loss": 0.0065,
      "step": 742
    },
    {
      "epoch": 2.8597590361445784,
      "grad_norm": 0.05772759020328522,
      "learning_rate": 2.8850000548115155e-07,
      "loss": 0.0067,
      "step": 743
    },
    {
      "epoch": 2.8636144578313254,
      "grad_norm": 0.062191132456064224,
      "learning_rate": 2.735498415054294e-07,
      "loss": 0.0017,
      "step": 744
    },
    {
      "epoch": 2.8674698795180724,
      "grad_norm": 0.07510855793952942,
      "learning_rate": 2.5899474724174313e-07,
      "loss": 0.0062,
      "step": 745
    },
    {
      "epoch": 2.8713253012048194,
      "grad_norm": 0.09228724986314774,
      "learning_rate": 2.4483501419093836e-07,
      "loss": 0.0052,
      "step": 746
    },
    {
      "epoch": 2.8751807228915665,
      "grad_norm": 0.07365840673446655,
      "learning_rate": 2.3107092593579905e-07,
      "loss": 0.0023,
      "step": 747
    },
    {
      "epoch": 2.879036144578313,
      "grad_norm": 0.04592515155673027,
      "learning_rate": 2.1770275813536746e-07,
      "loss": 0.0025,
      "step": 748
    },
    {
      "epoch": 2.8828915662650605,
      "grad_norm": 0.10120226442813873,
      "learning_rate": 2.0473077851942858e-07,
      "loss": 0.0031,
      "step": 749
    },
    {
      "epoch": 2.886746987951807,
      "grad_norm": 0.17792415618896484,
      "learning_rate": 1.9215524688313893e-07,
      "loss": 0.014,
      "step": 750
    },
    {
      "epoch": 2.890602409638554,
      "grad_norm": 0.06266997009515762,
      "learning_rate": 1.799764150818306e-07,
      "loss": 0.002,
      "step": 751
    },
    {
      "epoch": 2.894457831325301,
      "grad_norm": 0.45810872316360474,
      "learning_rate": 1.6819452702596882e-07,
      "loss": 0.0146,
      "step": 752
    },
    {
      "epoch": 2.898313253012048,
      "grad_norm": 0.05916321277618408,
      "learning_rate": 1.5680981867625566e-07,
      "loss": 0.0015,
      "step": 753
    },
    {
      "epoch": 2.902168674698795,
      "grad_norm": 0.03866809979081154,
      "learning_rate": 1.4582251803892055e-07,
      "loss": 0.0016,
      "step": 754
    },
    {
      "epoch": 2.906024096385542,
      "grad_norm": 0.036693356931209564,
      "learning_rate": 1.3523284516113955e-07,
      "loss": 0.0012,
      "step": 755
    },
    {
      "epoch": 2.909879518072289,
      "grad_norm": 0.05886024609208107,
      "learning_rate": 1.2504101212664099e-07,
      "loss": 0.0045,
      "step": 756
    },
    {
      "epoch": 2.913734939759036,
      "grad_norm": 0.040891118347644806,
      "learning_rate": 1.1524722305144231e-07,
      "loss": 0.0016,
      "step": 757
    },
    {
      "epoch": 2.917590361445783,
      "grad_norm": 0.041443172842264175,
      "learning_rate": 1.058516740797777e-07,
      "loss": 0.0013,
      "step": 758
    },
    {
      "epoch": 2.9214457831325302,
      "grad_norm": 0.054761920124292374,
      "learning_rate": 9.685455338016347e-08,
      "loss": 0.0089,
      "step": 759
    },
    {
      "epoch": 2.9253012048192772,
      "grad_norm": 0.07846648246049881,
      "learning_rate": 8.825604114162556e-08,
      "loss": 0.0039,
      "step": 760
    },
    {
      "epoch": 2.929156626506024,
      "grad_norm": 0.056512586772441864,
      "learning_rate": 8.005630957010014e-08,
      "loss": 0.0027,
      "step": 761
    },
    {
      "epoch": 2.9330120481927713,
      "grad_norm": 0.060711685568094254,
      "learning_rate": 7.225552288497195e-08,
      "loss": 0.0024,
      "step": 762
    },
    {
      "epoch": 2.936867469879518,
      "grad_norm": 0.0869465172290802,
      "learning_rate": 6.485383731580142e-08,
      "loss": 0.0044,
      "step": 763
    },
    {
      "epoch": 2.9407228915662653,
      "grad_norm": 0.05840534344315529,
      "learning_rate": 5.785140109917819e-08,
      "loss": 0.0018,
      "step": 764
    },
    {
      "epoch": 2.944578313253012,
      "grad_norm": 0.06657087057828903,
      "learning_rate": 5.1248354475768034e-08,
      "loss": 0.0018,
      "step": 765
    },
    {
      "epoch": 2.948433734939759,
      "grad_norm": 0.14245258271694183,
      "learning_rate": 4.5044829687497236e-08,
      "loss": 0.0064,
      "step": 766
    },
    {
      "epoch": 2.952289156626506,
      "grad_norm": 0.08578988164663315,
      "learning_rate": 3.924095097489922e-08,
      "loss": 0.0024,
      "step": 767
    },
    {
      "epoch": 2.956144578313253,
      "grad_norm": 0.06886732578277588,
      "learning_rate": 3.383683457463649e-08,
      "loss": 0.0025,
      "step": 768
    },
    {
      "epoch": 2.96,
      "grad_norm": 0.04008497670292854,
      "learning_rate": 2.8832588717164766e-08,
      "loss": 0.0011,
      "step": 769
    },
    {
      "epoch": 2.963855421686747,
      "grad_norm": 0.16563624143600464,
      "learning_rate": 2.4228313624568013e-08,
      "loss": 0.0037,
      "step": 770
    },
    {
      "epoch": 2.967710843373494,
      "grad_norm": 0.08181282877922058,
      "learning_rate": 2.0024101508555604e-08,
      "loss": 0.0054,
      "step": 771
    },
    {
      "epoch": 2.971566265060241,
      "grad_norm": 0.03823351487517357,
      "learning_rate": 1.6220036568608265e-08,
      "loss": 0.0012,
      "step": 772
    },
    {
      "epoch": 2.975421686746988,
      "grad_norm": 0.05286383256316185,
      "learning_rate": 1.281619499029274e-08,
      "loss": 0.0017,
      "step": 773
    },
    {
      "epoch": 2.979277108433735,
      "grad_norm": 0.06881670653820038,
      "learning_rate": 9.812644943745231e-09,
      "loss": 0.0018,
      "step": 774
    },
    {
      "epoch": 2.983132530120482,
      "grad_norm": 0.057672757655382156,
      "learning_rate": 7.209446582292501e-09,
      "loss": 0.0015,
      "step": 775
    },
    {
      "epoch": 2.9869879518072286,
      "grad_norm": 0.04903646931052208,
      "learning_rate": 5.006652041255055e-09,
      "loss": 0.0022,
      "step": 776
    },
    {
      "epoch": 2.990843373493976,
      "grad_norm": 0.08570756763219833,
      "learning_rate": 3.2043054369057523e-09,
      "loss": 0.0015,
      "step": 777
    },
    {
      "epoch": 2.9946987951807227,
      "grad_norm": 0.16793213784694672,
      "learning_rate": 1.8024428655794012e-09,
      "loss": 0.0141,
      "step": 778
    },
    {
      "epoch": 2.99855421686747,
      "grad_norm": 0.058249011635780334,
      "learning_rate": 8.010924029533406e-10,
      "loss": 0.0067,
      "step": 779
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.03606564551591873,
      "learning_rate": 2.002741034812239e-10,
      "loss": 0.0009,
      "step": 780
    },
    {
      "epoch": 3.0,
      "step": 780,
      "total_flos": 5.918873234148688e+18,
      "train_loss": 0.010715998370104875,
      "train_runtime": 13024.8275,
      "train_samples_per_second": 3.822,
      "train_steps_per_second": 0.06
    }
  ],
  "logging_steps": 1,
  "max_steps": 780,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 92,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 5.918873234148688e+18,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}