{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 3475,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0014388489208633094,
      "grad_norm": 2.4105954227484694,
      "learning_rate": 2.2988505747126437e-07,
      "loss": 0.7695,
      "step": 1
    },
    {
      "epoch": 0.0028776978417266188,
      "grad_norm": 2.491958180835405,
      "learning_rate": 4.5977011494252875e-07,
      "loss": 0.7668,
      "step": 2
    },
    {
      "epoch": 0.004316546762589928,
      "grad_norm": 2.4515145211929483,
      "learning_rate": 6.896551724137931e-07,
      "loss": 0.778,
      "step": 3
    },
    {
      "epoch": 0.0057553956834532375,
      "grad_norm": 2.464306284194726,
      "learning_rate": 9.195402298850575e-07,
      "loss": 0.7771,
      "step": 4
    },
    {
      "epoch": 0.007194244604316547,
      "grad_norm": 2.4381995924967113,
      "learning_rate": 1.1494252873563219e-06,
      "loss": 0.7718,
      "step": 5
    },
    {
      "epoch": 0.008633093525179856,
      "grad_norm": 2.2863926582720127,
      "learning_rate": 1.3793103448275862e-06,
      "loss": 0.7634,
      "step": 6
    },
    {
      "epoch": 0.010071942446043165,
      "grad_norm": 2.347937860490624,
      "learning_rate": 1.6091954022988506e-06,
      "loss": 0.7653,
      "step": 7
    },
    {
      "epoch": 0.011510791366906475,
      "grad_norm": 1.9546287546043806,
      "learning_rate": 1.839080459770115e-06,
      "loss": 0.7385,
      "step": 8
    },
    {
      "epoch": 0.012949640287769784,
      "grad_norm": 1.85957724143468,
      "learning_rate": 2.0689655172413796e-06,
      "loss": 0.7353,
      "step": 9
    },
    {
      "epoch": 0.014388489208633094,
      "grad_norm": 1.3128121565973474,
      "learning_rate": 2.2988505747126437e-06,
      "loss": 0.7172,
      "step": 10
    },
    {
      "epoch": 0.015827338129496403,
      "grad_norm": 1.2823467031872628,
      "learning_rate": 2.5287356321839083e-06,
      "loss": 0.7171,
      "step": 11
    },
    {
      "epoch": 0.017266187050359712,
      "grad_norm": 1.1777835656701032,
      "learning_rate": 2.7586206896551725e-06,
      "loss": 0.7044,
      "step": 12
    },
    {
      "epoch": 0.01870503597122302,
      "grad_norm": 1.016282103804724,
      "learning_rate": 2.988505747126437e-06,
      "loss": 0.6922,
      "step": 13
    },
    {
      "epoch": 0.02014388489208633,
      "grad_norm": 1.124753468365821,
      "learning_rate": 3.2183908045977012e-06,
      "loss": 0.6778,
      "step": 14
    },
    {
      "epoch": 0.02158273381294964,
      "grad_norm": 1.0897618312033936,
      "learning_rate": 3.448275862068966e-06,
      "loss": 0.6684,
      "step": 15
    },
    {
      "epoch": 0.02302158273381295,
      "grad_norm": 0.9987802726468263,
      "learning_rate": 3.67816091954023e-06,
      "loss": 0.6666,
      "step": 16
    },
    {
      "epoch": 0.02446043165467626,
      "grad_norm": 0.9119063742889313,
      "learning_rate": 3.908045977011495e-06,
      "loss": 0.6553,
      "step": 17
    },
    {
      "epoch": 0.025899280575539568,
      "grad_norm": 0.6845690635011229,
      "learning_rate": 4.137931034482759e-06,
      "loss": 0.6285,
      "step": 18
    },
    {
      "epoch": 0.027338129496402876,
      "grad_norm": 0.8002047902305283,
      "learning_rate": 4.367816091954023e-06,
      "loss": 0.6057,
      "step": 19
    },
    {
      "epoch": 0.02877697841726619,
      "grad_norm": 0.743409237994264,
      "learning_rate": 4.5977011494252875e-06,
      "loss": 0.5936,
      "step": 20
    },
    {
      "epoch": 0.030215827338129497,
      "grad_norm": 0.6250234115670129,
      "learning_rate": 4.8275862068965525e-06,
      "loss": 0.593,
      "step": 21
    },
    {
      "epoch": 0.031654676258992806,
      "grad_norm": 0.5164780369653312,
      "learning_rate": 5.057471264367817e-06,
      "loss": 0.5858,
      "step": 22
    },
    {
      "epoch": 0.033093525179856115,
      "grad_norm": 0.5234892365042625,
      "learning_rate": 5.287356321839081e-06,
      "loss": 0.5919,
      "step": 23
    },
    {
      "epoch": 0.034532374100719423,
      "grad_norm": 0.5072797188187648,
      "learning_rate": 5.517241379310345e-06,
      "loss": 0.5745,
      "step": 24
    },
    {
      "epoch": 0.03597122302158273,
      "grad_norm": 0.6212376093865479,
      "learning_rate": 5.747126436781609e-06,
      "loss": 0.5645,
      "step": 25
    },
    {
      "epoch": 0.03741007194244604,
      "grad_norm": 0.6306527534732702,
      "learning_rate": 5.977011494252874e-06,
      "loss": 0.5592,
      "step": 26
    },
    {
      "epoch": 0.03884892086330935,
      "grad_norm": 0.5440901874907684,
      "learning_rate": 6.206896551724138e-06,
      "loss": 0.5634,
      "step": 27
    },
    {
      "epoch": 0.04028776978417266,
      "grad_norm": 0.4763183956174834,
      "learning_rate": 6.4367816091954025e-06,
      "loss": 0.5588,
      "step": 28
    },
    {
      "epoch": 0.041726618705035974,
      "grad_norm": 0.5298392880409118,
      "learning_rate": 6.666666666666667e-06,
      "loss": 0.5372,
      "step": 29
    },
    {
      "epoch": 0.04316546762589928,
      "grad_norm": 0.43602918957586,
      "learning_rate": 6.896551724137932e-06,
      "loss": 0.5359,
      "step": 30
    },
    {
      "epoch": 0.04460431654676259,
      "grad_norm": 0.39977253671051743,
      "learning_rate": 7.126436781609196e-06,
      "loss": 0.5391,
      "step": 31
    },
    {
      "epoch": 0.0460431654676259,
      "grad_norm": 0.40590313043219917,
      "learning_rate": 7.35632183908046e-06,
      "loss": 0.5315,
      "step": 32
    },
    {
      "epoch": 0.04748201438848921,
      "grad_norm": 0.3360824835256222,
      "learning_rate": 7.586206896551724e-06,
      "loss": 0.5276,
      "step": 33
    },
    {
      "epoch": 0.04892086330935252,
      "grad_norm": 0.2659791474022181,
      "learning_rate": 7.81609195402299e-06,
      "loss": 0.5205,
      "step": 34
    },
    {
      "epoch": 0.050359712230215826,
      "grad_norm": 0.2761746224449288,
      "learning_rate": 8.045977011494253e-06,
      "loss": 0.5231,
      "step": 35
    },
    {
      "epoch": 0.051798561151079135,
      "grad_norm": 0.27773851379862535,
      "learning_rate": 8.275862068965518e-06,
      "loss": 0.5212,
      "step": 36
    },
    {
      "epoch": 0.053237410071942444,
      "grad_norm": 0.27271770524700484,
      "learning_rate": 8.505747126436782e-06,
      "loss": 0.5081,
      "step": 37
    },
    {
      "epoch": 0.05467625899280575,
      "grad_norm": 0.27581155555156306,
      "learning_rate": 8.735632183908047e-06,
      "loss": 0.5151,
      "step": 38
    },
    {
      "epoch": 0.05611510791366906,
      "grad_norm": 0.21882673011712764,
      "learning_rate": 8.965517241379312e-06,
      "loss": 0.5062,
      "step": 39
    },
    {
      "epoch": 0.05755395683453238,
      "grad_norm": 0.1992687663487771,
      "learning_rate": 9.195402298850575e-06,
      "loss": 0.5222,
      "step": 40
    },
    {
      "epoch": 0.058992805755395686,
      "grad_norm": 0.20624278775881216,
      "learning_rate": 9.42528735632184e-06,
      "loss": 0.4948,
      "step": 41
    },
    {
      "epoch": 0.060431654676258995,
      "grad_norm": 0.20069347615852262,
      "learning_rate": 9.655172413793105e-06,
      "loss": 0.5078,
      "step": 42
    },
    {
      "epoch": 0.0618705035971223,
      "grad_norm": 0.1845034563605306,
      "learning_rate": 9.885057471264368e-06,
      "loss": 0.5051,
      "step": 43
    },
    {
      "epoch": 0.06330935251798561,
      "grad_norm": 0.17534127903696262,
      "learning_rate": 1.0114942528735633e-05,
      "loss": 0.4967,
      "step": 44
    },
    {
      "epoch": 0.06474820143884892,
      "grad_norm": 0.1750837167101689,
      "learning_rate": 1.0344827586206898e-05,
      "loss": 0.4903,
      "step": 45
    },
    {
      "epoch": 0.06618705035971223,
      "grad_norm": 0.16672130368854124,
      "learning_rate": 1.0574712643678162e-05,
      "loss": 0.4873,
      "step": 46
    },
    {
      "epoch": 0.06762589928057554,
      "grad_norm": 0.15519991857027912,
      "learning_rate": 1.0804597701149427e-05,
      "loss": 0.499,
      "step": 47
    },
    {
      "epoch": 0.06906474820143885,
      "grad_norm": 0.15004507712336698,
      "learning_rate": 1.103448275862069e-05,
      "loss": 0.4813,
      "step": 48
    },
    {
      "epoch": 0.07050359712230216,
      "grad_norm": 0.15344504306667656,
      "learning_rate": 1.1264367816091955e-05,
      "loss": 0.4813,
      "step": 49
    },
    {
      "epoch": 0.07194244604316546,
      "grad_norm": 0.14368931627859793,
      "learning_rate": 1.1494252873563218e-05,
      "loss": 0.4858,
      "step": 50
    },
    {
      "epoch": 0.07338129496402877,
      "grad_norm": 0.1304589493399049,
      "learning_rate": 1.1724137931034483e-05,
      "loss": 0.4854,
      "step": 51
    },
    {
      "epoch": 0.07482014388489208,
      "grad_norm": 0.1377449195642743,
      "learning_rate": 1.1954022988505748e-05,
      "loss": 0.4789,
      "step": 52
    },
    {
      "epoch": 0.07625899280575539,
      "grad_norm": 0.15496411963731735,
      "learning_rate": 1.2183908045977013e-05,
      "loss": 0.4853,
      "step": 53
    },
    {
      "epoch": 0.0776978417266187,
      "grad_norm": 0.13357411915265002,
      "learning_rate": 1.2413793103448277e-05,
      "loss": 0.4802,
      "step": 54
    },
    {
      "epoch": 0.07913669064748201,
      "grad_norm": 0.14332033734170296,
      "learning_rate": 1.2643678160919542e-05,
      "loss": 0.4922,
      "step": 55
    },
    {
      "epoch": 0.08057553956834532,
      "grad_norm": 0.1292592407744019,
      "learning_rate": 1.2873563218390805e-05,
      "loss": 0.4835,
      "step": 56
    },
    {
      "epoch": 0.08201438848920864,
      "grad_norm": 0.1279098304442974,
      "learning_rate": 1.310344827586207e-05,
      "loss": 0.4772,
      "step": 57
    },
    {
      "epoch": 0.08345323741007195,
      "grad_norm": 0.12546369838992738,
      "learning_rate": 1.3333333333333333e-05,
      "loss": 0.4758,
      "step": 58
    },
    {
      "epoch": 0.08489208633093526,
      "grad_norm": 0.11644069764775271,
      "learning_rate": 1.3563218390804598e-05,
      "loss": 0.4683,
      "step": 59
    },
    {
      "epoch": 0.08633093525179857,
      "grad_norm": 0.119357895427173,
      "learning_rate": 1.3793103448275863e-05,
      "loss": 0.478,
      "step": 60
    },
    {
      "epoch": 0.08776978417266187,
      "grad_norm": 0.10056032693404654,
      "learning_rate": 1.4022988505747128e-05,
      "loss": 0.4697,
      "step": 61
    },
    {
      "epoch": 0.08920863309352518,
      "grad_norm": 0.1168386481490424,
      "learning_rate": 1.4252873563218392e-05,
      "loss": 0.4645,
      "step": 62
    },
    {
      "epoch": 0.09064748201438849,
      "grad_norm": 5.064701562877305,
      "learning_rate": 1.4482758620689657e-05,
      "loss": 0.4704,
      "step": 63
    },
    {
      "epoch": 0.0920863309352518,
      "grad_norm": 0.13239631855143605,
      "learning_rate": 1.471264367816092e-05,
      "loss": 0.4666,
      "step": 64
    },
    {
      "epoch": 0.09352517985611511,
      "grad_norm": 0.10626289271608953,
      "learning_rate": 1.4942528735632185e-05,
      "loss": 0.4633,
      "step": 65
    },
    {
      "epoch": 0.09496402877697842,
      "grad_norm": 0.11998633127665166,
      "learning_rate": 1.5172413793103448e-05,
      "loss": 0.4669,
      "step": 66
    },
    {
      "epoch": 0.09640287769784173,
      "grad_norm": 0.12948805456339993,
      "learning_rate": 1.540229885057471e-05,
      "loss": 0.464,
      "step": 67
    },
    {
      "epoch": 0.09784172661870504,
      "grad_norm": 0.11881891499790373,
      "learning_rate": 1.563218390804598e-05,
      "loss": 0.4523,
      "step": 68
    },
    {
      "epoch": 0.09928057553956834,
      "grad_norm": 0.1153614566652368,
      "learning_rate": 1.586206896551724e-05,
      "loss": 0.4545,
      "step": 69
    },
    {
      "epoch": 0.10071942446043165,
      "grad_norm": 0.12797301351533724,
      "learning_rate": 1.6091954022988507e-05,
      "loss": 0.4553,
      "step": 70
    },
    {
      "epoch": 0.10215827338129496,
      "grad_norm": 0.10554962913118272,
      "learning_rate": 1.632183908045977e-05,
      "loss": 0.4641,
      "step": 71
    },
    {
      "epoch": 0.10359712230215827,
      "grad_norm": 0.12020889100383784,
      "learning_rate": 1.6551724137931037e-05,
      "loss": 0.458,
      "step": 72
    },
    {
      "epoch": 0.10503597122302158,
      "grad_norm": 0.12992964208869473,
      "learning_rate": 1.6781609195402298e-05,
      "loss": 0.4486,
      "step": 73
    },
    {
      "epoch": 0.10647482014388489,
      "grad_norm": 0.12529418437433326,
      "learning_rate": 1.7011494252873563e-05,
      "loss": 0.4487,
      "step": 74
    },
    {
      "epoch": 0.1079136690647482,
      "grad_norm": 0.13838801655443417,
      "learning_rate": 1.7241379310344828e-05,
      "loss": 0.4471,
      "step": 75
    },
    {
      "epoch": 0.1093525179856115,
      "grad_norm": 0.19416526233841697,
      "learning_rate": 1.7471264367816093e-05,
      "loss": 0.4493,
      "step": 76
    },
    {
      "epoch": 0.11079136690647481,
      "grad_norm": 0.23172341553218395,
      "learning_rate": 1.770114942528736e-05,
      "loss": 0.456,
      "step": 77
    },
    {
      "epoch": 0.11223021582733812,
      "grad_norm": 0.25887867069435166,
      "learning_rate": 1.7931034482758623e-05,
      "loss": 0.4617,
      "step": 78
    },
    {
      "epoch": 0.11366906474820145,
      "grad_norm": 0.34859196212947413,
      "learning_rate": 1.8160919540229885e-05,
      "loss": 0.4513,
      "step": 79
    },
    {
      "epoch": 0.11510791366906475,
      "grad_norm": 0.29709836278105234,
      "learning_rate": 1.839080459770115e-05,
      "loss": 0.4453,
      "step": 80
    },
    {
      "epoch": 0.11654676258992806,
      "grad_norm": 0.22415770328366957,
      "learning_rate": 1.8620689655172415e-05,
      "loss": 0.4473,
      "step": 81
    },
    {
      "epoch": 0.11798561151079137,
      "grad_norm": 0.1832543617929052,
      "learning_rate": 1.885057471264368e-05,
      "loss": 0.4504,
      "step": 82
    },
    {
      "epoch": 0.11942446043165468,
      "grad_norm": 0.16025347616908636,
      "learning_rate": 1.908045977011494e-05,
      "loss": 0.4571,
      "step": 83
    },
    {
      "epoch": 0.12086330935251799,
      "grad_norm": 0.155497884160083,
      "learning_rate": 1.931034482758621e-05,
      "loss": 0.4416,
      "step": 84
    },
    {
      "epoch": 0.1223021582733813,
      "grad_norm": 0.19185065201692705,
      "learning_rate": 1.9540229885057475e-05,
      "loss": 0.4479,
      "step": 85
    },
    {
      "epoch": 0.1237410071942446,
      "grad_norm": 0.2389793638238605,
      "learning_rate": 1.9770114942528737e-05,
      "loss": 0.4548,
      "step": 86
    },
    {
      "epoch": 0.1251798561151079,
      "grad_norm": 0.2371356137047519,
      "learning_rate": 2e-05,
      "loss": 0.4498,
      "step": 87
    },
    {
      "epoch": 0.12661870503597122,
      "grad_norm": 0.2056326072963956,
      "learning_rate": 2.0229885057471267e-05,
      "loss": 0.4441,
      "step": 88
    },
    {
      "epoch": 0.12805755395683452,
      "grad_norm": 0.1392259208106514,
      "learning_rate": 2.0459770114942528e-05,
      "loss": 0.4411,
      "step": 89
    },
    {
      "epoch": 0.12949640287769784,
      "grad_norm": 0.14101469473373407,
      "learning_rate": 2.0689655172413797e-05,
      "loss": 0.4432,
      "step": 90
    },
    {
      "epoch": 0.13093525179856116,
      "grad_norm": 0.1701408324626712,
      "learning_rate": 2.0919540229885058e-05,
      "loss": 0.4465,
      "step": 91
    },
    {
      "epoch": 0.13237410071942446,
      "grad_norm": 0.20477036899270176,
      "learning_rate": 2.1149425287356323e-05,
      "loss": 0.4422,
      "step": 92
    },
    {
      "epoch": 0.13381294964028778,
      "grad_norm": 0.19497091900720495,
      "learning_rate": 2.1379310344827585e-05,
      "loss": 0.4408,
      "step": 93
    },
    {
      "epoch": 0.13525179856115108,
      "grad_norm": 0.19755905692235068,
      "learning_rate": 2.1609195402298853e-05,
      "loss": 0.4386,
      "step": 94
    },
    {
      "epoch": 0.1366906474820144,
      "grad_norm": 0.2538534216995381,
      "learning_rate": 2.183908045977012e-05,
      "loss": 0.4411,
      "step": 95
    },
    {
      "epoch": 0.1381294964028777,
      "grad_norm": 0.33652040167995695,
      "learning_rate": 2.206896551724138e-05,
      "loss": 0.4534,
      "step": 96
    },
    {
      "epoch": 0.13956834532374102,
      "grad_norm": 0.4499557903583448,
      "learning_rate": 2.229885057471265e-05,
      "loss": 0.4337,
      "step": 97
    },
    {
      "epoch": 0.1410071942446043,
      "grad_norm": 0.4787848785232312,
      "learning_rate": 2.252873563218391e-05,
      "loss": 0.4483,
      "step": 98
    },
    {
      "epoch": 0.14244604316546763,
      "grad_norm": 0.3652726105292967,
      "learning_rate": 2.2758620689655175e-05,
      "loss": 0.4448,
      "step": 99
    },
    {
      "epoch": 0.14388489208633093,
      "grad_norm": 0.1850523458420075,
      "learning_rate": 2.2988505747126437e-05,
      "loss": 0.436,
      "step": 100
    },
    {
      "epoch": 0.14532374100719425,
      "grad_norm": 0.28013625574877,
      "learning_rate": 2.3218390804597705e-05,
      "loss": 0.4378,
      "step": 101
    },
    {
      "epoch": 0.14676258992805755,
      "grad_norm": 0.34859128737402756,
      "learning_rate": 2.3448275862068967e-05,
      "loss": 0.4485,
      "step": 102
    },
    {
      "epoch": 0.14820143884892087,
      "grad_norm": 0.2386322213525175,
      "learning_rate": 2.367816091954023e-05,
      "loss": 0.4386,
      "step": 103
    },
    {
      "epoch": 0.14964028776978416,
      "grad_norm": 0.21644678275392948,
      "learning_rate": 2.3908045977011497e-05,
      "loss": 0.4342,
      "step": 104
    },
    {
      "epoch": 0.1510791366906475,
      "grad_norm": 0.3563969211116265,
      "learning_rate": 2.413793103448276e-05,
      "loss": 0.4382,
      "step": 105
    },
    {
      "epoch": 0.15251798561151078,
      "grad_norm": 0.29362358534583805,
      "learning_rate": 2.4367816091954027e-05,
      "loss": 0.4422,
      "step": 106
    },
    {
      "epoch": 0.1539568345323741,
      "grad_norm": 0.1966080484385412,
      "learning_rate": 2.4597701149425288e-05,
      "loss": 0.436,
      "step": 107
    },
    {
      "epoch": 0.1553956834532374,
      "grad_norm": 0.3027821801910231,
      "learning_rate": 2.4827586206896553e-05,
      "loss": 0.4431,
      "step": 108
    },
    {
      "epoch": 0.15683453237410072,
      "grad_norm": 0.3459544142754879,
      "learning_rate": 2.5057471264367815e-05,
      "loss": 0.4394,
      "step": 109
    },
    {
      "epoch": 0.15827338129496402,
      "grad_norm": 0.29343882759370143,
      "learning_rate": 2.5287356321839083e-05,
      "loss": 0.4368,
      "step": 110
    },
    {
      "epoch": 0.15971223021582734,
      "grad_norm": 0.3049054900232845,
      "learning_rate": 2.551724137931035e-05,
      "loss": 0.4364,
      "step": 111
    },
    {
      "epoch": 0.16115107913669063,
      "grad_norm": 0.4170403918987231,
      "learning_rate": 2.574712643678161e-05,
      "loss": 0.4407,
      "step": 112
    },
    {
      "epoch": 0.16258992805755396,
      "grad_norm": 0.6096237752048831,
      "learning_rate": 2.597701149425288e-05,
      "loss": 0.4436,
      "step": 113
    },
    {
      "epoch": 0.16402877697841728,
      "grad_norm": 0.872737579366971,
      "learning_rate": 2.620689655172414e-05,
      "loss": 0.4469,
      "step": 114
    },
    {
      "epoch": 0.16546762589928057,
      "grad_norm": 0.7464229294451004,
      "learning_rate": 2.6436781609195405e-05,
      "loss": 0.4385,
      "step": 115
    },
    {
      "epoch": 0.1669064748201439,
      "grad_norm": 0.3430285738778335,
      "learning_rate": 2.6666666666666667e-05,
      "loss": 0.4307,
      "step": 116
    },
    {
      "epoch": 0.1683453237410072,
      "grad_norm": 0.5713328379650735,
      "learning_rate": 2.6896551724137935e-05,
      "loss": 0.4389,
      "step": 117
    },
    {
      "epoch": 0.1697841726618705,
      "grad_norm": 0.5230987770214107,
      "learning_rate": 2.7126436781609197e-05,
      "loss": 0.4335,
      "step": 118
    },
    {
      "epoch": 0.1712230215827338,
      "grad_norm": 0.5925850527079884,
      "learning_rate": 2.735632183908046e-05,
      "loss": 0.4423,
      "step": 119
    },
    {
      "epoch": 0.17266187050359713,
      "grad_norm": 0.5490079016610805,
      "learning_rate": 2.7586206896551727e-05,
      "loss": 0.4427,
      "step": 120
    },
    {
      "epoch": 0.17410071942446043,
      "grad_norm": 0.4270003450740747,
      "learning_rate": 2.781609195402299e-05,
      "loss": 0.4304,
      "step": 121
    },
    {
      "epoch": 0.17553956834532375,
      "grad_norm": 0.6019702794042444,
      "learning_rate": 2.8045977011494257e-05,
      "loss": 0.4354,
      "step": 122
    },
    {
      "epoch": 0.17697841726618704,
      "grad_norm": 0.334370783331933,
      "learning_rate": 2.8275862068965518e-05,
      "loss": 0.4429,
      "step": 123
    },
    {
      "epoch": 0.17841726618705037,
      "grad_norm": 0.47395349878838183,
      "learning_rate": 2.8505747126436783e-05,
      "loss": 0.4287,
      "step": 124
    },
    {
      "epoch": 0.17985611510791366,
      "grad_norm": 0.3038871824866802,
      "learning_rate": 2.8735632183908045e-05,
      "loss": 0.4248,
      "step": 125
    },
    {
      "epoch": 0.18129496402877698,
      "grad_norm": 0.47453715308720656,
      "learning_rate": 2.8965517241379313e-05,
      "loss": 0.4343,
      "step": 126
    },
    {
      "epoch": 0.18273381294964028,
      "grad_norm": 0.31926301463397466,
      "learning_rate": 2.919540229885058e-05,
      "loss": 0.4353,
      "step": 127
    },
    {
      "epoch": 0.1841726618705036,
      "grad_norm": 0.33941879358451277,
      "learning_rate": 2.942528735632184e-05,
      "loss": 0.4295,
      "step": 128
    },
    {
      "epoch": 0.1856115107913669,
      "grad_norm": 0.27089134597336756,
      "learning_rate": 2.965517241379311e-05,
      "loss": 0.4384,
      "step": 129
    },
    {
      "epoch": 0.18705035971223022,
      "grad_norm": 0.31129593176521775,
      "learning_rate": 2.988505747126437e-05,
      "loss": 0.4203,
      "step": 130
    },
    {
      "epoch": 0.1884892086330935,
      "grad_norm": 0.283986981943612,
      "learning_rate": 3.0114942528735635e-05,
      "loss": 0.4308,
      "step": 131
    },
    {
      "epoch": 0.18992805755395684,
      "grad_norm": 0.25674815225905595,
      "learning_rate": 3.0344827586206897e-05,
      "loss": 0.4196,
      "step": 132
    },
    {
      "epoch": 0.19136690647482013,
      "grad_norm": 0.28252583896309685,
      "learning_rate": 3.057471264367816e-05,
      "loss": 0.4237,
      "step": 133
    },
    {
      "epoch": 0.19280575539568345,
      "grad_norm": 0.26632639356638954,
      "learning_rate": 3.080459770114942e-05,
      "loss": 0.4232,
      "step": 134
    },
    {
      "epoch": 0.19424460431654678,
      "grad_norm": 0.30656043922613224,
      "learning_rate": 3.103448275862069e-05,
      "loss": 0.4289,
      "step": 135
    },
    {
      "epoch": 0.19568345323741007,
      "grad_norm": 0.25927043256426724,
      "learning_rate": 3.126436781609196e-05,
      "loss": 0.427,
      "step": 136
    },
    {
      "epoch": 0.1971223021582734,
      "grad_norm": 0.24798130007018074,
      "learning_rate": 3.149425287356322e-05,
      "loss": 0.4345,
      "step": 137
    },
    {
      "epoch": 0.1985611510791367,
      "grad_norm": 0.23475327425958922,
      "learning_rate": 3.172413793103448e-05,
      "loss": 0.4287,
      "step": 138
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.2377350899828933,
      "learning_rate": 3.195402298850575e-05,
      "loss": 0.4273,
      "step": 139
    },
    {
      "epoch": 0.2014388489208633,
      "grad_norm": 0.3071734095000456,
      "learning_rate": 3.218390804597701e-05,
      "loss": 0.4248,
      "step": 140
    },
    {
      "epoch": 0.20287769784172663,
      "grad_norm": 0.3350793570904282,
      "learning_rate": 3.2413793103448275e-05,
      "loss": 0.4261,
      "step": 141
    },
    {
      "epoch": 0.20431654676258992,
      "grad_norm": 0.43440066658930093,
      "learning_rate": 3.264367816091954e-05,
      "loss": 0.4434,
      "step": 142
    },
    {
      "epoch": 0.20575539568345325,
      "grad_norm": 0.5538333001404411,
      "learning_rate": 3.287356321839081e-05,
      "loss": 0.4326,
      "step": 143
    },
    {
      "epoch": 0.20719424460431654,
      "grad_norm": 0.5881012244147088,
      "learning_rate": 3.310344827586207e-05,
      "loss": 0.4297,
      "step": 144
    },
    {
      "epoch": 0.20863309352517986,
      "grad_norm": 0.4374203738344661,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 0.4342,
      "step": 145
    },
    {
      "epoch": 0.21007194244604316,
      "grad_norm": 0.4160120130456441,
      "learning_rate": 3.3563218390804597e-05,
      "loss": 0.4251,
      "step": 146
    },
    {
      "epoch": 0.21151079136690648,
      "grad_norm": 0.5392228648425607,
      "learning_rate": 3.3793103448275865e-05,
      "loss": 0.439,
      "step": 147
    },
    {
      "epoch": 0.21294964028776978,
      "grad_norm": 0.4290554187950581,
      "learning_rate": 3.4022988505747127e-05,
      "loss": 0.4263,
      "step": 148
    },
    {
      "epoch": 0.2143884892086331,
      "grad_norm": 0.445806647816008,
      "learning_rate": 3.4252873563218395e-05,
      "loss": 0.4266,
      "step": 149
    },
    {
      "epoch": 0.2158273381294964,
      "grad_norm": 0.5143056157416875,
      "learning_rate": 3.4482758620689657e-05,
      "loss": 0.4302,
      "step": 150
    },
    {
      "epoch": 0.21726618705035972,
      "grad_norm": 0.31334789130815455,
      "learning_rate": 3.4712643678160925e-05,
      "loss": 0.4334,
      "step": 151
    },
    {
      "epoch": 0.218705035971223,
      "grad_norm": 0.4208883505046265,
      "learning_rate": 3.4942528735632187e-05,
      "loss": 0.4311,
      "step": 152
    },
    {
      "epoch": 0.22014388489208633,
      "grad_norm": 0.34323260654772186,
      "learning_rate": 3.517241379310345e-05,
      "loss": 0.432,
      "step": 153
    },
    {
      "epoch": 0.22158273381294963,
      "grad_norm": 0.3711247664184252,
      "learning_rate": 3.540229885057472e-05,
      "loss": 0.427,
      "step": 154
    },
    {
      "epoch": 0.22302158273381295,
      "grad_norm": 0.41996688614643096,
      "learning_rate": 3.563218390804598e-05,
      "loss": 0.4256,
      "step": 155
    },
    {
      "epoch": 0.22446043165467625,
      "grad_norm": 0.4405013162802177,
      "learning_rate": 3.586206896551725e-05,
      "loss": 0.4234,
      "step": 156
    },
    {
      "epoch": 0.22589928057553957,
      "grad_norm": 0.440866752806929,
      "learning_rate": 3.609195402298851e-05,
      "loss": 0.4211,
      "step": 157
    },
    {
      "epoch": 0.2273381294964029,
      "grad_norm": 0.3803054960625481,
      "learning_rate": 3.632183908045977e-05,
      "loss": 0.4291,
      "step": 158
    },
    {
      "epoch": 0.22877697841726619,
      "grad_norm": 0.47391874678742224,
      "learning_rate": 3.655172413793104e-05,
      "loss": 0.4259,
      "step": 159
    },
    {
      "epoch": 0.2302158273381295,
      "grad_norm": 0.4766341025501577,
      "learning_rate": 3.67816091954023e-05,
      "loss": 0.4284,
      "step": 160
    },
    {
      "epoch": 0.2316546762589928,
      "grad_norm": 0.7700120448534205,
      "learning_rate": 3.701149425287357e-05,
      "loss": 0.4216,
      "step": 161
    },
    {
      "epoch": 0.23309352517985613,
      "grad_norm": 0.7842053963294792,
      "learning_rate": 3.724137931034483e-05,
      "loss": 0.4303,
      "step": 162
    },
    {
      "epoch": 0.23453237410071942,
      "grad_norm": 0.4967000795881762,
      "learning_rate": 3.74712643678161e-05,
      "loss": 0.4191,
      "step": 163
    },
    {
      "epoch": 0.23597122302158274,
      "grad_norm": 0.49658628735384647,
      "learning_rate": 3.770114942528736e-05,
      "loss": 0.4224,
      "step": 164
    },
    {
      "epoch": 0.23741007194244604,
      "grad_norm": 0.39151704923998104,
      "learning_rate": 3.793103448275862e-05,
      "loss": 0.4255,
      "step": 165
    },
    {
      "epoch": 0.23884892086330936,
      "grad_norm": 0.5367521813703079,
      "learning_rate": 3.816091954022988e-05,
      "loss": 0.4174,
      "step": 166
    },
    {
      "epoch": 0.24028776978417266,
      "grad_norm": 0.3717546418391886,
      "learning_rate": 3.839080459770115e-05,
      "loss": 0.4245,
      "step": 167
    },
    {
      "epoch": 0.24172661870503598,
      "grad_norm": 0.36917102874526303,
      "learning_rate": 3.862068965517242e-05,
      "loss": 0.4175,
      "step": 168
    },
    {
      "epoch": 0.24316546762589927,
      "grad_norm": 0.3939287560108561,
      "learning_rate": 3.885057471264368e-05,
      "loss": 0.4254,
      "step": 169
    },
    {
      "epoch": 0.2446043165467626,
      "grad_norm": 0.3705339045689148,
      "learning_rate": 3.908045977011495e-05,
      "loss": 0.4183,
      "step": 170
    },
    {
      "epoch": 0.2460431654676259,
      "grad_norm": 0.3977076335757685,
      "learning_rate": 3.931034482758621e-05,
      "loss": 0.4359,
      "step": 171
    },
    {
      "epoch": 0.2474820143884892,
      "grad_norm": 0.3239142876763284,
      "learning_rate": 3.954022988505747e-05,
      "loss": 0.4131,
      "step": 172
    },
    {
      "epoch": 0.2489208633093525,
      "grad_norm": 0.4004560562375386,
      "learning_rate": 3.9770114942528735e-05,
      "loss": 0.4154,
      "step": 173
    },
    {
      "epoch": 0.2503597122302158,
      "grad_norm": 0.3482246073839111,
      "learning_rate": 4e-05,
      "loss": 0.4191,
      "step": 174
    },
    {
      "epoch": 0.2517985611510791,
      "grad_norm": 0.30916823043183983,
      "learning_rate": 4.022988505747127e-05,
      "loss": 0.4296,
      "step": 175
    },
    {
      "epoch": 0.25323741007194245,
      "grad_norm": 0.3926285238360458,
      "learning_rate": 4.045977011494253e-05,
      "loss": 0.4298,
      "step": 176
    },
    {
      "epoch": 0.25467625899280577,
      "grad_norm": 0.3473702277340475,
      "learning_rate": 4.0689655172413795e-05,
      "loss": 0.4276,
      "step": 177
    },
    {
      "epoch": 0.25611510791366904,
      "grad_norm": 0.39203554127421786,
      "learning_rate": 4.0919540229885057e-05,
      "loss": 0.4294,
      "step": 178
    },
    {
      "epoch": 0.25755395683453236,
      "grad_norm": 0.36858650708676194,
      "learning_rate": 4.1149425287356325e-05,
      "loss": 0.4272,
      "step": 179
    },
    {
      "epoch": 0.2589928057553957,
      "grad_norm": 0.35331416241514024,
      "learning_rate": 4.137931034482759e-05,
      "loss": 0.4165,
      "step": 180
    },
    {
      "epoch": 0.260431654676259,
      "grad_norm": 0.3650719681137465,
      "learning_rate": 4.160919540229885e-05,
      "loss": 0.4246,
      "step": 181
    },
    {
      "epoch": 0.26187050359712233,
      "grad_norm": 0.396532271347279,
      "learning_rate": 4.1839080459770117e-05,
      "loss": 0.4077,
      "step": 182
    },
    {
      "epoch": 0.2633093525179856,
      "grad_norm": 0.5027339023410375,
      "learning_rate": 4.2068965517241385e-05,
      "loss": 0.4171,
      "step": 183
    },
    {
      "epoch": 0.2647482014388489,
      "grad_norm": 0.5398000888843012,
      "learning_rate": 4.2298850574712647e-05,
      "loss": 0.4183,
      "step": 184
    },
    {
      "epoch": 0.26618705035971224,
      "grad_norm": 0.388305209367546,
      "learning_rate": 4.2528735632183915e-05,
      "loss": 0.4215,
      "step": 185
    },
    {
      "epoch": 0.26762589928057556,
      "grad_norm": 0.3594204739218431,
      "learning_rate": 4.275862068965517e-05,
      "loss": 0.4192,
      "step": 186
    },
    {
      "epoch": 0.26906474820143883,
      "grad_norm": 0.40748119218221474,
      "learning_rate": 4.298850574712644e-05,
      "loss": 0.4267,
      "step": 187
    },
    {
      "epoch": 0.27050359712230215,
      "grad_norm": 0.4831661157601768,
      "learning_rate": 4.321839080459771e-05,
      "loss": 0.4161,
      "step": 188
    },
    {
      "epoch": 0.2719424460431655,
      "grad_norm": 0.6431491770515896,
      "learning_rate": 4.344827586206897e-05,
      "loss": 0.4214,
      "step": 189
    },
    {
      "epoch": 0.2733812949640288,
      "grad_norm": 0.7689279391826508,
      "learning_rate": 4.367816091954024e-05,
      "loss": 0.4326,
      "step": 190
    },
    {
      "epoch": 0.27482014388489207,
      "grad_norm": 0.8194099131112732,
      "learning_rate": 4.39080459770115e-05,
      "loss": 0.4224,
      "step": 191
    },
    {
      "epoch": 0.2762589928057554,
      "grad_norm": 0.5719341880685053,
      "learning_rate": 4.413793103448276e-05,
      "loss": 0.4142,
      "step": 192
    },
    {
      "epoch": 0.2776978417266187,
      "grad_norm": 0.48369772529062066,
      "learning_rate": 4.436781609195403e-05,
      "loss": 0.4178,
      "step": 193
    },
    {
      "epoch": 0.27913669064748203,
      "grad_norm": 0.5930946177166372,
      "learning_rate": 4.45977011494253e-05,
      "loss": 0.4179,
      "step": 194
    },
    {
      "epoch": 0.2805755395683453,
      "grad_norm": 0.4258909411652574,
      "learning_rate": 4.482758620689655e-05,
      "loss": 0.4234,
      "step": 195
    },
    {
      "epoch": 0.2820143884892086,
      "grad_norm": 0.38845248341947575,
      "learning_rate": 4.505747126436782e-05,
      "loss": 0.4177,
      "step": 196
    },
    {
      "epoch": 0.28345323741007195,
      "grad_norm": 0.5374461347132847,
      "learning_rate": 4.528735632183908e-05,
      "loss": 0.4186,
      "step": 197
    },
    {
      "epoch": 0.28489208633093527,
      "grad_norm": 0.414301384735006,
      "learning_rate": 4.551724137931035e-05,
      "loss": 0.4123,
      "step": 198
    },
    {
      "epoch": 0.28633093525179854,
      "grad_norm": 0.3421929046493428,
      "learning_rate": 4.574712643678162e-05,
      "loss": 0.4116,
      "step": 199
    },
    {
      "epoch": 0.28776978417266186,
      "grad_norm": 0.4720206403909787,
      "learning_rate": 4.597701149425287e-05,
      "loss": 0.4212,
      "step": 200
    },
    {
      "epoch": 0.2892086330935252,
      "grad_norm": 0.41305656015425807,
      "learning_rate": 4.620689655172414e-05,
      "loss": 0.4112,
      "step": 201
    },
    {
      "epoch": 0.2906474820143885,
      "grad_norm": 0.303624198533764,
      "learning_rate": 4.643678160919541e-05,
      "loss": 0.4066,
      "step": 202
    },
    {
      "epoch": 0.2920863309352518,
      "grad_norm": 0.4023935675809489,
      "learning_rate": 4.666666666666667e-05,
      "loss": 0.4195,
      "step": 203
    },
    {
      "epoch": 0.2935251798561151,
      "grad_norm": 0.4065299617485997,
      "learning_rate": 4.689655172413793e-05,
      "loss": 0.416,
      "step": 204
    },
    {
      "epoch": 0.2949640287769784,
      "grad_norm": 0.4328671507220006,
      "learning_rate": 4.7126436781609195e-05,
      "loss": 0.4193,
      "step": 205
    },
    {
      "epoch": 0.29640287769784174,
      "grad_norm": 0.5357838410994751,
      "learning_rate": 4.735632183908046e-05,
      "loss": 0.4117,
      "step": 206
    },
    {
      "epoch": 0.29784172661870506,
      "grad_norm": 0.4771204429978033,
      "learning_rate": 4.758620689655173e-05,
      "loss": 0.412,
      "step": 207
    },
    {
      "epoch": 0.2992805755395683,
      "grad_norm": 0.437631567776442,
      "learning_rate": 4.781609195402299e-05,
      "loss": 0.4159,
      "step": 208
    },
    {
      "epoch": 0.30071942446043165,
      "grad_norm": 0.5244670541769951,
      "learning_rate": 4.8045977011494255e-05,
      "loss": 0.4065,
      "step": 209
    },
    {
      "epoch": 0.302158273381295,
      "grad_norm": 0.47470433749275237,
      "learning_rate": 4.827586206896552e-05,
      "loss": 0.4165,
      "step": 210
    },
    {
      "epoch": 0.3035971223021583,
      "grad_norm": 0.37019131484391476,
      "learning_rate": 4.8505747126436785e-05,
      "loss": 0.4027,
      "step": 211
    },
    {
      "epoch": 0.30503597122302156,
      "grad_norm": 0.3677924799998598,
      "learning_rate": 4.873563218390805e-05,
      "loss": 0.4221,
      "step": 212
    },
    {
      "epoch": 0.3064748201438849,
      "grad_norm": 0.4213221362717633,
      "learning_rate": 4.896551724137931e-05,
      "loss": 0.4136,
      "step": 213
    },
    {
      "epoch": 0.3079136690647482,
      "grad_norm": 0.5005616556129306,
      "learning_rate": 4.9195402298850577e-05,
      "loss": 0.4146,
      "step": 214
    },
    {
      "epoch": 0.30935251798561153,
      "grad_norm": 0.5660260676340962,
      "learning_rate": 4.9425287356321845e-05,
      "loss": 0.415,
      "step": 215
    },
    {
      "epoch": 0.3107913669064748,
      "grad_norm": 0.6288560002121155,
      "learning_rate": 4.9655172413793107e-05,
      "loss": 0.4127,
      "step": 216
    },
    {
      "epoch": 0.3122302158273381,
      "grad_norm": 0.629138529219548,
      "learning_rate": 4.9885057471264375e-05,
      "loss": 0.4139,
      "step": 217
    },
    {
      "epoch": 0.31366906474820144,
      "grad_norm": 0.423519291278689,
      "learning_rate": 5.011494252873563e-05,
      "loss": 0.4143,
      "step": 218
    },
    {
      "epoch": 0.31510791366906477,
      "grad_norm": 0.34837499698479096,
      "learning_rate": 5.03448275862069e-05,
      "loss": 0.4143,
      "step": 219
    },
    {
      "epoch": 0.31654676258992803,
      "grad_norm": 0.4615319088669839,
      "learning_rate": 5.057471264367817e-05,
      "loss": 0.4126,
      "step": 220
    },
    {
      "epoch": 0.31798561151079136,
      "grad_norm": 0.4937186470978348,
      "learning_rate": 5.0804597701149435e-05,
      "loss": 0.4138,
      "step": 221
    },
    {
      "epoch": 0.3194244604316547,
      "grad_norm": 0.414189734087535,
      "learning_rate": 5.10344827586207e-05,
      "loss": 0.4217,
      "step": 222
    },
    {
      "epoch": 0.320863309352518,
      "grad_norm": 0.28717519451044027,
      "learning_rate": 5.126436781609196e-05,
      "loss": 0.4143,
      "step": 223
    },
    {
      "epoch": 0.32230215827338127,
      "grad_norm": 0.48031210981271294,
      "learning_rate": 5.149425287356322e-05,
      "loss": 0.4211,
      "step": 224
    },
    {
      "epoch": 0.3237410071942446,
      "grad_norm": 0.4729273176257217,
      "learning_rate": 5.172413793103449e-05,
      "loss": 0.4188,
      "step": 225
    },
    {
      "epoch": 0.3251798561151079,
      "grad_norm": 0.25093035536711344,
      "learning_rate": 5.195402298850576e-05,
      "loss": 0.4087,
      "step": 226
    },
    {
      "epoch": 0.32661870503597124,
      "grad_norm": 0.3013410763134809,
      "learning_rate": 5.218390804597701e-05,
      "loss": 0.4158,
      "step": 227
    },
    {
      "epoch": 0.32805755395683456,
      "grad_norm": 0.44716508261402804,
      "learning_rate": 5.241379310344828e-05,
      "loss": 0.4099,
      "step": 228
    },
    {
      "epoch": 0.3294964028776978,
      "grad_norm": 0.3809050702014656,
      "learning_rate": 5.264367816091954e-05,
      "loss": 0.4113,
      "step": 229
    },
    {
      "epoch": 0.33093525179856115,
      "grad_norm": 0.20636335745276008,
      "learning_rate": 5.287356321839081e-05,
      "loss": 0.4105,
      "step": 230
    },
    {
      "epoch": 0.33237410071942447,
      "grad_norm": 0.3511948529002789,
      "learning_rate": 5.310344827586208e-05,
      "loss": 0.4185,
      "step": 231
    },
    {
      "epoch": 0.3338129496402878,
      "grad_norm": 0.40331439495888555,
      "learning_rate": 5.333333333333333e-05,
      "loss": 0.4221,
      "step": 232
    },
    {
      "epoch": 0.33525179856115106,
      "grad_norm": 0.34719092742015417,
      "learning_rate": 5.35632183908046e-05,
      "loss": 0.4145,
      "step": 233
    },
    {
      "epoch": 0.3366906474820144,
      "grad_norm": 0.5114677709909795,
      "learning_rate": 5.379310344827587e-05,
      "loss": 0.4112,
      "step": 234
    },
    {
      "epoch": 0.3381294964028777,
      "grad_norm": 0.7803945133643926,
      "learning_rate": 5.402298850574713e-05,
      "loss": 0.4165,
      "step": 235
    },
    {
      "epoch": 0.339568345323741,
      "grad_norm": 0.9609656354531556,
      "learning_rate": 5.425287356321839e-05,
      "loss": 0.4212,
      "step": 236
    },
    {
      "epoch": 0.3410071942446043,
      "grad_norm": 0.8077612130876954,
      "learning_rate": 5.4482758620689655e-05,
      "loss": 0.4195,
      "step": 237
    },
    {
      "epoch": 0.3424460431654676,
      "grad_norm": 0.44530673257470504,
      "learning_rate": 5.471264367816092e-05,
      "loss": 0.415,
      "step": 238
    },
    {
      "epoch": 0.34388489208633094,
      "grad_norm": 1.1511565326195332,
      "learning_rate": 5.494252873563219e-05,
      "loss": 0.4222,
      "step": 239
    },
    {
      "epoch": 0.34532374100719426,
      "grad_norm": 0.6381960947846417,
      "learning_rate": 5.517241379310345e-05,
      "loss": 0.4246,
      "step": 240
    },
    {
      "epoch": 0.34676258992805753,
      "grad_norm": 0.49519466414702334,
      "learning_rate": 5.5402298850574715e-05,
      "loss": 0.4146,
      "step": 241
    },
    {
      "epoch": 0.34820143884892085,
      "grad_norm": 0.5237504473154212,
      "learning_rate": 5.563218390804598e-05,
      "loss": 0.4192,
      "step": 242
    },
    {
      "epoch": 0.3496402877697842,
      "grad_norm": 0.5941024386408629,
      "learning_rate": 5.5862068965517245e-05,
      "loss": 0.4262,
      "step": 243
    },
    {
      "epoch": 0.3510791366906475,
      "grad_norm": 0.5466007109154629,
      "learning_rate": 5.609195402298851e-05,
      "loss": 0.4207,
      "step": 244
    },
    {
      "epoch": 0.35251798561151076,
      "grad_norm": 0.5070998561917002,
      "learning_rate": 5.632183908045977e-05,
      "loss": 0.4178,
      "step": 245
    },
    {
      "epoch": 0.3539568345323741,
      "grad_norm": 0.5532223210228329,
      "learning_rate": 5.6551724137931037e-05,
      "loss": 0.4204,
      "step": 246
    },
    {
      "epoch": 0.3553956834532374,
      "grad_norm": 0.5476749200012132,
      "learning_rate": 5.6781609195402305e-05,
      "loss": 0.4173,
      "step": 247
    },
    {
      "epoch": 0.35683453237410073,
      "grad_norm": 0.4702862030858147,
      "learning_rate": 5.7011494252873567e-05,
      "loss": 0.4208,
      "step": 248
    },
    {
      "epoch": 0.35827338129496406,
      "grad_norm": 0.4491819673301651,
      "learning_rate": 5.7241379310344835e-05,
      "loss": 0.4189,
      "step": 249
    },
    {
      "epoch": 0.3597122302158273,
      "grad_norm": 0.47023442603839466,
      "learning_rate": 5.747126436781609e-05,
      "loss": 0.4223,
      "step": 250
    },
    {
      "epoch": 0.36115107913669064,
      "grad_norm": 0.39940693933184623,
      "learning_rate": 5.770114942528736e-05,
      "loss": 0.4097,
      "step": 251
    },
    {
      "epoch": 0.36258992805755397,
      "grad_norm": 0.3758340984896894,
      "learning_rate": 5.7931034482758627e-05,
      "loss": 0.4172,
      "step": 252
    },
    {
      "epoch": 0.3640287769784173,
      "grad_norm": 0.40155533823476564,
      "learning_rate": 5.8160919540229895e-05,
      "loss": 0.4151,
      "step": 253
    },
    {
      "epoch": 0.36546762589928056,
      "grad_norm": 0.34253263674719914,
      "learning_rate": 5.839080459770116e-05,
      "loss": 0.4203,
      "step": 254
    },
    {
      "epoch": 0.3669064748201439,
      "grad_norm": 0.36390271929883083,
      "learning_rate": 5.862068965517242e-05,
      "loss": 0.4134,
      "step": 255
    },
    {
      "epoch": 0.3683453237410072,
      "grad_norm": 0.33231707017315143,
      "learning_rate": 5.885057471264368e-05,
      "loss": 0.419,
      "step": 256
    },
    {
      "epoch": 0.3697841726618705,
      "grad_norm": 0.33117090224747675,
      "learning_rate": 5.908045977011495e-05,
      "loss": 0.4105,
      "step": 257
    },
    {
      "epoch": 0.3712230215827338,
      "grad_norm": 0.3823966378687382,
      "learning_rate": 5.931034482758622e-05,
      "loss": 0.4271,
      "step": 258
    },
    {
      "epoch": 0.3726618705035971,
      "grad_norm": 0.3940172882104445,
      "learning_rate": 5.954022988505747e-05,
      "loss": 0.4157,
      "step": 259
    },
    {
      "epoch": 0.37410071942446044,
      "grad_norm": 0.3871761128804785,
      "learning_rate": 5.977011494252874e-05,
      "loss": 0.4193,
      "step": 260
    },
    {
      "epoch": 0.37553956834532376,
      "grad_norm": 0.5047843144990062,
      "learning_rate": 6.000000000000001e-05,
      "loss": 0.4194,
      "step": 261
    },
    {
      "epoch": 0.376978417266187,
      "grad_norm": 0.703100287188996,
      "learning_rate": 6.022988505747127e-05,
      "loss": 0.4153,
      "step": 262
    },
    {
      "epoch": 0.37841726618705035,
      "grad_norm": 0.8316949370232966,
      "learning_rate": 6.045977011494254e-05,
      "loss": 0.423,
      "step": 263
    },
    {
      "epoch": 0.37985611510791367,
      "grad_norm": 0.6133608576551307,
      "learning_rate": 6.068965517241379e-05,
      "loss": 0.4203,
      "step": 264
    },
    {
      "epoch": 0.381294964028777,
      "grad_norm": 0.3077241840978202,
      "learning_rate": 6.091954022988506e-05,
      "loss": 0.4138,
      "step": 265
    },
    {
      "epoch": 0.38273381294964026,
      "grad_norm": 0.574646804178916,
      "learning_rate": 6.114942528735632e-05,
      "loss": 0.4126,
      "step": 266
    },
    {
      "epoch": 0.3841726618705036,
      "grad_norm": 0.5005718982159811,
      "learning_rate": 6.137931034482759e-05,
      "loss": 0.4171,
      "step": 267
    },
    {
      "epoch": 0.3856115107913669,
      "grad_norm": 0.3374353201004547,
      "learning_rate": 6.160919540229885e-05,
      "loss": 0.4084,
      "step": 268
    },
    {
      "epoch": 0.38705035971223023,
      "grad_norm": 0.4562006878600622,
      "learning_rate": 6.183908045977011e-05,
      "loss": 0.4163,
      "step": 269
    },
    {
      "epoch": 0.38848920863309355,
      "grad_norm": 0.48033384427276327,
      "learning_rate": 6.206896551724138e-05,
      "loss": 0.4065,
      "step": 270
    },
    {
      "epoch": 0.3899280575539568,
      "grad_norm": 0.32216192622007456,
      "learning_rate": 6.229885057471265e-05,
      "loss": 0.4112,
      "step": 271
    },
    {
      "epoch": 0.39136690647482014,
      "grad_norm": 0.44669616782797317,
      "learning_rate": 6.252873563218392e-05,
      "loss": 0.4079,
      "step": 272
    },
    {
      "epoch": 0.39280575539568346,
      "grad_norm": 0.4476950772763014,
      "learning_rate": 6.275862068965517e-05,
      "loss": 0.4111,
      "step": 273
    },
    {
      "epoch": 0.3942446043165468,
      "grad_norm": 0.2886411172690327,
      "learning_rate": 6.298850574712644e-05,
      "loss": 0.4043,
      "step": 274
    },
    {
      "epoch": 0.39568345323741005,
      "grad_norm": 0.3702149250397203,
      "learning_rate": 6.321839080459771e-05,
      "loss": 0.402,
      "step": 275
    },
    {
      "epoch": 0.3971223021582734,
      "grad_norm": 0.3735855064136341,
      "learning_rate": 6.344827586206897e-05,
      "loss": 0.4097,
      "step": 276
    },
    {
      "epoch": 0.3985611510791367,
      "grad_norm": 0.2578446895913922,
      "learning_rate": 6.367816091954023e-05,
      "loss": 0.4085,
      "step": 277
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.36680592540229395,
      "learning_rate": 6.39080459770115e-05,
      "loss": 0.41,
      "step": 278
    },
    {
      "epoch": 0.4014388489208633,
      "grad_norm": 0.2847257307786734,
      "learning_rate": 6.413793103448276e-05,
      "loss": 0.4083,
      "step": 279
    },
    {
      "epoch": 0.4028776978417266,
      "grad_norm": 0.31354966366517834,
      "learning_rate": 6.436781609195403e-05,
      "loss": 0.4079,
      "step": 280
    },
    {
      "epoch": 0.40431654676258993,
      "grad_norm": 0.34653289172604856,
      "learning_rate": 6.45977011494253e-05,
      "loss": 0.4108,
      "step": 281
    },
    {
      "epoch": 0.40575539568345326,
      "grad_norm": 0.24017294677401724,
      "learning_rate": 6.482758620689655e-05,
      "loss": 0.3986,
      "step": 282
    },
    {
      "epoch": 0.4071942446043165,
      "grad_norm": 0.26441107666341884,
      "learning_rate": 6.505747126436782e-05,
      "loss": 0.4123,
      "step": 283
    },
    {
      "epoch": 0.40863309352517985,
      "grad_norm": 0.2911074896661384,
      "learning_rate": 6.528735632183909e-05,
      "loss": 0.4196,
      "step": 284
    },
    {
      "epoch": 0.41007194244604317,
      "grad_norm": 0.338550336780261,
      "learning_rate": 6.551724137931035e-05,
      "loss": 0.4039,
      "step": 285
    },
    {
      "epoch": 0.4115107913669065,
      "grad_norm": 0.4212636484391325,
      "learning_rate": 6.574712643678162e-05,
      "loss": 0.4191,
      "step": 286
    },
    {
      "epoch": 0.41294964028776976,
      "grad_norm": 0.4720387151383318,
      "learning_rate": 6.597701149425288e-05,
      "loss": 0.3996,
      "step": 287
    },
    {
      "epoch": 0.4143884892086331,
      "grad_norm": 0.5665310272932965,
      "learning_rate": 6.620689655172415e-05,
      "loss": 0.4168,
      "step": 288
    },
    {
      "epoch": 0.4158273381294964,
      "grad_norm": 0.6879396891468814,
      "learning_rate": 6.643678160919542e-05,
      "loss": 0.4113,
      "step": 289
    },
    {
      "epoch": 0.4172661870503597,
      "grad_norm": 0.9471828010742368,
      "learning_rate": 6.666666666666667e-05,
      "loss": 0.4192,
      "step": 290
    },
    {
      "epoch": 0.418705035971223,
      "grad_norm": 0.589750102641128,
      "learning_rate": 6.689655172413794e-05,
      "loss": 0.3972,
      "step": 291
    },
    {
      "epoch": 0.4201438848920863,
      "grad_norm": 0.512995567918449,
      "learning_rate": 6.712643678160919e-05,
      "loss": 0.4174,
      "step": 292
    },
    {
      "epoch": 0.42158273381294964,
      "grad_norm": 0.6129621783395007,
      "learning_rate": 6.735632183908046e-05,
      "loss": 0.4188,
      "step": 293
    },
    {
      "epoch": 0.42302158273381296,
      "grad_norm": 0.6662371744372287,
      "learning_rate": 6.758620689655173e-05,
      "loss": 0.4128,
      "step": 294
    },
    {
      "epoch": 0.4244604316546763,
      "grad_norm": 0.49955639912919664,
      "learning_rate": 6.7816091954023e-05,
      "loss": 0.4103,
      "step": 295
    },
    {
      "epoch": 0.42589928057553955,
      "grad_norm": 0.659145883377898,
      "learning_rate": 6.804597701149425e-05,
      "loss": 0.4103,
      "step": 296
    },
    {
      "epoch": 0.4273381294964029,
      "grad_norm": 0.5342696552963402,
      "learning_rate": 6.827586206896552e-05,
      "loss": 0.4125,
      "step": 297
    },
    {
      "epoch": 0.4287769784172662,
      "grad_norm": 0.5954892014584885,
      "learning_rate": 6.850574712643679e-05,
      "loss": 0.4085,
      "step": 298
    },
    {
      "epoch": 0.4302158273381295,
      "grad_norm": 0.5668914244262512,
      "learning_rate": 6.873563218390806e-05,
      "loss": 0.4121,
      "step": 299
    },
    {
      "epoch": 0.4316546762589928,
      "grad_norm": 0.5107590244214486,
      "learning_rate": 6.896551724137931e-05,
      "loss": 0.4106,
      "step": 300
    },
    {
      "epoch": 0.4330935251798561,
      "grad_norm": 0.6755319559179165,
      "learning_rate": 6.919540229885058e-05,
      "loss": 0.4117,
      "step": 301
    },
    {
      "epoch": 0.43453237410071943,
      "grad_norm": 0.4179083230881996,
      "learning_rate": 6.942528735632185e-05,
      "loss": 0.4046,
      "step": 302
    },
    {
      "epoch": 0.43597122302158275,
      "grad_norm": 0.5770119275066495,
      "learning_rate": 6.96551724137931e-05,
      "loss": 0.4123,
      "step": 303
    },
    {
      "epoch": 0.437410071942446,
      "grad_norm": 0.4400220981417635,
      "learning_rate": 6.988505747126437e-05,
      "loss": 0.414,
      "step": 304
    },
    {
      "epoch": 0.43884892086330934,
      "grad_norm": 0.7787505087188255,
      "learning_rate": 7.011494252873563e-05,
      "loss": 0.4154,
      "step": 305
    },
    {
      "epoch": 0.44028776978417267,
      "grad_norm": 0.5438951189611558,
      "learning_rate": 7.03448275862069e-05,
      "loss": 0.4103,
      "step": 306
    },
    {
      "epoch": 0.441726618705036,
      "grad_norm": 0.7182301959610059,
      "learning_rate": 7.057471264367816e-05,
      "loss": 0.4248,
      "step": 307
    },
    {
      "epoch": 0.44316546762589926,
      "grad_norm": 0.784357548693455,
      "learning_rate": 7.080459770114943e-05,
      "loss": 0.4248,
      "step": 308
    },
    {
      "epoch": 0.4446043165467626,
      "grad_norm": 1.4074663898040851,
      "learning_rate": 7.10344827586207e-05,
      "loss": 0.4331,
      "step": 309
    },
    {
      "epoch": 0.4460431654676259,
      "grad_norm": 0.6547039263701318,
      "learning_rate": 7.126436781609196e-05,
      "loss": 0.4134,
      "step": 310
    },
    {
      "epoch": 0.4474820143884892,
      "grad_norm": 0.7020988819775665,
      "learning_rate": 7.149425287356322e-05,
      "loss": 0.4173,
      "step": 311
    },
    {
      "epoch": 0.4489208633093525,
      "grad_norm": 0.5880726013717825,
      "learning_rate": 7.17241379310345e-05,
      "loss": 0.4129,
      "step": 312
    },
    {
      "epoch": 0.4503597122302158,
      "grad_norm": 0.5521119116996107,
      "learning_rate": 7.195402298850576e-05,
      "loss": 0.4153,
      "step": 313
    },
    {
      "epoch": 0.45179856115107914,
      "grad_norm": 3.9881369791462293,
      "learning_rate": 7.218390804597702e-05,
      "loss": 0.4232,
      "step": 314
    },
    {
      "epoch": 0.45323741007194246,
      "grad_norm": 0.7734742568120329,
      "learning_rate": 7.241379310344828e-05,
      "loss": 0.4254,
      "step": 315
    },
    {
      "epoch": 0.4546762589928058,
      "grad_norm": 0.6392054913190713,
      "learning_rate": 7.264367816091954e-05,
      "loss": 0.4154,
      "step": 316
    },
    {
      "epoch": 0.45611510791366905,
      "grad_norm": 0.5343121642978103,
      "learning_rate": 7.287356321839081e-05,
      "loss": 0.4183,
      "step": 317
    },
    {
      "epoch": 0.45755395683453237,
      "grad_norm": 0.534464146362328,
      "learning_rate": 7.310344827586208e-05,
      "loss": 0.4176,
      "step": 318
    },
    {
      "epoch": 0.4589928057553957,
      "grad_norm": 0.4591671523934069,
      "learning_rate": 7.333333333333333e-05,
      "loss": 0.4065,
      "step": 319
    },
    {
      "epoch": 0.460431654676259,
      "grad_norm": 0.4570818633965499,
      "learning_rate": 7.35632183908046e-05,
      "loss": 0.4183,
      "step": 320
    },
    {
      "epoch": 0.4618705035971223,
      "grad_norm": 0.4108070417925211,
      "learning_rate": 7.379310344827587e-05,
      "loss": 0.4233,
      "step": 321
    },
    {
      "epoch": 0.4633093525179856,
      "grad_norm": 0.40859086843130993,
      "learning_rate": 7.402298850574714e-05,
      "loss": 0.4126,
      "step": 322
    },
    {
      "epoch": 0.46474820143884893,
      "grad_norm": 0.46481655751060136,
      "learning_rate": 7.425287356321839e-05,
      "loss": 0.4162,
      "step": 323
    },
    {
      "epoch": 0.46618705035971225,
      "grad_norm": 0.35680701465011644,
      "learning_rate": 7.448275862068966e-05,
      "loss": 0.4112,
      "step": 324
    },
    {
      "epoch": 0.4676258992805755,
      "grad_norm": 0.39342147794665006,
      "learning_rate": 7.471264367816093e-05,
      "loss": 0.4087,
      "step": 325
    },
    {
      "epoch": 0.46906474820143884,
      "grad_norm": 0.343319754952168,
      "learning_rate": 7.49425287356322e-05,
      "loss": 0.4033,
      "step": 326
    },
    {
      "epoch": 0.47050359712230216,
      "grad_norm": 0.32852062718493036,
      "learning_rate": 7.517241379310345e-05,
      "loss": 0.4135,
      "step": 327
    },
    {
      "epoch": 0.4719424460431655,
      "grad_norm": 0.3022324788382363,
      "learning_rate": 7.540229885057472e-05,
      "loss": 0.4046,
      "step": 328
    },
    {
      "epoch": 0.47338129496402875,
      "grad_norm": 0.326440648315175,
      "learning_rate": 7.563218390804599e-05,
      "loss": 0.4215,
      "step": 329
    },
    {
      "epoch": 0.4748201438848921,
      "grad_norm": 0.31595727192347783,
      "learning_rate": 7.586206896551724e-05,
      "loss": 0.4141,
      "step": 330
    },
    {
      "epoch": 0.4762589928057554,
      "grad_norm": 0.2663905387567008,
      "learning_rate": 7.609195402298851e-05,
      "loss": 0.4081,
      "step": 331
    },
    {
      "epoch": 0.4776978417266187,
      "grad_norm": 0.278590485517421,
      "learning_rate": 7.632183908045977e-05,
      "loss": 0.4047,
      "step": 332
    },
    {
      "epoch": 0.479136690647482,
      "grad_norm": 0.23279241923894028,
      "learning_rate": 7.655172413793103e-05,
      "loss": 0.3964,
      "step": 333
    },
    {
      "epoch": 0.4805755395683453,
      "grad_norm": 0.29081713635668505,
      "learning_rate": 7.67816091954023e-05,
      "loss": 0.4087,
      "step": 334
    },
    {
      "epoch": 0.48201438848920863,
      "grad_norm": 0.21801609645937103,
      "learning_rate": 7.701149425287357e-05,
      "loss": 0.4096,
      "step": 335
    },
    {
      "epoch": 0.48345323741007196,
      "grad_norm": 0.2100066170024453,
      "learning_rate": 7.724137931034484e-05,
      "loss": 0.4019,
      "step": 336
    },
    {
      "epoch": 0.4848920863309353,
      "grad_norm": 0.21436855692470322,
      "learning_rate": 7.74712643678161e-05,
      "loss": 0.403,
      "step": 337
    },
    {
      "epoch": 0.48633093525179855,
      "grad_norm": 0.20241965859420183,
      "learning_rate": 7.770114942528736e-05,
      "loss": 0.4069,
      "step": 338
    },
    {
      "epoch": 0.48776978417266187,
      "grad_norm": 0.19119778436069676,
      "learning_rate": 7.793103448275863e-05,
      "loss": 0.4143,
      "step": 339
    },
    {
      "epoch": 0.4892086330935252,
      "grad_norm": 0.20104293091307832,
      "learning_rate": 7.81609195402299e-05,
      "loss": 0.4015,
      "step": 340
    },
    {
      "epoch": 0.4906474820143885,
      "grad_norm": 0.20525291633468387,
      "learning_rate": 7.839080459770115e-05,
      "loss": 0.4007,
      "step": 341
    },
    {
      "epoch": 0.4920863309352518,
      "grad_norm": 0.3065758658427838,
      "learning_rate": 7.862068965517242e-05,
      "loss": 0.3938,
      "step": 342
    },
    {
      "epoch": 0.4935251798561151,
      "grad_norm": 0.42866554620728164,
      "learning_rate": 7.885057471264368e-05,
      "loss": 0.4059,
      "step": 343
    },
    {
      "epoch": 0.4949640287769784,
      "grad_norm": 0.5089904520435118,
      "learning_rate": 7.908045977011495e-05,
      "loss": 0.401,
      "step": 344
    },
    {
      "epoch": 0.49640287769784175,
      "grad_norm": 0.6058763556754064,
      "learning_rate": 7.931034482758621e-05,
      "loss": 0.4002,
      "step": 345
    },
    {
      "epoch": 0.497841726618705,
      "grad_norm": 0.7046159485332273,
      "learning_rate": 7.954022988505747e-05,
      "loss": 0.4133,
      "step": 346
    },
    {
      "epoch": 0.49928057553956834,
      "grad_norm": 0.7368240450982135,
      "learning_rate": 7.977011494252874e-05,
      "loss": 0.4102,
      "step": 347
    },
    {
      "epoch": 0.5007194244604316,
      "grad_norm": 0.5207562288254234,
      "learning_rate": 8e-05,
      "loss": 0.4091,
      "step": 348
    },
    {
      "epoch": 0.5021582733812949,
      "grad_norm": 0.29358142633665374,
      "learning_rate": 7.999997981289966e-05,
      "loss": 0.3977,
      "step": 349
    },
    {
      "epoch": 0.5035971223021583,
      "grad_norm": 0.606870412872592,
      "learning_rate": 7.999991925161896e-05,
      "loss": 0.4094,
      "step": 350
    },
    {
      "epoch": 0.5050359712230216,
      "grad_norm": 0.5733670548773588,
      "learning_rate": 7.999981831621906e-05,
      "loss": 0.4057,
      "step": 351
    },
    {
      "epoch": 0.5064748201438849,
      "grad_norm": 0.2769785062462958,
      "learning_rate": 7.999967700680183e-05,
      "loss": 0.4063,
      "step": 352
    },
    {
      "epoch": 0.5079136690647482,
      "grad_norm": 0.6086587925627267,
      "learning_rate": 7.99994953235099e-05,
      "loss": 0.4029,
      "step": 353
    },
    {
      "epoch": 0.5093525179856115,
      "grad_norm": 0.5923654056002113,
      "learning_rate": 7.999927326652667e-05,
      "loss": 0.4108,
      "step": 354
    },
    {
      "epoch": 0.5107913669064749,
      "grad_norm": 0.39072319098055625,
      "learning_rate": 7.999901083607624e-05,
      "loss": 0.4049,
      "step": 355
    },
    {
      "epoch": 0.5122302158273381,
      "grad_norm": 0.49546215109939884,
      "learning_rate": 7.99987080324235e-05,
      "loss": 0.4052,
      "step": 356
    },
    {
      "epoch": 0.5136690647482014,
      "grad_norm": 0.4142270363441574,
      "learning_rate": 7.999836485587415e-05,
      "loss": 0.4115,
      "step": 357
    },
    {
      "epoch": 0.5151079136690647,
      "grad_norm": 0.34303257732768067,
      "learning_rate": 7.99979813067745e-05,
      "loss": 0.4105,
      "step": 358
    },
    {
      "epoch": 0.516546762589928,
      "grad_norm": 0.37193739236992807,
      "learning_rate": 7.999755738551171e-05,
      "loss": 0.4003,
      "step": 359
    },
    {
      "epoch": 0.5179856115107914,
      "grad_norm": 0.3618689619234623,
      "learning_rate": 7.999709309251368e-05,
      "loss": 0.4038,
      "step": 360
    },
    {
      "epoch": 0.5194244604316547,
      "grad_norm": 0.3527534652754947,
      "learning_rate": 7.999658842824904e-05,
      "loss": 0.3951,
      "step": 361
    },
    {
      "epoch": 0.520863309352518,
      "grad_norm": 0.31037934934255207,
      "learning_rate": 7.999604339322717e-05,
      "loss": 0.4049,
      "step": 362
    },
    {
      "epoch": 0.5223021582733813,
      "grad_norm": 0.2960883056882076,
      "learning_rate": 7.999545798799823e-05,
      "loss": 0.4079,
      "step": 363
    },
    {
      "epoch": 0.5237410071942447,
      "grad_norm": 0.3253854973145754,
      "learning_rate": 7.999483221315307e-05,
      "loss": 0.3932,
      "step": 364
    },
    {
      "epoch": 0.5251798561151079,
      "grad_norm": 0.2638110731347279,
      "learning_rate": 7.999416606932331e-05,
      "loss": 0.4076,
      "step": 365
    },
    {
      "epoch": 0.5266187050359712,
      "grad_norm": 0.29904682553238354,
      "learning_rate": 7.999345955718136e-05,
      "loss": 0.3919,
      "step": 366
    },
    {
      "epoch": 0.5280575539568345,
      "grad_norm": 0.35910718776043077,
      "learning_rate": 7.999271267744033e-05,
      "loss": 0.4028,
      "step": 367
    },
    {
      "epoch": 0.5294964028776978,
      "grad_norm": 0.2693411427809221,
      "learning_rate": 7.999192543085407e-05,
      "loss": 0.4036,
      "step": 368
    },
    {
      "epoch": 0.5309352517985612,
      "grad_norm": 0.3550220152156702,
      "learning_rate": 7.999109781821722e-05,
      "loss": 0.4079,
      "step": 369
    },
    {
      "epoch": 0.5323741007194245,
      "grad_norm": 0.3160549144611672,
      "learning_rate": 7.999022984036512e-05,
      "loss": 0.4111,
      "step": 370
    },
    {
      "epoch": 0.5338129496402878,
      "grad_norm": 0.2066309224309784,
      "learning_rate": 7.998932149817386e-05,
      "loss": 0.3995,
      "step": 371
    },
    {
      "epoch": 0.5352517985611511,
      "grad_norm": 0.2919697687299495,
      "learning_rate": 7.998837279256028e-05,
      "loss": 0.4004,
      "step": 372
    },
    {
      "epoch": 0.5366906474820143,
      "grad_norm": 0.268649195427655,
      "learning_rate": 7.998738372448196e-05,
      "loss": 0.4066,
      "step": 373
    },
    {
      "epoch": 0.5381294964028777,
      "grad_norm": 0.26476642665463757,
      "learning_rate": 7.998635429493726e-05,
      "loss": 0.3949,
      "step": 374
    },
    {
      "epoch": 0.539568345323741,
      "grad_norm": 0.36794156033362396,
      "learning_rate": 7.998528450496519e-05,
      "loss": 0.3976,
      "step": 375
    },
    {
      "epoch": 0.5410071942446043,
      "grad_norm": 0.40919097296904383,
      "learning_rate": 7.998417435564557e-05,
      "loss": 0.407,
      "step": 376
    },
    {
      "epoch": 0.5424460431654676,
      "grad_norm": 0.4526556851986713,
      "learning_rate": 7.998302384809893e-05,
      "loss": 0.4076,
      "step": 377
    },
    {
      "epoch": 0.543884892086331,
      "grad_norm": 0.4701472448170464,
      "learning_rate": 7.998183298348654e-05,
      "loss": 0.4046,
      "step": 378
    },
    {
      "epoch": 0.5453237410071943,
      "grad_norm": 0.44907946268323634,
      "learning_rate": 7.998060176301041e-05,
      "loss": 0.4121,
      "step": 379
    },
    {
      "epoch": 0.5467625899280576,
      "grad_norm": 0.26057683513153385,
      "learning_rate": 7.997933018791327e-05,
      "loss": 0.3997,
      "step": 380
    },
    {
      "epoch": 0.5482014388489208,
      "grad_norm": 0.2229074658092342,
      "learning_rate": 7.99780182594786e-05,
      "loss": 0.4018,
      "step": 381
    },
    {
      "epoch": 0.5496402877697841,
      "grad_norm": 0.31888109362416694,
      "learning_rate": 7.99766659790306e-05,
      "loss": 0.3984,
      "step": 382
    },
    {
      "epoch": 0.5510791366906475,
      "grad_norm": 0.3334807967394207,
      "learning_rate": 7.997527334793419e-05,
      "loss": 0.4038,
      "step": 383
    },
    {
      "epoch": 0.5525179856115108,
      "grad_norm": 0.20443061990304592,
      "learning_rate": 7.997384036759505e-05,
      "loss": 0.4077,
      "step": 384
    },
    {
      "epoch": 0.5539568345323741,
      "grad_norm": 0.2983478446351588,
      "learning_rate": 7.997236703945955e-05,
      "loss": 0.3985,
      "step": 385
    },
    {
      "epoch": 0.5553956834532374,
      "grad_norm": 0.3399128576179451,
      "learning_rate": 7.99708533650148e-05,
      "loss": 0.4133,
      "step": 386
    },
    {
      "epoch": 0.5568345323741007,
      "grad_norm": 0.24599737198322885,
      "learning_rate": 7.996929934578864e-05,
      "loss": 0.3985,
      "step": 387
    },
    {
      "epoch": 0.5582733812949641,
      "grad_norm": 0.25455223818502287,
      "learning_rate": 7.996770498334963e-05,
      "loss": 0.4051,
      "step": 388
    },
    {
      "epoch": 0.5597122302158274,
      "grad_norm": 0.2876531237701059,
      "learning_rate": 7.996607027930705e-05,
      "loss": 0.4056,
      "step": 389
    },
    {
      "epoch": 0.5611510791366906,
      "grad_norm": 0.23219710717556477,
      "learning_rate": 7.996439523531088e-05,
      "loss": 0.4052,
      "step": 390
    },
    {
      "epoch": 0.5625899280575539,
      "grad_norm": 0.29661563286854414,
      "learning_rate": 7.996267985305186e-05,
      "loss": 0.4071,
      "step": 391
    },
    {
      "epoch": 0.5640287769784172,
      "grad_norm": 0.3381873329962419,
      "learning_rate": 7.99609241342614e-05,
      "loss": 0.4035,
      "step": 392
    },
    {
      "epoch": 0.5654676258992806,
      "grad_norm": 0.3183331177671327,
      "learning_rate": 7.995912808071164e-05,
      "loss": 0.3979,
      "step": 393
    },
    {
      "epoch": 0.5669064748201439,
      "grad_norm": 0.362667614282157,
      "learning_rate": 7.995729169421545e-05,
      "loss": 0.4064,
      "step": 394
    },
    {
      "epoch": 0.5683453237410072,
      "grad_norm": 0.36848162295344455,
      "learning_rate": 7.99554149766264e-05,
      "loss": 0.4067,
      "step": 395
    },
    {
      "epoch": 0.5697841726618705,
      "grad_norm": 0.2309864523681632,
      "learning_rate": 7.995349792983874e-05,
      "loss": 0.3936,
      "step": 396
    },
    {
      "epoch": 0.5712230215827339,
      "grad_norm": 0.18573579497904213,
      "learning_rate": 7.995154055578748e-05,
      "loss": 0.4016,
      "step": 397
    },
    {
      "epoch": 0.5726618705035971,
      "grad_norm": 0.22848869369322164,
      "learning_rate": 7.994954285644827e-05,
      "loss": 0.4022,
      "step": 398
    },
    {
      "epoch": 0.5741007194244604,
      "grad_norm": 1.242889611634213,
      "learning_rate": 7.994750483383753e-05,
      "loss": 0.4074,
      "step": 399
    },
    {
      "epoch": 0.5755395683453237,
      "grad_norm": 0.15040364980189874,
      "learning_rate": 7.994542649001235e-05,
      "loss": 0.4005,
      "step": 400
    },
    {
      "epoch": 0.576978417266187,
      "grad_norm": 0.238032913741372,
      "learning_rate": 7.994330782707048e-05,
      "loss": 0.4092,
      "step": 401
    },
    {
      "epoch": 0.5784172661870504,
      "grad_norm": 0.23976492117285375,
      "learning_rate": 7.994114884715045e-05,
      "loss": 0.3969,
      "step": 402
    },
    {
      "epoch": 0.5798561151079137,
      "grad_norm": 0.27796861633714126,
      "learning_rate": 7.99389495524314e-05,
      "loss": 0.4069,
      "step": 403
    },
    {
      "epoch": 0.581294964028777,
      "grad_norm": 0.3870100203847514,
      "learning_rate": 7.993670994513321e-05,
      "loss": 0.4021,
      "step": 404
    },
    {
      "epoch": 0.5827338129496403,
      "grad_norm": 0.6297558741722189,
      "learning_rate": 7.993443002751646e-05,
      "loss": 0.4045,
      "step": 405
    },
    {
      "epoch": 0.5841726618705037,
      "grad_norm": 0.7519493513775716,
      "learning_rate": 7.993210980188236e-05,
      "loss": 0.4199,
      "step": 406
    },
    {
      "epoch": 0.5856115107913669,
      "grad_norm": 0.5848170414817211,
      "learning_rate": 7.992974927057287e-05,
      "loss": 0.4027,
      "step": 407
    },
    {
      "epoch": 0.5870503597122302,
      "grad_norm": 0.4741533309366105,
      "learning_rate": 7.992734843597058e-05,
      "loss": 0.397,
      "step": 408
    },
    {
      "epoch": 0.5884892086330935,
      "grad_norm": 0.525392793642585,
      "learning_rate": 7.992490730049881e-05,
      "loss": 0.4048,
      "step": 409
    },
    {
      "epoch": 0.5899280575539568,
      "grad_norm": 0.43840628431903,
      "learning_rate": 7.992242586662152e-05,
      "loss": 0.401,
      "step": 410
    },
    {
      "epoch": 0.5913669064748202,
      "grad_norm": 0.44899208218013437,
      "learning_rate": 7.991990413684336e-05,
      "loss": 0.4102,
      "step": 411
    },
    {
      "epoch": 0.5928057553956835,
      "grad_norm": 0.418529631501033,
      "learning_rate": 7.991734211370965e-05,
      "loss": 0.411,
      "step": 412
    },
    {
      "epoch": 0.5942446043165468,
      "grad_norm": 0.4023009688954088,
      "learning_rate": 7.991473979980637e-05,
      "loss": 0.406,
      "step": 413
    },
    {
      "epoch": 0.5956834532374101,
      "grad_norm": 0.3340069426859754,
      "learning_rate": 7.99120971977602e-05,
      "loss": 0.406,
      "step": 414
    },
    {
      "epoch": 0.5971223021582733,
      "grad_norm": 0.3122487477412848,
      "learning_rate": 7.990941431023844e-05,
      "loss": 0.4064,
      "step": 415
    },
    {
      "epoch": 0.5985611510791367,
      "grad_norm": 0.35328758358148876,
      "learning_rate": 7.990669113994911e-05,
      "loss": 0.3945,
      "step": 416
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.22445809971932562,
      "learning_rate": 7.99039276896408e-05,
      "loss": 0.3978,
      "step": 417
    },
    {
      "epoch": 0.6014388489208633,
      "grad_norm": 0.3117911528173465,
      "learning_rate": 7.990112396210288e-05,
      "loss": 0.4079,
      "step": 418
    },
    {
      "epoch": 0.6028776978417266,
      "grad_norm": 0.27110674043698635,
      "learning_rate": 7.989827996016525e-05,
      "loss": 0.3978,
      "step": 419
    },
    {
      "epoch": 0.60431654676259,
      "grad_norm": 0.28342593104905844,
      "learning_rate": 7.989539568669856e-05,
      "loss": 0.3945,
      "step": 420
    },
    {
      "epoch": 0.6057553956834533,
      "grad_norm": 0.2335891065020172,
      "learning_rate": 7.989247114461403e-05,
      "loss": 0.4012,
      "step": 421
    },
    {
      "epoch": 0.6071942446043166,
      "grad_norm": 0.34132845531397416,
      "learning_rate": 7.988950633686358e-05,
      "loss": 0.4059,
      "step": 422
    },
    {
      "epoch": 0.6086330935251798,
      "grad_norm": 0.3793753760188709,
      "learning_rate": 7.988650126643976e-05,
      "loss": 0.4083,
      "step": 423
    },
    {
      "epoch": 0.6100719424460431,
      "grad_norm": 0.24033707461795106,
      "learning_rate": 7.988345593637572e-05,
      "loss": 0.3927,
      "step": 424
    },
    {
      "epoch": 0.6115107913669064,
      "grad_norm": 0.26549905398007445,
      "learning_rate": 7.988037034974532e-05,
      "loss": 0.3936,
      "step": 425
    },
    {
      "epoch": 0.6129496402877698,
      "grad_norm": 0.30599802789608876,
      "learning_rate": 7.9877244509663e-05,
      "loss": 0.399,
      "step": 426
    },
    {
      "epoch": 0.6143884892086331,
      "grad_norm": 0.250416843178887,
      "learning_rate": 7.987407841928384e-05,
      "loss": 0.4016,
      "step": 427
    },
    {
      "epoch": 0.6158273381294964,
      "grad_norm": 0.3261688992689598,
      "learning_rate": 7.987087208180355e-05,
      "loss": 0.4034,
      "step": 428
    },
    {
      "epoch": 0.6172661870503597,
      "grad_norm": 0.2401440595172318,
      "learning_rate": 7.986762550045844e-05,
      "loss": 0.3891,
      "step": 429
    },
    {
      "epoch": 0.6187050359712231,
      "grad_norm": 0.2743844101931455,
      "learning_rate": 7.98643386785255e-05,
      "loss": 0.3981,
      "step": 430
    },
    {
      "epoch": 0.6201438848920864,
      "grad_norm": 0.2758490427203629,
      "learning_rate": 7.986101161932227e-05,
      "loss": 0.399,
      "step": 431
    },
    {
      "epoch": 0.6215827338129496,
      "grad_norm": 0.24764514548640124,
      "learning_rate": 7.985764432620695e-05,
      "loss": 0.3958,
      "step": 432
    },
    {
      "epoch": 0.6230215827338129,
      "grad_norm": 0.31498435577719125,
      "learning_rate": 7.985423680257833e-05,
      "loss": 0.4056,
      "step": 433
    },
    {
      "epoch": 0.6244604316546762,
      "grad_norm": 0.3071596705693149,
      "learning_rate": 7.985078905187582e-05,
      "loss": 0.3932,
      "step": 434
    },
    {
      "epoch": 0.6258992805755396,
      "grad_norm": 0.23562150322180472,
      "learning_rate": 7.984730107757942e-05,
      "loss": 0.391,
      "step": 435
    },
    {
      "epoch": 0.6273381294964029,
      "grad_norm": 0.2106502372853218,
      "learning_rate": 7.984377288320973e-05,
      "loss": 0.3866,
      "step": 436
    },
    {
      "epoch": 0.6287769784172662,
      "grad_norm": 0.18180087469996958,
      "learning_rate": 7.984020447232795e-05,
      "loss": 0.3948,
      "step": 437
    },
    {
      "epoch": 0.6302158273381295,
      "grad_norm": 0.1697981029409121,
      "learning_rate": 7.983659584853586e-05,
      "loss": 0.402,
      "step": 438
    },
    {
      "epoch": 0.6316546762589929,
      "grad_norm": 0.1599496406363699,
      "learning_rate": 7.983294701547588e-05,
      "loss": 0.3961,
      "step": 439
    },
    {
      "epoch": 0.6330935251798561,
      "grad_norm": 0.14858876445054642,
      "learning_rate": 7.982925797683095e-05,
      "loss": 0.398,
      "step": 440
    },
    {
      "epoch": 0.6345323741007194,
      "grad_norm": 0.15847796647568324,
      "learning_rate": 7.982552873632461e-05,
      "loss": 0.4042,
      "step": 441
    },
    {
      "epoch": 0.6359712230215827,
      "grad_norm": 0.19809172933544417,
      "learning_rate": 7.982175929772102e-05,
      "loss": 0.3864,
      "step": 442
    },
    {
      "epoch": 0.637410071942446,
      "grad_norm": 0.1811174238613106,
      "learning_rate": 7.981794966482486e-05,
      "loss": 0.3855,
      "step": 443
    },
    {
      "epoch": 0.6388489208633094,
      "grad_norm": 0.1675328143083596,
      "learning_rate": 7.98140998414814e-05,
      "loss": 0.3971,
      "step": 444
    },
    {
      "epoch": 0.6402877697841727,
      "grad_norm": 0.20034503405113696,
      "learning_rate": 7.98102098315765e-05,
      "loss": 0.399,
      "step": 445
    },
    {
      "epoch": 0.641726618705036,
      "grad_norm": 0.211833894086791,
      "learning_rate": 7.980627963903654e-05,
      "loss": 0.3908,
      "step": 446
    },
    {
      "epoch": 0.6431654676258993,
      "grad_norm": 0.23408497340134393,
      "learning_rate": 7.980230926782848e-05,
      "loss": 0.4003,
      "step": 447
    },
    {
      "epoch": 0.6446043165467625,
      "grad_norm": 0.23077143132838263,
      "learning_rate": 7.979829872195984e-05,
      "loss": 0.3962,
      "step": 448
    },
    {
      "epoch": 0.6460431654676259,
      "grad_norm": 0.2834731638435603,
      "learning_rate": 7.979424800547869e-05,
      "loss": 0.3995,
      "step": 449
    },
    {
      "epoch": 0.6474820143884892,
      "grad_norm": 0.36818575331884823,
      "learning_rate": 7.979015712247365e-05,
      "loss": 0.4002,
      "step": 450
    },
    {
      "epoch": 0.6489208633093525,
      "grad_norm": 0.44203983146087344,
      "learning_rate": 7.978602607707383e-05,
      "loss": 0.3948,
      "step": 451
    },
    {
      "epoch": 0.6503597122302158,
      "grad_norm": 0.442645682673057,
      "learning_rate": 7.978185487344897e-05,
      "loss": 0.4012,
      "step": 452
    },
    {
      "epoch": 0.6517985611510791,
      "grad_norm": 0.43444069228623855,
      "learning_rate": 7.977764351580928e-05,
      "loss": 0.4081,
      "step": 453
    },
    {
      "epoch": 0.6532374100719425,
      "grad_norm": 0.39467412539574687,
      "learning_rate": 7.97733920084055e-05,
      "loss": 0.4003,
      "step": 454
    },
    {
      "epoch": 0.6546762589928058,
      "grad_norm": 0.38297176846303266,
      "learning_rate": 7.976910035552892e-05,
      "loss": 0.393,
      "step": 455
    },
    {
      "epoch": 0.6561151079136691,
      "grad_norm": 0.3990027114029954,
      "learning_rate": 7.976476856151134e-05,
      "loss": 0.3967,
      "step": 456
    },
    {
      "epoch": 0.6575539568345323,
      "grad_norm": 0.38837742399412434,
      "learning_rate": 7.976039663072509e-05,
      "loss": 0.3908,
      "step": 457
    },
    {
      "epoch": 0.6589928057553956,
      "grad_norm": 0.290277744761257,
      "learning_rate": 7.975598456758298e-05,
      "loss": 0.4067,
      "step": 458
    },
    {
      "epoch": 0.660431654676259,
      "grad_norm": 0.21844368211822543,
      "learning_rate": 7.975153237653836e-05,
      "loss": 0.3988,
      "step": 459
    },
    {
      "epoch": 0.6618705035971223,
      "grad_norm": 0.3289265056281082,
      "learning_rate": 7.974704006208509e-05,
      "loss": 0.3981,
      "step": 460
    },
    {
      "epoch": 0.6633093525179856,
      "grad_norm": 0.42331452138501785,
      "learning_rate": 7.974250762875747e-05,
      "loss": 0.4028,
      "step": 461
    },
    {
      "epoch": 0.6647482014388489,
      "grad_norm": 0.3580448522286446,
      "learning_rate": 7.973793508113035e-05,
      "loss": 0.4046,
      "step": 462
    },
    {
      "epoch": 0.6661870503597123,
      "grad_norm": 0.168092987350451,
      "learning_rate": 7.973332242381908e-05,
      "loss": 0.3967,
      "step": 463
    },
    {
      "epoch": 0.6676258992805756,
      "grad_norm": 0.24235827768404627,
      "learning_rate": 7.972866966147942e-05,
      "loss": 0.4027,
      "step": 464
    },
    {
      "epoch": 0.6690647482014388,
      "grad_norm": 0.32639102896298916,
      "learning_rate": 7.972397679880771e-05,
      "loss": 0.4019,
      "step": 465
    },
    {
      "epoch": 0.6705035971223021,
      "grad_norm": 0.300427946195229,
      "learning_rate": 7.971924384054068e-05,
      "loss": 0.391,
      "step": 466
    },
    {
      "epoch": 0.6719424460431654,
      "grad_norm": 0.2047814273496808,
      "learning_rate": 7.971447079145557e-05,
      "loss": 0.3924,
      "step": 467
    },
    {
      "epoch": 0.6733812949640288,
      "grad_norm": 0.19410806738189568,
      "learning_rate": 7.970965765637011e-05,
      "loss": 0.3905,
      "step": 468
    },
    {
      "epoch": 0.6748201438848921,
      "grad_norm": 0.2461909540823566,
      "learning_rate": 7.970480444014244e-05,
      "loss": 0.3947,
      "step": 469
    },
    {
      "epoch": 0.6762589928057554,
      "grad_norm": 0.23520468572683895,
      "learning_rate": 7.969991114767114e-05,
      "loss": 0.4047,
      "step": 470
    },
    {
      "epoch": 0.6776978417266187,
      "grad_norm": 0.2288733575688635,
      "learning_rate": 7.969497778389534e-05,
      "loss": 0.3931,
      "step": 471
    },
    {
      "epoch": 0.679136690647482,
      "grad_norm": 0.22993980558386445,
      "learning_rate": 7.969000435379454e-05,
      "loss": 0.3947,
      "step": 472
    },
    {
      "epoch": 0.6805755395683454,
      "grad_norm": 0.23754711246173846,
      "learning_rate": 7.968499086238867e-05,
      "loss": 0.4004,
      "step": 473
    },
    {
      "epoch": 0.6820143884892086,
      "grad_norm": 0.20366152685811229,
      "learning_rate": 7.967993731473815e-05,
      "loss": 0.396,
      "step": 474
    },
    {
      "epoch": 0.6834532374100719,
      "grad_norm": 0.17920563839246975,
      "learning_rate": 7.96748437159438e-05,
      "loss": 0.3993,
      "step": 475
    },
    {
      "epoch": 0.6848920863309352,
      "grad_norm": 0.18952993481603614,
      "learning_rate": 7.966971007114686e-05,
      "loss": 0.3928,
      "step": 476
    },
    {
      "epoch": 0.6863309352517986,
      "grad_norm": 0.2172472942767943,
      "learning_rate": 7.966453638552901e-05,
      "loss": 0.3942,
      "step": 477
    },
    {
      "epoch": 0.6877697841726619,
      "grad_norm": 0.2098368899511751,
      "learning_rate": 7.965932266431232e-05,
      "loss": 0.3933,
      "step": 478
    },
    {
      "epoch": 0.6892086330935252,
      "grad_norm": 0.21759755000537567,
      "learning_rate": 7.96540689127593e-05,
      "loss": 0.3938,
      "step": 479
    },
    {
      "epoch": 0.6906474820143885,
      "grad_norm": 0.27684281206502115,
      "learning_rate": 7.964877513617285e-05,
      "loss": 0.3908,
      "step": 480
    },
    {
      "epoch": 0.6920863309352518,
      "grad_norm": 0.3955996614331896,
      "learning_rate": 7.964344133989627e-05,
      "loss": 0.3933,
      "step": 481
    },
    {
      "epoch": 0.6935251798561151,
      "grad_norm": 0.3936258614387365,
      "learning_rate": 7.963806752931324e-05,
      "loss": 0.3998,
      "step": 482
    },
    {
      "epoch": 0.6949640287769784,
      "grad_norm": 0.2970999309051372,
      "learning_rate": 7.963265370984786e-05,
      "loss": 0.3994,
      "step": 483
    },
    {
      "epoch": 0.6964028776978417,
      "grad_norm": 0.3279006569452077,
      "learning_rate": 7.962719988696458e-05,
      "loss": 0.3965,
      "step": 484
    },
    {
      "epoch": 0.697841726618705,
      "grad_norm": 0.40135146596828136,
      "learning_rate": 7.962170606616826e-05,
      "loss": 0.3986,
      "step": 485
    },
    {
      "epoch": 0.6992805755395683,
      "grad_norm": 0.41780843703588655,
      "learning_rate": 7.96161722530041e-05,
      "loss": 0.3965,
      "step": 486
    },
    {
      "epoch": 0.7007194244604317,
      "grad_norm": 0.4053211081313924,
      "learning_rate": 7.96105984530577e-05,
      "loss": 0.3774,
      "step": 487
    },
    {
      "epoch": 0.702158273381295,
      "grad_norm": 0.4103765238612219,
      "learning_rate": 7.9604984671955e-05,
      "loss": 0.3866,
      "step": 488
    },
    {
      "epoch": 0.7035971223021583,
      "grad_norm": 0.34964703097638195,
      "learning_rate": 7.959933091536227e-05,
      "loss": 0.3958,
      "step": 489
    },
    {
      "epoch": 0.7050359712230215,
      "grad_norm": 0.25846513601369847,
      "learning_rate": 7.95936371889862e-05,
      "loss": 0.3938,
      "step": 490
    },
    {
      "epoch": 0.7064748201438849,
      "grad_norm": 0.274682346218589,
      "learning_rate": 7.958790349857375e-05,
      "loss": 0.3891,
      "step": 491
    },
    {
      "epoch": 0.7079136690647482,
      "grad_norm": 0.3113217352383488,
      "learning_rate": 7.958212984991226e-05,
      "loss": 0.3974,
      "step": 492
    },
    {
      "epoch": 0.7093525179856115,
      "grad_norm": 0.324416073045715,
      "learning_rate": 7.957631624882938e-05,
      "loss": 0.3978,
      "step": 493
    },
    {
      "epoch": 0.7107913669064748,
      "grad_norm": 0.2591042835794199,
      "learning_rate": 7.957046270119313e-05,
      "loss": 0.3979,
      "step": 494
    },
    {
      "epoch": 0.7122302158273381,
      "grad_norm": 0.2393859657662622,
      "learning_rate": 7.956456921291178e-05,
      "loss": 0.3952,
      "step": 495
    },
    {
      "epoch": 0.7136690647482015,
      "grad_norm": 0.3290105193793378,
      "learning_rate": 7.955863578993396e-05,
      "loss": 0.3982,
      "step": 496
    },
    {
      "epoch": 0.7151079136690648,
      "grad_norm": 0.29739496982271724,
      "learning_rate": 7.955266243824864e-05,
      "loss": 0.3905,
      "step": 497
    },
    {
      "epoch": 0.7165467625899281,
      "grad_norm": 0.20314300140842406,
      "learning_rate": 7.954664916388499e-05,
      "loss": 0.3966,
      "step": 498
    },
    {
      "epoch": 0.7179856115107913,
      "grad_norm": 0.28034465884458026,
      "learning_rate": 7.954059597291257e-05,
      "loss": 0.3966,
      "step": 499
    },
    {
      "epoch": 0.7194244604316546,
      "grad_norm": 0.2959049118620304,
      "learning_rate": 7.953450287144121e-05,
      "loss": 0.4031,
      "step": 500
    },
    {
      "epoch": 0.720863309352518,
      "grad_norm": 0.2486789124385139,
      "learning_rate": 7.952836986562099e-05,
      "loss": 0.3921,
      "step": 501
    },
    {
      "epoch": 0.7223021582733813,
      "grad_norm": 0.27972167275170323,
      "learning_rate": 7.952219696164231e-05,
      "loss": 0.3942,
      "step": 502
    },
    {
      "epoch": 0.7237410071942446,
      "grad_norm": 0.24393341332660898,
      "learning_rate": 7.95159841657358e-05,
      "loss": 0.3899,
      "step": 503
    },
    {
      "epoch": 0.7251798561151079,
      "grad_norm": 0.21560853713804515,
      "learning_rate": 7.950973148417239e-05,
      "loss": 0.4051,
      "step": 504
    },
    {
      "epoch": 0.7266187050359713,
      "grad_norm": 0.24353818532928312,
      "learning_rate": 7.950343892326327e-05,
      "loss": 0.3992,
      "step": 505
    },
    {
      "epoch": 0.7280575539568346,
      "grad_norm": 0.25016763270320097,
      "learning_rate": 7.949710648935984e-05,
      "loss": 0.3995,
      "step": 506
    },
    {
      "epoch": 0.7294964028776978,
      "grad_norm": 0.2162108624913337,
      "learning_rate": 7.949073418885378e-05,
      "loss": 0.3966,
      "step": 507
    },
    {
      "epoch": 0.7309352517985611,
      "grad_norm": 0.24692064398043742,
      "learning_rate": 7.948432202817703e-05,
      "loss": 0.3976,
      "step": 508
    },
    {
      "epoch": 0.7323741007194244,
      "grad_norm": 0.29102943123049935,
      "learning_rate": 7.94778700138017e-05,
      "loss": 0.3899,
      "step": 509
    },
    {
      "epoch": 0.7338129496402878,
      "grad_norm": 0.26026846085626515,
      "learning_rate": 7.947137815224018e-05,
      "loss": 0.3937,
      "step": 510
    },
    {
      "epoch": 0.7352517985611511,
      "grad_norm": 0.277160471430053,
      "learning_rate": 7.946484645004508e-05,
      "loss": 0.3933,
      "step": 511
    },
    {
      "epoch": 0.7366906474820144,
      "grad_norm": 0.3580864764505278,
      "learning_rate": 7.945827491380916e-05,
      "loss": 0.3924,
      "step": 512
    },
    {
      "epoch": 0.7381294964028777,
      "grad_norm": 0.4100211305198917,
      "learning_rate": 7.945166355016548e-05,
      "loss": 0.3921,
      "step": 513
    },
    {
      "epoch": 0.739568345323741,
      "grad_norm": 0.3752077589402531,
      "learning_rate": 7.944501236578722e-05,
      "loss": 0.3834,
      "step": 514
    },
    {
      "epoch": 0.7410071942446043,
      "grad_norm": 0.35021889724147953,
      "learning_rate": 7.943832136738783e-05,
      "loss": 0.3894,
      "step": 515
    },
    {
      "epoch": 0.7424460431654676,
      "grad_norm": 0.3295101949914119,
      "learning_rate": 7.943159056172084e-05,
      "loss": 0.3987,
      "step": 516
    },
    {
      "epoch": 0.7438848920863309,
      "grad_norm": 0.29331915864233477,
      "learning_rate": 7.942481995558007e-05,
      "loss": 0.3867,
      "step": 517
    },
    {
      "epoch": 0.7453237410071942,
      "grad_norm": 0.2853388653685031,
      "learning_rate": 7.941800955579946e-05,
      "loss": 0.3891,
      "step": 518
    },
    {
      "epoch": 0.7467625899280576,
      "grad_norm": 0.2706330302215832,
      "learning_rate": 7.941115936925311e-05,
      "loss": 0.3929,
      "step": 519
    },
    {
      "epoch": 0.7482014388489209,
      "grad_norm": 0.23279517043371323,
      "learning_rate": 7.940426940285529e-05,
      "loss": 0.3913,
      "step": 520
    },
    {
      "epoch": 0.7496402877697842,
      "grad_norm": 0.2220900334502715,
      "learning_rate": 7.939733966356042e-05,
      "loss": 0.39,
      "step": 521
    },
    {
      "epoch": 0.7510791366906475,
      "grad_norm": 0.23271376043917275,
      "learning_rate": 7.939037015836308e-05,
      "loss": 0.3931,
      "step": 522
    },
    {
      "epoch": 0.7525179856115108,
      "grad_norm": 0.22106957223746632,
      "learning_rate": 7.938336089429796e-05,
      "loss": 0.3891,
      "step": 523
    },
    {
      "epoch": 0.753956834532374,
      "grad_norm": 0.24197907887885128,
      "learning_rate": 7.937631187843991e-05,
      "loss": 0.3902,
      "step": 524
    },
    {
      "epoch": 0.7553956834532374,
      "grad_norm": 0.22673980441736877,
      "learning_rate": 7.936922311790388e-05,
      "loss": 0.3973,
      "step": 525
    },
    {
      "epoch": 0.7568345323741007,
      "grad_norm": 0.1573949178411432,
      "learning_rate": 7.936209461984495e-05,
      "loss": 0.3931,
      "step": 526
    },
    {
      "epoch": 0.758273381294964,
      "grad_norm": 0.16989105140678423,
      "learning_rate": 7.935492639145831e-05,
      "loss": 0.3945,
      "step": 527
    },
    {
      "epoch": 0.7597122302158273,
      "grad_norm": 0.16198079458205,
      "learning_rate": 7.934771843997922e-05,
      "loss": 0.3938,
      "step": 528
    },
    {
      "epoch": 0.7611510791366907,
      "grad_norm": 0.14474801638735643,
      "learning_rate": 7.934047077268311e-05,
      "loss": 0.3932,
      "step": 529
    },
    {
      "epoch": 0.762589928057554,
      "grad_norm": 0.19071898012891114,
      "learning_rate": 7.93331833968854e-05,
      "loss": 0.3963,
      "step": 530
    },
    {
      "epoch": 0.7640287769784173,
      "grad_norm": 0.23415263060538,
      "learning_rate": 7.932585631994168e-05,
      "loss": 0.3995,
      "step": 531
    },
    {
      "epoch": 0.7654676258992805,
      "grad_norm": 0.25461113306253436,
      "learning_rate": 7.931848954924754e-05,
      "loss": 0.3877,
      "step": 532
    },
    {
      "epoch": 0.7669064748201438,
      "grad_norm": 0.27845571117851337,
      "learning_rate": 7.931108309223868e-05,
      "loss": 0.3955,
      "step": 533
    },
    {
      "epoch": 0.7683453237410072,
      "grad_norm": 0.29125838184684827,
      "learning_rate": 7.930363695639085e-05,
      "loss": 0.403,
      "step": 534
    },
    {
      "epoch": 0.7697841726618705,
      "grad_norm": 0.4030485435958359,
      "learning_rate": 7.929615114921984e-05,
      "loss": 0.3954,
      "step": 535
    },
    {
      "epoch": 0.7712230215827338,
      "grad_norm": 0.5501020153467006,
      "learning_rate": 7.92886256782815e-05,
      "loss": 0.4011,
      "step": 536
    },
    {
      "epoch": 0.7726618705035971,
      "grad_norm": 0.5591809285898809,
      "learning_rate": 7.928106055117168e-05,
      "loss": 0.3956,
      "step": 537
    },
    {
      "epoch": 0.7741007194244605,
      "grad_norm": 0.4561015670623354,
      "learning_rate": 7.927345577552627e-05,
      "loss": 0.3939,
      "step": 538
    },
    {
      "epoch": 0.7755395683453238,
      "grad_norm": 0.3534835719458935,
      "learning_rate": 7.926581135902122e-05,
      "loss": 0.3982,
      "step": 539
    },
    {
      "epoch": 0.7769784172661871,
      "grad_norm": 0.29817006306468613,
      "learning_rate": 7.925812730937245e-05,
      "loss": 0.3987,
      "step": 540
    },
    {
      "epoch": 0.7784172661870503,
      "grad_norm": 0.33721551414512624,
      "learning_rate": 7.92504036343359e-05,
      "loss": 0.3925,
      "step": 541
    },
    {
      "epoch": 0.7798561151079136,
      "grad_norm": 0.3534990722780468,
      "learning_rate": 7.924264034170747e-05,
      "loss": 0.4035,
      "step": 542
    },
    {
      "epoch": 0.781294964028777,
      "grad_norm": 0.28500786280558943,
      "learning_rate": 7.923483743932311e-05,
      "loss": 0.3841,
      "step": 543
    },
    {
      "epoch": 0.7827338129496403,
      "grad_norm": 0.2935353003885132,
      "learning_rate": 7.922699493505871e-05,
      "loss": 0.3979,
      "step": 544
    },
    {
      "epoch": 0.7841726618705036,
      "grad_norm": 0.2824720757373356,
      "learning_rate": 7.921911283683013e-05,
      "loss": 0.3812,
      "step": 545
    },
    {
      "epoch": 0.7856115107913669,
      "grad_norm": 0.21131745701377888,
      "learning_rate": 7.921119115259322e-05,
      "loss": 0.3889,
      "step": 546
    },
    {
      "epoch": 0.7870503597122303,
      "grad_norm": 0.17006553976446184,
      "learning_rate": 7.920322989034377e-05,
      "loss": 0.4014,
      "step": 547
    },
    {
      "epoch": 0.7884892086330936,
      "grad_norm": 0.22941668905471813,
      "learning_rate": 7.919522905811752e-05,
      "loss": 0.395,
      "step": 548
    },
    {
      "epoch": 0.7899280575539568,
      "grad_norm": 0.2595782530775398,
      "learning_rate": 7.918718866399012e-05,
      "loss": 0.3902,
      "step": 549
    },
    {
      "epoch": 0.7913669064748201,
      "grad_norm": 0.2241314542850331,
      "learning_rate": 7.917910871607723e-05,
      "loss": 0.3962,
      "step": 550
    },
    {
      "epoch": 0.7928057553956834,
      "grad_norm": 0.19297615840666787,
      "learning_rate": 7.917098922253436e-05,
      "loss": 0.3922,
      "step": 551
    },
    {
      "epoch": 0.7942446043165468,
      "grad_norm": 0.24337496404512848,
      "learning_rate": 7.916283019155696e-05,
      "loss": 0.3938,
      "step": 552
    },
    {
      "epoch": 0.7956834532374101,
      "grad_norm": 0.23584358774435407,
      "learning_rate": 7.915463163138041e-05,
      "loss": 0.3809,
      "step": 553
    },
    {
      "epoch": 0.7971223021582734,
      "grad_norm": 0.2005090428309887,
      "learning_rate": 7.914639355027995e-05,
      "loss": 0.386,
      "step": 554
    },
    {
      "epoch": 0.7985611510791367,
      "grad_norm": 0.20358168719753014,
      "learning_rate": 7.913811595657072e-05,
      "loss": 0.4024,
      "step": 555
    },
    {
      "epoch": 0.8,
      "grad_norm": 0.23516057951207506,
      "learning_rate": 7.912979885860776e-05,
      "loss": 0.3946,
      "step": 556
    },
    {
      "epoch": 0.8014388489208633,
      "grad_norm": 0.2436118231209871,
      "learning_rate": 7.912144226478598e-05,
      "loss": 0.394,
      "step": 557
    },
    {
      "epoch": 0.8028776978417266,
      "grad_norm": 0.3057793105311322,
      "learning_rate": 7.911304618354015e-05,
      "loss": 0.403,
      "step": 558
    },
    {
      "epoch": 0.8043165467625899,
      "grad_norm": 0.4044375403682651,
      "learning_rate": 7.910461062334488e-05,
      "loss": 0.3963,
      "step": 559
    },
    {
      "epoch": 0.8057553956834532,
      "grad_norm": 0.4561042808982624,
      "learning_rate": 7.909613559271467e-05,
      "loss": 0.3926,
      "step": 560
    },
    {
      "epoch": 0.8071942446043165,
      "grad_norm": 0.4063174953150093,
      "learning_rate": 7.908762110020382e-05,
      "loss": 0.4026,
      "step": 561
    },
    {
      "epoch": 0.8086330935251799,
      "grad_norm": 0.3189075553153762,
      "learning_rate": 7.907906715440649e-05,
      "loss": 0.3875,
      "step": 562
    },
    {
      "epoch": 0.8100719424460432,
      "grad_norm": 0.3138978072985748,
      "learning_rate": 7.907047376395661e-05,
      "loss": 0.4015,
      "step": 563
    },
    {
      "epoch": 0.8115107913669065,
      "grad_norm": 0.3131591493013707,
      "learning_rate": 7.906184093752801e-05,
      "loss": 0.3837,
      "step": 564
    },
    {
      "epoch": 0.8129496402877698,
      "grad_norm": 0.3206070232195477,
      "learning_rate": 7.905316868383425e-05,
      "loss": 0.4034,
      "step": 565
    },
    {
      "epoch": 0.814388489208633,
      "grad_norm": 0.34778976345979296,
      "learning_rate": 7.904445701162872e-05,
      "loss": 0.3929,
      "step": 566
    },
    {
      "epoch": 0.8158273381294964,
      "grad_norm": 0.30227284798904663,
      "learning_rate": 7.903570592970458e-05,
      "loss": 0.3902,
      "step": 567
    },
    {
      "epoch": 0.8172661870503597,
      "grad_norm": 0.19280019472036092,
      "learning_rate": 7.902691544689479e-05,
      "loss": 0.3823,
      "step": 568
    },
    {
      "epoch": 0.818705035971223,
      "grad_norm": 0.2160741916005615,
      "learning_rate": 7.901808557207206e-05,
      "loss": 0.391,
      "step": 569
    },
    {
      "epoch": 0.8201438848920863,
      "grad_norm": 0.2543543474640509,
      "learning_rate": 7.900921631414887e-05,
      "loss": 0.3964,
      "step": 570
    },
    {
      "epoch": 0.8215827338129497,
      "grad_norm": 0.22143844825733844,
      "learning_rate": 7.900030768207746e-05,
      "loss": 0.3865,
      "step": 571
    },
    {
      "epoch": 0.823021582733813,
      "grad_norm": 0.20666393285687437,
      "learning_rate": 7.899135968484979e-05,
      "loss": 0.378,
      "step": 572
    },
    {
      "epoch": 0.8244604316546763,
      "grad_norm": 0.2066782394153934,
      "learning_rate": 7.898237233149758e-05,
      "loss": 0.3962,
      "step": 573
    },
    {
      "epoch": 0.8258992805755395,
      "grad_norm": 0.1938863528224781,
      "learning_rate": 7.897334563109225e-05,
      "loss": 0.3918,
      "step": 574
    },
    {
      "epoch": 0.8273381294964028,
      "grad_norm": 0.20336201484248936,
      "learning_rate": 7.896427959274494e-05,
      "loss": 0.4001,
      "step": 575
    },
    {
      "epoch": 0.8287769784172662,
      "grad_norm": 0.2258744950644811,
      "learning_rate": 7.895517422560651e-05,
      "loss": 0.3871,
      "step": 576
    },
    {
      "epoch": 0.8302158273381295,
      "grad_norm": 0.23843133875317776,
      "learning_rate": 7.89460295388675e-05,
      "loss": 0.3955,
      "step": 577
    },
    {
      "epoch": 0.8316546762589928,
      "grad_norm": 0.2312577140122766,
      "learning_rate": 7.893684554175817e-05,
      "loss": 0.3898,
      "step": 578
    },
    {
      "epoch": 0.8330935251798561,
      "grad_norm": 0.27908521142623455,
      "learning_rate": 7.892762224354839e-05,
      "loss": 0.4008,
      "step": 579
    },
    {
      "epoch": 0.8345323741007195,
      "grad_norm": 0.2999958408864618,
      "learning_rate": 7.891835965354778e-05,
      "loss": 0.3965,
      "step": 580
    },
    {
      "epoch": 0.8359712230215828,
      "grad_norm": 0.26756124634390244,
      "learning_rate": 7.890905778110557e-05,
      "loss": 0.3849,
      "step": 581
    },
    {
      "epoch": 0.837410071942446,
      "grad_norm": 0.2799734483589846,
      "learning_rate": 7.889971663561065e-05,
      "loss": 0.4019,
      "step": 582
    },
    {
      "epoch": 0.8388489208633093,
      "grad_norm": 0.3158878705989888,
      "learning_rate": 7.889033622649155e-05,
      "loss": 0.3853,
      "step": 583
    },
    {
      "epoch": 0.8402877697841726,
      "grad_norm": 0.2982911349211642,
      "learning_rate": 7.888091656321644e-05,
      "loss": 0.397,
      "step": 584
    },
    {
      "epoch": 0.841726618705036,
      "grad_norm": 0.2053738696369239,
      "learning_rate": 7.88714576552931e-05,
      "loss": 0.3918,
      "step": 585
    },
    {
      "epoch": 0.8431654676258993,
      "grad_norm": 0.17663025292577625,
      "learning_rate": 7.886195951226892e-05,
      "loss": 0.3832,
      "step": 586
    },
    {
      "epoch": 0.8446043165467626,
      "grad_norm": 0.2826510855902402,
      "learning_rate": 7.885242214373091e-05,
      "loss": 0.3979,
      "step": 587
    },
    {
      "epoch": 0.8460431654676259,
      "grad_norm": 0.3492728858016357,
      "learning_rate": 7.884284555930564e-05,
      "loss": 0.3906,
      "step": 588
    },
    {
      "epoch": 0.8474820143884892,
      "grad_norm": 0.3158540804191533,
      "learning_rate": 7.883322976865932e-05,
      "loss": 0.3947,
      "step": 589
    },
    {
      "epoch": 0.8489208633093526,
      "grad_norm": 0.25409408381096943,
      "learning_rate": 7.882357478149767e-05,
      "loss": 0.3969,
      "step": 590
    },
    {
      "epoch": 0.8503597122302158,
      "grad_norm": 0.2075916331813354,
      "learning_rate": 7.8813880607566e-05,
      "loss": 0.3953,
      "step": 591
    },
    {
      "epoch": 0.8517985611510791,
      "grad_norm": 0.21423174653614885,
      "learning_rate": 7.880414725664918e-05,
      "loss": 0.3942,
      "step": 592
    },
    {
      "epoch": 0.8532374100719424,
      "grad_norm": 0.21588649263672796,
      "learning_rate": 7.879437473857161e-05,
      "loss": 0.3887,
      "step": 593
    },
    {
      "epoch": 0.8546762589928057,
      "grad_norm": 0.16259134395034616,
      "learning_rate": 7.878456306319723e-05,
      "loss": 0.391,
      "step": 594
    },
    {
      "epoch": 0.8561151079136691,
      "grad_norm": 0.186899603789238,
      "learning_rate": 7.877471224042952e-05,
      "loss": 0.3878,
      "step": 595
    },
    {
      "epoch": 0.8575539568345324,
      "grad_norm": 0.19396290760022455,
      "learning_rate": 7.876482228021144e-05,
      "loss": 0.3846,
      "step": 596
    },
    {
      "epoch": 0.8589928057553957,
      "grad_norm": 0.22712154579665747,
      "learning_rate": 7.875489319252549e-05,
      "loss": 0.3892,
      "step": 597
    },
    {
      "epoch": 0.860431654676259,
      "grad_norm": 0.24463632384626877,
      "learning_rate": 7.874492498739362e-05,
      "loss": 0.3932,
      "step": 598
    },
    {
      "epoch": 0.8618705035971223,
      "grad_norm": 0.2612460691086307,
      "learning_rate": 7.87349176748773e-05,
      "loss": 0.3882,
      "step": 599
    },
    {
      "epoch": 0.8633093525179856,
      "grad_norm": 0.2974439892363158,
      "learning_rate": 7.872487126507747e-05,
      "loss": 0.3925,
      "step": 600
    },
    {
      "epoch": 0.8647482014388489,
      "grad_norm": 0.3338655060234199,
      "learning_rate": 7.87147857681345e-05,
      "loss": 0.4075,
      "step": 601
    },
    {
      "epoch": 0.8661870503597122,
      "grad_norm": 0.35575445911282527,
      "learning_rate": 7.870466119422826e-05,
      "loss": 0.3902,
      "step": 602
    },
    {
      "epoch": 0.8676258992805755,
      "grad_norm": 0.3368046090318334,
      "learning_rate": 7.869449755357803e-05,
      "loss": 0.3898,
      "step": 603
    },
    {
      "epoch": 0.8690647482014389,
      "grad_norm": 0.2883199718243486,
      "learning_rate": 7.868429485644252e-05,
      "loss": 0.3919,
      "step": 604
    },
    {
      "epoch": 0.8705035971223022,
      "grad_norm": 0.26703764250144163,
      "learning_rate": 7.86740531131199e-05,
      "loss": 0.3975,
      "step": 605
    },
    {
      "epoch": 0.8719424460431655,
      "grad_norm": 0.30522825564675843,
      "learning_rate": 7.866377233394771e-05,
      "loss": 0.3936,
      "step": 606
    },
    {
      "epoch": 0.8733812949640288,
      "grad_norm": 0.32195739971921056,
      "learning_rate": 7.865345252930291e-05,
      "loss": 0.3953,
      "step": 607
    },
    {
      "epoch": 0.874820143884892,
      "grad_norm": 0.280171972891727,
      "learning_rate": 7.864309370960184e-05,
      "loss": 0.392,
      "step": 608
    },
    {
      "epoch": 0.8762589928057554,
      "grad_norm": 0.26936379941264654,
      "learning_rate": 7.863269588530023e-05,
      "loss": 0.3837,
      "step": 609
    },
    {
      "epoch": 0.8776978417266187,
      "grad_norm": 0.3070891060623235,
      "learning_rate": 7.862225906689319e-05,
      "loss": 0.3847,
      "step": 610
    },
    {
      "epoch": 0.879136690647482,
      "grad_norm": 0.3199770059266285,
      "learning_rate": 7.861178326491514e-05,
      "loss": 0.4021,
      "step": 611
    },
    {
      "epoch": 0.8805755395683453,
      "grad_norm": 0.29592006497124823,
      "learning_rate": 7.860126848993992e-05,
      "loss": 0.3941,
      "step": 612
    },
    {
      "epoch": 0.8820143884892087,
      "grad_norm": 0.2906273468610675,
      "learning_rate": 7.859071475258065e-05,
      "loss": 0.3927,
      "step": 613
    },
    {
      "epoch": 0.883453237410072,
      "grad_norm": 0.2831202573537749,
      "learning_rate": 7.85801220634898e-05,
      "loss": 0.3895,
      "step": 614
    },
    {
      "epoch": 0.8848920863309353,
      "grad_norm": 0.21640349754386537,
      "learning_rate": 7.856949043335917e-05,
      "loss": 0.388,
      "step": 615
    },
    {
      "epoch": 0.8863309352517985,
      "grad_norm": 0.16980883227629529,
      "learning_rate": 7.855881987291983e-05,
      "loss": 0.3841,
      "step": 616
    },
    {
      "epoch": 0.8877697841726618,
      "grad_norm": 0.2206574043305558,
      "learning_rate": 7.854811039294216e-05,
      "loss": 0.3995,
      "step": 617
    },
    {
      "epoch": 0.8892086330935252,
      "grad_norm": 0.22489015667648418,
      "learning_rate": 7.853736200423584e-05,
      "loss": 0.3839,
      "step": 618
    },
    {
      "epoch": 0.8906474820143885,
      "grad_norm": 0.19746269430804855,
      "learning_rate": 7.852657471764983e-05,
      "loss": 0.4014,
      "step": 619
    },
    {
      "epoch": 0.8920863309352518,
      "grad_norm": 0.16512005648681502,
      "learning_rate": 7.851574854407228e-05,
      "loss": 0.3943,
      "step": 620
    },
    {
      "epoch": 0.8935251798561151,
      "grad_norm": 0.2058837706591543,
      "learning_rate": 7.85048834944307e-05,
      "loss": 0.3913,
      "step": 621
    },
    {
      "epoch": 0.8949640287769784,
      "grad_norm": 0.2377523174843737,
      "learning_rate": 7.849397957969173e-05,
      "loss": 0.3867,
      "step": 622
    },
    {
      "epoch": 0.8964028776978418,
      "grad_norm": 0.2412960219227453,
      "learning_rate": 7.848303681086134e-05,
      "loss": 0.3885,
      "step": 623
    },
    {
      "epoch": 0.897841726618705,
      "grad_norm": 0.2109556511078225,
      "learning_rate": 7.847205519898461e-05,
      "loss": 0.3831,
      "step": 624
    },
    {
      "epoch": 0.8992805755395683,
      "grad_norm": 0.18051444186197507,
      "learning_rate": 7.846103475514595e-05,
      "loss": 0.3904,
      "step": 625
    },
    {
      "epoch": 0.9007194244604316,
      "grad_norm": 0.2324178742893254,
      "learning_rate": 7.844997549046886e-05,
      "loss": 0.3888,
      "step": 626
    },
    {
      "epoch": 0.902158273381295,
      "grad_norm": 0.34544320106868226,
      "learning_rate": 7.843887741611608e-05,
      "loss": 0.398,
      "step": 627
    },
    {
      "epoch": 0.9035971223021583,
      "grad_norm": 0.3834819492352676,
      "learning_rate": 7.842774054328949e-05,
      "loss": 0.3804,
      "step": 628
    },
    {
      "epoch": 0.9050359712230216,
      "grad_norm": 0.37512679464727644,
      "learning_rate": 7.841656488323017e-05,
      "loss": 0.3942,
      "step": 629
    },
    {
      "epoch": 0.9064748201438849,
      "grad_norm": 0.33642676880227634,
      "learning_rate": 7.840535044721832e-05,
      "loss": 0.3906,
      "step": 630
    },
    {
      "epoch": 0.9079136690647482,
      "grad_norm": 0.2776328123356685,
      "learning_rate": 7.839409724657327e-05,
      "loss": 0.39,
      "step": 631
    },
    {
      "epoch": 0.9093525179856116,
      "grad_norm": 0.264229711157273,
      "learning_rate": 7.838280529265353e-05,
      "loss": 0.3928,
      "step": 632
    },
    {
      "epoch": 0.9107913669064748,
      "grad_norm": 0.22856035604344255,
      "learning_rate": 7.837147459685666e-05,
      "loss": 0.38,
      "step": 633
    },
    {
      "epoch": 0.9122302158273381,
      "grad_norm": 0.19021078124482907,
      "learning_rate": 7.836010517061937e-05,
      "loss": 0.3964,
      "step": 634
    },
    {
      "epoch": 0.9136690647482014,
      "grad_norm": 0.22192646472597122,
      "learning_rate": 7.834869702541742e-05,
      "loss": 0.3918,
      "step": 635
    },
    {
      "epoch": 0.9151079136690647,
      "grad_norm": 0.2247343099527254,
      "learning_rate": 7.833725017276573e-05,
      "loss": 0.3924,
      "step": 636
    },
    {
      "epoch": 0.9165467625899281,
      "grad_norm": 0.20099820906716964,
      "learning_rate": 7.83257646242182e-05,
      "loss": 0.3886,
      "step": 637
    },
    {
      "epoch": 0.9179856115107914,
      "grad_norm": 0.2006531932455901,
      "learning_rate": 7.831424039136783e-05,
      "loss": 0.3948,
      "step": 638
    },
    {
      "epoch": 0.9194244604316547,
      "grad_norm": 0.2135581623713477,
      "learning_rate": 7.830267748584666e-05,
      "loss": 0.3909,
      "step": 639
    },
    {
      "epoch": 0.920863309352518,
      "grad_norm": 0.2930249173439451,
      "learning_rate": 7.829107591932578e-05,
      "loss": 0.3834,
      "step": 640
    },
    {
      "epoch": 0.9223021582733812,
      "grad_norm": 0.3510158419143827,
      "learning_rate": 7.82794357035153e-05,
      "loss": 0.3855,
      "step": 641
    },
    {
      "epoch": 0.9237410071942446,
      "grad_norm": 0.334990714639028,
      "learning_rate": 7.82677568501643e-05,
      "loss": 0.3889,
      "step": 642
    },
    {
      "epoch": 0.9251798561151079,
      "grad_norm": 0.3292222176629868,
      "learning_rate": 7.82560393710609e-05,
      "loss": 0.386,
      "step": 643
    },
    {
      "epoch": 0.9266187050359712,
      "grad_norm": 0.33404852035613686,
      "learning_rate": 7.824428327803221e-05,
      "loss": 0.3919,
      "step": 644
    },
    {
      "epoch": 0.9280575539568345,
      "grad_norm": 0.3436229866403737,
      "learning_rate": 7.823248858294428e-05,
      "loss": 0.3969,
      "step": 645
    },
    {
      "epoch": 0.9294964028776979,
      "grad_norm": 0.34608808373655575,
      "learning_rate": 7.822065529770216e-05,
      "loss": 0.3918,
      "step": 646
    },
    {
      "epoch": 0.9309352517985612,
      "grad_norm": 0.3568453570260214,
      "learning_rate": 7.820878343424984e-05,
      "loss": 0.3827,
      "step": 647
    },
    {
      "epoch": 0.9323741007194245,
      "grad_norm": 0.4080463861388677,
      "learning_rate": 7.819687300457021e-05,
      "loss": 0.384,
      "step": 648
    },
    {
      "epoch": 0.9338129496402877,
      "grad_norm": 0.4398849386803914,
      "learning_rate": 7.818492402068517e-05,
      "loss": 0.3936,
      "step": 649
    },
    {
      "epoch": 0.935251798561151,
      "grad_norm": 0.4242173179698511,
      "learning_rate": 7.817293649465546e-05,
      "loss": 0.3946,
      "step": 650
    },
    {
      "epoch": 0.9366906474820144,
      "grad_norm": 0.3000781182109413,
      "learning_rate": 7.816091043858076e-05,
      "loss": 0.3927,
      "step": 651
    },
    {
      "epoch": 0.9381294964028777,
      "grad_norm": 0.2671248146821321,
      "learning_rate": 7.814884586459962e-05,
      "loss": 0.3927,
      "step": 652
    },
    {
      "epoch": 0.939568345323741,
      "grad_norm": 0.3110113659917515,
      "learning_rate": 7.813674278488949e-05,
      "loss": 0.3853,
      "step": 653
    },
    {
      "epoch": 0.9410071942446043,
      "grad_norm": 0.32913498505284944,
      "learning_rate": 7.812460121166666e-05,
      "loss": 0.4011,
      "step": 654
    },
    {
      "epoch": 0.9424460431654677,
      "grad_norm": 0.2560732729098207,
      "learning_rate": 7.81124211571863e-05,
      "loss": 0.3849,
      "step": 655
    },
    {
      "epoch": 0.943884892086331,
      "grad_norm": 0.23473288445415613,
      "learning_rate": 7.810020263374239e-05,
      "loss": 0.3849,
      "step": 656
    },
    {
      "epoch": 0.9453237410071943,
      "grad_norm": 0.25502262467547715,
      "learning_rate": 7.808794565366778e-05,
      "loss": 0.3841,
      "step": 657
    },
    {
      "epoch": 0.9467625899280575,
      "grad_norm": 0.2675264909920931,
      "learning_rate": 7.807565022933412e-05,
      "loss": 0.3807,
      "step": 658
    },
    {
      "epoch": 0.9482014388489208,
      "grad_norm": 0.32022221655981714,
      "learning_rate": 7.806331637315183e-05,
      "loss": 0.3845,
      "step": 659
    },
    {
      "epoch": 0.9496402877697842,
      "grad_norm": 0.3358875172701553,
      "learning_rate": 7.805094409757017e-05,
      "loss": 0.3898,
      "step": 660
    },
    {
      "epoch": 0.9510791366906475,
      "grad_norm": 0.22558263781799218,
      "learning_rate": 7.803853341507715e-05,
      "loss": 0.3947,
      "step": 661
    },
    {
      "epoch": 0.9525179856115108,
      "grad_norm": 0.22029002177773868,
      "learning_rate": 7.802608433819957e-05,
      "loss": 0.3944,
      "step": 662
    },
    {
      "epoch": 0.9539568345323741,
      "grad_norm": 0.3013802495912675,
      "learning_rate": 7.801359687950292e-05,
      "loss": 0.4043,
      "step": 663
    },
    {
      "epoch": 0.9553956834532374,
      "grad_norm": 0.2732169010636322,
      "learning_rate": 7.800107105159155e-05,
      "loss": 0.3878,
      "step": 664
    },
    {
      "epoch": 0.9568345323741008,
      "grad_norm": 0.15983177234294135,
      "learning_rate": 7.798850686710841e-05,
      "loss": 0.3965,
      "step": 665
    },
    {
      "epoch": 0.958273381294964,
      "grad_norm": 0.24494999138716617,
      "learning_rate": 7.797590433873526e-05,
      "loss": 0.388,
      "step": 666
    },
    {
      "epoch": 0.9597122302158273,
      "grad_norm": 0.32383354768412387,
      "learning_rate": 7.79632634791925e-05,
      "loss": 0.387,
      "step": 667
    },
    {
      "epoch": 0.9611510791366906,
      "grad_norm": 0.2620658766378879,
      "learning_rate": 7.795058430123925e-05,
      "loss": 0.3962,
      "step": 668
    },
    {
      "epoch": 0.962589928057554,
      "grad_norm": 0.22541523028018362,
      "learning_rate": 7.793786681767333e-05,
      "loss": 0.3872,
      "step": 669
    },
    {
      "epoch": 0.9640287769784173,
      "grad_norm": 0.22687402775204735,
      "learning_rate": 7.792511104133117e-05,
      "loss": 0.381,
      "step": 670
    },
    {
      "epoch": 0.9654676258992806,
      "grad_norm": 0.25659393392471547,
      "learning_rate": 7.791231698508786e-05,
      "loss": 0.3876,
      "step": 671
    },
    {
      "epoch": 0.9669064748201439,
      "grad_norm": 0.3045763973091339,
      "learning_rate": 7.789948466185718e-05,
      "loss": 0.3708,
      "step": 672
    },
    {
      "epoch": 0.9683453237410072,
      "grad_norm": 0.2566836653530128,
      "learning_rate": 7.788661408459146e-05,
      "loss": 0.3886,
      "step": 673
    },
    {
      "epoch": 0.9697841726618706,
      "grad_norm": 0.23594214728583138,
      "learning_rate": 7.787370526628173e-05,
      "loss": 0.4012,
      "step": 674
    },
    {
      "epoch": 0.9712230215827338,
      "grad_norm": 0.24609866995203958,
      "learning_rate": 7.786075821995754e-05,
      "loss": 0.3934,
      "step": 675
    },
    {
      "epoch": 0.9726618705035971,
      "grad_norm": 0.3197191898312636,
      "learning_rate": 7.784777295868706e-05,
      "loss": 0.3894,
      "step": 676
    },
    {
      "epoch": 0.9741007194244604,
      "grad_norm": 0.4005845521559889,
      "learning_rate": 7.783474949557704e-05,
      "loss": 0.3856,
      "step": 677
    },
    {
      "epoch": 0.9755395683453237,
      "grad_norm": 0.4425181618821247,
      "learning_rate": 7.782168784377276e-05,
      "loss": 0.3959,
      "step": 678
    },
    {
      "epoch": 0.9769784172661871,
      "grad_norm": 0.4332856052063338,
      "learning_rate": 7.780858801645806e-05,
      "loss": 0.3885,
      "step": 679
    },
    {
      "epoch": 0.9784172661870504,
      "grad_norm": 0.3952524434700546,
      "learning_rate": 7.779545002685535e-05,
      "loss": 0.393,
      "step": 680
    },
    {
      "epoch": 0.9798561151079137,
      "grad_norm": 0.33755031917203393,
      "learning_rate": 7.778227388822552e-05,
      "loss": 0.389,
      "step": 681
    },
    {
      "epoch": 0.981294964028777,
      "grad_norm": 0.273520410377852,
      "learning_rate": 7.776905961386793e-05,
      "loss": 0.3939,
      "step": 682
    },
    {
      "epoch": 0.9827338129496402,
      "grad_norm": 0.2906169847789615,
      "learning_rate": 7.77558072171205e-05,
      "loss": 0.3961,
      "step": 683
    },
    {
      "epoch": 0.9841726618705036,
      "grad_norm": 0.29722724301916326,
      "learning_rate": 7.774251671135961e-05,
      "loss": 0.3812,
      "step": 684
    },
    {
      "epoch": 0.9856115107913669,
      "grad_norm": 0.2520810803176553,
      "learning_rate": 7.77291881100001e-05,
      "loss": 0.391,
      "step": 685
    },
    {
      "epoch": 0.9870503597122302,
      "grad_norm": 0.18539473632642314,
      "learning_rate": 7.771582142649523e-05,
      "loss": 0.3854,
      "step": 686
    },
    {
      "epoch": 0.9884892086330935,
      "grad_norm": 0.23387948076579174,
      "learning_rate": 7.770241667433677e-05,
      "loss": 0.3938,
      "step": 687
    },
    {
      "epoch": 0.9899280575539569,
      "grad_norm": 0.32818364161003727,
      "learning_rate": 7.768897386705488e-05,
      "loss": 0.3881,
      "step": 688
    },
    {
      "epoch": 0.9913669064748202,
      "grad_norm": 0.3151278015842664,
      "learning_rate": 7.767549301821807e-05,
      "loss": 0.3891,
      "step": 689
    },
    {
      "epoch": 0.9928057553956835,
      "grad_norm": 0.1978620567014893,
      "learning_rate": 7.766197414143333e-05,
      "loss": 0.3837,
      "step": 690
    },
    {
      "epoch": 0.9942446043165467,
      "grad_norm": 0.22559586800348716,
      "learning_rate": 7.764841725034602e-05,
      "loss": 0.3925,
      "step": 691
    },
    {
      "epoch": 0.99568345323741,
      "grad_norm": 0.28410829917691,
      "learning_rate": 7.763482235863985e-05,
      "loss": 0.3892,
      "step": 692
    },
    {
      "epoch": 0.9971223021582734,
      "grad_norm": 0.2648279300455164,
      "learning_rate": 7.762118948003688e-05,
      "loss": 0.3907,
      "step": 693
    },
    {
      "epoch": 0.9985611510791367,
      "grad_norm": 0.19968267788630498,
      "learning_rate": 7.760751862829754e-05,
      "loss": 0.3839,
      "step": 694
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.2618737125473869,
      "learning_rate": 7.759380981722055e-05,
      "loss": 0.3857,
      "step": 695
    },
    {
      "epoch": 1.0014388489208632,
      "grad_norm": 0.31553007501168995,
      "learning_rate": 7.758006306064301e-05,
      "loss": 0.3717,
      "step": 696
    },
    {
      "epoch": 1.0028776978417266,
      "grad_norm": 0.33965441725150664,
      "learning_rate": 7.756627837244023e-05,
      "loss": 0.3734,
      "step": 697
    },
    {
      "epoch": 1.0043165467625899,
      "grad_norm": 0.40641548452468257,
      "learning_rate": 7.755245576652588e-05,
      "loss": 0.3791,
      "step": 698
    },
    {
      "epoch": 1.0057553956834533,
      "grad_norm": 0.43018797825921173,
      "learning_rate": 7.753859525685187e-05,
      "loss": 0.3713,
      "step": 699
    },
    {
      "epoch": 1.0071942446043165,
      "grad_norm": 0.38735977677912703,
      "learning_rate": 7.752469685740838e-05,
      "loss": 0.379,
      "step": 700
    },
    {
      "epoch": 1.00863309352518,
      "grad_norm": 0.34346977718231,
      "learning_rate": 7.751076058222381e-05,
      "loss": 0.3666,
      "step": 701
    },
    {
      "epoch": 1.0100719424460431,
      "grad_norm": 0.2758009922986033,
      "learning_rate": 7.749678644536485e-05,
      "loss": 0.364,
      "step": 702
    },
    {
      "epoch": 1.0115107913669066,
      "grad_norm": 0.2508222065726072,
      "learning_rate": 7.748277446093631e-05,
      "loss": 0.3722,
      "step": 703
    },
    {
      "epoch": 1.0129496402877698,
      "grad_norm": 0.2581419457116818,
      "learning_rate": 7.746872464308131e-05,
      "loss": 0.3694,
      "step": 704
    },
    {
      "epoch": 1.014388489208633,
      "grad_norm": 0.2663971309609672,
      "learning_rate": 7.745463700598108e-05,
      "loss": 0.3723,
      "step": 705
    },
    {
      "epoch": 1.0158273381294964,
      "grad_norm": 0.24874575875769642,
      "learning_rate": 7.744051156385503e-05,
      "loss": 0.3707,
      "step": 706
    },
    {
      "epoch": 1.0172661870503596,
      "grad_norm": 0.2274756545689248,
      "learning_rate": 7.742634833096077e-05,
      "loss": 0.373,
      "step": 707
    },
    {
      "epoch": 1.018705035971223,
      "grad_norm": 0.22046600028071656,
      "learning_rate": 7.741214732159403e-05,
      "loss": 0.3793,
      "step": 708
    },
    {
      "epoch": 1.0201438848920863,
      "grad_norm": 0.23300885777665975,
      "learning_rate": 7.739790855008867e-05,
      "loss": 0.3733,
      "step": 709
    },
    {
      "epoch": 1.0215827338129497,
      "grad_norm": 0.24905008563481562,
      "learning_rate": 7.738363203081664e-05,
      "loss": 0.3747,
      "step": 710
    },
    {
      "epoch": 1.023021582733813,
      "grad_norm": 0.2678606420343796,
      "learning_rate": 7.736931777818805e-05,
      "loss": 0.3851,
      "step": 711
    },
    {
      "epoch": 1.0244604316546762,
      "grad_norm": 0.24026951579855546,
      "learning_rate": 7.735496580665105e-05,
      "loss": 0.3803,
      "step": 712
    },
    {
      "epoch": 1.0258992805755396,
      "grad_norm": 0.27398099644606966,
      "learning_rate": 7.734057613069188e-05,
      "loss": 0.3711,
      "step": 713
    },
    {
      "epoch": 1.0273381294964028,
      "grad_norm": 0.2690255911738012,
      "learning_rate": 7.73261487648348e-05,
      "loss": 0.3737,
      "step": 714
    },
    {
      "epoch": 1.0287769784172662,
      "grad_norm": 0.22042977642621647,
      "learning_rate": 7.731168372364219e-05,
      "loss": 0.3657,
      "step": 715
    },
    {
      "epoch": 1.0302158273381294,
      "grad_norm": 0.23820484191579894,
      "learning_rate": 7.729718102171438e-05,
      "loss": 0.3755,
      "step": 716
    },
    {
      "epoch": 1.0316546762589929,
      "grad_norm": 0.26180916905216817,
      "learning_rate": 7.728264067368976e-05,
      "loss": 0.3718,
      "step": 717
    },
    {
      "epoch": 1.033093525179856,
      "grad_norm": 0.27001803149465553,
      "learning_rate": 7.726806269424469e-05,
      "loss": 0.3657,
      "step": 718
    },
    {
      "epoch": 1.0345323741007195,
      "grad_norm": 0.2937530686224215,
      "learning_rate": 7.725344709809355e-05,
      "loss": 0.376,
      "step": 719
    },
    {
      "epoch": 1.0359712230215827,
      "grad_norm": 0.2973441276813663,
      "learning_rate": 7.723879389998864e-05,
      "loss": 0.3806,
      "step": 720
    },
    {
      "epoch": 1.037410071942446,
      "grad_norm": 0.25307126825041176,
      "learning_rate": 7.722410311472026e-05,
      "loss": 0.3678,
      "step": 721
    },
    {
      "epoch": 1.0388489208633094,
      "grad_norm": 0.19698468151176968,
      "learning_rate": 7.72093747571166e-05,
      "loss": 0.3688,
      "step": 722
    },
    {
      "epoch": 1.0402877697841726,
      "grad_norm": 0.17878109299499642,
      "learning_rate": 7.719460884204383e-05,
      "loss": 0.368,
      "step": 723
    },
    {
      "epoch": 1.041726618705036,
      "grad_norm": 0.22288327986868614,
      "learning_rate": 7.717980538440599e-05,
      "loss": 0.3683,
      "step": 724
    },
    {
      "epoch": 1.0431654676258992,
      "grad_norm": 0.2320159277967116,
      "learning_rate": 7.716496439914502e-05,
      "loss": 0.3645,
      "step": 725
    },
    {
      "epoch": 1.0446043165467627,
      "grad_norm": 0.21976899032515732,
      "learning_rate": 7.715008590124076e-05,
      "loss": 0.3783,
      "step": 726
    },
    {
      "epoch": 1.0460431654676259,
      "grad_norm": 0.2629062946538339,
      "learning_rate": 7.713516990571088e-05,
      "loss": 0.3704,
      "step": 727
    },
    {
      "epoch": 1.0474820143884893,
      "grad_norm": 0.28509364456056624,
      "learning_rate": 7.71202164276109e-05,
      "loss": 0.3721,
      "step": 728
    },
    {
      "epoch": 1.0489208633093525,
      "grad_norm": 0.2454798623600427,
      "learning_rate": 7.710522548203424e-05,
      "loss": 0.3751,
      "step": 729
    },
    {
      "epoch": 1.0503597122302157,
      "grad_norm": 0.2354289666700434,
      "learning_rate": 7.709019708411202e-05,
      "loss": 0.3758,
      "step": 730
    },
    {
      "epoch": 1.0517985611510792,
      "grad_norm": 0.24512748729268605,
      "learning_rate": 7.707513124901327e-05,
      "loss": 0.3666,
      "step": 731
    },
    {
      "epoch": 1.0532374100719424,
      "grad_norm": 0.3008060196418346,
      "learning_rate": 7.706002799194476e-05,
      "loss": 0.3647,
      "step": 732
    },
    {
      "epoch": 1.0546762589928058,
      "grad_norm": 0.34124747562157987,
      "learning_rate": 7.704488732815105e-05,
      "loss": 0.3674,
      "step": 733
    },
    {
      "epoch": 1.056115107913669,
      "grad_norm": 0.2934930694625762,
      "learning_rate": 7.702970927291442e-05,
      "loss": 0.3718,
      "step": 734
    },
    {
      "epoch": 1.0575539568345325,
      "grad_norm": 0.19598463515263995,
      "learning_rate": 7.701449384155492e-05,
      "loss": 0.3808,
      "step": 735
    },
    {
      "epoch": 1.0589928057553957,
      "grad_norm": 0.21231686194205143,
      "learning_rate": 7.699924104943033e-05,
      "loss": 0.3791,
      "step": 736
    },
    {
      "epoch": 1.0604316546762589,
      "grad_norm": 0.2793791463484476,
      "learning_rate": 7.698395091193615e-05,
      "loss": 0.3678,
      "step": 737
    },
    {
      "epoch": 1.0618705035971223,
      "grad_norm": 0.25077747032290765,
      "learning_rate": 7.696862344450553e-05,
      "loss": 0.3659,
      "step": 738
    },
    {
      "epoch": 1.0633093525179855,
      "grad_norm": 0.2698416723588037,
      "learning_rate": 7.695325866260932e-05,
      "loss": 0.3773,
      "step": 739
    },
    {
      "epoch": 1.064748201438849,
      "grad_norm": 0.25574664417176746,
      "learning_rate": 7.693785658175607e-05,
      "loss": 0.3634,
      "step": 740
    },
    {
      "epoch": 1.0661870503597122,
      "grad_norm": 0.2120620821263056,
      "learning_rate": 7.692241721749194e-05,
      "loss": 0.3707,
      "step": 741
    },
    {
      "epoch": 1.0676258992805756,
      "grad_norm": 0.2364943418652984,
      "learning_rate": 7.69069405854007e-05,
      "loss": 0.3737,
      "step": 742
    },
    {
      "epoch": 1.0690647482014388,
      "grad_norm": 0.22536048745525106,
      "learning_rate": 7.68914267011038e-05,
      "loss": 0.3694,
      "step": 743
    },
    {
      "epoch": 1.0705035971223023,
      "grad_norm": 0.22723369017031214,
      "learning_rate": 7.687587558026024e-05,
      "loss": 0.371,
      "step": 744
    },
    {
      "epoch": 1.0719424460431655,
      "grad_norm": 0.21568294234696916,
      "learning_rate": 7.686028723856664e-05,
      "loss": 0.3688,
      "step": 745
    },
    {
      "epoch": 1.0733812949640287,
      "grad_norm": 0.2528091241920581,
      "learning_rate": 7.684466169175714e-05,
      "loss": 0.3739,
      "step": 746
    },
    {
      "epoch": 1.074820143884892,
      "grad_norm": 0.2829084202430895,
      "learning_rate": 7.68289989556035e-05,
      "loss": 0.3642,
      "step": 747
    },
    {
      "epoch": 1.0762589928057553,
      "grad_norm": 0.2889957548752893,
      "learning_rate": 7.681329904591495e-05,
      "loss": 0.365,
      "step": 748
    },
    {
      "epoch": 1.0776978417266188,
      "grad_norm": 0.30667124621667996,
      "learning_rate": 7.67975619785383e-05,
      "loss": 0.3681,
      "step": 749
    },
    {
      "epoch": 1.079136690647482,
      "grad_norm": 0.29156344248640786,
      "learning_rate": 7.678178776935781e-05,
      "loss": 0.3717,
      "step": 750
    },
    {
      "epoch": 1.0805755395683454,
      "grad_norm": 0.26343478914536866,
      "learning_rate": 7.676597643429528e-05,
      "loss": 0.3739,
      "step": 751
    },
    {
      "epoch": 1.0820143884892086,
      "grad_norm": 0.26201198141793924,
      "learning_rate": 7.675012798930994e-05,
      "loss": 0.3742,
      "step": 752
    },
    {
      "epoch": 1.083453237410072,
      "grad_norm": 0.2762651821216201,
      "learning_rate": 7.673424245039852e-05,
      "loss": 0.3715,
      "step": 753
    },
    {
      "epoch": 1.0848920863309353,
      "grad_norm": 0.22200173851448812,
      "learning_rate": 7.671831983359515e-05,
      "loss": 0.3761,
      "step": 754
    },
    {
      "epoch": 1.0863309352517985,
      "grad_norm": 0.22087460184150698,
      "learning_rate": 7.670236015497141e-05,
      "loss": 0.3671,
      "step": 755
    },
    {
      "epoch": 1.087769784172662,
      "grad_norm": 0.266194809910504,
      "learning_rate": 7.668636343063628e-05,
      "loss": 0.3651,
      "step": 756
    },
    {
      "epoch": 1.0892086330935251,
      "grad_norm": 0.30392490240355563,
      "learning_rate": 7.667032967673614e-05,
      "loss": 0.3726,
      "step": 757
    },
    {
      "epoch": 1.0906474820143885,
      "grad_norm": 0.32390718405302116,
      "learning_rate": 7.665425890945474e-05,
      "loss": 0.3756,
      "step": 758
    },
    {
      "epoch": 1.0920863309352518,
      "grad_norm": 0.313310504767798,
      "learning_rate": 7.663815114501319e-05,
      "loss": 0.3685,
      "step": 759
    },
    {
      "epoch": 1.0935251798561152,
      "grad_norm": 0.275653965646266,
      "learning_rate": 7.662200639966992e-05,
      "loss": 0.3611,
      "step": 760
    },
    {
      "epoch": 1.0949640287769784,
      "grad_norm": 0.2797921397183036,
      "learning_rate": 7.660582468972074e-05,
      "loss": 0.3804,
      "step": 761
    },
    {
      "epoch": 1.0964028776978418,
      "grad_norm": 0.34469485096564356,
      "learning_rate": 7.658960603149873e-05,
      "loss": 0.3654,
      "step": 762
    },
    {
      "epoch": 1.097841726618705,
      "grad_norm": 0.34903953742928767,
      "learning_rate": 7.657335044137427e-05,
      "loss": 0.3698,
      "step": 763
    },
    {
      "epoch": 1.0992805755395683,
      "grad_norm": 0.27823069653353727,
      "learning_rate": 7.655705793575504e-05,
      "loss": 0.366,
      "step": 764
    },
    {
      "epoch": 1.1007194244604317,
      "grad_norm": 0.24822074396235053,
      "learning_rate": 7.654072853108592e-05,
      "loss": 0.3717,
      "step": 765
    },
    {
      "epoch": 1.102158273381295,
      "grad_norm": 0.25840628636978735,
      "learning_rate": 7.652436224384911e-05,
      "loss": 0.3703,
      "step": 766
    },
    {
      "epoch": 1.1035971223021583,
      "grad_norm": 0.2529631240696829,
      "learning_rate": 7.6507959090564e-05,
      "loss": 0.3722,
      "step": 767
    },
    {
      "epoch": 1.1050359712230216,
      "grad_norm": 0.24929231191974152,
      "learning_rate": 7.649151908778721e-05,
      "loss": 0.3722,
      "step": 768
    },
    {
      "epoch": 1.106474820143885,
      "grad_norm": 0.2269118319413078,
      "learning_rate": 7.64750422521125e-05,
      "loss": 0.3746,
      "step": 769
    },
    {
      "epoch": 1.1079136690647482,
      "grad_norm": 0.21647296032756777,
      "learning_rate": 7.645852860017086e-05,
      "loss": 0.367,
      "step": 770
    },
    {
      "epoch": 1.1093525179856114,
      "grad_norm": 0.2471192399071732,
      "learning_rate": 7.644197814863045e-05,
      "loss": 0.3684,
      "step": 771
    },
    {
      "epoch": 1.1107913669064748,
      "grad_norm": 0.25791371757926,
      "learning_rate": 7.642539091419654e-05,
      "loss": 0.3771,
      "step": 772
    },
    {
      "epoch": 1.112230215827338,
      "grad_norm": 0.21350379236032627,
      "learning_rate": 7.640876691361152e-05,
      "loss": 0.372,
      "step": 773
    },
    {
      "epoch": 1.1136690647482015,
      "grad_norm": 0.19325186323404028,
      "learning_rate": 7.639210616365494e-05,
      "loss": 0.3747,
      "step": 774
    },
    {
      "epoch": 1.1151079136690647,
      "grad_norm": 0.2329585900087163,
      "learning_rate": 7.637540868114338e-05,
      "loss": 0.3767,
      "step": 775
    },
    {
      "epoch": 1.1165467625899281,
      "grad_norm": 0.2312114835528738,
      "learning_rate": 7.635867448293056e-05,
      "loss": 0.3699,
      "step": 776
    },
    {
      "epoch": 1.1179856115107913,
      "grad_norm": 0.22412843889749232,
      "learning_rate": 7.63419035859072e-05,
      "loss": 0.3687,
      "step": 777
    },
    {
      "epoch": 1.1194244604316548,
      "grad_norm": 0.1692205798761735,
      "learning_rate": 7.63250960070011e-05,
      "loss": 0.3722,
      "step": 778
    },
    {
      "epoch": 1.120863309352518,
      "grad_norm": 0.16114224565456758,
      "learning_rate": 7.630825176317707e-05,
      "loss": 0.3697,
      "step": 779
    },
    {
      "epoch": 1.1223021582733812,
      "grad_norm": 0.2213503208640909,
      "learning_rate": 7.629137087143693e-05,
      "loss": 0.3738,
      "step": 780
    },
    {
      "epoch": 1.1237410071942446,
      "grad_norm": 0.24905814357039824,
      "learning_rate": 7.627445334881951e-05,
      "loss": 0.3738,
      "step": 781
    },
    {
      "epoch": 1.1251798561151078,
      "grad_norm": 0.27876416934327075,
      "learning_rate": 7.625749921240058e-05,
      "loss": 0.3702,
      "step": 782
    },
    {
      "epoch": 1.1266187050359713,
      "grad_norm": 0.2907151065971304,
      "learning_rate": 7.62405084792929e-05,
      "loss": 0.3805,
      "step": 783
    },
    {
      "epoch": 1.1280575539568345,
      "grad_norm": 0.3183459859308742,
      "learning_rate": 7.622348116664611e-05,
      "loss": 0.3761,
      "step": 784
    },
    {
      "epoch": 1.129496402877698,
      "grad_norm": 0.28647369901052755,
      "learning_rate": 7.620641729164686e-05,
      "loss": 0.3804,
      "step": 785
    },
    {
      "epoch": 1.1309352517985611,
      "grad_norm": 0.26695805060206323,
      "learning_rate": 7.618931687151863e-05,
      "loss": 0.3851,
      "step": 786
    },
    {
      "epoch": 1.1323741007194243,
      "grad_norm": 0.21157618631799555,
      "learning_rate": 7.617217992352183e-05,
      "loss": 0.3712,
      "step": 787
    },
    {
      "epoch": 1.1338129496402878,
      "grad_norm": 0.17102518487708407,
      "learning_rate": 7.615500646495373e-05,
      "loss": 0.3617,
      "step": 788
    },
    {
      "epoch": 1.135251798561151,
      "grad_norm": 0.2327103761272104,
      "learning_rate": 7.613779651314841e-05,
      "loss": 0.3597,
      "step": 789
    },
    {
      "epoch": 1.1366906474820144,
      "grad_norm": 0.2855044981002147,
      "learning_rate": 7.612055008547688e-05,
      "loss": 0.3778,
      "step": 790
    },
    {
      "epoch": 1.1381294964028776,
      "grad_norm": 0.28780307592946286,
      "learning_rate": 7.610326719934685e-05,
      "loss": 0.3718,
      "step": 791
    },
    {
      "epoch": 1.139568345323741,
      "grad_norm": 0.2829437935854506,
      "learning_rate": 7.608594787220292e-05,
      "loss": 0.3694,
      "step": 792
    },
    {
      "epoch": 1.1410071942446043,
      "grad_norm": 0.28165833621378605,
      "learning_rate": 7.606859212152644e-05,
      "loss": 0.3734,
      "step": 793
    },
    {
      "epoch": 1.1424460431654677,
      "grad_norm": 0.266286805674129,
      "learning_rate": 7.605119996483551e-05,
      "loss": 0.3679,
      "step": 794
    },
    {
      "epoch": 1.143884892086331,
      "grad_norm": 0.2471187084073408,
      "learning_rate": 7.6033771419685e-05,
      "loss": 0.3724,
      "step": 795
    },
    {
      "epoch": 1.1453237410071941,
      "grad_norm": 0.2983714367288603,
      "learning_rate": 7.601630650366648e-05,
      "loss": 0.3755,
      "step": 796
    },
    {
      "epoch": 1.1467625899280576,
      "grad_norm": 0.3492960796329427,
      "learning_rate": 7.59988052344083e-05,
      "loss": 0.37,
      "step": 797
    },
    {
      "epoch": 1.1482014388489208,
      "grad_norm": 0.3751692962462577,
      "learning_rate": 7.59812676295754e-05,
      "loss": 0.3713,
      "step": 798
    },
    {
      "epoch": 1.1496402877697842,
      "grad_norm": 0.39396196639251957,
      "learning_rate": 7.596369370686947e-05,
      "loss": 0.3783,
      "step": 799
    },
    {
      "epoch": 1.1510791366906474,
      "grad_norm": 0.3445354561658745,
      "learning_rate": 7.594608348402885e-05,
      "loss": 0.3804,
      "step": 800
    },
    {
      "epoch": 1.1525179856115109,
      "grad_norm": 0.2735280407069058,
      "learning_rate": 7.592843697882848e-05,
      "loss": 0.3764,
      "step": 801
    },
    {
      "epoch": 1.153956834532374,
      "grad_norm": 0.2726335410361186,
      "learning_rate": 7.591075420907997e-05,
      "loss": 0.3727,
      "step": 802
    },
    {
      "epoch": 1.1553956834532375,
      "grad_norm": 0.2622606934047612,
      "learning_rate": 7.589303519263151e-05,
      "loss": 0.3684,
      "step": 803
    },
    {
      "epoch": 1.1568345323741007,
      "grad_norm": 0.2148203795515544,
      "learning_rate": 7.587527994736787e-05,
      "loss": 0.3683,
      "step": 804
    },
    {
      "epoch": 1.158273381294964,
      "grad_norm": 0.23153893883516943,
      "learning_rate": 7.58574884912104e-05,
      "loss": 0.3696,
      "step": 805
    },
    {
      "epoch": 1.1597122302158274,
      "grad_norm": 0.25522408205885655,
      "learning_rate": 7.5839660842117e-05,
      "loss": 0.3759,
      "step": 806
    },
    {
      "epoch": 1.1611510791366906,
      "grad_norm": 0.22696954301069733,
      "learning_rate": 7.582179701808208e-05,
      "loss": 0.3709,
      "step": 807
    },
    {
      "epoch": 1.162589928057554,
      "grad_norm": 0.20794558001687014,
      "learning_rate": 7.580389703713661e-05,
      "loss": 0.3672,
      "step": 808
    },
    {
      "epoch": 1.1640287769784172,
      "grad_norm": 0.23369271592611313,
      "learning_rate": 7.5785960917348e-05,
      "loss": 0.3638,
      "step": 809
    },
    {
      "epoch": 1.1654676258992807,
      "grad_norm": 0.2386353589713116,
      "learning_rate": 7.576798867682018e-05,
      "loss": 0.374,
      "step": 810
    },
    {
      "epoch": 1.1669064748201439,
      "grad_norm": 0.21507342856828493,
      "learning_rate": 7.574998033369349e-05,
      "loss": 0.3654,
      "step": 811
    },
    {
      "epoch": 1.1683453237410073,
      "grad_norm": 0.21609386965419786,
      "learning_rate": 7.573193590614479e-05,
      "loss": 0.3719,
      "step": 812
    },
    {
      "epoch": 1.1697841726618705,
      "grad_norm": 0.24765489873299398,
      "learning_rate": 7.571385541238727e-05,
      "loss": 0.3763,
      "step": 813
    },
    {
      "epoch": 1.1712230215827337,
      "grad_norm": 0.2508495575608087,
      "learning_rate": 7.569573887067059e-05,
      "loss": 0.3675,
      "step": 814
    },
    {
      "epoch": 1.1726618705035972,
      "grad_norm": 0.19808518068361278,
      "learning_rate": 7.567758629928076e-05,
      "loss": 0.3791,
      "step": 815
    },
    {
      "epoch": 1.1741007194244604,
      "grad_norm": 0.1979656718305221,
      "learning_rate": 7.565939771654018e-05,
      "loss": 0.3681,
      "step": 816
    },
    {
      "epoch": 1.1755395683453238,
      "grad_norm": 0.19541210281020807,
      "learning_rate": 7.564117314080758e-05,
      "loss": 0.3656,
      "step": 817
    },
    {
      "epoch": 1.176978417266187,
      "grad_norm": 0.24875723856508886,
      "learning_rate": 7.562291259047804e-05,
      "loss": 0.366,
      "step": 818
    },
    {
      "epoch": 1.1784172661870504,
      "grad_norm": 0.2593364294380872,
      "learning_rate": 7.560461608398292e-05,
      "loss": 0.3673,
      "step": 819
    },
    {
      "epoch": 1.1798561151079137,
      "grad_norm": 0.2037316550239657,
      "learning_rate": 7.558628363978991e-05,
      "loss": 0.3697,
      "step": 820
    },
    {
      "epoch": 1.181294964028777,
      "grad_norm": 0.24957260089896505,
      "learning_rate": 7.556791527640292e-05,
      "loss": 0.3691,
      "step": 821
    },
    {
      "epoch": 1.1827338129496403,
      "grad_norm": 0.2765448339671684,
      "learning_rate": 7.554951101236219e-05,
      "loss": 0.3725,
      "step": 822
    },
    {
      "epoch": 1.1841726618705035,
      "grad_norm": 0.20202979074251823,
      "learning_rate": 7.553107086624413e-05,
      "loss": 0.3595,
      "step": 823
    },
    {
      "epoch": 1.185611510791367,
      "grad_norm": 0.1740436926149169,
      "learning_rate": 7.551259485666141e-05,
      "loss": 0.3635,
      "step": 824
    },
    {
      "epoch": 1.1870503597122302,
      "grad_norm": 0.21631024506455465,
      "learning_rate": 7.549408300226287e-05,
      "loss": 0.3712,
      "step": 825
    },
    {
      "epoch": 1.1884892086330936,
      "grad_norm": 0.25177375486379844,
      "learning_rate": 7.547553532173356e-05,
      "loss": 0.3699,
      "step": 826
    },
    {
      "epoch": 1.1899280575539568,
      "grad_norm": 0.20550097869119144,
      "learning_rate": 7.545695183379465e-05,
      "loss": 0.3662,
      "step": 827
    },
    {
      "epoch": 1.19136690647482,
      "grad_norm": 0.1833546056638634,
      "learning_rate": 7.54383325572035e-05,
      "loss": 0.3717,
      "step": 828
    },
    {
      "epoch": 1.1928057553956835,
      "grad_norm": 0.21642428138305977,
      "learning_rate": 7.541967751075354e-05,
      "loss": 0.3779,
      "step": 829
    },
    {
      "epoch": 1.1942446043165469,
      "grad_norm": 0.22311333328763097,
      "learning_rate": 7.540098671327438e-05,
      "loss": 0.3672,
      "step": 830
    },
    {
      "epoch": 1.19568345323741,
      "grad_norm": 0.2230284422426472,
      "learning_rate": 7.538226018363164e-05,
      "loss": 0.3704,
      "step": 831
    },
    {
      "epoch": 1.1971223021582733,
      "grad_norm": 0.2233858353142221,
      "learning_rate": 7.536349794072705e-05,
      "loss": 0.3661,
      "step": 832
    },
    {
      "epoch": 1.1985611510791367,
      "grad_norm": 0.24431933896310673,
      "learning_rate": 7.534470000349835e-05,
      "loss": 0.3696,
      "step": 833
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.20570153241892766,
      "learning_rate": 7.532586639091936e-05,
      "loss": 0.378,
      "step": 834
    },
    {
      "epoch": 1.2014388489208634,
      "grad_norm": 0.23363348010581506,
      "learning_rate": 7.530699712199985e-05,
      "loss": 0.3638,
      "step": 835
    },
    {
      "epoch": 1.2028776978417266,
      "grad_norm": 0.2551061431484015,
      "learning_rate": 7.528809221578565e-05,
      "loss": 0.3663,
      "step": 836
    },
    {
      "epoch": 1.2043165467625898,
      "grad_norm": 0.20603774908286682,
      "learning_rate": 7.52691516913585e-05,
      "loss": 0.3804,
      "step": 837
    },
    {
      "epoch": 1.2057553956834532,
      "grad_norm": 0.230745543757371,
      "learning_rate": 7.525017556783612e-05,
      "loss": 0.3748,
      "step": 838
    },
    {
      "epoch": 1.2071942446043165,
      "grad_norm": 0.27823629877405837,
      "learning_rate": 7.523116386437216e-05,
      "loss": 0.3636,
      "step": 839
    },
    {
      "epoch": 1.20863309352518,
      "grad_norm": 0.27983234823070235,
      "learning_rate": 7.521211660015615e-05,
      "loss": 0.3713,
      "step": 840
    },
    {
      "epoch": 1.210071942446043,
      "grad_norm": 0.20857782351282705,
      "learning_rate": 7.519303379441357e-05,
      "loss": 0.3682,
      "step": 841
    },
    {
      "epoch": 1.2115107913669065,
      "grad_norm": 0.1809768437352649,
      "learning_rate": 7.517391546640573e-05,
      "loss": 0.374,
      "step": 842
    },
    {
      "epoch": 1.2129496402877697,
      "grad_norm": 0.21711202508277216,
      "learning_rate": 7.515476163542982e-05,
      "loss": 0.3686,
      "step": 843
    },
    {
      "epoch": 1.2143884892086332,
      "grad_norm": 0.1921623166614579,
      "learning_rate": 7.513557232081887e-05,
      "loss": 0.3694,
      "step": 844
    },
    {
      "epoch": 1.2158273381294964,
      "grad_norm": 0.15316247122528281,
      "learning_rate": 7.511634754194168e-05,
      "loss": 0.3727,
      "step": 845
    },
    {
      "epoch": 1.2172661870503596,
      "grad_norm": 0.18834756267417926,
      "learning_rate": 7.50970873182029e-05,
      "loss": 0.3692,
      "step": 846
    },
    {
      "epoch": 1.218705035971223,
      "grad_norm": 0.23686774578710018,
      "learning_rate": 7.507779166904292e-05,
      "loss": 0.3711,
      "step": 847
    },
    {
      "epoch": 1.2201438848920863,
      "grad_norm": 0.24771915125409488,
      "learning_rate": 7.50584606139379e-05,
      "loss": 0.3763,
      "step": 848
    },
    {
      "epoch": 1.2215827338129497,
      "grad_norm": 0.24131346498923673,
      "learning_rate": 7.503909417239975e-05,
      "loss": 0.3725,
      "step": 849
    },
    {
      "epoch": 1.223021582733813,
      "grad_norm": 0.25443424568582057,
      "learning_rate": 7.501969236397607e-05,
      "loss": 0.3716,
      "step": 850
    },
    {
      "epoch": 1.2244604316546763,
      "grad_norm": 0.28102454004883953,
      "learning_rate": 7.500025520825018e-05,
      "loss": 0.3655,
      "step": 851
    },
    {
      "epoch": 1.2258992805755395,
      "grad_norm": 0.2731890197889495,
      "learning_rate": 7.498078272484108e-05,
      "loss": 0.3723,
      "step": 852
    },
    {
      "epoch": 1.227338129496403,
      "grad_norm": 0.24784435562954357,
      "learning_rate": 7.496127493340341e-05,
      "loss": 0.3811,
      "step": 853
    },
    {
      "epoch": 1.2287769784172662,
      "grad_norm": 0.2603058743485281,
      "learning_rate": 7.494173185362745e-05,
      "loss": 0.3776,
      "step": 854
    },
    {
      "epoch": 1.2302158273381294,
      "grad_norm": 0.24634148328332406,
      "learning_rate": 7.492215350523913e-05,
      "loss": 0.3685,
      "step": 855
    },
    {
      "epoch": 1.2316546762589928,
      "grad_norm": 0.2529437285257151,
      "learning_rate": 7.490253990799991e-05,
      "loss": 0.3769,
      "step": 856
    },
    {
      "epoch": 1.233093525179856,
      "grad_norm": 0.21585577519657578,
      "learning_rate": 7.488289108170692e-05,
      "loss": 0.371,
      "step": 857
    },
    {
      "epoch": 1.2345323741007195,
      "grad_norm": 0.20070593268415463,
      "learning_rate": 7.486320704619276e-05,
      "loss": 0.362,
      "step": 858
    },
    {
      "epoch": 1.2359712230215827,
      "grad_norm": 0.33278265943848906,
      "learning_rate": 7.484348782132565e-05,
      "loss": 0.3719,
      "step": 859
    },
    {
      "epoch": 1.2374100719424461,
      "grad_norm": 0.39253677405101345,
      "learning_rate": 7.482373342700927e-05,
      "loss": 0.3699,
      "step": 860
    },
    {
      "epoch": 1.2388489208633093,
      "grad_norm": 0.348245063526455,
      "learning_rate": 7.48039438831828e-05,
      "loss": 0.3668,
      "step": 861
    },
    {
      "epoch": 1.2402877697841728,
      "grad_norm": 0.27252448611358787,
      "learning_rate": 7.478411920982095e-05,
      "loss": 0.3745,
      "step": 862
    },
    {
      "epoch": 1.241726618705036,
      "grad_norm": 0.20338714632236685,
      "learning_rate": 7.476425942693382e-05,
      "loss": 0.3767,
      "step": 863
    },
    {
      "epoch": 1.2431654676258992,
      "grad_norm": 0.21236877270999938,
      "learning_rate": 7.474436455456701e-05,
      "loss": 0.3735,
      "step": 864
    },
    {
      "epoch": 1.2446043165467626,
      "grad_norm": 0.21143808627777527,
      "learning_rate": 7.472443461280149e-05,
      "loss": 0.3727,
      "step": 865
    },
    {
      "epoch": 1.2460431654676258,
      "grad_norm": 0.22982317906740338,
      "learning_rate": 7.470446962175367e-05,
      "loss": 0.3764,
      "step": 866
    },
    {
      "epoch": 1.2474820143884893,
      "grad_norm": 0.22235343681142253,
      "learning_rate": 7.468446960157527e-05,
      "loss": 0.3669,
      "step": 867
    },
    {
      "epoch": 1.2489208633093525,
      "grad_norm": 0.18991249022815596,
      "learning_rate": 7.466443457245344e-05,
      "loss": 0.3772,
      "step": 868
    },
    {
      "epoch": 1.2503597122302157,
      "grad_norm": 0.20548444109653444,
      "learning_rate": 7.464436455461066e-05,
      "loss": 0.3692,
      "step": 869
    },
    {
      "epoch": 1.2517985611510791,
      "grad_norm": 0.24937818557657385,
      "learning_rate": 7.462425956830466e-05,
      "loss": 0.3733,
      "step": 870
    },
    {
      "epoch": 1.2532374100719426,
      "grad_norm": 0.29764989918438073,
      "learning_rate": 7.460411963382853e-05,
      "loss": 0.3685,
      "step": 871
    },
    {
      "epoch": 1.2546762589928058,
      "grad_norm": 0.31750242257491057,
      "learning_rate": 7.45839447715106e-05,
      "loss": 0.3652,
      "step": 872
    },
    {
      "epoch": 1.256115107913669,
      "grad_norm": 0.3134852958397248,
      "learning_rate": 7.456373500171449e-05,
      "loss": 0.3644,
      "step": 873
    },
    {
      "epoch": 1.2575539568345324,
      "grad_norm": 0.30230759591716544,
      "learning_rate": 7.454349034483903e-05,
      "loss": 0.3686,
      "step": 874
    },
    {
      "epoch": 1.2589928057553956,
      "grad_norm": 0.318563757610303,
      "learning_rate": 7.452321082131824e-05,
      "loss": 0.3776,
      "step": 875
    },
    {
      "epoch": 1.260431654676259,
      "grad_norm": 0.27034438572857183,
      "learning_rate": 7.450289645162138e-05,
      "loss": 0.3671,
      "step": 876
    },
    {
      "epoch": 1.2618705035971223,
      "grad_norm": 0.20339298565882605,
      "learning_rate": 7.448254725625287e-05,
      "loss": 0.3711,
      "step": 877
    },
    {
      "epoch": 1.2633093525179855,
      "grad_norm": 0.1882757355289913,
      "learning_rate": 7.446216325575225e-05,
      "loss": 0.3765,
      "step": 878
    },
    {
      "epoch": 1.264748201438849,
      "grad_norm": 0.2282191365184483,
      "learning_rate": 7.444174447069423e-05,
      "loss": 0.3732,
      "step": 879
    },
    {
      "epoch": 1.2661870503597124,
      "grad_norm": 0.24506792063831453,
      "learning_rate": 7.442129092168859e-05,
      "loss": 0.3764,
      "step": 880
    },
    {
      "epoch": 1.2676258992805756,
      "grad_norm": 0.22089288271493143,
      "learning_rate": 7.440080262938026e-05,
      "loss": 0.3681,
      "step": 881
    },
    {
      "epoch": 1.2690647482014388,
      "grad_norm": 0.2301162172592513,
      "learning_rate": 7.438027961444916e-05,
      "loss": 0.3748,
      "step": 882
    },
    {
      "epoch": 1.2705035971223022,
      "grad_norm": 0.2542452738575605,
      "learning_rate": 7.435972189761033e-05,
      "loss": 0.3688,
      "step": 883
    },
    {
      "epoch": 1.2719424460431654,
      "grad_norm": 0.22470200136134963,
      "learning_rate": 7.43391294996138e-05,
      "loss": 0.3611,
      "step": 884
    },
    {
      "epoch": 1.2733812949640289,
      "grad_norm": 0.1898198683091336,
      "learning_rate": 7.431850244124459e-05,
      "loss": 0.3739,
      "step": 885
    },
    {
      "epoch": 1.274820143884892,
      "grad_norm": 0.1820491835452753,
      "learning_rate": 7.429784074332274e-05,
      "loss": 0.3712,
      "step": 886
    },
    {
      "epoch": 1.2762589928057553,
      "grad_norm": 0.25402600893388405,
      "learning_rate": 7.427714442670324e-05,
      "loss": 0.3609,
      "step": 887
    },
    {
      "epoch": 1.2776978417266187,
      "grad_norm": 0.2932596065482579,
      "learning_rate": 7.425641351227602e-05,
      "loss": 0.3735,
      "step": 888
    },
    {
      "epoch": 1.2791366906474821,
      "grad_norm": 0.254759984310122,
      "learning_rate": 7.423564802096592e-05,
      "loss": 0.3706,
      "step": 889
    },
    {
      "epoch": 1.2805755395683454,
      "grad_norm": 0.15825456679635624,
      "learning_rate": 7.42148479737327e-05,
      "loss": 0.3732,
      "step": 890
    },
    {
      "epoch": 1.2820143884892086,
      "grad_norm": 0.19782704607500964,
      "learning_rate": 7.419401339157099e-05,
      "loss": 0.3666,
      "step": 891
    },
    {
      "epoch": 1.283453237410072,
      "grad_norm": 0.22692372888409243,
      "learning_rate": 7.41731442955103e-05,
      "loss": 0.3727,
      "step": 892
    },
    {
      "epoch": 1.2848920863309352,
      "grad_norm": 0.22599493246768218,
      "learning_rate": 7.415224070661492e-05,
      "loss": 0.3735,
      "step": 893
    },
    {
      "epoch": 1.2863309352517986,
      "grad_norm": 0.20090376197165424,
      "learning_rate": 7.413130264598404e-05,
      "loss": 0.3742,
      "step": 894
    },
    {
      "epoch": 1.2877697841726619,
      "grad_norm": 0.18062614457652348,
      "learning_rate": 7.411033013475156e-05,
      "loss": 0.3661,
      "step": 895
    },
    {
      "epoch": 1.289208633093525,
      "grad_norm": 0.22557471773322108,
      "learning_rate": 7.408932319408619e-05,
      "loss": 0.3706,
      "step": 896
    },
    {
      "epoch": 1.2906474820143885,
      "grad_norm": 0.21324522732601672,
      "learning_rate": 7.406828184519141e-05,
      "loss": 0.373,
      "step": 897
    },
    {
      "epoch": 1.292086330935252,
      "grad_norm": 0.20198963604788683,
      "learning_rate": 7.40472061093054e-05,
      "loss": 0.3727,
      "step": 898
    },
    {
      "epoch": 1.2935251798561151,
      "grad_norm": 0.19423555758666586,
      "learning_rate": 7.402609600770104e-05,
      "loss": 0.369,
      "step": 899
    },
    {
      "epoch": 1.2949640287769784,
      "grad_norm": 0.20589478044466908,
      "learning_rate": 7.400495156168596e-05,
      "loss": 0.3728,
      "step": 900
    },
    {
      "epoch": 1.2964028776978418,
      "grad_norm": 0.25023522722452557,
      "learning_rate": 7.39837727926024e-05,
      "loss": 0.3636,
      "step": 901
    },
    {
      "epoch": 1.297841726618705,
      "grad_norm": 0.2919924081876397,
      "learning_rate": 7.396255972182723e-05,
      "loss": 0.3675,
      "step": 902
    },
    {
      "epoch": 1.2992805755395684,
      "grad_norm": 0.2561789804875008,
      "learning_rate": 7.394131237077199e-05,
      "loss": 0.3722,
      "step": 903
    },
    {
      "epoch": 1.3007194244604317,
      "grad_norm": 0.2476025541732896,
      "learning_rate": 7.39200307608828e-05,
      "loss": 0.3724,
      "step": 904
    },
    {
      "epoch": 1.3021582733812949,
      "grad_norm": 0.2756821793834618,
      "learning_rate": 7.389871491364036e-05,
      "loss": 0.3712,
      "step": 905
    },
    {
      "epoch": 1.3035971223021583,
      "grad_norm": 0.2725376081819577,
      "learning_rate": 7.387736485055993e-05,
      "loss": 0.3691,
      "step": 906
    },
    {
      "epoch": 1.3050359712230215,
      "grad_norm": 0.2632588761126749,
      "learning_rate": 7.385598059319129e-05,
      "loss": 0.3806,
      "step": 907
    },
    {
      "epoch": 1.306474820143885,
      "grad_norm": 0.274824077738147,
      "learning_rate": 7.383456216311875e-05,
      "loss": 0.38,
      "step": 908
    },
    {
      "epoch": 1.3079136690647482,
      "grad_norm": 0.2976218077528491,
      "learning_rate": 7.381310958196112e-05,
      "loss": 0.3745,
      "step": 909
    },
    {
      "epoch": 1.3093525179856116,
      "grad_norm": 0.2740150521185998,
      "learning_rate": 7.379162287137167e-05,
      "loss": 0.3641,
      "step": 910
    },
    {
      "epoch": 1.3107913669064748,
      "grad_norm": 0.18624314318474133,
      "learning_rate": 7.37701020530381e-05,
      "loss": 0.3656,
      "step": 911
    },
    {
      "epoch": 1.3122302158273382,
      "grad_norm": 0.23514671142831362,
      "learning_rate": 7.374854714868259e-05,
      "loss": 0.3619,
      "step": 912
    },
    {
      "epoch": 1.3136690647482014,
      "grad_norm": 0.2640162079181578,
      "learning_rate": 7.372695818006167e-05,
      "loss": 0.3737,
      "step": 913
    },
    {
      "epoch": 1.3151079136690647,
      "grad_norm": 0.21895715772510793,
      "learning_rate": 7.370533516896627e-05,
      "loss": 0.3748,
      "step": 914
    },
    {
      "epoch": 1.316546762589928,
      "grad_norm": 0.19204811742035638,
      "learning_rate": 7.368367813722169e-05,
      "loss": 0.3762,
      "step": 915
    },
    {
      "epoch": 1.3179856115107913,
      "grad_norm": 0.23551727025175465,
      "learning_rate": 7.366198710668755e-05,
      "loss": 0.371,
      "step": 916
    },
    {
      "epoch": 1.3194244604316547,
      "grad_norm": 0.21097683299141343,
      "learning_rate": 7.364026209925783e-05,
      "loss": 0.3749,
      "step": 917
    },
    {
      "epoch": 1.320863309352518,
      "grad_norm": 0.18362234695221757,
      "learning_rate": 7.361850313686076e-05,
      "loss": 0.3728,
      "step": 918
    },
    {
      "epoch": 1.3223021582733812,
      "grad_norm": 0.21469391732409804,
      "learning_rate": 7.359671024145886e-05,
      "loss": 0.3656,
      "step": 919
    },
    {
      "epoch": 1.3237410071942446,
      "grad_norm": 0.21885651922020025,
      "learning_rate": 7.35748834350489e-05,
      "loss": 0.3671,
      "step": 920
    },
    {
      "epoch": 1.325179856115108,
      "grad_norm": 0.1970601176982681,
      "learning_rate": 7.355302273966186e-05,
      "loss": 0.3716,
      "step": 921
    },
    {
      "epoch": 1.3266187050359712,
      "grad_norm": 0.2417575513009536,
      "learning_rate": 7.353112817736295e-05,
      "loss": 0.3659,
      "step": 922
    },
    {
      "epoch": 1.3280575539568344,
      "grad_norm": 0.22902836288856157,
      "learning_rate": 7.350919977025157e-05,
      "loss": 0.3689,
      "step": 923
    },
    {
      "epoch": 1.3294964028776979,
      "grad_norm": 0.18807707508033616,
      "learning_rate": 7.348723754046127e-05,
      "loss": 0.3607,
      "step": 924
    },
    {
      "epoch": 1.330935251798561,
      "grad_norm": 0.19350319400188104,
      "learning_rate": 7.34652415101597e-05,
      "loss": 0.3663,
      "step": 925
    },
    {
      "epoch": 1.3323741007194245,
      "grad_norm": 0.22217192896787372,
      "learning_rate": 7.344321170154871e-05,
      "loss": 0.3689,
      "step": 926
    },
    {
      "epoch": 1.3338129496402877,
      "grad_norm": 0.27901108408156117,
      "learning_rate": 7.342114813686419e-05,
      "loss": 0.3723,
      "step": 927
    },
    {
      "epoch": 1.335251798561151,
      "grad_norm": 0.2869745209501443,
      "learning_rate": 7.339905083837608e-05,
      "loss": 0.3831,
      "step": 928
    },
    {
      "epoch": 1.3366906474820144,
      "grad_norm": 0.22988927418717225,
      "learning_rate": 7.337691982838841e-05,
      "loss": 0.3633,
      "step": 929
    },
    {
      "epoch": 1.3381294964028778,
      "grad_norm": 0.21230789850039003,
      "learning_rate": 7.335475512923924e-05,
      "loss": 0.3642,
      "step": 930
    },
    {
      "epoch": 1.339568345323741,
      "grad_norm": 0.23346971236746394,
      "learning_rate": 7.33325567633006e-05,
      "loss": 0.3768,
      "step": 931
    },
    {
      "epoch": 1.3410071942446042,
      "grad_norm": 0.17917347614029558,
      "learning_rate": 7.331032475297855e-05,
      "loss": 0.3684,
      "step": 932
    },
    {
      "epoch": 1.3424460431654677,
      "grad_norm": 0.15564979709111454,
      "learning_rate": 7.328805912071307e-05,
      "loss": 0.3648,
      "step": 933
    },
    {
      "epoch": 1.3438848920863309,
      "grad_norm": 0.18785506120243278,
      "learning_rate": 7.326575988897807e-05,
      "loss": 0.3752,
      "step": 934
    },
    {
      "epoch": 1.3453237410071943,
      "grad_norm": 0.1965931833564018,
      "learning_rate": 7.324342708028141e-05,
      "loss": 0.3673,
      "step": 935
    },
    {
      "epoch": 1.3467625899280575,
      "grad_norm": 0.20849109538010188,
      "learning_rate": 7.322106071716483e-05,
      "loss": 0.3618,
      "step": 936
    },
    {
      "epoch": 1.3482014388489207,
      "grad_norm": 0.19947202708650083,
      "learning_rate": 7.319866082220388e-05,
      "loss": 0.3696,
      "step": 937
    },
    {
      "epoch": 1.3496402877697842,
      "grad_norm": 0.16626015484382997,
      "learning_rate": 7.317622741800808e-05,
      "loss": 0.3869,
      "step": 938
    },
    {
      "epoch": 1.3510791366906476,
      "grad_norm": 0.18899282070305004,
      "learning_rate": 7.315376052722065e-05,
      "loss": 0.369,
      "step": 939
    },
    {
      "epoch": 1.3525179856115108,
      "grad_norm": 0.1743281306881679,
      "learning_rate": 7.313126017251868e-05,
      "loss": 0.3589,
      "step": 940
    },
    {
      "epoch": 1.353956834532374,
      "grad_norm": 0.1450761098854889,
      "learning_rate": 7.3108726376613e-05,
      "loss": 0.3722,
      "step": 941
    },
    {
      "epoch": 1.3553956834532375,
      "grad_norm": 0.15543250685925575,
      "learning_rate": 7.308615916224823e-05,
      "loss": 0.3634,
      "step": 942
    },
    {
      "epoch": 1.3568345323741007,
      "grad_norm": 0.2076568865911509,
      "learning_rate": 7.306355855220267e-05,
      "loss": 0.3749,
      "step": 943
    },
    {
      "epoch": 1.358273381294964,
      "grad_norm": 0.2164167100448256,
      "learning_rate": 7.30409245692884e-05,
      "loss": 0.3699,
      "step": 944
    },
    {
      "epoch": 1.3597122302158273,
      "grad_norm": 0.20103550217189015,
      "learning_rate": 7.301825723635111e-05,
      "loss": 0.3653,
      "step": 945
    },
    {
      "epoch": 1.3611510791366905,
      "grad_norm": 0.22906493617217463,
      "learning_rate": 7.299555657627021e-05,
      "loss": 0.3659,
      "step": 946
    },
    {
      "epoch": 1.362589928057554,
      "grad_norm": 0.24804845625514324,
      "learning_rate": 7.29728226119587e-05,
      "loss": 0.3843,
      "step": 947
    },
    {
      "epoch": 1.3640287769784174,
      "grad_norm": 0.2574010821529766,
      "learning_rate": 7.295005536636325e-05,
      "loss": 0.3694,
      "step": 948
    },
    {
      "epoch": 1.3654676258992806,
      "grad_norm": 0.27319217453544853,
      "learning_rate": 7.292725486246407e-05,
      "loss": 0.3691,
      "step": 949
    },
    {
      "epoch": 1.3669064748201438,
      "grad_norm": 0.3137763417062616,
      "learning_rate": 7.290442112327498e-05,
      "loss": 0.3728,
      "step": 950
    },
    {
      "epoch": 1.3683453237410073,
      "grad_norm": 0.31367627699699374,
      "learning_rate": 7.288155417184331e-05,
      "loss": 0.3755,
      "step": 951
    },
    {
      "epoch": 1.3697841726618705,
      "grad_norm": 0.31054301331184375,
      "learning_rate": 7.285865403124995e-05,
      "loss": 0.3701,
      "step": 952
    },
    {
      "epoch": 1.371223021582734,
      "grad_norm": 0.264865486382744,
      "learning_rate": 7.283572072460927e-05,
      "loss": 0.3637,
      "step": 953
    },
    {
      "epoch": 1.3726618705035971,
      "grad_norm": 0.24832730740175757,
      "learning_rate": 7.28127542750691e-05,
      "loss": 0.3666,
      "step": 954
    },
    {
      "epoch": 1.3741007194244603,
      "grad_norm": 0.24633003699743544,
      "learning_rate": 7.278975470581076e-05,
      "loss": 0.3744,
      "step": 955
    },
    {
      "epoch": 1.3755395683453238,
      "grad_norm": 0.23944991878988073,
      "learning_rate": 7.276672204004898e-05,
      "loss": 0.3723,
      "step": 956
    },
    {
      "epoch": 1.376978417266187,
      "grad_norm": 0.23658561224616861,
      "learning_rate": 7.274365630103189e-05,
      "loss": 0.3683,
      "step": 957
    },
    {
      "epoch": 1.3784172661870504,
      "grad_norm": 0.2592946740203759,
      "learning_rate": 7.2720557512041e-05,
      "loss": 0.3748,
      "step": 958
    },
    {
      "epoch": 1.3798561151079136,
      "grad_norm": 0.23670621727444183,
      "learning_rate": 7.269742569639121e-05,
      "loss": 0.3744,
      "step": 959
    },
    {
      "epoch": 1.381294964028777,
      "grad_norm": 0.2292635420385005,
      "learning_rate": 7.267426087743073e-05,
      "loss": 0.3768,
      "step": 960
    },
    {
      "epoch": 1.3827338129496403,
      "grad_norm": 0.2268629950072278,
      "learning_rate": 7.265106307854107e-05,
      "loss": 0.3738,
      "step": 961
    },
    {
      "epoch": 1.3841726618705037,
      "grad_norm": 0.22747961927056887,
      "learning_rate": 7.262783232313706e-05,
      "loss": 0.3639,
      "step": 962
    },
    {
      "epoch": 1.385611510791367,
      "grad_norm": 0.22375027684102045,
      "learning_rate": 7.260456863466676e-05,
      "loss": 0.3693,
      "step": 963
    },
    {
      "epoch": 1.3870503597122301,
      "grad_norm": 0.26756377015293353,
      "learning_rate": 7.258127203661153e-05,
      "loss": 0.3836,
      "step": 964
    },
    {
      "epoch": 1.3884892086330936,
      "grad_norm": 0.2641592742373027,
      "learning_rate": 7.255794255248587e-05,
      "loss": 0.3687,
      "step": 965
    },
    {
      "epoch": 1.3899280575539568,
      "grad_norm": 0.1741039452690814,
      "learning_rate": 7.253458020583752e-05,
      "loss": 0.365,
      "step": 966
    },
    {
      "epoch": 1.3913669064748202,
      "grad_norm": 0.24619697055268608,
      "learning_rate": 7.25111850202474e-05,
      "loss": 0.3755,
      "step": 967
    },
    {
      "epoch": 1.3928057553956834,
      "grad_norm": 0.29978393559314154,
      "learning_rate": 7.248775701932953e-05,
      "loss": 0.3806,
      "step": 968
    },
    {
      "epoch": 1.3942446043165468,
      "grad_norm": 0.2198291463813615,
      "learning_rate": 7.246429622673111e-05,
      "loss": 0.3794,
      "step": 969
    },
    {
      "epoch": 1.39568345323741,
      "grad_norm": 0.149923625810162,
      "learning_rate": 7.244080266613238e-05,
      "loss": 0.3758,
      "step": 970
    },
    {
      "epoch": 1.3971223021582735,
      "grad_norm": 0.1727076360812617,
      "learning_rate": 7.241727636124671e-05,
      "loss": 0.3718,
      "step": 971
    },
    {
      "epoch": 1.3985611510791367,
      "grad_norm": 0.2094557826048381,
      "learning_rate": 7.239371733582047e-05,
      "loss": 0.3776,
      "step": 972
    },
    {
      "epoch": 1.4,
      "grad_norm": 0.15811082164113044,
      "learning_rate": 7.23701256136331e-05,
      "loss": 0.3623,
      "step": 973
    },
    {
      "epoch": 1.4014388489208633,
      "grad_norm": 0.13596205161356006,
      "learning_rate": 7.2346501218497e-05,
      "loss": 0.3679,
      "step": 974
    },
    {
      "epoch": 1.4028776978417266,
      "grad_norm": 0.1547108001168075,
      "learning_rate": 7.23228441742576e-05,
      "loss": 0.3694,
      "step": 975
    },
    {
      "epoch": 1.40431654676259,
      "grad_norm": 0.1515226364651108,
      "learning_rate": 7.229915450479324e-05,
      "loss": 0.366,
      "step": 976
    },
    {
      "epoch": 1.4057553956834532,
      "grad_norm": 0.13965019541850962,
      "learning_rate": 7.227543223401522e-05,
      "loss": 0.3707,
      "step": 977
    },
    {
      "epoch": 1.4071942446043164,
      "grad_norm": 0.18867540462886592,
      "learning_rate": 7.225167738586772e-05,
      "loss": 0.3707,
      "step": 978
    },
    {
      "epoch": 1.4086330935251798,
      "grad_norm": 0.2690397878961195,
      "learning_rate": 7.22278899843278e-05,
      "loss": 0.3727,
      "step": 979
    },
    {
      "epoch": 1.4100719424460433,
      "grad_norm": 0.3127286924758779,
      "learning_rate": 7.220407005340542e-05,
      "loss": 0.3702,
      "step": 980
    },
    {
      "epoch": 1.4115107913669065,
      "grad_norm": 0.35116182021408393,
      "learning_rate": 7.218021761714336e-05,
      "loss": 0.3752,
      "step": 981
    },
    {
      "epoch": 1.4129496402877697,
      "grad_norm": 0.3130530174839103,
      "learning_rate": 7.215633269961714e-05,
      "loss": 0.3665,
      "step": 982
    },
    {
      "epoch": 1.4143884892086331,
      "grad_norm": 0.254603450015429,
      "learning_rate": 7.213241532493516e-05,
      "loss": 0.3653,
      "step": 983
    },
    {
      "epoch": 1.4158273381294963,
      "grad_norm": 0.26860387604138586,
      "learning_rate": 7.210846551723855e-05,
      "loss": 0.3679,
      "step": 984
    },
    {
      "epoch": 1.4172661870503598,
      "grad_norm": 0.29070538996667433,
      "learning_rate": 7.208448330070116e-05,
      "loss": 0.3766,
      "step": 985
    },
    {
      "epoch": 1.418705035971223,
      "grad_norm": 0.31265244907033013,
      "learning_rate": 7.206046869952954e-05,
      "loss": 0.3824,
      "step": 986
    },
    {
      "epoch": 1.4201438848920862,
      "grad_norm": 0.2763893862741098,
      "learning_rate": 7.203642173796298e-05,
      "loss": 0.3673,
      "step": 987
    },
    {
      "epoch": 1.4215827338129496,
      "grad_norm": 0.22370884435098334,
      "learning_rate": 7.201234244027338e-05,
      "loss": 0.3709,
      "step": 988
    },
    {
      "epoch": 1.423021582733813,
      "grad_norm": 0.19105592811118327,
      "learning_rate": 7.19882308307653e-05,
      "loss": 0.3636,
      "step": 989
    },
    {
      "epoch": 1.4244604316546763,
      "grad_norm": 0.20107475993801432,
      "learning_rate": 7.196408693377594e-05,
      "loss": 0.3684,
      "step": 990
    },
    {
      "epoch": 1.4258992805755395,
      "grad_norm": 0.26309111130685686,
      "learning_rate": 7.193991077367501e-05,
      "loss": 0.3778,
      "step": 991
    },
    {
      "epoch": 1.427338129496403,
      "grad_norm": 0.28549866613414865,
      "learning_rate": 7.19157023748649e-05,
      "loss": 0.372,
      "step": 992
    },
    {
      "epoch": 1.4287769784172661,
      "grad_norm": 0.287943613682877,
      "learning_rate": 7.189146176178044e-05,
      "loss": 0.376,
      "step": 993
    },
    {
      "epoch": 1.4302158273381296,
      "grad_norm": 0.24113088730328244,
      "learning_rate": 7.186718895888904e-05,
      "loss": 0.3663,
      "step": 994
    },
    {
      "epoch": 1.4316546762589928,
      "grad_norm": 0.2138198760304494,
      "learning_rate": 7.184288399069054e-05,
      "loss": 0.3671,
      "step": 995
    },
    {
      "epoch": 1.433093525179856,
      "grad_norm": 0.22492272003674318,
      "learning_rate": 7.181854688171732e-05,
      "loss": 0.3708,
      "step": 996
    },
    {
      "epoch": 1.4345323741007194,
      "grad_norm": 0.20104573143639234,
      "learning_rate": 7.179417765653413e-05,
      "loss": 0.3695,
      "step": 997
    },
    {
      "epoch": 1.4359712230215829,
      "grad_norm": 0.16938615838863322,
      "learning_rate": 7.17697763397382e-05,
      "loss": 0.3662,
      "step": 998
    },
    {
      "epoch": 1.437410071942446,
      "grad_norm": 0.14805036489703202,
      "learning_rate": 7.174534295595911e-05,
      "loss": 0.3648,
      "step": 999
    },
    {
      "epoch": 1.4388489208633093,
      "grad_norm": 0.17266866558604405,
      "learning_rate": 7.17208775298588e-05,
      "loss": 0.3721,
      "step": 1000
    },
    {
      "epoch": 1.4402877697841727,
      "grad_norm": 0.19524748761472635,
      "learning_rate": 7.169638008613158e-05,
      "loss": 0.3731,
      "step": 1001
    },
    {
      "epoch": 1.441726618705036,
      "grad_norm": 0.220526684507708,
      "learning_rate": 7.16718506495041e-05,
      "loss": 0.372,
      "step": 1002
    },
    {
      "epoch": 1.4431654676258994,
      "grad_norm": 0.21718586501066883,
      "learning_rate": 7.164728924473522e-05,
      "loss": 0.3666,
      "step": 1003
    },
    {
      "epoch": 1.4446043165467626,
      "grad_norm": 0.199258907361367,
      "learning_rate": 7.162269589661614e-05,
      "loss": 0.3735,
      "step": 1004
    },
    {
      "epoch": 1.4460431654676258,
      "grad_norm": 0.2314267753915402,
      "learning_rate": 7.15980706299703e-05,
      "loss": 0.3569,
      "step": 1005
    },
    {
      "epoch": 1.4474820143884892,
      "grad_norm": 0.26551029161174233,
      "learning_rate": 7.15734134696533e-05,
      "loss": 0.3759,
      "step": 1006
    },
    {
      "epoch": 1.4489208633093524,
      "grad_norm": 0.22711136350517233,
      "learning_rate": 7.1548724440553e-05,
      "loss": 0.3576,
      "step": 1007
    },
    {
      "epoch": 1.4503597122302159,
      "grad_norm": 0.17628765409818595,
      "learning_rate": 7.152400356758937e-05,
      "loss": 0.3721,
      "step": 1008
    },
    {
      "epoch": 1.451798561151079,
      "grad_norm": 0.20206966855772898,
      "learning_rate": 7.149925087571456e-05,
      "loss": 0.3744,
      "step": 1009
    },
    {
      "epoch": 1.4532374100719425,
      "grad_norm": 0.1865332111696231,
      "learning_rate": 7.147446638991283e-05,
      "loss": 0.362,
      "step": 1010
    },
    {
      "epoch": 1.4546762589928057,
      "grad_norm": 0.16890414484534108,
      "learning_rate": 7.14496501352005e-05,
      "loss": 0.3672,
      "step": 1011
    },
    {
      "epoch": 1.4561151079136692,
      "grad_norm": 0.17979472458157772,
      "learning_rate": 7.1424802136626e-05,
      "loss": 0.3719,
      "step": 1012
    },
    {
      "epoch": 1.4575539568345324,
      "grad_norm": 0.15033256457936278,
      "learning_rate": 7.139992241926978e-05,
      "loss": 0.3607,
      "step": 1013
    },
    {
      "epoch": 1.4589928057553956,
      "grad_norm": 0.1863289313868577,
      "learning_rate": 7.137501100824432e-05,
      "loss": 0.3698,
      "step": 1014
    },
    {
      "epoch": 1.460431654676259,
      "grad_norm": 0.2298324813762039,
      "learning_rate": 7.135006792869405e-05,
      "loss": 0.3698,
      "step": 1015
    },
    {
      "epoch": 1.4618705035971222,
      "grad_norm": 0.17501429166945917,
      "learning_rate": 7.132509320579542e-05,
      "loss": 0.3691,
      "step": 1016
    },
    {
      "epoch": 1.4633093525179857,
      "grad_norm": 0.1568780008946555,
      "learning_rate": 7.130008686475677e-05,
      "loss": 0.3771,
      "step": 1017
    },
    {
      "epoch": 1.4647482014388489,
      "grad_norm": 0.18883894264720721,
      "learning_rate": 7.127504893081839e-05,
      "loss": 0.361,
      "step": 1018
    },
    {
      "epoch": 1.4661870503597123,
      "grad_norm": 0.21335909285528223,
      "learning_rate": 7.124997942925244e-05,
      "loss": 0.3716,
      "step": 1019
    },
    {
      "epoch": 1.4676258992805755,
      "grad_norm": 0.2465896954841192,
      "learning_rate": 7.122487838536295e-05,
      "loss": 0.3624,
      "step": 1020
    },
    {
      "epoch": 1.469064748201439,
      "grad_norm": 0.2432164708019936,
      "learning_rate": 7.119974582448577e-05,
      "loss": 0.366,
      "step": 1021
    },
    {
      "epoch": 1.4705035971223022,
      "grad_norm": 0.2251933489831929,
      "learning_rate": 7.11745817719886e-05,
      "loss": 0.3675,
      "step": 1022
    },
    {
      "epoch": 1.4719424460431654,
      "grad_norm": 0.21640744171179588,
      "learning_rate": 7.114938625327088e-05,
      "loss": 0.3644,
      "step": 1023
    },
    {
      "epoch": 1.4733812949640288,
      "grad_norm": 0.20407010222209512,
      "learning_rate": 7.112415929376385e-05,
      "loss": 0.3785,
      "step": 1024
    },
    {
      "epoch": 1.474820143884892,
      "grad_norm": 0.22433584927431985,
      "learning_rate": 7.109890091893047e-05,
      "loss": 0.3747,
      "step": 1025
    },
    {
      "epoch": 1.4762589928057555,
      "grad_norm": 0.20891736414872672,
      "learning_rate": 7.107361115426537e-05,
      "loss": 0.3636,
      "step": 1026
    },
    {
      "epoch": 1.4776978417266187,
      "grad_norm": 0.1846405972029036,
      "learning_rate": 7.104829002529496e-05,
      "loss": 0.3652,
      "step": 1027
    },
    {
      "epoch": 1.4791366906474819,
      "grad_norm": 0.18316835473870297,
      "learning_rate": 7.102293755757721e-05,
      "loss": 0.3742,
      "step": 1028
    },
    {
      "epoch": 1.4805755395683453,
      "grad_norm": 0.18940001949154145,
      "learning_rate": 7.099755377670177e-05,
      "loss": 0.3787,
      "step": 1029
    },
    {
      "epoch": 1.4820143884892087,
      "grad_norm": 0.202197442529192,
      "learning_rate": 7.097213870828989e-05,
      "loss": 0.3682,
      "step": 1030
    },
    {
      "epoch": 1.483453237410072,
      "grad_norm": 0.23653540527897343,
      "learning_rate": 7.094669237799437e-05,
      "loss": 0.365,
      "step": 1031
    },
    {
      "epoch": 1.4848920863309352,
      "grad_norm": 0.25273303274457065,
      "learning_rate": 7.092121481149964e-05,
      "loss": 0.3816,
      "step": 1032
    },
    {
      "epoch": 1.4863309352517986,
      "grad_norm": 0.2541430913646154,
      "learning_rate": 7.089570603452157e-05,
      "loss": 0.3571,
      "step": 1033
    },
    {
      "epoch": 1.4877697841726618,
      "grad_norm": 0.23731631571248973,
      "learning_rate": 7.087016607280758e-05,
      "loss": 0.3654,
      "step": 1034
    },
    {
      "epoch": 1.4892086330935252,
      "grad_norm": 0.2102000795141157,
      "learning_rate": 7.084459495213658e-05,
      "loss": 0.369,
      "step": 1035
    },
    {
      "epoch": 1.4906474820143885,
      "grad_norm": 0.20018786941562838,
      "learning_rate": 7.081899269831888e-05,
      "loss": 0.3726,
      "step": 1036
    },
    {
      "epoch": 1.4920863309352517,
      "grad_norm": 0.18174622680531183,
      "learning_rate": 7.079335933719625e-05,
      "loss": 0.3703,
      "step": 1037
    },
    {
      "epoch": 1.493525179856115,
      "grad_norm": 0.1739627444880869,
      "learning_rate": 7.076769489464188e-05,
      "loss": 0.3689,
      "step": 1038
    },
    {
      "epoch": 1.4949640287769785,
      "grad_norm": 0.19045338757680166,
      "learning_rate": 7.074199939656027e-05,
      "loss": 0.3684,
      "step": 1039
    },
    {
      "epoch": 1.4964028776978417,
      "grad_norm": 0.17323825783489552,
      "learning_rate": 7.071627286888731e-05,
      "loss": 0.3592,
      "step": 1040
    },
    {
      "epoch": 1.497841726618705,
      "grad_norm": 0.17301111473128455,
      "learning_rate": 7.06905153375902e-05,
      "loss": 0.3646,
      "step": 1041
    },
    {
      "epoch": 1.4992805755395684,
      "grad_norm": 0.212194491495326,
      "learning_rate": 7.066472682866744e-05,
      "loss": 0.3654,
      "step": 1042
    },
    {
      "epoch": 1.5007194244604316,
      "grad_norm": 0.24175569970243074,
      "learning_rate": 7.063890736814878e-05,
      "loss": 0.369,
      "step": 1043
    },
    {
      "epoch": 1.502158273381295,
      "grad_norm": 0.19559789804626979,
      "learning_rate": 7.061305698209524e-05,
      "loss": 0.3592,
      "step": 1044
    },
    {
      "epoch": 1.5035971223021583,
      "grad_norm": 0.1674547024455131,
      "learning_rate": 7.058717569659901e-05,
      "loss": 0.3686,
      "step": 1045
    },
    {
      "epoch": 1.5050359712230215,
      "grad_norm": 0.21656195173077553,
      "learning_rate": 7.05612635377835e-05,
      "loss": 0.3665,
      "step": 1046
    },
    {
      "epoch": 1.506474820143885,
      "grad_norm": 0.25183894141407687,
      "learning_rate": 7.053532053180332e-05,
      "loss": 0.373,
      "step": 1047
    },
    {
      "epoch": 1.5079136690647483,
      "grad_norm": 0.22848196273677326,
      "learning_rate": 7.050934670484413e-05,
      "loss": 0.3649,
      "step": 1048
    },
    {
      "epoch": 1.5093525179856115,
      "grad_norm": 0.20612656601373666,
      "learning_rate": 7.048334208312273e-05,
      "loss": 0.3762,
      "step": 1049
    },
    {
      "epoch": 1.5107913669064748,
      "grad_norm": 0.2012284069329427,
      "learning_rate": 7.045730669288706e-05,
      "loss": 0.3739,
      "step": 1050
    },
    {
      "epoch": 1.512230215827338,
      "grad_norm": 0.19784816330810212,
      "learning_rate": 7.043124056041606e-05,
      "loss": 0.3764,
      "step": 1051
    },
    {
      "epoch": 1.5136690647482014,
      "grad_norm": 0.20070931859918947,
      "learning_rate": 7.040514371201969e-05,
      "loss": 0.3686,
      "step": 1052
    },
    {
      "epoch": 1.5151079136690648,
      "grad_norm": 0.21327088238852762,
      "learning_rate": 7.037901617403894e-05,
      "loss": 0.3719,
      "step": 1053
    },
    {
      "epoch": 1.516546762589928,
      "grad_norm": 0.18889305816471605,
      "learning_rate": 7.035285797284578e-05,
      "loss": 0.3629,
      "step": 1054
    },
    {
      "epoch": 1.5179856115107913,
      "grad_norm": 0.1717405383978451,
      "learning_rate": 7.032666913484313e-05,
      "loss": 0.366,
      "step": 1055
    },
    {
      "epoch": 1.5194244604316547,
      "grad_norm": 0.17505137907376786,
      "learning_rate": 7.030044968646481e-05,
      "loss": 0.3752,
      "step": 1056
    },
    {
      "epoch": 1.5208633093525181,
      "grad_norm": 0.17506375531474533,
      "learning_rate": 7.027419965417556e-05,
      "loss": 0.362,
      "step": 1057
    },
    {
      "epoch": 1.5223021582733813,
      "grad_norm": 0.19851104466139188,
      "learning_rate": 7.024791906447098e-05,
      "loss": 0.3642,
      "step": 1058
    },
    {
      "epoch": 1.5237410071942445,
      "grad_norm": 0.20949914509907405,
      "learning_rate": 7.022160794387751e-05,
      "loss": 0.3672,
      "step": 1059
    },
    {
      "epoch": 1.5251798561151078,
      "grad_norm": 0.22099560063799745,
      "learning_rate": 7.019526631895242e-05,
      "loss": 0.3655,
      "step": 1060
    },
    {
      "epoch": 1.5266187050359712,
      "grad_norm": 0.19446322850465173,
      "learning_rate": 7.016889421628374e-05,
      "loss": 0.3729,
      "step": 1061
    },
    {
      "epoch": 1.5280575539568346,
      "grad_norm": 0.19321062750086784,
      "learning_rate": 7.014249166249032e-05,
      "loss": 0.3769,
      "step": 1062
    },
    {
      "epoch": 1.5294964028776978,
      "grad_norm": 0.1830637878435879,
      "learning_rate": 7.011605868422168e-05,
      "loss": 0.3733,
      "step": 1063
    },
    {
      "epoch": 1.530935251798561,
      "grad_norm": 0.15814437083114577,
      "learning_rate": 7.00895953081581e-05,
      "loss": 0.363,
      "step": 1064
    },
    {
      "epoch": 1.5323741007194245,
      "grad_norm": 0.13505172891850736,
      "learning_rate": 7.00631015610105e-05,
      "loss": 0.3643,
      "step": 1065
    },
    {
      "epoch": 1.533812949640288,
      "grad_norm": 0.21924398456152785,
      "learning_rate": 7.00365774695205e-05,
      "loss": 0.3654,
      "step": 1066
    },
    {
      "epoch": 1.5352517985611511,
      "grad_norm": 0.26690718573582306,
      "learning_rate": 7.001002306046031e-05,
      "loss": 0.3719,
      "step": 1067
    },
    {
      "epoch": 1.5366906474820143,
      "grad_norm": 0.2733501865057393,
      "learning_rate": 6.998343836063276e-05,
      "loss": 0.364,
      "step": 1068
    },
    {
      "epoch": 1.5381294964028775,
      "grad_norm": 0.3003023341903461,
      "learning_rate": 6.995682339687125e-05,
      "loss": 0.3698,
      "step": 1069
    },
    {
      "epoch": 1.539568345323741,
      "grad_norm": 0.3855597047106891,
      "learning_rate": 6.993017819603973e-05,
      "loss": 0.3792,
      "step": 1070
    },
    {
      "epoch": 1.5410071942446044,
      "grad_norm": 0.3930859327750899,
      "learning_rate": 6.990350278503267e-05,
      "loss": 0.3696,
      "step": 1071
    },
    {
      "epoch": 1.5424460431654676,
      "grad_norm": 0.29835743193969017,
      "learning_rate": 6.9876797190775e-05,
      "loss": 0.3613,
      "step": 1072
    },
    {
      "epoch": 1.5438848920863308,
      "grad_norm": 0.17425106011105304,
      "learning_rate": 6.985006144022219e-05,
      "loss": 0.3673,
      "step": 1073
    },
    {
      "epoch": 1.5453237410071943,
      "grad_norm": 0.1512964787752433,
      "learning_rate": 6.982329556036007e-05,
      "loss": 0.3655,
      "step": 1074
    },
    {
      "epoch": 1.5467625899280577,
      "grad_norm": 0.21301259298164918,
      "learning_rate": 6.979649957820494e-05,
      "loss": 0.3751,
      "step": 1075
    },
    {
      "epoch": 1.548201438848921,
      "grad_norm": 0.27811188745530113,
      "learning_rate": 6.976967352080345e-05,
      "loss": 0.3628,
      "step": 1076
    },
    {
      "epoch": 1.5496402877697841,
      "grad_norm": 0.2682485717058034,
      "learning_rate": 6.974281741523259e-05,
      "loss": 0.3649,
      "step": 1077
    },
    {
      "epoch": 1.5510791366906473,
      "grad_norm": 0.2132643819684934,
      "learning_rate": 6.971593128859974e-05,
      "loss": 0.375,
      "step": 1078
    },
    {
      "epoch": 1.5525179856115108,
      "grad_norm": 0.14719184623094933,
      "learning_rate": 6.968901516804254e-05,
      "loss": 0.3648,
      "step": 1079
    },
    {
      "epoch": 1.5539568345323742,
      "grad_norm": 0.15999419220097852,
      "learning_rate": 6.966206908072891e-05,
      "loss": 0.3628,
      "step": 1080
    },
    {
      "epoch": 1.5553956834532374,
      "grad_norm": 0.16389494346907643,
      "learning_rate": 6.963509305385701e-05,
      "loss": 0.37,
      "step": 1081
    },
    {
      "epoch": 1.5568345323741006,
      "grad_norm": 0.16612092719907467,
      "learning_rate": 6.960808711465524e-05,
      "loss": 0.363,
      "step": 1082
    },
    {
      "epoch": 1.558273381294964,
      "grad_norm": 0.17113416113904897,
      "learning_rate": 6.958105129038216e-05,
      "loss": 0.3642,
      "step": 1083
    },
    {
      "epoch": 1.5597122302158275,
      "grad_norm": 0.19938671064970884,
      "learning_rate": 6.955398560832654e-05,
      "loss": 0.3707,
      "step": 1084
    },
    {
      "epoch": 1.5611510791366907,
      "grad_norm": 0.19592331868330812,
      "learning_rate": 6.952689009580724e-05,
      "loss": 0.3715,
      "step": 1085
    },
    {
      "epoch": 1.562589928057554,
      "grad_norm": 0.1593070837808773,
      "learning_rate": 6.949976478017327e-05,
      "loss": 0.3651,
      "step": 1086
    },
    {
      "epoch": 1.5640287769784171,
      "grad_norm": 0.1729324239186276,
      "learning_rate": 6.947260968880369e-05,
      "loss": 0.3686,
      "step": 1087
    },
    {
      "epoch": 1.5654676258992806,
      "grad_norm": 0.20192957746825058,
      "learning_rate": 6.944542484910763e-05,
      "loss": 0.3705,
      "step": 1088
    },
    {
      "epoch": 1.566906474820144,
      "grad_norm": 0.20153030072972658,
      "learning_rate": 6.941821028852424e-05,
      "loss": 0.3656,
      "step": 1089
    },
    {
      "epoch": 1.5683453237410072,
      "grad_norm": 0.18388173358459498,
      "learning_rate": 6.939096603452269e-05,
      "loss": 0.3786,
      "step": 1090
    },
    {
      "epoch": 1.5697841726618704,
      "grad_norm": 0.20279535476864027,
      "learning_rate": 6.93636921146021e-05,
      "loss": 0.3695,
      "step": 1091
    },
    {
      "epoch": 1.5712230215827339,
      "grad_norm": 0.2564529735884279,
      "learning_rate": 6.933638855629153e-05,
      "loss": 0.3706,
      "step": 1092
    },
    {
      "epoch": 1.572661870503597,
      "grad_norm": 0.2592608372047231,
      "learning_rate": 6.930905538714995e-05,
      "loss": 0.3737,
      "step": 1093
    },
    {
      "epoch": 1.5741007194244605,
      "grad_norm": 0.26381299512115763,
      "learning_rate": 6.928169263476628e-05,
      "loss": 0.3714,
      "step": 1094
    },
    {
      "epoch": 1.5755395683453237,
      "grad_norm": 0.25979589437396916,
      "learning_rate": 6.92543003267592e-05,
      "loss": 0.3725,
      "step": 1095
    },
    {
      "epoch": 1.576978417266187,
      "grad_norm": 0.22649384845684808,
      "learning_rate": 6.922687849077729e-05,
      "loss": 0.3757,
      "step": 1096
    },
    {
      "epoch": 1.5784172661870504,
      "grad_norm": 0.23969878317497015,
      "learning_rate": 6.919942715449893e-05,
      "loss": 0.3734,
      "step": 1097
    },
    {
      "epoch": 1.5798561151079138,
      "grad_norm": 0.24479553315852776,
      "learning_rate": 6.917194634563225e-05,
      "loss": 0.3711,
      "step": 1098
    },
    {
      "epoch": 1.581294964028777,
      "grad_norm": 0.23940246332303425,
      "learning_rate": 6.914443609191514e-05,
      "loss": 0.37,
      "step": 1099
    },
    {
      "epoch": 1.5827338129496402,
      "grad_norm": 0.2605546110742917,
      "learning_rate": 6.911689642111523e-05,
      "loss": 0.3653,
      "step": 1100
    },
    {
      "epoch": 1.5841726618705037,
      "grad_norm": 0.2302503038672148,
      "learning_rate": 6.90893273610298e-05,
      "loss": 0.3743,
      "step": 1101
    },
    {
      "epoch": 1.5856115107913669,
      "grad_norm": 0.18357282616918016,
      "learning_rate": 6.906172893948585e-05,
      "loss": 0.372,
      "step": 1102
    },
    {
      "epoch": 1.5870503597122303,
      "grad_norm": 0.15529606711487257,
      "learning_rate": 6.903410118433996e-05,
      "loss": 0.3689,
      "step": 1103
    },
    {
      "epoch": 1.5884892086330935,
      "grad_norm": 0.18466658574544775,
      "learning_rate": 6.900644412347836e-05,
      "loss": 0.375,
      "step": 1104
    },
    {
      "epoch": 1.5899280575539567,
      "grad_norm": 0.19475119813923158,
      "learning_rate": 6.897875778481682e-05,
      "loss": 0.3618,
      "step": 1105
    },
    {
      "epoch": 1.5913669064748202,
      "grad_norm": 0.17366024888705028,
      "learning_rate": 6.89510421963007e-05,
      "loss": 0.3712,
      "step": 1106
    },
    {
      "epoch": 1.5928057553956836,
      "grad_norm": 0.15676286754267493,
      "learning_rate": 6.892329738590489e-05,
      "loss": 0.3683,
      "step": 1107
    },
    {
      "epoch": 1.5942446043165468,
      "grad_norm": 0.2188595760487771,
      "learning_rate": 6.889552338163372e-05,
      "loss": 0.3707,
      "step": 1108
    },
    {
      "epoch": 1.59568345323741,
      "grad_norm": 0.24277985356006634,
      "learning_rate": 6.886772021152104e-05,
      "loss": 0.3777,
      "step": 1109
    },
    {
      "epoch": 1.5971223021582732,
      "grad_norm": 0.18816245997249675,
      "learning_rate": 6.883988790363009e-05,
      "loss": 0.3727,
      "step": 1110
    },
    {
      "epoch": 1.5985611510791367,
      "grad_norm": 0.17503299519995805,
      "learning_rate": 6.881202648605359e-05,
      "loss": 0.3772,
      "step": 1111
    },
    {
      "epoch": 1.6,
      "grad_norm": 0.16511086634598757,
      "learning_rate": 6.878413598691358e-05,
      "loss": 0.377,
      "step": 1112
    },
    {
      "epoch": 1.6014388489208633,
      "grad_norm": 0.15203174372301287,
      "learning_rate": 6.875621643436147e-05,
      "loss": 0.3656,
      "step": 1113
    },
    {
      "epoch": 1.6028776978417265,
      "grad_norm": 0.16503153227139583,
      "learning_rate": 6.872826785657802e-05,
      "loss": 0.3655,
      "step": 1114
    },
    {
      "epoch": 1.60431654676259,
      "grad_norm": 0.15592639608997116,
      "learning_rate": 6.870029028177324e-05,
      "loss": 0.3689,
      "step": 1115
    },
    {
      "epoch": 1.6057553956834534,
      "grad_norm": 0.1641954675900134,
      "learning_rate": 6.867228373818648e-05,
      "loss": 0.3665,
      "step": 1116
    },
    {
      "epoch": 1.6071942446043166,
      "grad_norm": 0.19296460226555923,
      "learning_rate": 6.864424825408624e-05,
      "loss": 0.3586,
      "step": 1117
    },
    {
      "epoch": 1.6086330935251798,
      "grad_norm": 0.2215282372407742,
      "learning_rate": 6.861618385777028e-05,
      "loss": 0.3781,
      "step": 1118
    },
    {
      "epoch": 1.610071942446043,
      "grad_norm": 0.18461010675491152,
      "learning_rate": 6.858809057756558e-05,
      "loss": 0.3662,
      "step": 1119
    },
    {
      "epoch": 1.6115107913669064,
      "grad_norm": 0.17631555123291315,
      "learning_rate": 6.855996844182819e-05,
      "loss": 0.3716,
      "step": 1120
    },
    {
      "epoch": 1.6129496402877699,
      "grad_norm": 0.23550309117468388,
      "learning_rate": 6.853181747894334e-05,
      "loss": 0.3609,
      "step": 1121
    },
    {
      "epoch": 1.614388489208633,
      "grad_norm": 0.24146841042380257,
      "learning_rate": 6.850363771732536e-05,
      "loss": 0.3708,
      "step": 1122
    },
    {
      "epoch": 1.6158273381294963,
      "grad_norm": 0.17650593443303575,
      "learning_rate": 6.847542918541762e-05,
      "loss": 0.3747,
      "step": 1123
    },
    {
      "epoch": 1.6172661870503597,
      "grad_norm": 0.17144861816967746,
      "learning_rate": 6.844719191169254e-05,
      "loss": 0.3686,
      "step": 1124
    },
    {
      "epoch": 1.6187050359712232,
      "grad_norm": 0.19574566897019172,
      "learning_rate": 6.841892592465158e-05,
      "loss": 0.3622,
      "step": 1125
    },
    {
      "epoch": 1.6201438848920864,
      "grad_norm": 0.2285187712932195,
      "learning_rate": 6.839063125282512e-05,
      "loss": 0.3717,
      "step": 1126
    },
    {
      "epoch": 1.6215827338129496,
      "grad_norm": 0.2397466316648997,
      "learning_rate": 6.836230792477256e-05,
      "loss": 0.3659,
      "step": 1127
    },
    {
      "epoch": 1.6230215827338128,
      "grad_norm": 0.2450769117093886,
      "learning_rate": 6.833395596908217e-05,
      "loss": 0.3711,
      "step": 1128
    },
    {
      "epoch": 1.6244604316546762,
      "grad_norm": 0.20283090074104773,
      "learning_rate": 6.830557541437114e-05,
      "loss": 0.3612,
      "step": 1129
    },
    {
      "epoch": 1.6258992805755397,
      "grad_norm": 0.1412473996698023,
      "learning_rate": 6.827716628928556e-05,
      "loss": 0.3704,
      "step": 1130
    },
    {
      "epoch": 1.6273381294964029,
      "grad_norm": 0.16775492920558058,
      "learning_rate": 6.824872862250028e-05,
      "loss": 0.375,
      "step": 1131
    },
    {
      "epoch": 1.628776978417266,
      "grad_norm": 0.21575772836874516,
      "learning_rate": 6.822026244271903e-05,
      "loss": 0.3674,
      "step": 1132
    },
    {
      "epoch": 1.6302158273381295,
      "grad_norm": 0.2039009705419886,
      "learning_rate": 6.819176777867425e-05,
      "loss": 0.3689,
      "step": 1133
    },
    {
      "epoch": 1.631654676258993,
      "grad_norm": 0.18811765846923917,
      "learning_rate": 6.816324465912723e-05,
      "loss": 0.3664,
      "step": 1134
    },
    {
      "epoch": 1.6330935251798562,
      "grad_norm": 0.18184626894267752,
      "learning_rate": 6.813469311286789e-05,
      "loss": 0.3734,
      "step": 1135
    },
    {
      "epoch": 1.6345323741007194,
      "grad_norm": 0.19266509419072342,
      "learning_rate": 6.810611316871488e-05,
      "loss": 0.3664,
      "step": 1136
    },
    {
      "epoch": 1.6359712230215826,
      "grad_norm": 0.19628417801952316,
      "learning_rate": 6.80775048555155e-05,
      "loss": 0.3648,
      "step": 1137
    },
    {
      "epoch": 1.637410071942446,
      "grad_norm": 0.18940598982443663,
      "learning_rate": 6.804886820214572e-05,
      "loss": 0.3693,
      "step": 1138
    },
    {
      "epoch": 1.6388489208633095,
      "grad_norm": 0.16004033404127446,
      "learning_rate": 6.802020323751008e-05,
      "loss": 0.3685,
      "step": 1139
    },
    {
      "epoch": 1.6402877697841727,
      "grad_norm": 0.13072688623804118,
      "learning_rate": 6.799150999054169e-05,
      "loss": 0.3755,
      "step": 1140
    },
    {
      "epoch": 1.641726618705036,
      "grad_norm": 0.13198554865782608,
      "learning_rate": 6.796278849020225e-05,
      "loss": 0.369,
      "step": 1141
    },
    {
      "epoch": 1.6431654676258993,
      "grad_norm": 0.1599842172754047,
      "learning_rate": 6.79340387654819e-05,
      "loss": 0.3642,
      "step": 1142
    },
    {
      "epoch": 1.6446043165467625,
      "grad_norm": 0.17771857869118973,
      "learning_rate": 6.790526084539939e-05,
      "loss": 0.371,
      "step": 1143
    },
    {
      "epoch": 1.646043165467626,
      "grad_norm": 0.15171821300344104,
      "learning_rate": 6.787645475900182e-05,
      "loss": 0.3622,
      "step": 1144
    },
    {
      "epoch": 1.6474820143884892,
      "grad_norm": 0.14137633171084957,
      "learning_rate": 6.784762053536475e-05,
      "loss": 0.3748,
      "step": 1145
    },
    {
      "epoch": 1.6489208633093524,
      "grad_norm": 0.15929284019782258,
      "learning_rate": 6.781875820359216e-05,
      "loss": 0.3634,
      "step": 1146
    },
    {
      "epoch": 1.6503597122302158,
      "grad_norm": 0.17031051029268407,
      "learning_rate": 6.778986779281639e-05,
      "loss": 0.3712,
      "step": 1147
    },
    {
      "epoch": 1.6517985611510793,
      "grad_norm": 0.21332579251788533,
      "learning_rate": 6.776094933219811e-05,
      "loss": 0.3616,
      "step": 1148
    },
    {
      "epoch": 1.6532374100719425,
      "grad_norm": 0.17755181249847204,
      "learning_rate": 6.773200285092633e-05,
      "loss": 0.3652,
      "step": 1149
    },
    {
      "epoch": 1.6546762589928057,
      "grad_norm": 0.1924293240221449,
      "learning_rate": 6.770302837821833e-05,
      "loss": 0.364,
      "step": 1150
    },
    {
      "epoch": 1.6561151079136691,
      "grad_norm": 0.21501293495283566,
      "learning_rate": 6.767402594331961e-05,
      "loss": 0.3652,
      "step": 1151
    },
    {
      "epoch": 1.6575539568345323,
      "grad_norm": 0.2001501970519388,
      "learning_rate": 6.764499557550396e-05,
      "loss": 0.3681,
      "step": 1152
    },
    {
      "epoch": 1.6589928057553958,
      "grad_norm": 0.18563372424586336,
      "learning_rate": 6.761593730407329e-05,
      "loss": 0.3723,
      "step": 1153
    },
    {
      "epoch": 1.660431654676259,
      "grad_norm": 0.1828232460234351,
      "learning_rate": 6.758685115835776e-05,
      "loss": 0.3685,
      "step": 1154
    },
    {
      "epoch": 1.6618705035971222,
      "grad_norm": 0.17699360766600628,
      "learning_rate": 6.755773716771555e-05,
      "loss": 0.3679,
      "step": 1155
    },
    {
      "epoch": 1.6633093525179856,
      "grad_norm": 0.20368342130435985,
      "learning_rate": 6.752859536153306e-05,
      "loss": 0.3618,
      "step": 1156
    },
    {
      "epoch": 1.664748201438849,
      "grad_norm": 0.17640051533425183,
      "learning_rate": 6.749942576922473e-05,
      "loss": 0.3619,
      "step": 1157
    },
    {
      "epoch": 1.6661870503597123,
      "grad_norm": 0.21632935911496773,
      "learning_rate": 6.7470228420233e-05,
      "loss": 0.3751,
      "step": 1158
    },
    {
      "epoch": 1.6676258992805755,
      "grad_norm": 0.3076754300359413,
      "learning_rate": 6.744100334402836e-05,
      "loss": 0.3665,
      "step": 1159
    },
    {
      "epoch": 1.6690647482014387,
      "grad_norm": 0.19521131323389862,
      "learning_rate": 6.741175057010932e-05,
      "loss": 0.3568,
      "step": 1160
    },
    {
      "epoch": 1.6705035971223021,
      "grad_norm": 0.19925854450194885,
      "learning_rate": 6.738247012800228e-05,
      "loss": 0.3651,
      "step": 1161
    },
    {
      "epoch": 1.6719424460431656,
      "grad_norm": 0.24512084833363884,
      "learning_rate": 6.735316204726163e-05,
      "loss": 0.3749,
      "step": 1162
    },
    {
      "epoch": 1.6733812949640288,
      "grad_norm": 0.20602257731161921,
      "learning_rate": 6.732382635746961e-05,
      "loss": 0.3666,
      "step": 1163
    },
    {
      "epoch": 1.674820143884892,
      "grad_norm": 0.18003047454062612,
      "learning_rate": 6.729446308823635e-05,
      "loss": 0.3599,
      "step": 1164
    },
    {
      "epoch": 1.6762589928057554,
      "grad_norm": 0.16513473718549246,
      "learning_rate": 6.72650722691998e-05,
      "loss": 0.3725,
      "step": 1165
    },
    {
      "epoch": 1.6776978417266188,
      "grad_norm": 0.19027418872001917,
      "learning_rate": 6.723565393002576e-05,
      "loss": 0.3784,
      "step": 1166
    },
    {
      "epoch": 1.679136690647482,
      "grad_norm": 0.20054960977344133,
      "learning_rate": 6.720620810040776e-05,
      "loss": 0.3697,
      "step": 1167
    },
    {
      "epoch": 1.6805755395683453,
      "grad_norm": 0.18110955789965194,
      "learning_rate": 6.717673481006709e-05,
      "loss": 0.3606,
      "step": 1168
    },
    {
      "epoch": 1.6820143884892085,
      "grad_norm": 0.20028353836227986,
      "learning_rate": 6.714723408875279e-05,
      "loss": 0.3703,
      "step": 1169
    },
    {
      "epoch": 1.683453237410072,
      "grad_norm": 0.24954652165243063,
      "learning_rate": 6.711770596624153e-05,
      "loss": 0.3726,
      "step": 1170
    },
    {
      "epoch": 1.6848920863309353,
      "grad_norm": 0.2517417857934957,
      "learning_rate": 6.708815047233768e-05,
      "loss": 0.3693,
      "step": 1171
    },
    {
      "epoch": 1.6863309352517986,
      "grad_norm": 0.2089087199180209,
      "learning_rate": 6.705856763687324e-05,
      "loss": 0.3704,
      "step": 1172
    },
    {
      "epoch": 1.6877697841726618,
      "grad_norm": 0.1742666318915172,
      "learning_rate": 6.702895748970776e-05,
      "loss": 0.3634,
      "step": 1173
    },
    {
      "epoch": 1.6892086330935252,
      "grad_norm": 0.18217385997746868,
      "learning_rate": 6.699932006072842e-05,
      "loss": 0.3756,
      "step": 1174
    },
    {
      "epoch": 1.6906474820143886,
      "grad_norm": 0.2625774979517684,
      "learning_rate": 6.69696553798499e-05,
      "loss": 0.3621,
      "step": 1175
    },
    {
      "epoch": 1.6920863309352518,
      "grad_norm": 0.2868987974011614,
      "learning_rate": 6.693996347701442e-05,
      "loss": 0.3697,
      "step": 1176
    },
    {
      "epoch": 1.693525179856115,
      "grad_norm": 0.2503251710843237,
      "learning_rate": 6.691024438219159e-05,
      "loss": 0.3789,
      "step": 1177
    },
    {
      "epoch": 1.6949640287769783,
      "grad_norm": 0.23069913579436455,
      "learning_rate": 6.688049812537857e-05,
      "loss": 0.3579,
      "step": 1178
    },
    {
      "epoch": 1.6964028776978417,
      "grad_norm": 0.22720764849016103,
      "learning_rate": 6.685072473659989e-05,
      "loss": 0.3619,
      "step": 1179
    },
    {
      "epoch": 1.6978417266187051,
      "grad_norm": 0.2096733860806973,
      "learning_rate": 6.682092424590747e-05,
      "loss": 0.3709,
      "step": 1180
    },
    {
      "epoch": 1.6992805755395683,
      "grad_norm": 0.25133324773649096,
      "learning_rate": 6.679109668338057e-05,
      "loss": 0.3808,
      "step": 1181
    },
    {
      "epoch": 1.7007194244604316,
      "grad_norm": 0.27145707465158153,
      "learning_rate": 6.676124207912582e-05,
      "loss": 0.3661,
      "step": 1182
    },
    {
      "epoch": 1.702158273381295,
      "grad_norm": 0.27304035934956816,
      "learning_rate": 6.673136046327707e-05,
      "loss": 0.3616,
      "step": 1183
    },
    {
      "epoch": 1.7035971223021584,
      "grad_norm": 0.2723443196995204,
      "learning_rate": 6.670145186599552e-05,
      "loss": 0.3665,
      "step": 1184
    },
    {
      "epoch": 1.7050359712230216,
      "grad_norm": 0.22953328221226027,
      "learning_rate": 6.667151631746953e-05,
      "loss": 0.3627,
      "step": 1185
    },
    {
      "epoch": 1.7064748201438849,
      "grad_norm": 0.1612118326533789,
      "learning_rate": 6.664155384791473e-05,
      "loss": 0.3589,
      "step": 1186
    },
    {
      "epoch": 1.707913669064748,
      "grad_norm": 0.14286583024419586,
      "learning_rate": 6.661156448757386e-05,
      "loss": 0.3615,
      "step": 1187
    },
    {
      "epoch": 1.7093525179856115,
      "grad_norm": 0.1388824638725049,
      "learning_rate": 6.658154826671685e-05,
      "loss": 0.3662,
      "step": 1188
    },
    {
      "epoch": 1.710791366906475,
      "grad_norm": 0.14299854170823556,
      "learning_rate": 6.655150521564072e-05,
      "loss": 0.363,
      "step": 1189
    },
    {
      "epoch": 1.7122302158273381,
      "grad_norm": 0.18627026369421645,
      "learning_rate": 6.652143536466955e-05,
      "loss": 0.3756,
      "step": 1190
    },
    {
      "epoch": 1.7136690647482014,
      "grad_norm": 0.25452604401196904,
      "learning_rate": 6.649133874415454e-05,
      "loss": 0.3668,
      "step": 1191
    },
    {
      "epoch": 1.7151079136690648,
      "grad_norm": 0.3430639021513708,
      "learning_rate": 6.646121538447382e-05,
      "loss": 0.3618,
      "step": 1192
    },
    {
      "epoch": 1.7165467625899282,
      "grad_norm": 0.3712491097759974,
      "learning_rate": 6.643106531603259e-05,
      "loss": 0.3602,
      "step": 1193
    },
    {
      "epoch": 1.7179856115107914,
      "grad_norm": 0.2362138858573847,
      "learning_rate": 6.640088856926294e-05,
      "loss": 0.364,
      "step": 1194
    },
    {
      "epoch": 1.7194244604316546,
      "grad_norm": 0.13478226211555927,
      "learning_rate": 6.637068517462395e-05,
      "loss": 0.3583,
      "step": 1195
    },
    {
      "epoch": 1.7208633093525179,
      "grad_norm": 0.20014562213498244,
      "learning_rate": 6.634045516260156e-05,
      "loss": 0.367,
      "step": 1196
    },
    {
      "epoch": 1.7223021582733813,
      "grad_norm": 0.17170578915671195,
      "learning_rate": 6.631019856370856e-05,
      "loss": 0.3688,
      "step": 1197
    },
    {
      "epoch": 1.7237410071942447,
      "grad_norm": 0.14360750660228105,
      "learning_rate": 6.627991540848464e-05,
      "loss": 0.3576,
      "step": 1198
    },
    {
      "epoch": 1.725179856115108,
      "grad_norm": 0.15177488956197097,
      "learning_rate": 6.624960572749622e-05,
      "loss": 0.3609,
      "step": 1199
    },
    {
      "epoch": 1.7266187050359711,
      "grad_norm": 0.19404811811754438,
      "learning_rate": 6.621926955133657e-05,
      "loss": 0.3679,
      "step": 1200
    },
    {
      "epoch": 1.7280575539568346,
      "grad_norm": 0.2172039496918918,
      "learning_rate": 6.618890691062561e-05,
      "loss": 0.3731,
      "step": 1201
    },
    {
      "epoch": 1.7294964028776978,
      "grad_norm": 0.208959610319358,
      "learning_rate": 6.615851783601006e-05,
      "loss": 0.3613,
      "step": 1202
    },
    {
      "epoch": 1.7309352517985612,
      "grad_norm": 0.1838832578768312,
      "learning_rate": 6.612810235816326e-05,
      "loss": 0.3653,
      "step": 1203
    },
    {
      "epoch": 1.7323741007194244,
      "grad_norm": 0.14991254472871976,
      "learning_rate": 6.609766050778525e-05,
      "loss": 0.3658,
      "step": 1204
    },
    {
      "epoch": 1.7338129496402876,
      "grad_norm": 0.15056586517037424,
      "learning_rate": 6.606719231560265e-05,
      "loss": 0.3683,
      "step": 1205
    },
    {
      "epoch": 1.735251798561151,
      "grad_norm": 0.15261729120006054,
      "learning_rate": 6.60366978123687e-05,
      "loss": 0.3647,
      "step": 1206
    },
    {
      "epoch": 1.7366906474820145,
      "grad_norm": 0.15674945015806585,
      "learning_rate": 6.600617702886314e-05,
      "loss": 0.3637,
      "step": 1207
    },
    {
      "epoch": 1.7381294964028777,
      "grad_norm": 0.217110724574672,
      "learning_rate": 6.597562999589233e-05,
      "loss": 0.365,
      "step": 1208
    },
    {
      "epoch": 1.739568345323741,
      "grad_norm": 0.18175264752569695,
      "learning_rate": 6.594505674428903e-05,
      "loss": 0.369,
      "step": 1209
    },
    {
      "epoch": 1.7410071942446042,
      "grad_norm": 0.1544655331884978,
      "learning_rate": 6.59144573049125e-05,
      "loss": 0.3627,
      "step": 1210
    },
    {
      "epoch": 1.7424460431654676,
      "grad_norm": 0.20292392125680517,
      "learning_rate": 6.588383170864849e-05,
      "loss": 0.3677,
      "step": 1211
    },
    {
      "epoch": 1.743884892086331,
      "grad_norm": 0.21311850938644913,
      "learning_rate": 6.585317998640903e-05,
      "loss": 0.3705,
      "step": 1212
    },
    {
      "epoch": 1.7453237410071942,
      "grad_norm": 0.1860092526443083,
      "learning_rate": 6.582250216913265e-05,
      "loss": 0.3595,
      "step": 1213
    },
    {
      "epoch": 1.7467625899280574,
      "grad_norm": 0.16196745930604664,
      "learning_rate": 6.579179828778414e-05,
      "loss": 0.3631,
      "step": 1214
    },
    {
      "epoch": 1.7482014388489209,
      "grad_norm": 0.15627090063490578,
      "learning_rate": 6.576106837335458e-05,
      "loss": 0.3633,
      "step": 1215
    },
    {
      "epoch": 1.7496402877697843,
      "grad_norm": 0.17278059621160283,
      "learning_rate": 6.573031245686142e-05,
      "loss": 0.3701,
      "step": 1216
    },
    {
      "epoch": 1.7510791366906475,
      "grad_norm": 0.22942428046256286,
      "learning_rate": 6.569953056934826e-05,
      "loss": 0.3674,
      "step": 1217
    },
    {
      "epoch": 1.7525179856115107,
      "grad_norm": 0.20358107561700375,
      "learning_rate": 6.566872274188496e-05,
      "loss": 0.3774,
      "step": 1218
    },
    {
      "epoch": 1.753956834532374,
      "grad_norm": 0.17835485713159435,
      "learning_rate": 6.563788900556756e-05,
      "loss": 0.3693,
      "step": 1219
    },
    {
      "epoch": 1.7553956834532374,
      "grad_norm": 0.17026695702516972,
      "learning_rate": 6.560702939151826e-05,
      "loss": 0.3695,
      "step": 1220
    },
    {
      "epoch": 1.7568345323741008,
      "grad_norm": 0.19569792601854083,
      "learning_rate": 6.557614393088534e-05,
      "loss": 0.3719,
      "step": 1221
    },
    {
      "epoch": 1.758273381294964,
      "grad_norm": 0.18425833648219234,
      "learning_rate": 6.554523265484321e-05,
      "loss": 0.3623,
      "step": 1222
    },
    {
      "epoch": 1.7597122302158272,
      "grad_norm": 0.19413952365363657,
      "learning_rate": 6.551429559459231e-05,
      "loss": 0.3656,
      "step": 1223
    },
    {
      "epoch": 1.7611510791366907,
      "grad_norm": 0.22492340242534004,
      "learning_rate": 6.548333278135915e-05,
      "loss": 0.3609,
      "step": 1224
    },
    {
      "epoch": 1.762589928057554,
      "grad_norm": 0.23250710762288015,
      "learning_rate": 6.545234424639616e-05,
      "loss": 0.3651,
      "step": 1225
    },
    {
      "epoch": 1.7640287769784173,
      "grad_norm": 0.2531774006159571,
      "learning_rate": 6.542133002098178e-05,
      "loss": 0.3714,
      "step": 1226
    },
    {
      "epoch": 1.7654676258992805,
      "grad_norm": 0.2851427168782333,
      "learning_rate": 6.53902901364204e-05,
      "loss": 0.3733,
      "step": 1227
    },
    {
      "epoch": 1.7669064748201437,
      "grad_norm": 0.22742460070704784,
      "learning_rate": 6.535922462404226e-05,
      "loss": 0.3606,
      "step": 1228
    },
    {
      "epoch": 1.7683453237410072,
      "grad_norm": 0.17288087034904095,
      "learning_rate": 6.53281335152035e-05,
      "loss": 0.3693,
      "step": 1229
    },
    {
      "epoch": 1.7697841726618706,
      "grad_norm": 0.19183373954992122,
      "learning_rate": 6.529701684128608e-05,
      "loss": 0.3639,
      "step": 1230
    },
    {
      "epoch": 1.7712230215827338,
      "grad_norm": 0.23127921588481407,
      "learning_rate": 6.526587463369779e-05,
      "loss": 0.3736,
      "step": 1231
    },
    {
      "epoch": 1.772661870503597,
      "grad_norm": 0.22274556172374138,
      "learning_rate": 6.523470692387215e-05,
      "loss": 0.3768,
      "step": 1232
    },
    {
      "epoch": 1.7741007194244605,
      "grad_norm": 0.15866051821034918,
      "learning_rate": 6.520351374326846e-05,
      "loss": 0.3661,
      "step": 1233
    },
    {
      "epoch": 1.775539568345324,
      "grad_norm": 0.19594762797331902,
      "learning_rate": 6.51722951233717e-05,
      "loss": 0.3746,
      "step": 1234
    },
    {
      "epoch": 1.776978417266187,
      "grad_norm": 0.2515028034448848,
      "learning_rate": 6.514105109569254e-05,
      "loss": 0.3667,
      "step": 1235
    },
    {
      "epoch": 1.7784172661870503,
      "grad_norm": 0.21371915658842414,
      "learning_rate": 6.510978169176731e-05,
      "loss": 0.3704,
      "step": 1236
    },
    {
      "epoch": 1.7798561151079135,
      "grad_norm": 0.21260882546744508,
      "learning_rate": 6.507848694315794e-05,
      "loss": 0.3605,
      "step": 1237
    },
    {
      "epoch": 1.781294964028777,
      "grad_norm": 0.21239711315848003,
      "learning_rate": 6.504716688145192e-05,
      "loss": 0.3638,
      "step": 1238
    },
    {
      "epoch": 1.7827338129496404,
      "grad_norm": 0.21190449910072812,
      "learning_rate": 6.501582153826235e-05,
      "loss": 0.3696,
      "step": 1239
    },
    {
      "epoch": 1.7841726618705036,
      "grad_norm": 0.2266603833816144,
      "learning_rate": 6.498445094522776e-05,
      "loss": 0.3682,
      "step": 1240
    },
    {
      "epoch": 1.7856115107913668,
      "grad_norm": 0.2229483162436417,
      "learning_rate": 6.495305513401226e-05,
      "loss": 0.3644,
      "step": 1241
    },
    {
      "epoch": 1.7870503597122303,
      "grad_norm": 0.231901254439104,
      "learning_rate": 6.492163413630534e-05,
      "loss": 0.3778,
      "step": 1242
    },
    {
      "epoch": 1.7884892086330937,
      "grad_norm": 0.2586259725233526,
      "learning_rate": 6.489018798382195e-05,
      "loss": 0.3749,
      "step": 1243
    },
    {
      "epoch": 1.789928057553957,
      "grad_norm": 0.25644600072677914,
      "learning_rate": 6.485871670830243e-05,
      "loss": 0.3729,
      "step": 1244
    },
    {
      "epoch": 1.79136690647482,
      "grad_norm": 0.20278941867009576,
      "learning_rate": 6.482722034151247e-05,
      "loss": 0.3657,
      "step": 1245
    },
    {
      "epoch": 1.7928057553956833,
      "grad_norm": 0.12947462225096898,
      "learning_rate": 6.479569891524307e-05,
      "loss": 0.3605,
      "step": 1246
    },
    {
      "epoch": 1.7942446043165468,
      "grad_norm": 0.19473353666555931,
      "learning_rate": 6.476415246131056e-05,
      "loss": 0.3573,
      "step": 1247
    },
    {
      "epoch": 1.7956834532374102,
      "grad_norm": 0.22175060764272925,
      "learning_rate": 6.47325810115565e-05,
      "loss": 0.3688,
      "step": 1248
    },
    {
      "epoch": 1.7971223021582734,
      "grad_norm": 0.19107613478091598,
      "learning_rate": 6.470098459784768e-05,
      "loss": 0.3607,
      "step": 1249
    },
    {
      "epoch": 1.7985611510791366,
      "grad_norm": 0.20692252089586538,
      "learning_rate": 6.466936325207612e-05,
      "loss": 0.3618,
      "step": 1250
    },
    {
      "epoch": 1.8,
      "grad_norm": 0.2216428206800112,
      "learning_rate": 6.463771700615898e-05,
      "loss": 0.3706,
      "step": 1251
    },
    {
      "epoch": 1.8014388489208633,
      "grad_norm": 0.21527489050122087,
      "learning_rate": 6.460604589203854e-05,
      "loss": 0.3642,
      "step": 1252
    },
    {
      "epoch": 1.8028776978417267,
      "grad_norm": 0.22442501298074238,
      "learning_rate": 6.457434994168224e-05,
      "loss": 0.3738,
      "step": 1253
    },
    {
      "epoch": 1.80431654676259,
      "grad_norm": 0.2334843631216741,
      "learning_rate": 6.454262918708247e-05,
      "loss": 0.3636,
      "step": 1254
    },
    {
      "epoch": 1.8057553956834531,
      "grad_norm": 0.17986454834962382,
      "learning_rate": 6.451088366025682e-05,
      "loss": 0.3606,
      "step": 1255
    },
    {
      "epoch": 1.8071942446043165,
      "grad_norm": 0.16168341390536148,
      "learning_rate": 6.447911339324773e-05,
      "loss": 0.3613,
      "step": 1256
    },
    {
      "epoch": 1.80863309352518,
      "grad_norm": 0.15127179324517867,
      "learning_rate": 6.444731841812274e-05,
      "loss": 0.3731,
      "step": 1257
    },
    {
      "epoch": 1.8100719424460432,
      "grad_norm": 0.13619364927493682,
      "learning_rate": 6.44154987669742e-05,
      "loss": 0.3603,
      "step": 1258
    },
    {
      "epoch": 1.8115107913669064,
      "grad_norm": 0.13911547113387118,
      "learning_rate": 6.438365447191947e-05,
      "loss": 0.3671,
      "step": 1259
    },
    {
      "epoch": 1.8129496402877698,
      "grad_norm": 0.1763023708267854,
      "learning_rate": 6.435178556510076e-05,
      "loss": 0.3667,
      "step": 1260
    },
    {
      "epoch": 1.814388489208633,
      "grad_norm": 0.1844999431533389,
      "learning_rate": 6.431989207868508e-05,
      "loss": 0.3676,
      "step": 1261
    },
    {
      "epoch": 1.8158273381294965,
      "grad_norm": 0.179685943790929,
      "learning_rate": 6.428797404486431e-05,
      "loss": 0.3723,
      "step": 1262
    },
    {
      "epoch": 1.8172661870503597,
      "grad_norm": 0.18079082090576035,
      "learning_rate": 6.425603149585507e-05,
      "loss": 0.3667,
      "step": 1263
    },
    {
      "epoch": 1.818705035971223,
      "grad_norm": 0.18067051290306504,
      "learning_rate": 6.422406446389872e-05,
      "loss": 0.3698,
      "step": 1264
    },
    {
      "epoch": 1.8201438848920863,
      "grad_norm": 0.18456235795127293,
      "learning_rate": 6.419207298126135e-05,
      "loss": 0.3624,
      "step": 1265
    },
    {
      "epoch": 1.8215827338129498,
      "grad_norm": 0.1989141620198725,
      "learning_rate": 6.416005708023372e-05,
      "loss": 0.36,
      "step": 1266
    },
    {
      "epoch": 1.823021582733813,
      "grad_norm": 0.17040604847465962,
      "learning_rate": 6.412801679313125e-05,
      "loss": 0.3594,
      "step": 1267
    },
    {
      "epoch": 1.8244604316546762,
      "grad_norm": 0.17522779502348249,
      "learning_rate": 6.409595215229397e-05,
      "loss": 0.3667,
      "step": 1268
    },
    {
      "epoch": 1.8258992805755394,
      "grad_norm": 0.167039688944147,
      "learning_rate": 6.406386319008647e-05,
      "loss": 0.3641,
      "step": 1269
    },
    {
      "epoch": 1.8273381294964028,
      "grad_norm": 0.1614900845830248,
      "learning_rate": 6.403174993889791e-05,
      "loss": 0.3715,
      "step": 1270
    },
    {
      "epoch": 1.8287769784172663,
      "grad_norm": 0.17026645396291346,
      "learning_rate": 6.399961243114197e-05,
      "loss": 0.3592,
      "step": 1271
    },
    {
      "epoch": 1.8302158273381295,
      "grad_norm": 0.16282300582633127,
      "learning_rate": 6.39674506992568e-05,
      "loss": 0.3648,
      "step": 1272
    },
    {
      "epoch": 1.8316546762589927,
      "grad_norm": 0.16423560483078464,
      "learning_rate": 6.393526477570499e-05,
      "loss": 0.3583,
      "step": 1273
    },
    {
      "epoch": 1.8330935251798561,
      "grad_norm": 0.15753644555304952,
      "learning_rate": 6.390305469297357e-05,
      "loss": 0.3666,
      "step": 1274
    },
    {
      "epoch": 1.8345323741007196,
      "grad_norm": 0.132931699103671,
      "learning_rate": 6.387082048357397e-05,
      "loss": 0.3588,
      "step": 1275
    },
    {
      "epoch": 1.8359712230215828,
      "grad_norm": 0.12381739999202473,
      "learning_rate": 6.383856218004193e-05,
      "loss": 0.3619,
      "step": 1276
    },
    {
      "epoch": 1.837410071942446,
      "grad_norm": 0.1408804747953695,
      "learning_rate": 6.380627981493753e-05,
      "loss": 0.3745,
      "step": 1277
    },
    {
      "epoch": 1.8388489208633092,
      "grad_norm": 0.15318148858885486,
      "learning_rate": 6.377397342084514e-05,
      "loss": 0.3641,
      "step": 1278
    },
    {
      "epoch": 1.8402877697841726,
      "grad_norm": 0.15896624901132014,
      "learning_rate": 6.37416430303734e-05,
      "loss": 0.359,
      "step": 1279
    },
    {
      "epoch": 1.841726618705036,
      "grad_norm": 0.16889794983736373,
      "learning_rate": 6.370928867615513e-05,
      "loss": 0.3604,
      "step": 1280
    },
    {
      "epoch": 1.8431654676258993,
      "grad_norm": 0.13534285200730897,
      "learning_rate": 6.367691039084736e-05,
      "loss": 0.3616,
      "step": 1281
    },
    {
      "epoch": 1.8446043165467625,
      "grad_norm": 0.15095602064699098,
      "learning_rate": 6.36445082071313e-05,
      "loss": 0.3658,
      "step": 1282
    },
    {
      "epoch": 1.846043165467626,
      "grad_norm": 0.16119564119857152,
      "learning_rate": 6.361208215771222e-05,
      "loss": 0.3681,
      "step": 1283
    },
    {
      "epoch": 1.8474820143884894,
      "grad_norm": 0.1687200328354194,
      "learning_rate": 6.357963227531954e-05,
      "loss": 0.3778,
      "step": 1284
    },
    {
      "epoch": 1.8489208633093526,
      "grad_norm": 0.1927316215580099,
      "learning_rate": 6.35471585927067e-05,
      "loss": 0.3709,
      "step": 1285
    },
    {
      "epoch": 1.8503597122302158,
      "grad_norm": 0.2255474866727223,
      "learning_rate": 6.351466114265118e-05,
      "loss": 0.3668,
      "step": 1286
    },
    {
      "epoch": 1.851798561151079,
      "grad_norm": 0.23023478951407433,
      "learning_rate": 6.348213995795445e-05,
      "loss": 0.3777,
      "step": 1287
    },
    {
      "epoch": 1.8532374100719424,
      "grad_norm": 0.2380129207576633,
      "learning_rate": 6.344959507144192e-05,
      "loss": 0.3684,
      "step": 1288
    },
    {
      "epoch": 1.8546762589928059,
      "grad_norm": 0.19321416153221121,
      "learning_rate": 6.341702651596293e-05,
      "loss": 0.3539,
      "step": 1289
    },
    {
      "epoch": 1.856115107913669,
      "grad_norm": 0.23807443862365457,
      "learning_rate": 6.338443432439074e-05,
      "loss": 0.3673,
      "step": 1290
    },
    {
      "epoch": 1.8575539568345323,
      "grad_norm": 0.246407917353753,
      "learning_rate": 6.335181852962242e-05,
      "loss": 0.3594,
      "step": 1291
    },
    {
      "epoch": 1.8589928057553957,
      "grad_norm": 0.239352433412642,
      "learning_rate": 6.331917916457889e-05,
      "loss": 0.3588,
      "step": 1292
    },
    {
      "epoch": 1.8604316546762591,
      "grad_norm": 0.24341180965224837,
      "learning_rate": 6.328651626220485e-05,
      "loss": 0.3736,
      "step": 1293
    },
    {
      "epoch": 1.8618705035971224,
      "grad_norm": 0.22174826280195295,
      "learning_rate": 6.325382985546879e-05,
      "loss": 0.3663,
      "step": 1294
    },
    {
      "epoch": 1.8633093525179856,
      "grad_norm": 0.18922955181189371,
      "learning_rate": 6.322111997736288e-05,
      "loss": 0.3711,
      "step": 1295
    },
    {
      "epoch": 1.8647482014388488,
      "grad_norm": 0.18508999656566633,
      "learning_rate": 6.3188386660903e-05,
      "loss": 0.3656,
      "step": 1296
    },
    {
      "epoch": 1.8661870503597122,
      "grad_norm": 0.16258187592426696,
      "learning_rate": 6.315562993912869e-05,
      "loss": 0.3684,
      "step": 1297
    },
    {
      "epoch": 1.8676258992805757,
      "grad_norm": 0.11809985776299295,
      "learning_rate": 6.31228498451031e-05,
      "loss": 0.3704,
      "step": 1298
    },
    {
      "epoch": 1.8690647482014389,
      "grad_norm": 0.1473174157257907,
      "learning_rate": 6.309004641191299e-05,
      "loss": 0.367,
      "step": 1299
    },
    {
      "epoch": 1.870503597122302,
      "grad_norm": 0.15981579791476963,
      "learning_rate": 6.305721967266869e-05,
      "loss": 0.3722,
      "step": 1300
    },
    {
      "epoch": 1.8719424460431655,
      "grad_norm": 0.15920152652070532,
      "learning_rate": 6.302436966050401e-05,
      "loss": 0.3614,
      "step": 1301
    },
    {
      "epoch": 1.873381294964029,
      "grad_norm": 0.13753742153654028,
      "learning_rate": 6.29914964085763e-05,
      "loss": 0.3635,
      "step": 1302
    },
    {
      "epoch": 1.8748201438848922,
      "grad_norm": 0.15314177588108846,
      "learning_rate": 6.295859995006629e-05,
      "loss": 0.3669,
      "step": 1303
    },
    {
      "epoch": 1.8762589928057554,
      "grad_norm": 0.14603492031310347,
      "learning_rate": 6.292568031817823e-05,
      "loss": 0.3673,
      "step": 1304
    },
    {
      "epoch": 1.8776978417266186,
      "grad_norm": 0.13655641297840962,
      "learning_rate": 6.28927375461397e-05,
      "loss": 0.3617,
      "step": 1305
    },
    {
      "epoch": 1.879136690647482,
      "grad_norm": 0.1682465239379107,
      "learning_rate": 6.285977166720166e-05,
      "loss": 0.3637,
      "step": 1306
    },
    {
      "epoch": 1.8805755395683454,
      "grad_norm": 0.21073997101971043,
      "learning_rate": 6.28267827146384e-05,
      "loss": 0.3565,
      "step": 1307
    },
    {
      "epoch": 1.8820143884892087,
      "grad_norm": 0.16044415201678497,
      "learning_rate": 6.279377072174744e-05,
      "loss": 0.3682,
      "step": 1308
    },
    {
      "epoch": 1.8834532374100719,
      "grad_norm": 0.19534802614931365,
      "learning_rate": 6.276073572184964e-05,
      "loss": 0.3717,
      "step": 1309
    },
    {
      "epoch": 1.8848920863309353,
      "grad_norm": 0.1905370702491253,
      "learning_rate": 6.272767774828903e-05,
      "loss": 0.3694,
      "step": 1310
    },
    {
      "epoch": 1.8863309352517985,
      "grad_norm": 0.22291681201611888,
      "learning_rate": 6.269459683443283e-05,
      "loss": 0.3683,
      "step": 1311
    },
    {
      "epoch": 1.887769784172662,
      "grad_norm": 0.22979965511133224,
      "learning_rate": 6.266149301367146e-05,
      "loss": 0.3685,
      "step": 1312
    },
    {
      "epoch": 1.8892086330935252,
      "grad_norm": 0.25371034308502655,
      "learning_rate": 6.262836631941839e-05,
      "loss": 0.3665,
      "step": 1313
    },
    {
      "epoch": 1.8906474820143884,
      "grad_norm": 0.23742213179615146,
      "learning_rate": 6.259521678511023e-05,
      "loss": 0.3682,
      "step": 1314
    },
    {
      "epoch": 1.8920863309352518,
      "grad_norm": 0.17859614363972742,
      "learning_rate": 6.256204444420663e-05,
      "loss": 0.3558,
      "step": 1315
    },
    {
      "epoch": 1.8935251798561152,
      "grad_norm": 0.15944292200948446,
      "learning_rate": 6.252884933019028e-05,
      "loss": 0.3664,
      "step": 1316
    },
    {
      "epoch": 1.8949640287769784,
      "grad_norm": 0.16886543532919862,
      "learning_rate": 6.249563147656679e-05,
      "loss": 0.3646,
      "step": 1317
    },
    {
      "epoch": 1.8964028776978417,
      "grad_norm": 0.17695565927950804,
      "learning_rate": 6.24623909168648e-05,
      "loss": 0.3634,
      "step": 1318
    },
    {
      "epoch": 1.8978417266187049,
      "grad_norm": 0.173219674595819,
      "learning_rate": 6.242912768463581e-05,
      "loss": 0.3707,
      "step": 1319
    },
    {
      "epoch": 1.8992805755395683,
      "grad_norm": 0.18810502872275187,
      "learning_rate": 6.239584181345426e-05,
      "loss": 0.3658,
      "step": 1320
    },
    {
      "epoch": 1.9007194244604317,
      "grad_norm": 0.18522148169382519,
      "learning_rate": 6.236253333691739e-05,
      "loss": 0.3657,
      "step": 1321
    },
    {
      "epoch": 1.902158273381295,
      "grad_norm": 0.17037815206677648,
      "learning_rate": 6.23292022886453e-05,
      "loss": 0.3617,
      "step": 1322
    },
    {
      "epoch": 1.9035971223021582,
      "grad_norm": 0.13651020757870233,
      "learning_rate": 6.229584870228083e-05,
      "loss": 0.3663,
      "step": 1323
    },
    {
      "epoch": 1.9050359712230216,
      "grad_norm": 0.13238836277044766,
      "learning_rate": 6.226247261148958e-05,
      "loss": 0.3674,
      "step": 1324
    },
    {
      "epoch": 1.906474820143885,
      "grad_norm": 0.17756725488705433,
      "learning_rate": 6.22290740499599e-05,
      "loss": 0.3802,
      "step": 1325
    },
    {
      "epoch": 1.9079136690647482,
      "grad_norm": 0.19697424948513603,
      "learning_rate": 6.21956530514028e-05,
      "loss": 0.3667,
      "step": 1326
    },
    {
      "epoch": 1.9093525179856115,
      "grad_norm": 0.18668259822133004,
      "learning_rate": 6.216220964955192e-05,
      "loss": 0.37,
      "step": 1327
    },
    {
      "epoch": 1.9107913669064747,
      "grad_norm": 0.18432614814837855,
      "learning_rate": 6.21287438781635e-05,
      "loss": 0.3587,
      "step": 1328
    },
    {
      "epoch": 1.912230215827338,
      "grad_norm": 0.21777479742632244,
      "learning_rate": 6.209525577101642e-05,
      "loss": 0.367,
      "step": 1329
    },
    {
      "epoch": 1.9136690647482015,
      "grad_norm": 0.1551943406063464,
      "learning_rate": 6.206174536191207e-05,
      "loss": 0.3663,
      "step": 1330
    },
    {
      "epoch": 1.9151079136690647,
      "grad_norm": 0.18586350947210925,
      "learning_rate": 6.202821268467433e-05,
      "loss": 0.376,
      "step": 1331
    },
    {
      "epoch": 1.916546762589928,
      "grad_norm": 0.17665307141723707,
      "learning_rate": 6.199465777314958e-05,
      "loss": 0.3703,
      "step": 1332
    },
    {
      "epoch": 1.9179856115107914,
      "grad_norm": 0.16332710030615785,
      "learning_rate": 6.196108066120663e-05,
      "loss": 0.3632,
      "step": 1333
    },
    {
      "epoch": 1.9194244604316548,
      "grad_norm": 0.16610604271465892,
      "learning_rate": 6.192748138273674e-05,
      "loss": 0.3677,
      "step": 1334
    },
    {
      "epoch": 1.920863309352518,
      "grad_norm": 0.14593869750353175,
      "learning_rate": 6.189385997165348e-05,
      "loss": 0.3676,
      "step": 1335
    },
    {
      "epoch": 1.9223021582733812,
      "grad_norm": 0.15808245007787355,
      "learning_rate": 6.186021646189281e-05,
      "loss": 0.3571,
      "step": 1336
    },
    {
      "epoch": 1.9237410071942445,
      "grad_norm": 0.19800729472492867,
      "learning_rate": 6.182655088741294e-05,
      "loss": 0.3672,
      "step": 1337
    },
    {
      "epoch": 1.925179856115108,
      "grad_norm": 0.2671922414742339,
      "learning_rate": 6.179286328219442e-05,
      "loss": 0.3613,
      "step": 1338
    },
    {
      "epoch": 1.9266187050359713,
      "grad_norm": 0.2986364081661207,
      "learning_rate": 6.175915368024e-05,
      "loss": 0.3702,
      "step": 1339
    },
    {
      "epoch": 1.9280575539568345,
      "grad_norm": 0.263811149892615,
      "learning_rate": 6.172542211557463e-05,
      "loss": 0.3626,
      "step": 1340
    },
    {
      "epoch": 1.9294964028776977,
      "grad_norm": 0.21786234216312614,
      "learning_rate": 6.169166862224542e-05,
      "loss": 0.3655,
      "step": 1341
    },
    {
      "epoch": 1.9309352517985612,
      "grad_norm": 0.2121469588015191,
      "learning_rate": 6.165789323432166e-05,
      "loss": 0.3656,
      "step": 1342
    },
    {
      "epoch": 1.9323741007194246,
      "grad_norm": 0.20357450615026512,
      "learning_rate": 6.162409598589467e-05,
      "loss": 0.372,
      "step": 1343
    },
    {
      "epoch": 1.9338129496402878,
      "grad_norm": 0.19609243094112727,
      "learning_rate": 6.159027691107791e-05,
      "loss": 0.3678,
      "step": 1344
    },
    {
      "epoch": 1.935251798561151,
      "grad_norm": 0.24947346016676034,
      "learning_rate": 6.15564360440068e-05,
      "loss": 0.367,
      "step": 1345
    },
    {
      "epoch": 1.9366906474820142,
      "grad_norm": 0.28117985290654207,
      "learning_rate": 6.15225734188388e-05,
      "loss": 0.3694,
      "step": 1346
    },
    {
      "epoch": 1.9381294964028777,
      "grad_norm": 0.23619911878557048,
      "learning_rate": 6.148868906975334e-05,
      "loss": 0.375,
      "step": 1347
    },
    {
      "epoch": 1.9395683453237411,
      "grad_norm": 0.1889537191471565,
      "learning_rate": 6.145478303095174e-05,
      "loss": 0.3588,
      "step": 1348
    },
    {
      "epoch": 1.9410071942446043,
      "grad_norm": 0.16659880981858904,
      "learning_rate": 6.142085533665722e-05,
      "loss": 0.3761,
      "step": 1349
    },
    {
      "epoch": 1.9424460431654675,
      "grad_norm": 0.1548161898642397,
      "learning_rate": 6.138690602111487e-05,
      "loss": 0.3628,
      "step": 1350
    },
    {
      "epoch": 1.943884892086331,
      "grad_norm": 0.16605248079429874,
      "learning_rate": 6.135293511859164e-05,
      "loss": 0.3622,
      "step": 1351
    },
    {
      "epoch": 1.9453237410071944,
      "grad_norm": 0.21011815374468795,
      "learning_rate": 6.131894266337618e-05,
      "loss": 0.3614,
      "step": 1352
    },
    {
      "epoch": 1.9467625899280576,
      "grad_norm": 0.26222420519160206,
      "learning_rate": 6.128492868977897e-05,
      "loss": 0.3745,
      "step": 1353
    },
    {
      "epoch": 1.9482014388489208,
      "grad_norm": 0.2761941631424964,
      "learning_rate": 6.12508932321322e-05,
      "loss": 0.3702,
      "step": 1354
    },
    {
      "epoch": 1.949640287769784,
      "grad_norm": 0.24631564378436294,
      "learning_rate": 6.12168363247897e-05,
      "loss": 0.3589,
      "step": 1355
    },
    {
      "epoch": 1.9510791366906475,
      "grad_norm": 0.21284330567492582,
      "learning_rate": 6.1182758002127e-05,
      "loss": 0.3639,
      "step": 1356
    },
    {
      "epoch": 1.952517985611511,
      "grad_norm": 0.2046714792956475,
      "learning_rate": 6.114865829854123e-05,
      "loss": 0.3761,
      "step": 1357
    },
    {
      "epoch": 1.9539568345323741,
      "grad_norm": 0.2095542825539175,
      "learning_rate": 6.111453724845106e-05,
      "loss": 0.37,
      "step": 1358
    },
    {
      "epoch": 1.9553956834532373,
      "grad_norm": 0.20867260956799893,
      "learning_rate": 6.108039488629679e-05,
      "loss": 0.3716,
      "step": 1359
    },
    {
      "epoch": 1.9568345323741008,
      "grad_norm": 0.204268029457269,
      "learning_rate": 6.104623124654016e-05,
      "loss": 0.3718,
      "step": 1360
    },
    {
      "epoch": 1.958273381294964,
      "grad_norm": 0.13842933139509825,
      "learning_rate": 6.101204636366441e-05,
      "loss": 0.3667,
      "step": 1361
    },
    {
      "epoch": 1.9597122302158274,
      "grad_norm": 0.14559876830826288,
      "learning_rate": 6.0977840272174224e-05,
      "loss": 0.3694,
      "step": 1362
    },
    {
      "epoch": 1.9611510791366906,
      "grad_norm": 0.21853957364813154,
      "learning_rate": 6.094361300659571e-05,
      "loss": 0.3661,
      "step": 1363
    },
    {
      "epoch": 1.9625899280575538,
      "grad_norm": 0.2803796152416274,
      "learning_rate": 6.090936460147632e-05,
      "loss": 0.3683,
      "step": 1364
    },
    {
      "epoch": 1.9640287769784173,
      "grad_norm": 0.24471699891354518,
      "learning_rate": 6.087509509138483e-05,
      "loss": 0.3567,
      "step": 1365
    },
    {
      "epoch": 1.9654676258992807,
      "grad_norm": 0.19905025886723116,
      "learning_rate": 6.0840804510911374e-05,
      "loss": 0.3661,
      "step": 1366
    },
    {
      "epoch": 1.966906474820144,
      "grad_norm": 0.1586018942006736,
      "learning_rate": 6.0806492894667315e-05,
      "loss": 0.355,
      "step": 1367
    },
    {
      "epoch": 1.9683453237410071,
      "grad_norm": 0.1966596533936432,
      "learning_rate": 6.077216027728524e-05,
      "loss": 0.3608,
      "step": 1368
    },
    {
      "epoch": 1.9697841726618706,
      "grad_norm": 0.1952894028757089,
      "learning_rate": 6.073780669341896e-05,
      "loss": 0.3605,
      "step": 1369
    },
    {
      "epoch": 1.9712230215827338,
      "grad_norm": 0.1739202507995161,
      "learning_rate": 6.070343217774343e-05,
      "loss": 0.3618,
      "step": 1370
    },
    {
      "epoch": 1.9726618705035972,
      "grad_norm": 0.18788051008043297,
      "learning_rate": 6.066903676495477e-05,
      "loss": 0.3682,
      "step": 1371
    },
    {
      "epoch": 1.9741007194244604,
      "grad_norm": 0.17297079370821708,
      "learning_rate": 6.063462048977011e-05,
      "loss": 0.3619,
      "step": 1372
    },
    {
      "epoch": 1.9755395683453236,
      "grad_norm": 0.14786038421172948,
      "learning_rate": 6.060018338692774e-05,
      "loss": 0.3704,
      "step": 1373
    },
    {
      "epoch": 1.976978417266187,
      "grad_norm": 0.1526032428610714,
      "learning_rate": 6.056572549118688e-05,
      "loss": 0.372,
      "step": 1374
    },
    {
      "epoch": 1.9784172661870505,
      "grad_norm": 0.15556697159769797,
      "learning_rate": 6.053124683732781e-05,
      "loss": 0.3815,
      "step": 1375
    },
    {
      "epoch": 1.9798561151079137,
      "grad_norm": 0.14307035561229411,
      "learning_rate": 6.049674746015172e-05,
      "loss": 0.3612,
      "step": 1376
    },
    {
      "epoch": 1.981294964028777,
      "grad_norm": 0.16303436232338575,
      "learning_rate": 6.046222739448075e-05,
      "loss": 0.3702,
      "step": 1377
    },
    {
      "epoch": 1.9827338129496401,
      "grad_norm": 0.20362078098379227,
      "learning_rate": 6.042768667515786e-05,
      "loss": 0.3676,
      "step": 1378
    },
    {
      "epoch": 1.9841726618705036,
      "grad_norm": 0.20224652092435189,
      "learning_rate": 6.039312533704692e-05,
      "loss": 0.3644,
      "step": 1379
    },
    {
      "epoch": 1.985611510791367,
      "grad_norm": 0.1918306268378846,
      "learning_rate": 6.0358543415032625e-05,
      "loss": 0.3644,
      "step": 1380
    },
    {
      "epoch": 1.9870503597122302,
      "grad_norm": 0.1977587698041893,
      "learning_rate": 6.032394094402035e-05,
      "loss": 0.3623,
      "step": 1381
    },
    {
      "epoch": 1.9884892086330934,
      "grad_norm": 0.19004846555269467,
      "learning_rate": 6.0289317958936305e-05,
      "loss": 0.3709,
      "step": 1382
    },
    {
      "epoch": 1.9899280575539569,
      "grad_norm": 0.189812208978719,
      "learning_rate": 6.0254674494727374e-05,
      "loss": 0.3596,
      "step": 1383
    },
    {
      "epoch": 1.9913669064748203,
      "grad_norm": 0.17927036123690349,
      "learning_rate": 6.022001058636111e-05,
      "loss": 0.3669,
      "step": 1384
    },
    {
      "epoch": 1.9928057553956835,
      "grad_norm": 0.12890285825777262,
      "learning_rate": 6.01853262688257e-05,
      "loss": 0.3645,
      "step": 1385
    },
    {
      "epoch": 1.9942446043165467,
      "grad_norm": 0.15473138461434743,
      "learning_rate": 6.0150621577129934e-05,
      "loss": 0.3681,
      "step": 1386
    },
    {
      "epoch": 1.99568345323741,
      "grad_norm": 0.168212010686727,
      "learning_rate": 6.011589654630318e-05,
      "loss": 0.3643,
      "step": 1387
    },
    {
      "epoch": 1.9971223021582734,
      "grad_norm": 0.17058104280627068,
      "learning_rate": 6.008115121139528e-05,
      "loss": 0.3605,
      "step": 1388
    },
    {
      "epoch": 1.9985611510791368,
      "grad_norm": 0.16992170065244036,
      "learning_rate": 6.0046385607476655e-05,
      "loss": 0.372,
      "step": 1389
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.13875453854266914,
      "learning_rate": 6.001159976963814e-05,
      "loss": 0.3767,
      "step": 1390
    },
    {
      "epoch": 2.001438848920863,
      "grad_norm": 0.148563996887187,
      "learning_rate": 5.9976793732990965e-05,
      "loss": 0.3344,
      "step": 1391
    },
    {
      "epoch": 2.0028776978417264,
      "grad_norm": 0.17757271066320204,
      "learning_rate": 5.9941967532666806e-05,
      "loss": 0.3414,
      "step": 1392
    },
    {
      "epoch": 2.00431654676259,
      "grad_norm": 0.25728320339880906,
      "learning_rate": 5.990712120381766e-05,
      "loss": 0.3512,
      "step": 1393
    },
    {
      "epoch": 2.0057553956834533,
      "grad_norm": 0.3018541720324142,
      "learning_rate": 5.987225478161583e-05,
      "loss": 0.3371,
      "step": 1394
    },
    {
      "epoch": 2.0071942446043165,
      "grad_norm": 0.3411629721917868,
      "learning_rate": 5.9837368301253905e-05,
      "loss": 0.3352,
      "step": 1395
    },
    {
      "epoch": 2.0086330935251797,
      "grad_norm": 0.36077816050495254,
      "learning_rate": 5.980246179794476e-05,
      "loss": 0.3481,
      "step": 1396
    },
    {
      "epoch": 2.0100719424460434,
      "grad_norm": 0.3325755806427443,
      "learning_rate": 5.976753530692144e-05,
      "loss": 0.3358,
      "step": 1397
    },
    {
      "epoch": 2.0115107913669066,
      "grad_norm": 0.28271431398627417,
      "learning_rate": 5.9732588863437155e-05,
      "loss": 0.3368,
      "step": 1398
    },
    {
      "epoch": 2.01294964028777,
      "grad_norm": 0.22383842416931668,
      "learning_rate": 5.96976225027653e-05,
      "loss": 0.3476,
      "step": 1399
    },
    {
      "epoch": 2.014388489208633,
      "grad_norm": 0.22964513577848436,
      "learning_rate": 5.966263626019932e-05,
      "loss": 0.3397,
      "step": 1400
    },
    {
      "epoch": 2.015827338129496,
      "grad_norm": 0.19952963432087853,
      "learning_rate": 5.9627630171052774e-05,
      "loss": 0.3545,
      "step": 1401
    },
    {
      "epoch": 2.01726618705036,
      "grad_norm": 0.215144227694722,
      "learning_rate": 5.9592604270659234e-05,
      "loss": 0.347,
      "step": 1402
    },
    {
      "epoch": 2.018705035971223,
      "grad_norm": 0.27146520260633705,
      "learning_rate": 5.955755859437225e-05,
      "loss": 0.3402,
      "step": 1403
    },
    {
      "epoch": 2.0201438848920863,
      "grad_norm": 0.2558100421691262,
      "learning_rate": 5.9522493177565366e-05,
      "loss": 0.3407,
      "step": 1404
    },
    {
      "epoch": 2.0215827338129495,
      "grad_norm": 0.19499628780137643,
      "learning_rate": 5.948740805563203e-05,
      "loss": 0.3363,
      "step": 1405
    },
    {
      "epoch": 2.023021582733813,
      "grad_norm": 0.25988959050213156,
      "learning_rate": 5.94523032639856e-05,
      "loss": 0.339,
      "step": 1406
    },
    {
      "epoch": 2.0244604316546764,
      "grad_norm": 0.2691665083854109,
      "learning_rate": 5.9417178838059254e-05,
      "loss": 0.3306,
      "step": 1407
    },
    {
      "epoch": 2.0258992805755396,
      "grad_norm": 0.16866973938494045,
      "learning_rate": 5.9382034813306014e-05,
      "loss": 0.3357,
      "step": 1408
    },
    {
      "epoch": 2.027338129496403,
      "grad_norm": 0.1980678367873882,
      "learning_rate": 5.934687122519868e-05,
      "loss": 0.343,
      "step": 1409
    },
    {
      "epoch": 2.028776978417266,
      "grad_norm": 0.24433337786013076,
      "learning_rate": 5.93116881092298e-05,
      "loss": 0.3354,
      "step": 1410
    },
    {
      "epoch": 2.0302158273381297,
      "grad_norm": 0.17072295105484814,
      "learning_rate": 5.927648550091162e-05,
      "loss": 0.3415,
      "step": 1411
    },
    {
      "epoch": 2.031654676258993,
      "grad_norm": 0.1471505339735923,
      "learning_rate": 5.9241263435776087e-05,
      "loss": 0.3363,
      "step": 1412
    },
    {
      "epoch": 2.033093525179856,
      "grad_norm": 0.17589140549291227,
      "learning_rate": 5.920602194937474e-05,
      "loss": 0.3478,
      "step": 1413
    },
    {
      "epoch": 2.0345323741007193,
      "grad_norm": 0.15623510804112709,
      "learning_rate": 5.9170761077278766e-05,
      "loss": 0.3396,
      "step": 1414
    },
    {
      "epoch": 2.0359712230215825,
      "grad_norm": 0.16624956542152658,
      "learning_rate": 5.9135480855078915e-05,
      "loss": 0.3421,
      "step": 1415
    },
    {
      "epoch": 2.037410071942446,
      "grad_norm": 0.16604428989665748,
      "learning_rate": 5.910018131838544e-05,
      "loss": 0.3458,
      "step": 1416
    },
    {
      "epoch": 2.0388489208633094,
      "grad_norm": 0.1481243103484568,
      "learning_rate": 5.906486250282811e-05,
      "loss": 0.3407,
      "step": 1417
    },
    {
      "epoch": 2.0402877697841726,
      "grad_norm": 0.16981494985197118,
      "learning_rate": 5.902952444405615e-05,
      "loss": 0.3437,
      "step": 1418
    },
    {
      "epoch": 2.041726618705036,
      "grad_norm": 0.18613786633593082,
      "learning_rate": 5.899416717773822e-05,
      "loss": 0.3471,
      "step": 1419
    },
    {
      "epoch": 2.0431654676258995,
      "grad_norm": 0.16572440895735488,
      "learning_rate": 5.8958790739562316e-05,
      "loss": 0.3387,
      "step": 1420
    },
    {
      "epoch": 2.0446043165467627,
      "grad_norm": 0.13683264362354047,
      "learning_rate": 5.892339516523586e-05,
      "loss": 0.3354,
      "step": 1421
    },
    {
      "epoch": 2.046043165467626,
      "grad_norm": 0.13439859226195303,
      "learning_rate": 5.8887980490485536e-05,
      "loss": 0.3338,
      "step": 1422
    },
    {
      "epoch": 2.047482014388489,
      "grad_norm": 0.1439569376941477,
      "learning_rate": 5.8852546751057337e-05,
      "loss": 0.3413,
      "step": 1423
    },
    {
      "epoch": 2.0489208633093523,
      "grad_norm": 0.1598768464407202,
      "learning_rate": 5.8817093982716455e-05,
      "loss": 0.3531,
      "step": 1424
    },
    {
      "epoch": 2.050359712230216,
      "grad_norm": 0.14821641303085784,
      "learning_rate": 5.878162222124735e-05,
      "loss": 0.3356,
      "step": 1425
    },
    {
      "epoch": 2.051798561151079,
      "grad_norm": 0.12491325288420484,
      "learning_rate": 5.8746131502453623e-05,
      "loss": 0.3385,
      "step": 1426
    },
    {
      "epoch": 2.0532374100719424,
      "grad_norm": 0.12083502206900046,
      "learning_rate": 5.871062186215799e-05,
      "loss": 0.3409,
      "step": 1427
    },
    {
      "epoch": 2.0546762589928056,
      "grad_norm": 0.13571360860778423,
      "learning_rate": 5.867509333620231e-05,
      "loss": 0.3342,
      "step": 1428
    },
    {
      "epoch": 2.0561151079136692,
      "grad_norm": 0.13984388418344307,
      "learning_rate": 5.863954596044744e-05,
      "loss": 0.3361,
      "step": 1429
    },
    {
      "epoch": 2.0575539568345325,
      "grad_norm": 0.11629574383383963,
      "learning_rate": 5.8603979770773344e-05,
      "loss": 0.3414,
      "step": 1430
    },
    {
      "epoch": 2.0589928057553957,
      "grad_norm": 0.13495699245456752,
      "learning_rate": 5.85683948030789e-05,
      "loss": 0.3406,
      "step": 1431
    },
    {
      "epoch": 2.060431654676259,
      "grad_norm": 0.13148052384917566,
      "learning_rate": 5.8532791093282e-05,
      "loss": 0.3357,
      "step": 1432
    },
    {
      "epoch": 2.061870503597122,
      "grad_norm": 0.14840204064521428,
      "learning_rate": 5.849716867731941e-05,
      "loss": 0.3416,
      "step": 1433
    },
    {
      "epoch": 2.0633093525179858,
      "grad_norm": 0.11713407890189519,
      "learning_rate": 5.84615275911468e-05,
      "loss": 0.3373,
      "step": 1434
    },
    {
      "epoch": 2.064748201438849,
      "grad_norm": 0.11695249228680646,
      "learning_rate": 5.8425867870738684e-05,
      "loss": 0.3387,
      "step": 1435
    },
    {
      "epoch": 2.066187050359712,
      "grad_norm": 0.12771762431703068,
      "learning_rate": 5.839018955208838e-05,
      "loss": 0.3404,
      "step": 1436
    },
    {
      "epoch": 2.0676258992805754,
      "grad_norm": 0.14190392315489428,
      "learning_rate": 5.835449267120796e-05,
      "loss": 0.3405,
      "step": 1437
    },
    {
      "epoch": 2.069064748201439,
      "grad_norm": 0.1314280361275215,
      "learning_rate": 5.831877726412827e-05,
      "loss": 0.3374,
      "step": 1438
    },
    {
      "epoch": 2.0705035971223023,
      "grad_norm": 0.11923833830476203,
      "learning_rate": 5.828304336689883e-05,
      "loss": 0.3419,
      "step": 1439
    },
    {
      "epoch": 2.0719424460431655,
      "grad_norm": 0.15719810095823256,
      "learning_rate": 5.824729101558781e-05,
      "loss": 0.3419,
      "step": 1440
    },
    {
      "epoch": 2.0733812949640287,
      "grad_norm": 0.1448222991005709,
      "learning_rate": 5.821152024628207e-05,
      "loss": 0.3338,
      "step": 1441
    },
    {
      "epoch": 2.074820143884892,
      "grad_norm": 0.16522879889055553,
      "learning_rate": 5.8175731095086974e-05,
      "loss": 0.3399,
      "step": 1442
    },
    {
      "epoch": 2.0762589928057555,
      "grad_norm": 0.21170071164270177,
      "learning_rate": 5.813992359812649e-05,
      "loss": 0.3433,
      "step": 1443
    },
    {
      "epoch": 2.0776978417266188,
      "grad_norm": 0.2202554706035317,
      "learning_rate": 5.8104097791543104e-05,
      "loss": 0.3376,
      "step": 1444
    },
    {
      "epoch": 2.079136690647482,
      "grad_norm": 0.17559261793029335,
      "learning_rate": 5.806825371149778e-05,
      "loss": 0.342,
      "step": 1445
    },
    {
      "epoch": 2.080575539568345,
      "grad_norm": 0.16582794991328403,
      "learning_rate": 5.803239139416989e-05,
      "loss": 0.3377,
      "step": 1446
    },
    {
      "epoch": 2.082014388489209,
      "grad_norm": 0.16252281453261838,
      "learning_rate": 5.799651087575728e-05,
      "loss": 0.3348,
      "step": 1447
    },
    {
      "epoch": 2.083453237410072,
      "grad_norm": 0.14014857137408293,
      "learning_rate": 5.7960612192476096e-05,
      "loss": 0.3362,
      "step": 1448
    },
    {
      "epoch": 2.0848920863309353,
      "grad_norm": 0.16596189212725054,
      "learning_rate": 5.792469538056089e-05,
      "loss": 0.3368,
      "step": 1449
    },
    {
      "epoch": 2.0863309352517985,
      "grad_norm": 0.23794659327159862,
      "learning_rate": 5.7888760476264445e-05,
      "loss": 0.3308,
      "step": 1450
    },
    {
      "epoch": 2.0877697841726617,
      "grad_norm": 0.2519812743121408,
      "learning_rate": 5.785280751585785e-05,
      "loss": 0.3453,
      "step": 1451
    },
    {
      "epoch": 2.0892086330935253,
      "grad_norm": 0.24031404589039398,
      "learning_rate": 5.7816836535630436e-05,
      "loss": 0.3387,
      "step": 1452
    },
    {
      "epoch": 2.0906474820143885,
      "grad_norm": 0.18973703655841345,
      "learning_rate": 5.7780847571889625e-05,
      "loss": 0.3378,
      "step": 1453
    },
    {
      "epoch": 2.0920863309352518,
      "grad_norm": 0.1682276408188406,
      "learning_rate": 5.7744840660961126e-05,
      "loss": 0.3399,
      "step": 1454
    },
    {
      "epoch": 2.093525179856115,
      "grad_norm": 0.17641722119061082,
      "learning_rate": 5.770881583918865e-05,
      "loss": 0.3394,
      "step": 1455
    },
    {
      "epoch": 2.0949640287769786,
      "grad_norm": 0.17089632987102674,
      "learning_rate": 5.767277314293404e-05,
      "loss": 0.3485,
      "step": 1456
    },
    {
      "epoch": 2.096402877697842,
      "grad_norm": 0.13352662398274492,
      "learning_rate": 5.76367126085772e-05,
      "loss": 0.3479,
      "step": 1457
    },
    {
      "epoch": 2.097841726618705,
      "grad_norm": 0.16704433169505478,
      "learning_rate": 5.760063427251599e-05,
      "loss": 0.3459,
      "step": 1458
    },
    {
      "epoch": 2.0992805755395683,
      "grad_norm": 0.19259136865967705,
      "learning_rate": 5.756453817116624e-05,
      "loss": 0.3413,
      "step": 1459
    },
    {
      "epoch": 2.1007194244604315,
      "grad_norm": 0.17415900235040346,
      "learning_rate": 5.752842434096176e-05,
      "loss": 0.3426,
      "step": 1460
    },
    {
      "epoch": 2.102158273381295,
      "grad_norm": 0.12734265587328836,
      "learning_rate": 5.7492292818354224e-05,
      "loss": 0.3491,
      "step": 1461
    },
    {
      "epoch": 2.1035971223021583,
      "grad_norm": 0.11675352615016288,
      "learning_rate": 5.745614363981316e-05,
      "loss": 0.338,
      "step": 1462
    },
    {
      "epoch": 2.1050359712230216,
      "grad_norm": 0.12091941181674576,
      "learning_rate": 5.741997684182591e-05,
      "loss": 0.3345,
      "step": 1463
    },
    {
      "epoch": 2.1064748201438848,
      "grad_norm": 0.14343425734938844,
      "learning_rate": 5.7383792460897626e-05,
      "loss": 0.3372,
      "step": 1464
    },
    {
      "epoch": 2.1079136690647484,
      "grad_norm": 0.15837441419013518,
      "learning_rate": 5.73475905335512e-05,
      "loss": 0.3372,
      "step": 1465
    },
    {
      "epoch": 2.1093525179856116,
      "grad_norm": 0.1703498771793665,
      "learning_rate": 5.731137109632722e-05,
      "loss": 0.3454,
      "step": 1466
    },
    {
      "epoch": 2.110791366906475,
      "grad_norm": 0.14049879617314373,
      "learning_rate": 5.727513418578397e-05,
      "loss": 0.3368,
      "step": 1467
    },
    {
      "epoch": 2.112230215827338,
      "grad_norm": 0.1500152804042965,
      "learning_rate": 5.723887983849732e-05,
      "loss": 0.3453,
      "step": 1468
    },
    {
      "epoch": 2.1136690647482013,
      "grad_norm": 0.14449565334649642,
      "learning_rate": 5.720260809106083e-05,
      "loss": 0.3436,
      "step": 1469
    },
    {
      "epoch": 2.115107913669065,
      "grad_norm": 0.1756781174580756,
      "learning_rate": 5.716631898008553e-05,
      "loss": 0.3467,
      "step": 1470
    },
    {
      "epoch": 2.116546762589928,
      "grad_norm": 0.18614967731122278,
      "learning_rate": 5.713001254220002e-05,
      "loss": 0.3351,
      "step": 1471
    },
    {
      "epoch": 2.1179856115107913,
      "grad_norm": 0.14261200052126147,
      "learning_rate": 5.7093688814050425e-05,
      "loss": 0.3332,
      "step": 1472
    },
    {
      "epoch": 2.1194244604316546,
      "grad_norm": 0.1675466491240586,
      "learning_rate": 5.705734783230022e-05,
      "loss": 0.3352,
      "step": 1473
    },
    {
      "epoch": 2.1208633093525178,
      "grad_norm": 0.19342404719601078,
      "learning_rate": 5.7020989633630414e-05,
      "loss": 0.3448,
      "step": 1474
    },
    {
      "epoch": 2.1223021582733814,
      "grad_norm": 0.1449713706015703,
      "learning_rate": 5.6984614254739306e-05,
      "loss": 0.3419,
      "step": 1475
    },
    {
      "epoch": 2.1237410071942446,
      "grad_norm": 0.13417399972592065,
      "learning_rate": 5.694822173234257e-05,
      "loss": 0.3365,
      "step": 1476
    },
    {
      "epoch": 2.125179856115108,
      "grad_norm": 0.14800438024460638,
      "learning_rate": 5.691181210317319e-05,
      "loss": 0.3307,
      "step": 1477
    },
    {
      "epoch": 2.126618705035971,
      "grad_norm": 0.1764029381697298,
      "learning_rate": 5.687538540398141e-05,
      "loss": 0.3371,
      "step": 1478
    },
    {
      "epoch": 2.1280575539568347,
      "grad_norm": 0.1831382465599354,
      "learning_rate": 5.683894167153468e-05,
      "loss": 0.3428,
      "step": 1479
    },
    {
      "epoch": 2.129496402877698,
      "grad_norm": 0.1768420571871011,
      "learning_rate": 5.680248094261769e-05,
      "loss": 0.3521,
      "step": 1480
    },
    {
      "epoch": 2.130935251798561,
      "grad_norm": 0.21501595835453444,
      "learning_rate": 5.676600325403224e-05,
      "loss": 0.334,
      "step": 1481
    },
    {
      "epoch": 2.1323741007194243,
      "grad_norm": 0.18877105575339637,
      "learning_rate": 5.672950864259729e-05,
      "loss": 0.338,
      "step": 1482
    },
    {
      "epoch": 2.133812949640288,
      "grad_norm": 0.1294937427982825,
      "learning_rate": 5.669299714514884e-05,
      "loss": 0.3422,
      "step": 1483
    },
    {
      "epoch": 2.135251798561151,
      "grad_norm": 0.1273382842813177,
      "learning_rate": 5.665646879853995e-05,
      "loss": 0.336,
      "step": 1484
    },
    {
      "epoch": 2.1366906474820144,
      "grad_norm": 0.13317103255146273,
      "learning_rate": 5.661992363964072e-05,
      "loss": 0.3405,
      "step": 1485
    },
    {
      "epoch": 2.1381294964028776,
      "grad_norm": 0.14529342520812324,
      "learning_rate": 5.658336170533814e-05,
      "loss": 0.3453,
      "step": 1486
    },
    {
      "epoch": 2.139568345323741,
      "grad_norm": 0.1597014572285992,
      "learning_rate": 5.654678303253624e-05,
      "loss": 0.3414,
      "step": 1487
    },
    {
      "epoch": 2.1410071942446045,
      "grad_norm": 0.1304551078464195,
      "learning_rate": 5.6510187658155846e-05,
      "loss": 0.3551,
      "step": 1488
    },
    {
      "epoch": 2.1424460431654677,
      "grad_norm": 0.11581885124410166,
      "learning_rate": 5.6473575619134686e-05,
      "loss": 0.3391,
      "step": 1489
    },
    {
      "epoch": 2.143884892086331,
      "grad_norm": 0.13725711766965948,
      "learning_rate": 5.643694695242731e-05,
      "loss": 0.338,
      "step": 1490
    },
    {
      "epoch": 2.145323741007194,
      "grad_norm": 0.1451690787664973,
      "learning_rate": 5.640030169500508e-05,
      "loss": 0.3314,
      "step": 1491
    },
    {
      "epoch": 2.1467625899280574,
      "grad_norm": 0.14030207891161744,
      "learning_rate": 5.636363988385601e-05,
      "loss": 0.343,
      "step": 1492
    },
    {
      "epoch": 2.148201438848921,
      "grad_norm": 0.1610816781622484,
      "learning_rate": 5.632696155598493e-05,
      "loss": 0.3502,
      "step": 1493
    },
    {
      "epoch": 2.149640287769784,
      "grad_norm": 0.12540923093393241,
      "learning_rate": 5.6290266748413266e-05,
      "loss": 0.3304,
      "step": 1494
    },
    {
      "epoch": 2.1510791366906474,
      "grad_norm": 0.13418605248153795,
      "learning_rate": 5.6253555498179124e-05,
      "loss": 0.3363,
      "step": 1495
    },
    {
      "epoch": 2.1525179856115106,
      "grad_norm": 0.14709879762829714,
      "learning_rate": 5.621682784233718e-05,
      "loss": 0.3336,
      "step": 1496
    },
    {
      "epoch": 2.1539568345323743,
      "grad_norm": 0.1563659130071038,
      "learning_rate": 5.618008381795868e-05,
      "loss": 0.3414,
      "step": 1497
    },
    {
      "epoch": 2.1553956834532375,
      "grad_norm": 0.18468733308318555,
      "learning_rate": 5.61433234621314e-05,
      "loss": 0.3375,
      "step": 1498
    },
    {
      "epoch": 2.1568345323741007,
      "grad_norm": 0.18787953455404294,
      "learning_rate": 5.610654681195957e-05,
      "loss": 0.3363,
      "step": 1499
    },
    {
      "epoch": 2.158273381294964,
      "grad_norm": 0.13890337995163107,
      "learning_rate": 5.606975390456391e-05,
      "loss": 0.337,
      "step": 1500
    },
    {
      "epoch": 2.159712230215827,
      "grad_norm": 0.14644128624879602,
      "learning_rate": 5.603294477708149e-05,
      "loss": 0.338,
      "step": 1501
    },
    {
      "epoch": 2.161151079136691,
      "grad_norm": 0.1501386328488287,
      "learning_rate": 5.599611946666581e-05,
      "loss": 0.3409,
      "step": 1502
    },
    {
      "epoch": 2.162589928057554,
      "grad_norm": 0.17246425668599588,
      "learning_rate": 5.595927801048669e-05,
      "loss": 0.3377,
      "step": 1503
    },
    {
      "epoch": 2.1640287769784172,
      "grad_norm": 0.2019670162251421,
      "learning_rate": 5.5922420445730245e-05,
      "loss": 0.3446,
      "step": 1504
    },
    {
      "epoch": 2.1654676258992804,
      "grad_norm": 0.17811936053004404,
      "learning_rate": 5.5885546809598805e-05,
      "loss": 0.3351,
      "step": 1505
    },
    {
      "epoch": 2.166906474820144,
      "grad_norm": 0.10886271085349164,
      "learning_rate": 5.584865713931098e-05,
      "loss": 0.3353,
      "step": 1506
    },
    {
      "epoch": 2.1683453237410073,
      "grad_norm": 0.16795292903684267,
      "learning_rate": 5.5811751472101564e-05,
      "loss": 0.3371,
      "step": 1507
    },
    {
      "epoch": 2.1697841726618705,
      "grad_norm": 0.1863404289138655,
      "learning_rate": 5.577482984522145e-05,
      "loss": 0.339,
      "step": 1508
    },
    {
      "epoch": 2.1712230215827337,
      "grad_norm": 0.183043593865869,
      "learning_rate": 5.573789229593767e-05,
      "loss": 0.337,
      "step": 1509
    },
    {
      "epoch": 2.172661870503597,
      "grad_norm": 0.17483783244947432,
      "learning_rate": 5.570093886153334e-05,
      "loss": 0.3486,
      "step": 1510
    },
    {
      "epoch": 2.1741007194244606,
      "grad_norm": 0.15053536278422408,
      "learning_rate": 5.5663969579307594e-05,
      "loss": 0.3358,
      "step": 1511
    },
    {
      "epoch": 2.175539568345324,
      "grad_norm": 0.1270283844779313,
      "learning_rate": 5.562698448657553e-05,
      "loss": 0.3512,
      "step": 1512
    },
    {
      "epoch": 2.176978417266187,
      "grad_norm": 0.13060290196603938,
      "learning_rate": 5.5589983620668286e-05,
      "loss": 0.3431,
      "step": 1513
    },
    {
      "epoch": 2.1784172661870502,
      "grad_norm": 0.13973132677886133,
      "learning_rate": 5.555296701893284e-05,
      "loss": 0.333,
      "step": 1514
    },
    {
      "epoch": 2.1798561151079134,
      "grad_norm": 0.12157919806508072,
      "learning_rate": 5.551593471873208e-05,
      "loss": 0.3269,
      "step": 1515
    },
    {
      "epoch": 2.181294964028777,
      "grad_norm": 0.15709986759465888,
      "learning_rate": 5.547888675744476e-05,
      "loss": 0.3472,
      "step": 1516
    },
    {
      "epoch": 2.1827338129496403,
      "grad_norm": 0.1397129088825444,
      "learning_rate": 5.5441823172465427e-05,
      "loss": 0.3429,
      "step": 1517
    },
    {
      "epoch": 2.1841726618705035,
      "grad_norm": 0.13661235608902214,
      "learning_rate": 5.540474400120438e-05,
      "loss": 0.3352,
      "step": 1518
    },
    {
      "epoch": 2.1856115107913667,
      "grad_norm": 0.15779330501808156,
      "learning_rate": 5.536764928108769e-05,
      "loss": 0.3414,
      "step": 1519
    },
    {
      "epoch": 2.1870503597122304,
      "grad_norm": 0.16129758993795465,
      "learning_rate": 5.533053904955709e-05,
      "loss": 0.3362,
      "step": 1520
    },
    {
      "epoch": 2.1884892086330936,
      "grad_norm": 0.15720442376879168,
      "learning_rate": 5.5293413344069964e-05,
      "loss": 0.3485,
      "step": 1521
    },
    {
      "epoch": 2.189928057553957,
      "grad_norm": 0.15507626372202452,
      "learning_rate": 5.525627220209934e-05,
      "loss": 0.3374,
      "step": 1522
    },
    {
      "epoch": 2.19136690647482,
      "grad_norm": 0.16565853659345364,
      "learning_rate": 5.5219115661133815e-05,
      "loss": 0.3431,
      "step": 1523
    },
    {
      "epoch": 2.1928057553956837,
      "grad_norm": 0.2042792580144636,
      "learning_rate": 5.518194375867754e-05,
      "loss": 0.346,
      "step": 1524
    },
    {
      "epoch": 2.194244604316547,
      "grad_norm": 0.19753686236847903,
      "learning_rate": 5.514475653225014e-05,
      "loss": 0.3425,
      "step": 1525
    },
    {
      "epoch": 2.19568345323741,
      "grad_norm": 0.17338936644955222,
      "learning_rate": 5.510755401938676e-05,
      "loss": 0.3442,
      "step": 1526
    },
    {
      "epoch": 2.1971223021582733,
      "grad_norm": 0.14526568967476972,
      "learning_rate": 5.5070336257637904e-05,
      "loss": 0.3332,
      "step": 1527
    },
    {
      "epoch": 2.1985611510791365,
      "grad_norm": 0.16128188345297867,
      "learning_rate": 5.503310328456953e-05,
      "loss": 0.34,
      "step": 1528
    },
    {
      "epoch": 2.2,
      "grad_norm": 0.17205429431240457,
      "learning_rate": 5.4995855137762926e-05,
      "loss": 0.3517,
      "step": 1529
    },
    {
      "epoch": 2.2014388489208634,
      "grad_norm": 0.17768080599054073,
      "learning_rate": 5.4958591854814695e-05,
      "loss": 0.349,
      "step": 1530
    },
    {
      "epoch": 2.2028776978417266,
      "grad_norm": 0.20441850917447524,
      "learning_rate": 5.492131347333671e-05,
      "loss": 0.3395,
      "step": 1531
    },
    {
      "epoch": 2.20431654676259,
      "grad_norm": 0.17639121611676273,
      "learning_rate": 5.48840200309561e-05,
      "loss": 0.3366,
      "step": 1532
    },
    {
      "epoch": 2.205755395683453,
      "grad_norm": 0.1167227355953685,
      "learning_rate": 5.484671156531519e-05,
      "loss": 0.3297,
      "step": 1533
    },
    {
      "epoch": 2.2071942446043167,
      "grad_norm": 0.13602920961715934,
      "learning_rate": 5.480938811407146e-05,
      "loss": 0.3415,
      "step": 1534
    },
    {
      "epoch": 2.20863309352518,
      "grad_norm": 0.15865350852770105,
      "learning_rate": 5.477204971489753e-05,
      "loss": 0.3408,
      "step": 1535
    },
    {
      "epoch": 2.210071942446043,
      "grad_norm": 0.1784279601859331,
      "learning_rate": 5.473469640548109e-05,
      "loss": 0.3444,
      "step": 1536
    },
    {
      "epoch": 2.2115107913669063,
      "grad_norm": 0.15677594496706615,
      "learning_rate": 5.469732822352491e-05,
      "loss": 0.34,
      "step": 1537
    },
    {
      "epoch": 2.21294964028777,
      "grad_norm": 0.1558089174768745,
      "learning_rate": 5.465994520674672e-05,
      "loss": 0.3323,
      "step": 1538
    },
    {
      "epoch": 2.214388489208633,
      "grad_norm": 0.17636357175020703,
      "learning_rate": 5.4622547392879295e-05,
      "loss": 0.3433,
      "step": 1539
    },
    {
      "epoch": 2.2158273381294964,
      "grad_norm": 0.15555718344338995,
      "learning_rate": 5.458513481967027e-05,
      "loss": 0.3448,
      "step": 1540
    },
    {
      "epoch": 2.2172661870503596,
      "grad_norm": 0.18393076876807044,
      "learning_rate": 5.454770752488223e-05,
      "loss": 0.3291,
      "step": 1541
    },
    {
      "epoch": 2.218705035971223,
      "grad_norm": 0.13756871917599844,
      "learning_rate": 5.4510265546292615e-05,
      "loss": 0.3444,
      "step": 1542
    },
    {
      "epoch": 2.2201438848920865,
      "grad_norm": 0.14744324982539372,
      "learning_rate": 5.4472808921693657e-05,
      "loss": 0.3414,
      "step": 1543
    },
    {
      "epoch": 2.2215827338129497,
      "grad_norm": 0.17324681650300475,
      "learning_rate": 5.4435337688892396e-05,
      "loss": 0.3348,
      "step": 1544
    },
    {
      "epoch": 2.223021582733813,
      "grad_norm": 0.21017507418299664,
      "learning_rate": 5.4397851885710595e-05,
      "loss": 0.3476,
      "step": 1545
    },
    {
      "epoch": 2.224460431654676,
      "grad_norm": 0.18560881927511783,
      "learning_rate": 5.4360351549984755e-05,
      "loss": 0.3395,
      "step": 1546
    },
    {
      "epoch": 2.2258992805755398,
      "grad_norm": 0.16057845613316232,
      "learning_rate": 5.432283671956601e-05,
      "loss": 0.3374,
      "step": 1547
    },
    {
      "epoch": 2.227338129496403,
      "grad_norm": 0.17400221956971498,
      "learning_rate": 5.428530743232016e-05,
      "loss": 0.3464,
      "step": 1548
    },
    {
      "epoch": 2.228776978417266,
      "grad_norm": 0.13688750875923683,
      "learning_rate": 5.4247763726127564e-05,
      "loss": 0.3412,
      "step": 1549
    },
    {
      "epoch": 2.2302158273381294,
      "grad_norm": 0.16895566858175826,
      "learning_rate": 5.421020563888317e-05,
      "loss": 0.3389,
      "step": 1550
    },
    {
      "epoch": 2.2316546762589926,
      "grad_norm": 0.16049543492332127,
      "learning_rate": 5.417263320849641e-05,
      "loss": 0.3302,
      "step": 1551
    },
    {
      "epoch": 2.2330935251798563,
      "grad_norm": 0.1529656726630707,
      "learning_rate": 5.4135046472891205e-05,
      "loss": 0.3388,
      "step": 1552
    },
    {
      "epoch": 2.2345323741007195,
      "grad_norm": 0.1661012439318489,
      "learning_rate": 5.409744547000591e-05,
      "loss": 0.3409,
      "step": 1553
    },
    {
      "epoch": 2.2359712230215827,
      "grad_norm": 0.1523798399883737,
      "learning_rate": 5.405983023779328e-05,
      "loss": 0.3364,
      "step": 1554
    },
    {
      "epoch": 2.237410071942446,
      "grad_norm": 0.14102246718693223,
      "learning_rate": 5.402220081422048e-05,
      "loss": 0.3408,
      "step": 1555
    },
    {
      "epoch": 2.2388489208633096,
      "grad_norm": 0.11049551216075727,
      "learning_rate": 5.3984557237268905e-05,
      "loss": 0.3433,
      "step": 1556
    },
    {
      "epoch": 2.2402877697841728,
      "grad_norm": 0.1399944478206375,
      "learning_rate": 5.394689954493432e-05,
      "loss": 0.3488,
      "step": 1557
    },
    {
      "epoch": 2.241726618705036,
      "grad_norm": 0.1479515174519836,
      "learning_rate": 5.390922777522669e-05,
      "loss": 0.3387,
      "step": 1558
    },
    {
      "epoch": 2.243165467625899,
      "grad_norm": 0.12763995560575422,
      "learning_rate": 5.3871541966170225e-05,
      "loss": 0.3446,
      "step": 1559
    },
    {
      "epoch": 2.2446043165467624,
      "grad_norm": 0.1309388399200539,
      "learning_rate": 5.383384215580326e-05,
      "loss": 0.3385,
      "step": 1560
    },
    {
      "epoch": 2.246043165467626,
      "grad_norm": 0.13535157925392624,
      "learning_rate": 5.37961283821783e-05,
      "loss": 0.3472,
      "step": 1561
    },
    {
      "epoch": 2.2474820143884893,
      "grad_norm": 0.14589245872554155,
      "learning_rate": 5.3758400683361926e-05,
      "loss": 0.3355,
      "step": 1562
    },
    {
      "epoch": 2.2489208633093525,
      "grad_norm": 0.14696443357983832,
      "learning_rate": 5.372065909743479e-05,
      "loss": 0.3456,
      "step": 1563
    },
    {
      "epoch": 2.2503597122302157,
      "grad_norm": 0.14858801303380195,
      "learning_rate": 5.368290366249155e-05,
      "loss": 0.3471,
      "step": 1564
    },
    {
      "epoch": 2.2517985611510793,
      "grad_norm": 0.12730617646963155,
      "learning_rate": 5.364513441664084e-05,
      "loss": 0.3374,
      "step": 1565
    },
    {
      "epoch": 2.2532374100719426,
      "grad_norm": 0.12323182979257986,
      "learning_rate": 5.3607351398005234e-05,
      "loss": 0.3381,
      "step": 1566
    },
    {
      "epoch": 2.2546762589928058,
      "grad_norm": 0.14275410896069196,
      "learning_rate": 5.356955464472121e-05,
      "loss": 0.335,
      "step": 1567
    },
    {
      "epoch": 2.256115107913669,
      "grad_norm": 0.16897913426284591,
      "learning_rate": 5.353174419493913e-05,
      "loss": 0.3375,
      "step": 1568
    },
    {
      "epoch": 2.257553956834532,
      "grad_norm": 0.1779313024622998,
      "learning_rate": 5.349392008682314e-05,
      "loss": 0.3467,
      "step": 1569
    },
    {
      "epoch": 2.258992805755396,
      "grad_norm": 0.14193982348022544,
      "learning_rate": 5.3456082358551204e-05,
      "loss": 0.335,
      "step": 1570
    },
    {
      "epoch": 2.260431654676259,
      "grad_norm": 0.12525506601160286,
      "learning_rate": 5.341823104831501e-05,
      "loss": 0.3398,
      "step": 1571
    },
    {
      "epoch": 2.2618705035971223,
      "grad_norm": 0.13843939017177903,
      "learning_rate": 5.338036619431999e-05,
      "loss": 0.3433,
      "step": 1572
    },
    {
      "epoch": 2.2633093525179855,
      "grad_norm": 0.13249616791552016,
      "learning_rate": 5.33424878347852e-05,
      "loss": 0.3345,
      "step": 1573
    },
    {
      "epoch": 2.2647482014388487,
      "grad_norm": 0.1681175570097203,
      "learning_rate": 5.330459600794337e-05,
      "loss": 0.3437,
      "step": 1574
    },
    {
      "epoch": 2.2661870503597124,
      "grad_norm": 0.13046711251361295,
      "learning_rate": 5.32666907520408e-05,
      "loss": 0.3388,
      "step": 1575
    },
    {
      "epoch": 2.2676258992805756,
      "grad_norm": 0.1294688814318854,
      "learning_rate": 5.322877210533735e-05,
      "loss": 0.3376,
      "step": 1576
    },
    {
      "epoch": 2.2690647482014388,
      "grad_norm": 0.1689404345986412,
      "learning_rate": 5.319084010610638e-05,
      "loss": 0.3402,
      "step": 1577
    },
    {
      "epoch": 2.270503597122302,
      "grad_norm": 0.13531193824777604,
      "learning_rate": 5.3152894792634785e-05,
      "loss": 0.3456,
      "step": 1578
    },
    {
      "epoch": 2.2719424460431656,
      "grad_norm": 0.13641412543988995,
      "learning_rate": 5.311493620322282e-05,
      "loss": 0.3417,
      "step": 1579
    },
    {
      "epoch": 2.273381294964029,
      "grad_norm": 0.14410478214500225,
      "learning_rate": 5.3076964376184186e-05,
      "loss": 0.3364,
      "step": 1580
    },
    {
      "epoch": 2.274820143884892,
      "grad_norm": 0.17981099543469098,
      "learning_rate": 5.303897934984595e-05,
      "loss": 0.3413,
      "step": 1581
    },
    {
      "epoch": 2.2762589928057553,
      "grad_norm": 0.1310151076813013,
      "learning_rate": 5.300098116254848e-05,
      "loss": 0.3358,
      "step": 1582
    },
    {
      "epoch": 2.277697841726619,
      "grad_norm": 0.1397793816458818,
      "learning_rate": 5.296296985264543e-05,
      "loss": 0.3412,
      "step": 1583
    },
    {
      "epoch": 2.279136690647482,
      "grad_norm": 0.14915863726588208,
      "learning_rate": 5.2924945458503713e-05,
      "loss": 0.3397,
      "step": 1584
    },
    {
      "epoch": 2.2805755395683454,
      "grad_norm": 0.10674266202915253,
      "learning_rate": 5.2886908018503454e-05,
      "loss": 0.3454,
      "step": 1585
    },
    {
      "epoch": 2.2820143884892086,
      "grad_norm": 0.12978752685607617,
      "learning_rate": 5.284885757103792e-05,
      "loss": 0.3509,
      "step": 1586
    },
    {
      "epoch": 2.283453237410072,
      "grad_norm": 0.15617892058617083,
      "learning_rate": 5.2810794154513503e-05,
      "loss": 0.3425,
      "step": 1587
    },
    {
      "epoch": 2.2848920863309354,
      "grad_norm": 0.1735752896636099,
      "learning_rate": 5.277271780734975e-05,
      "loss": 0.3362,
      "step": 1588
    },
    {
      "epoch": 2.2863309352517986,
      "grad_norm": 0.15900631594222825,
      "learning_rate": 5.273462856797918e-05,
      "loss": 0.3442,
      "step": 1589
    },
    {
      "epoch": 2.287769784172662,
      "grad_norm": 0.11309688348190682,
      "learning_rate": 5.269652647484735e-05,
      "loss": 0.3391,
      "step": 1590
    },
    {
      "epoch": 2.289208633093525,
      "grad_norm": 0.14290407695721685,
      "learning_rate": 5.2658411566412837e-05,
      "loss": 0.3346,
      "step": 1591
    },
    {
      "epoch": 2.2906474820143883,
      "grad_norm": 0.15093684348741013,
      "learning_rate": 5.262028388114708e-05,
      "loss": 0.3402,
      "step": 1592
    },
    {
      "epoch": 2.292086330935252,
      "grad_norm": 0.15631978580232964,
      "learning_rate": 5.258214345753446e-05,
      "loss": 0.3439,
      "step": 1593
    },
    {
      "epoch": 2.293525179856115,
      "grad_norm": 0.12869733397073002,
      "learning_rate": 5.254399033407221e-05,
      "loss": 0.3424,
      "step": 1594
    },
    {
      "epoch": 2.2949640287769784,
      "grad_norm": 0.11517532335773004,
      "learning_rate": 5.250582454927037e-05,
      "loss": 0.3403,
      "step": 1595
    },
    {
      "epoch": 2.2964028776978416,
      "grad_norm": 0.1340681459546274,
      "learning_rate": 5.2467646141651764e-05,
      "loss": 0.3377,
      "step": 1596
    },
    {
      "epoch": 2.2978417266187052,
      "grad_norm": 0.12046792802010879,
      "learning_rate": 5.2429455149751976e-05,
      "loss": 0.3438,
      "step": 1597
    },
    {
      "epoch": 2.2992805755395684,
      "grad_norm": 0.13079680094789867,
      "learning_rate": 5.2391251612119256e-05,
      "loss": 0.3452,
      "step": 1598
    },
    {
      "epoch": 2.3007194244604317,
      "grad_norm": 0.16640190235613248,
      "learning_rate": 5.235303556731456e-05,
      "loss": 0.3461,
      "step": 1599
    },
    {
      "epoch": 2.302158273381295,
      "grad_norm": 0.1577686684215034,
      "learning_rate": 5.23148070539114e-05,
      "loss": 0.341,
      "step": 1600
    },
    {
      "epoch": 2.3035971223021585,
      "grad_norm": 0.14925437585718365,
      "learning_rate": 5.227656611049598e-05,
      "loss": 0.3387,
      "step": 1601
    },
    {
      "epoch": 2.3050359712230217,
      "grad_norm": 0.13710986596933333,
      "learning_rate": 5.2238312775666935e-05,
      "loss": 0.3368,
      "step": 1602
    },
    {
      "epoch": 2.306474820143885,
      "grad_norm": 0.1500167813711351,
      "learning_rate": 5.220004708803548e-05,
      "loss": 0.3375,
      "step": 1603
    },
    {
      "epoch": 2.307913669064748,
      "grad_norm": 0.1566653927707589,
      "learning_rate": 5.216176908622528e-05,
      "loss": 0.3416,
      "step": 1604
    },
    {
      "epoch": 2.3093525179856114,
      "grad_norm": 0.13490532368742505,
      "learning_rate": 5.2123478808872436e-05,
      "loss": 0.332,
      "step": 1605
    },
    {
      "epoch": 2.310791366906475,
      "grad_norm": 0.14186561037817463,
      "learning_rate": 5.208517629462541e-05,
      "loss": 0.338,
      "step": 1606
    },
    {
      "epoch": 2.3122302158273382,
      "grad_norm": 0.1738923611376932,
      "learning_rate": 5.204686158214507e-05,
      "loss": 0.3347,
      "step": 1607
    },
    {
      "epoch": 2.3136690647482014,
      "grad_norm": 0.1643249308537781,
      "learning_rate": 5.200853471010453e-05,
      "loss": 0.3408,
      "step": 1608
    },
    {
      "epoch": 2.3151079136690647,
      "grad_norm": 0.11412779665907977,
      "learning_rate": 5.197019571718921e-05,
      "loss": 0.3399,
      "step": 1609
    },
    {
      "epoch": 2.316546762589928,
      "grad_norm": 0.12569340024427234,
      "learning_rate": 5.19318446420968e-05,
      "loss": 0.3453,
      "step": 1610
    },
    {
      "epoch": 2.3179856115107915,
      "grad_norm": 0.11776087089400157,
      "learning_rate": 5.189348152353712e-05,
      "loss": 0.338,
      "step": 1611
    },
    {
      "epoch": 2.3194244604316547,
      "grad_norm": 0.13584830504004294,
      "learning_rate": 5.1855106400232196e-05,
      "loss": 0.3444,
      "step": 1612
    },
    {
      "epoch": 2.320863309352518,
      "grad_norm": 0.12917173409302307,
      "learning_rate": 5.181671931091612e-05,
      "loss": 0.3411,
      "step": 1613
    },
    {
      "epoch": 2.322302158273381,
      "grad_norm": 0.10729175450876592,
      "learning_rate": 5.1778320294335126e-05,
      "loss": 0.3499,
      "step": 1614
    },
    {
      "epoch": 2.3237410071942444,
      "grad_norm": 0.11881215722162419,
      "learning_rate": 5.1739909389247445e-05,
      "loss": 0.33,
      "step": 1615
    },
    {
      "epoch": 2.325179856115108,
      "grad_norm": 0.13059841815215834,
      "learning_rate": 5.17014866344233e-05,
      "loss": 0.341,
      "step": 1616
    },
    {
      "epoch": 2.3266187050359712,
      "grad_norm": 0.12307382947679618,
      "learning_rate": 5.166305206864492e-05,
      "loss": 0.337,
      "step": 1617
    },
    {
      "epoch": 2.3280575539568344,
      "grad_norm": 0.13075069201133485,
      "learning_rate": 5.162460573070642e-05,
      "loss": 0.342,
      "step": 1618
    },
    {
      "epoch": 2.3294964028776977,
      "grad_norm": 0.14771166131358693,
      "learning_rate": 5.158614765941376e-05,
      "loss": 0.3317,
      "step": 1619
    },
    {
      "epoch": 2.3309352517985613,
      "grad_norm": 0.1507808153078834,
      "learning_rate": 5.1547677893584846e-05,
      "loss": 0.343,
      "step": 1620
    },
    {
      "epoch": 2.3323741007194245,
      "grad_norm": 0.153815934182145,
      "learning_rate": 5.15091964720493e-05,
      "loss": 0.341,
      "step": 1621
    },
    {
      "epoch": 2.3338129496402877,
      "grad_norm": 0.09981420149445622,
      "learning_rate": 5.1470703433648556e-05,
      "loss": 0.3342,
      "step": 1622
    },
    {
      "epoch": 2.335251798561151,
      "grad_norm": 0.12819863045020796,
      "learning_rate": 5.143219881723573e-05,
      "loss": 0.3463,
      "step": 1623
    },
    {
      "epoch": 2.3366906474820146,
      "grad_norm": 0.14046083524838301,
      "learning_rate": 5.139368266167567e-05,
      "loss": 0.3408,
      "step": 1624
    },
    {
      "epoch": 2.338129496402878,
      "grad_norm": 0.12589480843570602,
      "learning_rate": 5.135515500584484e-05,
      "loss": 0.3397,
      "step": 1625
    },
    {
      "epoch": 2.339568345323741,
      "grad_norm": 0.1426606838423531,
      "learning_rate": 5.131661588863132e-05,
      "loss": 0.3375,
      "step": 1626
    },
    {
      "epoch": 2.3410071942446042,
      "grad_norm": 0.15078541911069537,
      "learning_rate": 5.1278065348934786e-05,
      "loss": 0.335,
      "step": 1627
    },
    {
      "epoch": 2.3424460431654675,
      "grad_norm": 0.19084042148697,
      "learning_rate": 5.123950342566639e-05,
      "loss": 0.3491,
      "step": 1628
    },
    {
      "epoch": 2.343884892086331,
      "grad_norm": 0.2067992073638373,
      "learning_rate": 5.120093015774882e-05,
      "loss": 0.3428,
      "step": 1629
    },
    {
      "epoch": 2.3453237410071943,
      "grad_norm": 0.15228406978948328,
      "learning_rate": 5.116234558411618e-05,
      "loss": 0.3439,
      "step": 1630
    },
    {
      "epoch": 2.3467625899280575,
      "grad_norm": 0.13129931838861417,
      "learning_rate": 5.1123749743714024e-05,
      "loss": 0.3337,
      "step": 1631
    },
    {
      "epoch": 2.3482014388489207,
      "grad_norm": 0.14308636666369418,
      "learning_rate": 5.1085142675499246e-05,
      "loss": 0.3441,
      "step": 1632
    },
    {
      "epoch": 2.349640287769784,
      "grad_norm": 0.15789616839818424,
      "learning_rate": 5.1046524418440075e-05,
      "loss": 0.3396,
      "step": 1633
    },
    {
      "epoch": 2.3510791366906476,
      "grad_norm": 0.1466083603631073,
      "learning_rate": 5.100789501151607e-05,
      "loss": 0.3369,
      "step": 1634
    },
    {
      "epoch": 2.352517985611511,
      "grad_norm": 0.13018873128250721,
      "learning_rate": 5.0969254493717996e-05,
      "loss": 0.3522,
      "step": 1635
    },
    {
      "epoch": 2.353956834532374,
      "grad_norm": 0.17119707649582805,
      "learning_rate": 5.093060290404785e-05,
      "loss": 0.3413,
      "step": 1636
    },
    {
      "epoch": 2.3553956834532372,
      "grad_norm": 0.20255061297486265,
      "learning_rate": 5.089194028151882e-05,
      "loss": 0.3462,
      "step": 1637
    },
    {
      "epoch": 2.356834532374101,
      "grad_norm": 0.15442557209983027,
      "learning_rate": 5.085326666515521e-05,
      "loss": 0.3436,
      "step": 1638
    },
    {
      "epoch": 2.358273381294964,
      "grad_norm": 0.1340549799852972,
      "learning_rate": 5.081458209399243e-05,
      "loss": 0.3408,
      "step": 1639
    },
    {
      "epoch": 2.3597122302158273,
      "grad_norm": 0.1328264324241929,
      "learning_rate": 5.0775886607076954e-05,
      "loss": 0.3341,
      "step": 1640
    },
    {
      "epoch": 2.3611510791366905,
      "grad_norm": 0.13228593919564147,
      "learning_rate": 5.073718024346626e-05,
      "loss": 0.3408,
      "step": 1641
    },
    {
      "epoch": 2.362589928057554,
      "grad_norm": 0.11427875745181809,
      "learning_rate": 5.06984630422288e-05,
      "loss": 0.3405,
      "step": 1642
    },
    {
      "epoch": 2.3640287769784174,
      "grad_norm": 0.12760861743227073,
      "learning_rate": 5.065973504244399e-05,
      "loss": 0.3395,
      "step": 1643
    },
    {
      "epoch": 2.3654676258992806,
      "grad_norm": 0.13863017553609638,
      "learning_rate": 5.062099628320213e-05,
      "loss": 0.3436,
      "step": 1644
    },
    {
      "epoch": 2.366906474820144,
      "grad_norm": 0.13931443184416437,
      "learning_rate": 5.058224680360438e-05,
      "loss": 0.3418,
      "step": 1645
    },
    {
      "epoch": 2.368345323741007,
      "grad_norm": 0.12265615954125163,
      "learning_rate": 5.054348664276271e-05,
      "loss": 0.344,
      "step": 1646
    },
    {
      "epoch": 2.3697841726618707,
      "grad_norm": 0.14603930332322596,
      "learning_rate": 5.05047158397999e-05,
      "loss": 0.3459,
      "step": 1647
    },
    {
      "epoch": 2.371223021582734,
      "grad_norm": 0.1440671894672473,
      "learning_rate": 5.046593443384945e-05,
      "loss": 0.3416,
      "step": 1648
    },
    {
      "epoch": 2.372661870503597,
      "grad_norm": 0.11840615315430422,
      "learning_rate": 5.042714246405555e-05,
      "loss": 0.3417,
      "step": 1649
    },
    {
      "epoch": 2.3741007194244603,
      "grad_norm": 0.141585181622913,
      "learning_rate": 5.038833996957309e-05,
      "loss": 0.3405,
      "step": 1650
    },
    {
      "epoch": 2.3755395683453235,
      "grad_norm": 0.1509373476302574,
      "learning_rate": 5.0349526989567546e-05,
      "loss": 0.3437,
      "step": 1651
    },
    {
      "epoch": 2.376978417266187,
      "grad_norm": 0.14972886167306537,
      "learning_rate": 5.0310703563215016e-05,
      "loss": 0.3418,
      "step": 1652
    },
    {
      "epoch": 2.3784172661870504,
      "grad_norm": 0.11796062644059152,
      "learning_rate": 5.027186972970211e-05,
      "loss": 0.3509,
      "step": 1653
    },
    {
      "epoch": 2.3798561151079136,
      "grad_norm": 0.12176269985179156,
      "learning_rate": 5.0233025528225934e-05,
      "loss": 0.3427,
      "step": 1654
    },
    {
      "epoch": 2.381294964028777,
      "grad_norm": 0.14452049677617043,
      "learning_rate": 5.01941709979941e-05,
      "loss": 0.3387,
      "step": 1655
    },
    {
      "epoch": 2.38273381294964,
      "grad_norm": 0.14278552999151747,
      "learning_rate": 5.015530617822462e-05,
      "loss": 0.3454,
      "step": 1656
    },
    {
      "epoch": 2.3841726618705037,
      "grad_norm": 0.12193205162577694,
      "learning_rate": 5.011643110814589e-05,
      "loss": 0.3412,
      "step": 1657
    },
    {
      "epoch": 2.385611510791367,
      "grad_norm": 0.11282073766836978,
      "learning_rate": 5.007754582699666e-05,
      "loss": 0.3382,
      "step": 1658
    },
    {
      "epoch": 2.38705035971223,
      "grad_norm": 0.11703149562420231,
      "learning_rate": 5.003865037402598e-05,
      "loss": 0.3409,
      "step": 1659
    },
    {
      "epoch": 2.3884892086330938,
      "grad_norm": 0.12885309449382898,
      "learning_rate": 4.999974478849319e-05,
      "loss": 0.3312,
      "step": 1660
    },
    {
      "epoch": 2.389928057553957,
      "grad_norm": 0.13183350461843132,
      "learning_rate": 4.99608291096678e-05,
      "loss": 0.3475,
      "step": 1661
    },
    {
      "epoch": 2.39136690647482,
      "grad_norm": 0.13776304321963204,
      "learning_rate": 4.9921903376829565e-05,
      "loss": 0.3384,
      "step": 1662
    },
    {
      "epoch": 2.3928057553956834,
      "grad_norm": 0.18077136229119795,
      "learning_rate": 4.988296762926838e-05,
      "loss": 0.3427,
      "step": 1663
    },
    {
      "epoch": 2.3942446043165466,
      "grad_norm": 0.15410079075328875,
      "learning_rate": 4.984402190628422e-05,
      "loss": 0.3432,
      "step": 1664
    },
    {
      "epoch": 2.3956834532374103,
      "grad_norm": 0.18997535918274108,
      "learning_rate": 4.980506624718716e-05,
      "loss": 0.3487,
      "step": 1665
    },
    {
      "epoch": 2.3971223021582735,
      "grad_norm": 0.2531239787919152,
      "learning_rate": 4.9766100691297284e-05,
      "loss": 0.3426,
      "step": 1666
    },
    {
      "epoch": 2.3985611510791367,
      "grad_norm": 0.1801137622984491,
      "learning_rate": 4.9727125277944675e-05,
      "loss": 0.3356,
      "step": 1667
    },
    {
      "epoch": 2.4,
      "grad_norm": 0.1551017120725682,
      "learning_rate": 4.968814004646934e-05,
      "loss": 0.3359,
      "step": 1668
    },
    {
      "epoch": 2.401438848920863,
      "grad_norm": 0.22300888322165083,
      "learning_rate": 4.964914503622126e-05,
      "loss": 0.3425,
      "step": 1669
    },
    {
      "epoch": 2.402877697841727,
      "grad_norm": 0.1652182883526435,
      "learning_rate": 4.961014028656021e-05,
      "loss": 0.3461,
      "step": 1670
    },
    {
      "epoch": 2.40431654676259,
      "grad_norm": 0.14298724843508484,
      "learning_rate": 4.9571125836855825e-05,
      "loss": 0.3472,
      "step": 1671
    },
    {
      "epoch": 2.405755395683453,
      "grad_norm": 0.16991455955822699,
      "learning_rate": 4.9532101726487564e-05,
      "loss": 0.3321,
      "step": 1672
    },
    {
      "epoch": 2.4071942446043164,
      "grad_norm": 0.14727464943168272,
      "learning_rate": 4.9493067994844606e-05,
      "loss": 0.3408,
      "step": 1673
    },
    {
      "epoch": 2.4086330935251796,
      "grad_norm": 0.13934054200481163,
      "learning_rate": 4.9454024681325815e-05,
      "loss": 0.3425,
      "step": 1674
    },
    {
      "epoch": 2.4100719424460433,
      "grad_norm": 0.14174089322523498,
      "learning_rate": 4.941497182533978e-05,
      "loss": 0.3432,
      "step": 1675
    },
    {
      "epoch": 2.4115107913669065,
      "grad_norm": 0.12575370552729948,
      "learning_rate": 4.937590946630469e-05,
      "loss": 0.3454,
      "step": 1676
    },
    {
      "epoch": 2.4129496402877697,
      "grad_norm": 0.1487581138720967,
      "learning_rate": 4.9336837643648335e-05,
      "loss": 0.336,
      "step": 1677
    },
    {
      "epoch": 2.414388489208633,
      "grad_norm": 0.16032016529401594,
      "learning_rate": 4.929775639680805e-05,
      "loss": 0.3345,
      "step": 1678
    },
    {
      "epoch": 2.4158273381294966,
      "grad_norm": 0.12110484420380835,
      "learning_rate": 4.925866576523069e-05,
      "loss": 0.3461,
      "step": 1679
    },
    {
      "epoch": 2.41726618705036,
      "grad_norm": 0.13464064234471756,
      "learning_rate": 4.921956578837259e-05,
      "loss": 0.3417,
      "step": 1680
    },
    {
      "epoch": 2.418705035971223,
      "grad_norm": 0.130308509890039,
      "learning_rate": 4.918045650569949e-05,
      "loss": 0.3414,
      "step": 1681
    },
    {
      "epoch": 2.420143884892086,
      "grad_norm": 0.11452785815593626,
      "learning_rate": 4.9141337956686564e-05,
      "loss": 0.3397,
      "step": 1682
    },
    {
      "epoch": 2.42158273381295,
      "grad_norm": 0.1305037076855578,
      "learning_rate": 4.91022101808183e-05,
      "loss": 0.3462,
      "step": 1683
    },
    {
      "epoch": 2.423021582733813,
      "grad_norm": 0.11371114866586443,
      "learning_rate": 4.90630732175885e-05,
      "loss": 0.3433,
      "step": 1684
    },
    {
      "epoch": 2.4244604316546763,
      "grad_norm": 0.13409494807980796,
      "learning_rate": 4.902392710650028e-05,
      "loss": 0.349,
      "step": 1685
    },
    {
      "epoch": 2.4258992805755395,
      "grad_norm": 0.12018496996475034,
      "learning_rate": 4.898477188706596e-05,
      "loss": 0.3428,
      "step": 1686
    },
    {
      "epoch": 2.4273381294964027,
      "grad_norm": 0.13131070325532423,
      "learning_rate": 4.894560759880705e-05,
      "loss": 0.3402,
      "step": 1687
    },
    {
      "epoch": 2.4287769784172664,
      "grad_norm": 0.15857659122130136,
      "learning_rate": 4.8906434281254223e-05,
      "loss": 0.3327,
      "step": 1688
    },
    {
      "epoch": 2.4302158273381296,
      "grad_norm": 0.13889386695086822,
      "learning_rate": 4.886725197394726e-05,
      "loss": 0.333,
      "step": 1689
    },
    {
      "epoch": 2.431654676258993,
      "grad_norm": 0.13073855148187585,
      "learning_rate": 4.882806071643503e-05,
      "loss": 0.3509,
      "step": 1690
    },
    {
      "epoch": 2.433093525179856,
      "grad_norm": 0.15118527435233725,
      "learning_rate": 4.878886054827541e-05,
      "loss": 0.3447,
      "step": 1691
    },
    {
      "epoch": 2.434532374100719,
      "grad_norm": 0.14452053762841285,
      "learning_rate": 4.874965150903529e-05,
      "loss": 0.3428,
      "step": 1692
    },
    {
      "epoch": 2.435971223021583,
      "grad_norm": 0.10431327788410687,
      "learning_rate": 4.871043363829053e-05,
      "loss": 0.3336,
      "step": 1693
    },
    {
      "epoch": 2.437410071942446,
      "grad_norm": 0.4213052230566776,
      "learning_rate": 4.8671206975625856e-05,
      "loss": 0.3526,
      "step": 1694
    },
    {
      "epoch": 2.4388489208633093,
      "grad_norm": 0.11184770364690542,
      "learning_rate": 4.863197156063492e-05,
      "loss": 0.3353,
      "step": 1695
    },
    {
      "epoch": 2.4402877697841725,
      "grad_norm": 0.12222149419458228,
      "learning_rate": 4.859272743292017e-05,
      "loss": 0.3429,
      "step": 1696
    },
    {
      "epoch": 2.441726618705036,
      "grad_norm": 0.11907575482636068,
      "learning_rate": 4.855347463209287e-05,
      "loss": 0.3481,
      "step": 1697
    },
    {
      "epoch": 2.4431654676258994,
      "grad_norm": 0.13084626104764957,
      "learning_rate": 4.851421319777304e-05,
      "loss": 0.3399,
      "step": 1698
    },
    {
      "epoch": 2.4446043165467626,
      "grad_norm": 0.1397141335067657,
      "learning_rate": 4.847494316958939e-05,
      "loss": 0.338,
      "step": 1699
    },
    {
      "epoch": 2.446043165467626,
      "grad_norm": 0.1319380742709877,
      "learning_rate": 4.8435664587179315e-05,
      "loss": 0.348,
      "step": 1700
    },
    {
      "epoch": 2.4474820143884894,
      "grad_norm": 0.1269627828943175,
      "learning_rate": 4.839637749018887e-05,
      "loss": 0.3372,
      "step": 1701
    },
    {
      "epoch": 2.4489208633093527,
      "grad_norm": 0.1153638843635166,
      "learning_rate": 4.835708191827268e-05,
      "loss": 0.3449,
      "step": 1702
    },
    {
      "epoch": 2.450359712230216,
      "grad_norm": 0.11531311545018066,
      "learning_rate": 4.831777791109392e-05,
      "loss": 0.3378,
      "step": 1703
    },
    {
      "epoch": 2.451798561151079,
      "grad_norm": 0.11940463375470235,
      "learning_rate": 4.827846550832428e-05,
      "loss": 0.3417,
      "step": 1704
    },
    {
      "epoch": 2.4532374100719423,
      "grad_norm": 0.1143421974967014,
      "learning_rate": 4.8239144749643936e-05,
      "loss": 0.3397,
      "step": 1705
    },
    {
      "epoch": 2.454676258992806,
      "grad_norm": 0.1127354412143374,
      "learning_rate": 4.819981567474152e-05,
      "loss": 0.3393,
      "step": 1706
    },
    {
      "epoch": 2.456115107913669,
      "grad_norm": 0.12792273576722857,
      "learning_rate": 4.8160478323313974e-05,
      "loss": 0.3454,
      "step": 1707
    },
    {
      "epoch": 2.4575539568345324,
      "grad_norm": 0.12738881142266212,
      "learning_rate": 4.812113273506671e-05,
      "loss": 0.3363,
      "step": 1708
    },
    {
      "epoch": 2.4589928057553956,
      "grad_norm": 0.1038933836953817,
      "learning_rate": 4.808177894971336e-05,
      "loss": 0.3429,
      "step": 1709
    },
    {
      "epoch": 2.460431654676259,
      "grad_norm": 0.15633130637066203,
      "learning_rate": 4.804241700697588e-05,
      "loss": 0.341,
      "step": 1710
    },
    {
      "epoch": 2.4618705035971225,
      "grad_norm": 0.1293726497004006,
      "learning_rate": 4.800304694658443e-05,
      "loss": 0.3409,
      "step": 1711
    },
    {
      "epoch": 2.4633093525179857,
      "grad_norm": 0.13166280142095088,
      "learning_rate": 4.796366880827739e-05,
      "loss": 0.336,
      "step": 1712
    },
    {
      "epoch": 2.464748201438849,
      "grad_norm": 0.14510994717815986,
      "learning_rate": 4.792428263180128e-05,
      "loss": 0.334,
      "step": 1713
    },
    {
      "epoch": 2.466187050359712,
      "grad_norm": 0.10196507070564377,
      "learning_rate": 4.7884888456910734e-05,
      "loss": 0.3475,
      "step": 1714
    },
    {
      "epoch": 2.4676258992805753,
      "grad_norm": 0.13058594222204808,
      "learning_rate": 4.784548632336846e-05,
      "loss": 0.3462,
      "step": 1715
    },
    {
      "epoch": 2.469064748201439,
      "grad_norm": 0.11063133459925471,
      "learning_rate": 4.7806076270945197e-05,
      "loss": 0.3375,
      "step": 1716
    },
    {
      "epoch": 2.470503597122302,
      "grad_norm": 0.10271160902895207,
      "learning_rate": 4.776665833941968e-05,
      "loss": 0.3394,
      "step": 1717
    },
    {
      "epoch": 2.4719424460431654,
      "grad_norm": 0.12607363146537412,
      "learning_rate": 4.772723256857859e-05,
      "loss": 0.3451,
      "step": 1718
    },
    {
      "epoch": 2.4733812949640286,
      "grad_norm": 0.1583496627807278,
      "learning_rate": 4.768779899821655e-05,
      "loss": 0.3366,
      "step": 1719
    },
    {
      "epoch": 2.4748201438848922,
      "grad_norm": 0.14234848814645065,
      "learning_rate": 4.7648357668135996e-05,
      "loss": 0.3432,
      "step": 1720
    },
    {
      "epoch": 2.4762589928057555,
      "grad_norm": 0.14076368701870684,
      "learning_rate": 4.760890861814726e-05,
      "loss": 0.3437,
      "step": 1721
    },
    {
      "epoch": 2.4776978417266187,
      "grad_norm": 0.16272716328653117,
      "learning_rate": 4.756945188806843e-05,
      "loss": 0.336,
      "step": 1722
    },
    {
      "epoch": 2.479136690647482,
      "grad_norm": 0.13042375807753362,
      "learning_rate": 4.752998751772536e-05,
      "loss": 0.3464,
      "step": 1723
    },
    {
      "epoch": 2.4805755395683455,
      "grad_norm": 0.10483141311526244,
      "learning_rate": 4.749051554695159e-05,
      "loss": 0.3359,
      "step": 1724
    },
    {
      "epoch": 2.4820143884892087,
      "grad_norm": 0.15640901733596446,
      "learning_rate": 4.745103601558838e-05,
      "loss": 0.3409,
      "step": 1725
    },
    {
      "epoch": 2.483453237410072,
      "grad_norm": 0.14828497121504203,
      "learning_rate": 4.741154896348458e-05,
      "loss": 0.3447,
      "step": 1726
    },
    {
      "epoch": 2.484892086330935,
      "grad_norm": 0.1103431235543875,
      "learning_rate": 4.7372054430496636e-05,
      "loss": 0.3364,
      "step": 1727
    },
    {
      "epoch": 2.4863309352517984,
      "grad_norm": 0.16230804063638982,
      "learning_rate": 4.733255245648857e-05,
      "loss": 0.3445,
      "step": 1728
    },
    {
      "epoch": 2.487769784172662,
      "grad_norm": 0.1351199508794201,
      "learning_rate": 4.729304308133189e-05,
      "loss": 0.3405,
      "step": 1729
    },
    {
      "epoch": 2.4892086330935252,
      "grad_norm": 0.10339238683809066,
      "learning_rate": 4.725352634490557e-05,
      "loss": 0.3452,
      "step": 1730
    },
    {
      "epoch": 2.4906474820143885,
      "grad_norm": 0.12454825077224349,
      "learning_rate": 4.7214002287096035e-05,
      "loss": 0.3411,
      "step": 1731
    },
    {
      "epoch": 2.4920863309352517,
      "grad_norm": 0.1365182483518328,
      "learning_rate": 4.7174470947797117e-05,
      "loss": 0.3385,
      "step": 1732
    },
    {
      "epoch": 2.493525179856115,
      "grad_norm": 0.11652871026144225,
      "learning_rate": 4.7134932366909915e-05,
      "loss": 0.3355,
      "step": 1733
    },
    {
      "epoch": 2.4949640287769785,
      "grad_norm": 0.1398221290455718,
      "learning_rate": 4.709538658434294e-05,
      "loss": 0.3421,
      "step": 1734
    },
    {
      "epoch": 2.4964028776978417,
      "grad_norm": 0.12643466864251143,
      "learning_rate": 4.705583364001192e-05,
      "loss": 0.3412,
      "step": 1735
    },
    {
      "epoch": 2.497841726618705,
      "grad_norm": 0.12663940225696427,
      "learning_rate": 4.701627357383981e-05,
      "loss": 0.3427,
      "step": 1736
    },
    {
      "epoch": 2.499280575539568,
      "grad_norm": 0.12867713456211932,
      "learning_rate": 4.697670642575675e-05,
      "loss": 0.3389,
      "step": 1737
    },
    {
      "epoch": 2.5007194244604314,
      "grad_norm": 0.09729215423207732,
      "learning_rate": 4.693713223570006e-05,
      "loss": 0.3396,
      "step": 1738
    },
    {
      "epoch": 2.502158273381295,
      "grad_norm": 0.10536205673373836,
      "learning_rate": 4.689755104361414e-05,
      "loss": 0.3358,
      "step": 1739
    },
    {
      "epoch": 2.5035971223021583,
      "grad_norm": 0.09470267440128205,
      "learning_rate": 4.685796288945046e-05,
      "loss": 0.3354,
      "step": 1740
    },
    {
      "epoch": 2.5050359712230215,
      "grad_norm": 0.10720684460790161,
      "learning_rate": 4.6818367813167535e-05,
      "loss": 0.3392,
      "step": 1741
    },
    {
      "epoch": 2.506474820143885,
      "grad_norm": 0.12457788316646835,
      "learning_rate": 4.6778765854730835e-05,
      "loss": 0.3369,
      "step": 1742
    },
    {
      "epoch": 2.5079136690647483,
      "grad_norm": 0.12249312745078086,
      "learning_rate": 4.673915705411281e-05,
      "loss": 0.3502,
      "step": 1743
    },
    {
      "epoch": 2.5093525179856115,
      "grad_norm": 0.12834266627532392,
      "learning_rate": 4.6699541451292786e-05,
      "loss": 0.3385,
      "step": 1744
    },
    {
      "epoch": 2.5107913669064748,
      "grad_norm": 0.12800319139369157,
      "learning_rate": 4.665991908625699e-05,
      "loss": 0.3434,
      "step": 1745
    },
    {
      "epoch": 2.512230215827338,
      "grad_norm": 0.11948627362015994,
      "learning_rate": 4.6620289998998445e-05,
      "loss": 0.3344,
      "step": 1746
    },
    {
      "epoch": 2.5136690647482016,
      "grad_norm": 0.13107752356793956,
      "learning_rate": 4.658065422951697e-05,
      "loss": 0.3359,
      "step": 1747
    },
    {
      "epoch": 2.515107913669065,
      "grad_norm": 0.11391152786479106,
      "learning_rate": 4.654101181781913e-05,
      "loss": 0.341,
      "step": 1748
    },
    {
      "epoch": 2.516546762589928,
      "grad_norm": 0.13063226539610714,
      "learning_rate": 4.650136280391818e-05,
      "loss": 0.3389,
      "step": 1749
    },
    {
      "epoch": 2.5179856115107913,
      "grad_norm": 0.1261484669423309,
      "learning_rate": 4.646170722783408e-05,
      "loss": 0.3291,
      "step": 1750
    },
    {
      "epoch": 2.5194244604316545,
      "grad_norm": 0.10139629190687235,
      "learning_rate": 4.6422045129593344e-05,
      "loss": 0.3355,
      "step": 1751
    },
    {
      "epoch": 2.520863309352518,
      "grad_norm": 0.14901272921951966,
      "learning_rate": 4.6382376549229146e-05,
      "loss": 0.3288,
      "step": 1752
    },
    {
      "epoch": 2.5223021582733813,
      "grad_norm": 0.1523215911606212,
      "learning_rate": 4.634270152678115e-05,
      "loss": 0.3355,
      "step": 1753
    },
    {
      "epoch": 2.5237410071942445,
      "grad_norm": 0.11328006772432715,
      "learning_rate": 4.630302010229555e-05,
      "loss": 0.3305,
      "step": 1754
    },
    {
      "epoch": 2.5251798561151078,
      "grad_norm": 0.14774464605253138,
      "learning_rate": 4.6263332315824964e-05,
      "loss": 0.3428,
      "step": 1755
    },
    {
      "epoch": 2.526618705035971,
      "grad_norm": 0.13300622516392707,
      "learning_rate": 4.622363820742848e-05,
      "loss": 0.3439,
      "step": 1756
    },
    {
      "epoch": 2.5280575539568346,
      "grad_norm": 0.1287918531160956,
      "learning_rate": 4.618393781717156e-05,
      "loss": 0.3378,
      "step": 1757
    },
    {
      "epoch": 2.529496402877698,
      "grad_norm": 0.11652999713195161,
      "learning_rate": 4.614423118512595e-05,
      "loss": 0.3492,
      "step": 1758
    },
    {
      "epoch": 2.530935251798561,
      "grad_norm": 0.11213280118872815,
      "learning_rate": 4.610451835136978e-05,
      "loss": 0.3437,
      "step": 1759
    },
    {
      "epoch": 2.5323741007194247,
      "grad_norm": 0.11185644660014751,
      "learning_rate": 4.606479935598738e-05,
      "loss": 0.3501,
      "step": 1760
    },
    {
      "epoch": 2.533812949640288,
      "grad_norm": 0.09436021263924985,
      "learning_rate": 4.602507423906931e-05,
      "loss": 0.3354,
      "step": 1761
    },
    {
      "epoch": 2.535251798561151,
      "grad_norm": 0.11104665481821985,
      "learning_rate": 4.598534304071233e-05,
      "loss": 0.3338,
      "step": 1762
    },
    {
      "epoch": 2.5366906474820143,
      "grad_norm": 0.10911876228788199,
      "learning_rate": 4.5945605801019315e-05,
      "loss": 0.3345,
      "step": 1763
    },
    {
      "epoch": 2.5381294964028775,
      "grad_norm": 0.09465719016284928,
      "learning_rate": 4.5905862560099255e-05,
      "loss": 0.3387,
      "step": 1764
    },
    {
      "epoch": 2.539568345323741,
      "grad_norm": 0.10650314757534553,
      "learning_rate": 4.5866113358067187e-05,
      "loss": 0.3389,
      "step": 1765
    },
    {
      "epoch": 2.5410071942446044,
      "grad_norm": 0.11984942002554853,
      "learning_rate": 4.582635823504416e-05,
      "loss": 0.3379,
      "step": 1766
    },
    {
      "epoch": 2.5424460431654676,
      "grad_norm": 0.11734119542612027,
      "learning_rate": 4.5786597231157214e-05,
      "loss": 0.3474,
      "step": 1767
    },
    {
      "epoch": 2.543884892086331,
      "grad_norm": 0.11015931760329653,
      "learning_rate": 4.574683038653932e-05,
      "loss": 0.3396,
      "step": 1768
    },
    {
      "epoch": 2.545323741007194,
      "grad_norm": 0.11844530280191058,
      "learning_rate": 4.5707057741329324e-05,
      "loss": 0.3391,
      "step": 1769
    },
    {
      "epoch": 2.5467625899280577,
      "grad_norm": 0.12190665595078562,
      "learning_rate": 4.5667279335671986e-05,
      "loss": 0.3445,
      "step": 1770
    },
    {
      "epoch": 2.548201438848921,
      "grad_norm": 0.11538645447979806,
      "learning_rate": 4.56274952097178e-05,
      "loss": 0.3369,
      "step": 1771
    },
    {
      "epoch": 2.549640287769784,
      "grad_norm": 0.13013491237388408,
      "learning_rate": 4.558770540362308e-05,
      "loss": 0.341,
      "step": 1772
    },
    {
      "epoch": 2.5510791366906473,
      "grad_norm": 0.11811645416708122,
      "learning_rate": 4.554790995754988e-05,
      "loss": 0.3421,
      "step": 1773
    },
    {
      "epoch": 2.5525179856115106,
      "grad_norm": 0.14139738127617787,
      "learning_rate": 4.5508108911665926e-05,
      "loss": 0.343,
      "step": 1774
    },
    {
      "epoch": 2.553956834532374,
      "grad_norm": 0.15176651595735202,
      "learning_rate": 4.5468302306144594e-05,
      "loss": 0.338,
      "step": 1775
    },
    {
      "epoch": 2.5553956834532374,
      "grad_norm": 0.10735008156962264,
      "learning_rate": 4.542849018116491e-05,
      "loss": 0.3393,
      "step": 1776
    },
    {
      "epoch": 2.5568345323741006,
      "grad_norm": 0.14879243298691355,
      "learning_rate": 4.538867257691141e-05,
      "loss": 0.3311,
      "step": 1777
    },
    {
      "epoch": 2.5582733812949643,
      "grad_norm": 0.12535624465428738,
      "learning_rate": 4.53488495335742e-05,
      "loss": 0.3482,
      "step": 1778
    },
    {
      "epoch": 2.5597122302158275,
      "grad_norm": 0.10908302463595965,
      "learning_rate": 4.5309021091348885e-05,
      "loss": 0.3427,
      "step": 1779
    },
    {
      "epoch": 2.5611510791366907,
      "grad_norm": 0.15905673972568443,
      "learning_rate": 4.5269187290436486e-05,
      "loss": 0.3357,
      "step": 1780
    },
    {
      "epoch": 2.562589928057554,
      "grad_norm": 0.13253803661911623,
      "learning_rate": 4.5229348171043466e-05,
      "loss": 0.3372,
      "step": 1781
    },
    {
      "epoch": 2.564028776978417,
      "grad_norm": 0.10950791156714575,
      "learning_rate": 4.51895037733816e-05,
      "loss": 0.3408,
      "step": 1782
    },
    {
      "epoch": 2.565467625899281,
      "grad_norm": 0.16002575202649105,
      "learning_rate": 4.5149654137668095e-05,
      "loss": 0.3408,
      "step": 1783
    },
    {
      "epoch": 2.566906474820144,
      "grad_norm": 0.17440279143410928,
      "learning_rate": 4.5109799304125333e-05,
      "loss": 0.3445,
      "step": 1784
    },
    {
      "epoch": 2.568345323741007,
      "grad_norm": 0.0900887649274377,
      "learning_rate": 4.5069939312981e-05,
      "loss": 0.3387,
      "step": 1785
    },
    {
      "epoch": 2.5697841726618704,
      "grad_norm": 0.12332539288262825,
      "learning_rate": 4.503007420446798e-05,
      "loss": 0.3352,
      "step": 1786
    },
    {
      "epoch": 2.5712230215827336,
      "grad_norm": 0.11515832396272689,
      "learning_rate": 4.499020401882433e-05,
      "loss": 0.3374,
      "step": 1787
    },
    {
      "epoch": 2.5726618705035973,
      "grad_norm": 0.11235829008442862,
      "learning_rate": 4.49503287962932e-05,
      "loss": 0.3388,
      "step": 1788
    },
    {
      "epoch": 2.5741007194244605,
      "grad_norm": 0.11147519106474965,
      "learning_rate": 4.491044857712288e-05,
      "loss": 0.3402,
      "step": 1789
    },
    {
      "epoch": 2.5755395683453237,
      "grad_norm": 0.1262425182010213,
      "learning_rate": 4.4870563401566634e-05,
      "loss": 0.3458,
      "step": 1790
    },
    {
      "epoch": 2.576978417266187,
      "grad_norm": 0.11324580140469244,
      "learning_rate": 4.483067330988278e-05,
      "loss": 0.3419,
      "step": 1791
    },
    {
      "epoch": 2.57841726618705,
      "grad_norm": 0.1289434899627111,
      "learning_rate": 4.479077834233458e-05,
      "loss": 0.3436,
      "step": 1792
    },
    {
      "epoch": 2.579856115107914,
      "grad_norm": 0.1122464263243746,
      "learning_rate": 4.475087853919023e-05,
      "loss": 0.3407,
      "step": 1793
    },
    {
      "epoch": 2.581294964028777,
      "grad_norm": 0.13063460751115946,
      "learning_rate": 4.4710973940722786e-05,
      "loss": 0.3456,
      "step": 1794
    },
    {
      "epoch": 2.58273381294964,
      "grad_norm": 0.13627796419509441,
      "learning_rate": 4.4671064587210146e-05,
      "loss": 0.3361,
      "step": 1795
    },
    {
      "epoch": 2.584172661870504,
      "grad_norm": 0.13508693136071181,
      "learning_rate": 4.4631150518935044e-05,
      "loss": 0.336,
      "step": 1796
    },
    {
      "epoch": 2.5856115107913666,
      "grad_norm": 0.12401876363541481,
      "learning_rate": 4.459123177618491e-05,
      "loss": 0.3379,
      "step": 1797
    },
    {
      "epoch": 2.5870503597122303,
      "grad_norm": 0.12941049453432754,
      "learning_rate": 4.455130839925195e-05,
      "loss": 0.3435,
      "step": 1798
    },
    {
      "epoch": 2.5884892086330935,
      "grad_norm": 0.14867071306096177,
      "learning_rate": 4.451138042843302e-05,
      "loss": 0.3425,
      "step": 1799
    },
    {
      "epoch": 2.5899280575539567,
      "grad_norm": 0.12093240923116741,
      "learning_rate": 4.447144790402963e-05,
      "loss": 0.3408,
      "step": 1800
    },
    {
      "epoch": 2.5913669064748204,
      "grad_norm": 0.12374790333094081,
      "learning_rate": 4.4431510866347837e-05,
      "loss": 0.3375,
      "step": 1801
    },
    {
      "epoch": 2.5928057553956836,
      "grad_norm": 0.14857921532385407,
      "learning_rate": 4.439156935569833e-05,
      "loss": 0.3416,
      "step": 1802
    },
    {
      "epoch": 2.594244604316547,
      "grad_norm": 0.12926069519248012,
      "learning_rate": 4.435162341239625e-05,
      "loss": 0.3435,
      "step": 1803
    },
    {
      "epoch": 2.59568345323741,
      "grad_norm": 0.11651180748397517,
      "learning_rate": 4.4311673076761254e-05,
      "loss": 0.3373,
      "step": 1804
    },
    {
      "epoch": 2.597122302158273,
      "grad_norm": 0.1507180724691262,
      "learning_rate": 4.42717183891174e-05,
      "loss": 0.3331,
      "step": 1805
    },
    {
      "epoch": 2.598561151079137,
      "grad_norm": 0.1414529505953001,
      "learning_rate": 4.4231759389793144e-05,
      "loss": 0.3439,
      "step": 1806
    },
    {
      "epoch": 2.6,
      "grad_norm": 0.11773868704444448,
      "learning_rate": 4.4191796119121335e-05,
      "loss": 0.3421,
      "step": 1807
    },
    {
      "epoch": 2.6014388489208633,
      "grad_norm": 0.14819298825521954,
      "learning_rate": 4.415182861743906e-05,
      "loss": 0.3476,
      "step": 1808
    },
    {
      "epoch": 2.6028776978417265,
      "grad_norm": 0.140434101586376,
      "learning_rate": 4.411185692508774e-05,
      "loss": 0.3414,
      "step": 1809
    },
    {
      "epoch": 2.6043165467625897,
      "grad_norm": 0.10932990039065672,
      "learning_rate": 4.4071881082413e-05,
      "loss": 0.3375,
      "step": 1810
    },
    {
      "epoch": 2.6057553956834534,
      "grad_norm": 0.13786451900966154,
      "learning_rate": 4.4031901129764665e-05,
      "loss": 0.3331,
      "step": 1811
    },
    {
      "epoch": 2.6071942446043166,
      "grad_norm": 0.12996289496226116,
      "learning_rate": 4.3991917107496695e-05,
      "loss": 0.3431,
      "step": 1812
    },
    {
      "epoch": 2.60863309352518,
      "grad_norm": 0.10387392325944972,
      "learning_rate": 4.395192905596716e-05,
      "loss": 0.3394,
      "step": 1813
    },
    {
      "epoch": 2.610071942446043,
      "grad_norm": 0.12503653841902704,
      "learning_rate": 4.3911937015538186e-05,
      "loss": 0.3413,
      "step": 1814
    },
    {
      "epoch": 2.6115107913669062,
      "grad_norm": 0.10971997429608449,
      "learning_rate": 4.3871941026575965e-05,
      "loss": 0.3374,
      "step": 1815
    },
    {
      "epoch": 2.61294964028777,
      "grad_norm": 0.10023185050648277,
      "learning_rate": 4.383194112945066e-05,
      "loss": 0.3421,
      "step": 1816
    },
    {
      "epoch": 2.614388489208633,
      "grad_norm": 0.11916928542493764,
      "learning_rate": 4.379193736453633e-05,
      "loss": 0.339,
      "step": 1817
    },
    {
      "epoch": 2.6158273381294963,
      "grad_norm": 0.10586309598590037,
      "learning_rate": 4.375192977221099e-05,
      "loss": 0.3484,
      "step": 1818
    },
    {
      "epoch": 2.61726618705036,
      "grad_norm": 0.1098671434915074,
      "learning_rate": 4.371191839285651e-05,
      "loss": 0.3354,
      "step": 1819
    },
    {
      "epoch": 2.618705035971223,
      "grad_norm": 0.13188890631748518,
      "learning_rate": 4.367190326685858e-05,
      "loss": 0.3413,
      "step": 1820
    },
    {
      "epoch": 2.6201438848920864,
      "grad_norm": 0.11743665747189141,
      "learning_rate": 4.363188443460666e-05,
      "loss": 0.3459,
      "step": 1821
    },
    {
      "epoch": 2.6215827338129496,
      "grad_norm": 0.12017290401897036,
      "learning_rate": 4.3591861936493964e-05,
      "loss": 0.3355,
      "step": 1822
    },
    {
      "epoch": 2.623021582733813,
      "grad_norm": 0.1146193053784997,
      "learning_rate": 4.3551835812917395e-05,
      "loss": 0.34,
      "step": 1823
    },
    {
      "epoch": 2.6244604316546765,
      "grad_norm": 0.09860614798113755,
      "learning_rate": 4.351180610427754e-05,
      "loss": 0.3415,
      "step": 1824
    },
    {
      "epoch": 2.6258992805755397,
      "grad_norm": 0.11114368863560233,
      "learning_rate": 4.347177285097855e-05,
      "loss": 0.3382,
      "step": 1825
    },
    {
      "epoch": 2.627338129496403,
      "grad_norm": 0.11665309354902487,
      "learning_rate": 4.343173609342822e-05,
      "loss": 0.3446,
      "step": 1826
    },
    {
      "epoch": 2.628776978417266,
      "grad_norm": 0.10277560427518727,
      "learning_rate": 4.339169587203785e-05,
      "loss": 0.3391,
      "step": 1827
    },
    {
      "epoch": 2.6302158273381293,
      "grad_norm": 0.09599960246248385,
      "learning_rate": 4.335165222722222e-05,
      "loss": 0.3287,
      "step": 1828
    },
    {
      "epoch": 2.631654676258993,
      "grad_norm": 0.11722916136015507,
      "learning_rate": 4.331160519939962e-05,
      "loss": 0.3328,
      "step": 1829
    },
    {
      "epoch": 2.633093525179856,
      "grad_norm": 0.11023207099145288,
      "learning_rate": 4.327155482899168e-05,
      "loss": 0.3378,
      "step": 1830
    },
    {
      "epoch": 2.6345323741007194,
      "grad_norm": 0.10202424983535885,
      "learning_rate": 4.323150115642346e-05,
      "loss": 0.3446,
      "step": 1831
    },
    {
      "epoch": 2.6359712230215826,
      "grad_norm": 0.10413403149240487,
      "learning_rate": 4.3191444222123326e-05,
      "loss": 0.3477,
      "step": 1832
    },
    {
      "epoch": 2.637410071942446,
      "grad_norm": 0.11491521916422581,
      "learning_rate": 4.3151384066522964e-05,
      "loss": 0.3367,
      "step": 1833
    },
    {
      "epoch": 2.6388489208633095,
      "grad_norm": 0.10822247599269093,
      "learning_rate": 4.311132073005727e-05,
      "loss": 0.3422,
      "step": 1834
    },
    {
      "epoch": 2.6402877697841727,
      "grad_norm": 0.12789621557709568,
      "learning_rate": 4.3071254253164395e-05,
      "loss": 0.339,
      "step": 1835
    },
    {
      "epoch": 2.641726618705036,
      "grad_norm": 0.10547673251689751,
      "learning_rate": 4.3031184676285625e-05,
      "loss": 0.3297,
      "step": 1836
    },
    {
      "epoch": 2.6431654676258995,
      "grad_norm": 0.12123644061016982,
      "learning_rate": 4.299111203986539e-05,
      "loss": 0.3312,
      "step": 1837
    },
    {
      "epoch": 2.6446043165467623,
      "grad_norm": 0.12844557785693508,
      "learning_rate": 4.29510363843512e-05,
      "loss": 0.3367,
      "step": 1838
    },
    {
      "epoch": 2.646043165467626,
      "grad_norm": 0.10624197613105688,
      "learning_rate": 4.291095775019364e-05,
      "loss": 0.3406,
      "step": 1839
    },
    {
      "epoch": 2.647482014388489,
      "grad_norm": 0.11173704018377709,
      "learning_rate": 4.287087617784627e-05,
      "loss": 0.3451,
      "step": 1840
    },
    {
      "epoch": 2.6489208633093524,
      "grad_norm": 0.10968755925479357,
      "learning_rate": 4.283079170776561e-05,
      "loss": 0.3384,
      "step": 1841
    },
    {
      "epoch": 2.650359712230216,
      "grad_norm": 0.11346023095700333,
      "learning_rate": 4.279070438041116e-05,
      "loss": 0.3373,
      "step": 1842
    },
    {
      "epoch": 2.6517985611510793,
      "grad_norm": 0.10204403289138868,
      "learning_rate": 4.275061423624522e-05,
      "loss": 0.3428,
      "step": 1843
    },
    {
      "epoch": 2.6532374100719425,
      "grad_norm": 0.1169231203643409,
      "learning_rate": 4.2710521315733e-05,
      "loss": 0.3413,
      "step": 1844
    },
    {
      "epoch": 2.6546762589928057,
      "grad_norm": 0.10943698461313435,
      "learning_rate": 4.26704256593425e-05,
      "loss": 0.3474,
      "step": 1845
    },
    {
      "epoch": 2.656115107913669,
      "grad_norm": 0.1220135814725209,
      "learning_rate": 4.2630327307544454e-05,
      "loss": 0.333,
      "step": 1846
    },
    {
      "epoch": 2.6575539568345325,
      "grad_norm": 0.109415482241854,
      "learning_rate": 4.2590226300812335e-05,
      "loss": 0.3381,
      "step": 1847
    },
    {
      "epoch": 2.6589928057553958,
      "grad_norm": 0.1179174317816068,
      "learning_rate": 4.255012267962232e-05,
      "loss": 0.3453,
      "step": 1848
    },
    {
      "epoch": 2.660431654676259,
      "grad_norm": 0.1275534930596529,
      "learning_rate": 4.251001648445317e-05,
      "loss": 0.3405,
      "step": 1849
    },
    {
      "epoch": 2.661870503597122,
      "grad_norm": 0.13711098030323063,
      "learning_rate": 4.246990775578628e-05,
      "loss": 0.3393,
      "step": 1850
    },
    {
      "epoch": 2.6633093525179854,
      "grad_norm": 0.12306157494688558,
      "learning_rate": 4.242979653410562e-05,
      "loss": 0.3397,
      "step": 1851
    },
    {
      "epoch": 2.664748201438849,
      "grad_norm": 0.10730585499334598,
      "learning_rate": 4.238968285989762e-05,
      "loss": 0.337,
      "step": 1852
    },
    {
      "epoch": 2.6661870503597123,
      "grad_norm": 0.15486322539275116,
      "learning_rate": 4.2349566773651236e-05,
      "loss": 0.336,
      "step": 1853
    },
    {
      "epoch": 2.6676258992805755,
      "grad_norm": 0.13889360595217953,
      "learning_rate": 4.2309448315857844e-05,
      "loss": 0.3419,
      "step": 1854
    },
    {
      "epoch": 2.6690647482014387,
      "grad_norm": 0.09999278397483768,
      "learning_rate": 4.226932752701122e-05,
      "loss": 0.3423,
      "step": 1855
    },
    {
      "epoch": 2.670503597122302,
      "grad_norm": 0.14533948392392976,
      "learning_rate": 4.2229204447607456e-05,
      "loss": 0.3345,
      "step": 1856
    },
    {
      "epoch": 2.6719424460431656,
      "grad_norm": 0.15935841097160405,
      "learning_rate": 4.2189079118145e-05,
      "loss": 0.3461,
      "step": 1857
    },
    {
      "epoch": 2.6733812949640288,
      "grad_norm": 0.10932579096393788,
      "learning_rate": 4.214895157912454e-05,
      "loss": 0.3429,
      "step": 1858
    },
    {
      "epoch": 2.674820143884892,
      "grad_norm": 0.13080358011840967,
      "learning_rate": 4.210882187104904e-05,
      "loss": 0.3366,
      "step": 1859
    },
    {
      "epoch": 2.6762589928057556,
      "grad_norm": 0.13269828722864208,
      "learning_rate": 4.206869003442358e-05,
      "loss": 0.3408,
      "step": 1860
    },
    {
      "epoch": 2.677697841726619,
      "grad_norm": 0.1305819791221825,
      "learning_rate": 4.2028556109755465e-05,
      "loss": 0.3391,
      "step": 1861
    },
    {
      "epoch": 2.679136690647482,
      "grad_norm": 0.09724856574709481,
      "learning_rate": 4.198842013755408e-05,
      "loss": 0.3397,
      "step": 1862
    },
    {
      "epoch": 2.6805755395683453,
      "grad_norm": 0.09635878900706464,
      "learning_rate": 4.194828215833082e-05,
      "loss": 0.3403,
      "step": 1863
    },
    {
      "epoch": 2.6820143884892085,
      "grad_norm": 0.09396676697509772,
      "learning_rate": 4.1908142212599206e-05,
      "loss": 0.3288,
      "step": 1864
    },
    {
      "epoch": 2.683453237410072,
      "grad_norm": 0.0974403067346001,
      "learning_rate": 4.1868000340874674e-05,
      "loss": 0.332,
      "step": 1865
    },
    {
      "epoch": 2.6848920863309353,
      "grad_norm": 0.09508337444864512,
      "learning_rate": 4.182785658367462e-05,
      "loss": 0.3427,
      "step": 1866
    },
    {
      "epoch": 2.6863309352517986,
      "grad_norm": 0.11136661124114632,
      "learning_rate": 4.178771098151835e-05,
      "loss": 0.3425,
      "step": 1867
    },
    {
      "epoch": 2.6877697841726618,
      "grad_norm": 0.09821683497577681,
      "learning_rate": 4.1747563574927034e-05,
      "loss": 0.3443,
      "step": 1868
    },
    {
      "epoch": 2.689208633093525,
      "grad_norm": 0.12174694705551985,
      "learning_rate": 4.170741440442366e-05,
      "loss": 0.3391,
      "step": 1869
    },
    {
      "epoch": 2.6906474820143886,
      "grad_norm": 0.11064487171338318,
      "learning_rate": 4.166726351053299e-05,
      "loss": 0.3436,
      "step": 1870
    },
    {
      "epoch": 2.692086330935252,
      "grad_norm": 0.11510251312155208,
      "learning_rate": 4.1627110933781515e-05,
      "loss": 0.3448,
      "step": 1871
    },
    {
      "epoch": 2.693525179856115,
      "grad_norm": 0.14301855964769836,
      "learning_rate": 4.158695671469746e-05,
      "loss": 0.3506,
      "step": 1872
    },
    {
      "epoch": 2.6949640287769783,
      "grad_norm": 0.13519930844792627,
      "learning_rate": 4.154680089381068e-05,
      "loss": 0.3455,
      "step": 1873
    },
    {
      "epoch": 2.6964028776978415,
      "grad_norm": 0.11251336867088967,
      "learning_rate": 4.150664351165266e-05,
      "loss": 0.3312,
      "step": 1874
    },
    {
      "epoch": 2.697841726618705,
      "grad_norm": 0.11106285385888814,
      "learning_rate": 4.146648460875646e-05,
      "loss": 0.3436,
      "step": 1875
    },
    {
      "epoch": 2.6992805755395683,
      "grad_norm": 0.09324584790437306,
      "learning_rate": 4.1426324225656644e-05,
      "loss": 0.3377,
      "step": 1876
    },
    {
      "epoch": 2.7007194244604316,
      "grad_norm": 0.10724805131104773,
      "learning_rate": 4.138616240288934e-05,
      "loss": 0.3434,
      "step": 1877
    },
    {
      "epoch": 2.702158273381295,
      "grad_norm": 0.0868909954045789,
      "learning_rate": 4.134599918099204e-05,
      "loss": 0.3436,
      "step": 1878
    },
    {
      "epoch": 2.7035971223021584,
      "grad_norm": 0.10193825613946285,
      "learning_rate": 4.130583460050371e-05,
      "loss": 0.3453,
      "step": 1879
    },
    {
      "epoch": 2.7050359712230216,
      "grad_norm": 0.09979947035317822,
      "learning_rate": 4.126566870196468e-05,
      "loss": 0.3469,
      "step": 1880
    },
    {
      "epoch": 2.706474820143885,
      "grad_norm": 0.10164179590282962,
      "learning_rate": 4.12255015259166e-05,
      "loss": 0.3402,
      "step": 1881
    },
    {
      "epoch": 2.707913669064748,
      "grad_norm": 0.10457175175599578,
      "learning_rate": 4.1185333112902394e-05,
      "loss": 0.3387,
      "step": 1882
    },
    {
      "epoch": 2.7093525179856117,
      "grad_norm": 0.10303970397298673,
      "learning_rate": 4.114516350346626e-05,
      "loss": 0.3342,
      "step": 1883
    },
    {
      "epoch": 2.710791366906475,
      "grad_norm": 0.09774648641990759,
      "learning_rate": 4.1104992738153616e-05,
      "loss": 0.3438,
      "step": 1884
    },
    {
      "epoch": 2.712230215827338,
      "grad_norm": 0.11011856691134238,
      "learning_rate": 4.1064820857511e-05,
      "loss": 0.3385,
      "step": 1885
    },
    {
      "epoch": 2.7136690647482014,
      "grad_norm": 0.10500214455171199,
      "learning_rate": 4.1024647902086107e-05,
      "loss": 0.3484,
      "step": 1886
    },
    {
      "epoch": 2.7151079136690646,
      "grad_norm": 0.09702583849457841,
      "learning_rate": 4.0984473912427706e-05,
      "loss": 0.3459,
      "step": 1887
    },
    {
      "epoch": 2.716546762589928,
      "grad_norm": 0.11101139153632772,
      "learning_rate": 4.0944298929085633e-05,
      "loss": 0.3362,
      "step": 1888
    },
    {
      "epoch": 2.7179856115107914,
      "grad_norm": 0.11344635607958369,
      "learning_rate": 4.090412299261068e-05,
      "loss": 0.3393,
      "step": 1889
    },
    {
      "epoch": 2.7194244604316546,
      "grad_norm": 0.0849582398459722,
      "learning_rate": 4.086394614355467e-05,
      "loss": 0.3331,
      "step": 1890
    },
    {
      "epoch": 2.720863309352518,
      "grad_norm": 0.09560759718322384,
      "learning_rate": 4.082376842247027e-05,
      "loss": 0.3416,
      "step": 1891
    },
    {
      "epoch": 2.722302158273381,
      "grad_norm": 0.11213224875463144,
      "learning_rate": 4.0783589869911074e-05,
      "loss": 0.3443,
      "step": 1892
    },
    {
      "epoch": 2.7237410071942447,
      "grad_norm": 0.1025268180157425,
      "learning_rate": 4.074341052643152e-05,
      "loss": 0.34,
      "step": 1893
    },
    {
      "epoch": 2.725179856115108,
      "grad_norm": 0.10061029461157417,
      "learning_rate": 4.070323043258683e-05,
      "loss": 0.3456,
      "step": 1894
    },
    {
      "epoch": 2.726618705035971,
      "grad_norm": 0.1315208555486423,
      "learning_rate": 4.066304962893297e-05,
      "loss": 0.3405,
      "step": 1895
    },
    {
      "epoch": 2.728057553956835,
      "grad_norm": 0.10202207975723282,
      "learning_rate": 4.062286815602661e-05,
      "loss": 0.3452,
      "step": 1896
    },
    {
      "epoch": 2.7294964028776976,
      "grad_norm": 0.10788336704356037,
      "learning_rate": 4.0582686054425196e-05,
      "loss": 0.3409,
      "step": 1897
    },
    {
      "epoch": 2.7309352517985612,
      "grad_norm": 0.10995402693608562,
      "learning_rate": 4.054250336468666e-05,
      "loss": 0.3361,
      "step": 1898
    },
    {
      "epoch": 2.7323741007194244,
      "grad_norm": 0.08228729974196912,
      "learning_rate": 4.050232012736964e-05,
      "loss": 0.3411,
      "step": 1899
    },
    {
      "epoch": 2.7338129496402876,
      "grad_norm": 0.11300246784808828,
      "learning_rate": 4.0462136383033285e-05,
      "loss": 0.3437,
      "step": 1900
    },
    {
      "epoch": 2.7352517985611513,
      "grad_norm": 0.10330823730912551,
      "learning_rate": 4.0421952172237254e-05,
      "loss": 0.3375,
      "step": 1901
    },
    {
      "epoch": 2.7366906474820145,
      "grad_norm": 0.10158770831666479,
      "learning_rate": 4.038176753554166e-05,
      "loss": 0.34,
      "step": 1902
    },
    {
      "epoch": 2.7381294964028777,
      "grad_norm": 0.09037962763345136,
      "learning_rate": 4.034158251350711e-05,
      "loss": 0.3334,
      "step": 1903
    },
    {
      "epoch": 2.739568345323741,
      "grad_norm": 0.12040518845184638,
      "learning_rate": 4.030139714669453e-05,
      "loss": 0.339,
      "step": 1904
    },
    {
      "epoch": 2.741007194244604,
      "grad_norm": 0.09999156206608065,
      "learning_rate": 4.026121147566522e-05,
      "loss": 0.3487,
      "step": 1905
    },
    {
      "epoch": 2.742446043165468,
      "grad_norm": 0.10962480095976812,
      "learning_rate": 4.02210255409808e-05,
      "loss": 0.3409,
      "step": 1906
    },
    {
      "epoch": 2.743884892086331,
      "grad_norm": 0.14188551617672068,
      "learning_rate": 4.018083938320314e-05,
      "loss": 0.3397,
      "step": 1907
    },
    {
      "epoch": 2.7453237410071942,
      "grad_norm": 0.11311796878797645,
      "learning_rate": 4.014065304289435e-05,
      "loss": 0.3351,
      "step": 1908
    },
    {
      "epoch": 2.7467625899280574,
      "grad_norm": 0.10314189465154625,
      "learning_rate": 4.010046656061669e-05,
      "loss": 0.3472,
      "step": 1909
    },
    {
      "epoch": 2.7482014388489207,
      "grad_norm": 0.13708723801063197,
      "learning_rate": 4.006027997693262e-05,
      "loss": 0.3357,
      "step": 1910
    },
    {
      "epoch": 2.7496402877697843,
      "grad_norm": 0.1116048166839257,
      "learning_rate": 4.002009333240465e-05,
      "loss": 0.3327,
      "step": 1911
    },
    {
      "epoch": 2.7510791366906475,
      "grad_norm": 0.11477952516508165,
      "learning_rate": 3.997990666759536e-05,
      "loss": 0.3459,
      "step": 1912
    },
    {
      "epoch": 2.7525179856115107,
      "grad_norm": 0.09994756450196751,
      "learning_rate": 3.99397200230674e-05,
      "loss": 0.3299,
      "step": 1913
    },
    {
      "epoch": 2.753956834532374,
      "grad_norm": 0.10474276010820825,
      "learning_rate": 3.989953343938331e-05,
      "loss": 0.3431,
      "step": 1914
    },
    {
      "epoch": 2.755395683453237,
      "grad_norm": 0.09656557000221296,
      "learning_rate": 3.985934695710566e-05,
      "loss": 0.3403,
      "step": 1915
    },
    {
      "epoch": 2.756834532374101,
      "grad_norm": 0.08824704913143855,
      "learning_rate": 3.9819160616796873e-05,
      "loss": 0.343,
      "step": 1916
    },
    {
      "epoch": 2.758273381294964,
      "grad_norm": 0.09649943443919938,
      "learning_rate": 3.977897445901922e-05,
      "loss": 0.3452,
      "step": 1917
    },
    {
      "epoch": 2.7597122302158272,
      "grad_norm": 0.10209351657601745,
      "learning_rate": 3.9738788524334794e-05,
      "loss": 0.3392,
      "step": 1918
    },
    {
      "epoch": 2.761151079136691,
      "grad_norm": 0.12454331122782362,
      "learning_rate": 3.969860285330549e-05,
      "loss": 0.3445,
      "step": 1919
    },
    {
      "epoch": 2.762589928057554,
      "grad_norm": 0.10092792322414952,
      "learning_rate": 3.965841748649291e-05,
      "loss": 0.3425,
      "step": 1920
    },
    {
      "epoch": 2.7640287769784173,
      "grad_norm": 0.1077236194758336,
      "learning_rate": 3.961823246445834e-05,
      "loss": 0.3386,
      "step": 1921
    },
    {
      "epoch": 2.7654676258992805,
      "grad_norm": 0.13388696323456153,
      "learning_rate": 3.957804782776276e-05,
      "loss": 0.3441,
      "step": 1922
    },
    {
      "epoch": 2.7669064748201437,
      "grad_norm": 0.10255911600542274,
      "learning_rate": 3.953786361696673e-05,
      "loss": 0.3394,
      "step": 1923
    },
    {
      "epoch": 2.7683453237410074,
      "grad_norm": 0.103328615587966,
      "learning_rate": 3.9497679872630366e-05,
      "loss": 0.3431,
      "step": 1924
    },
    {
      "epoch": 2.7697841726618706,
      "grad_norm": 0.14211785564146165,
      "learning_rate": 3.945749663531334e-05,
      "loss": 0.3425,
      "step": 1925
    },
    {
      "epoch": 2.771223021582734,
      "grad_norm": 0.12904341075634934,
      "learning_rate": 3.941731394557482e-05,
      "loss": 0.3407,
      "step": 1926
    },
    {
      "epoch": 2.772661870503597,
      "grad_norm": 0.11197846800896227,
      "learning_rate": 3.9377131843973394e-05,
      "loss": 0.339,
      "step": 1927
    },
    {
      "epoch": 2.7741007194244602,
      "grad_norm": 0.1429322126035399,
      "learning_rate": 3.933695037106705e-05,
      "loss": 0.3348,
      "step": 1928
    },
    {
      "epoch": 2.775539568345324,
      "grad_norm": 0.1106949557977696,
      "learning_rate": 3.9296769567413177e-05,
      "loss": 0.3485,
      "step": 1929
    },
    {
      "epoch": 2.776978417266187,
      "grad_norm": 0.1510109014335804,
      "learning_rate": 3.925658947356849e-05,
      "loss": 0.3366,
      "step": 1930
    },
    {
      "epoch": 2.7784172661870503,
      "grad_norm": 0.11513010328064764,
      "learning_rate": 3.921641013008893e-05,
      "loss": 0.3378,
      "step": 1931
    },
    {
      "epoch": 2.7798561151079135,
      "grad_norm": 0.09727543648341261,
      "learning_rate": 3.9176231577529734e-05,
      "loss": 0.3465,
      "step": 1932
    },
    {
      "epoch": 2.7812949640287767,
      "grad_norm": 0.0973114578702156,
      "learning_rate": 3.913605385644535e-05,
      "loss": 0.3394,
      "step": 1933
    },
    {
      "epoch": 2.7827338129496404,
      "grad_norm": 0.08922787063098898,
      "learning_rate": 3.909587700738933e-05,
      "loss": 0.3343,
      "step": 1934
    },
    {
      "epoch": 2.7841726618705036,
      "grad_norm": 0.1028230823021992,
      "learning_rate": 3.9055701070914393e-05,
      "loss": 0.3464,
      "step": 1935
    },
    {
      "epoch": 2.785611510791367,
      "grad_norm": 0.09200732441336425,
      "learning_rate": 3.90155260875723e-05,
      "loss": 0.3322,
      "step": 1936
    },
    {
      "epoch": 2.7870503597122305,
      "grad_norm": 0.108749718978638,
      "learning_rate": 3.8975352097913914e-05,
      "loss": 0.3428,
      "step": 1937
    },
    {
      "epoch": 2.7884892086330937,
      "grad_norm": 0.10129148106185752,
      "learning_rate": 3.8935179142489016e-05,
      "loss": 0.3401,
      "step": 1938
    },
    {
      "epoch": 2.789928057553957,
      "grad_norm": 0.0929517364536396,
      "learning_rate": 3.8895007261846404e-05,
      "loss": 0.3376,
      "step": 1939
    },
    {
      "epoch": 2.79136690647482,
      "grad_norm": 0.12330365574595786,
      "learning_rate": 3.885483649653374e-05,
      "loss": 0.3401,
      "step": 1940
    },
    {
      "epoch": 2.7928057553956833,
      "grad_norm": 0.11359836606019805,
      "learning_rate": 3.881466688709761e-05,
      "loss": 0.34,
      "step": 1941
    },
    {
      "epoch": 2.794244604316547,
      "grad_norm": 0.11152528505935266,
      "learning_rate": 3.877449847408342e-05,
      "loss": 0.3456,
      "step": 1942
    },
    {
      "epoch": 2.79568345323741,
      "grad_norm": 0.15288234424007455,
      "learning_rate": 3.873433129803532e-05,
      "loss": 0.3371,
      "step": 1943
    },
    {
      "epoch": 2.7971223021582734,
      "grad_norm": 0.12451350839843649,
      "learning_rate": 3.86941653994963e-05,
      "loss": 0.3499,
      "step": 1944
    },
    {
      "epoch": 2.7985611510791366,
      "grad_norm": 0.13729679818380444,
      "learning_rate": 3.8654000819007974e-05,
      "loss": 0.3486,
      "step": 1945
    },
    {
      "epoch": 2.8,
      "grad_norm": 0.13096931728184064,
      "learning_rate": 3.8613837597110686e-05,
      "loss": 0.3443,
      "step": 1946
    },
    {
      "epoch": 2.8014388489208635,
      "grad_norm": 0.0874959006555631,
      "learning_rate": 3.8573675774343356e-05,
      "loss": 0.3418,
      "step": 1947
    },
    {
      "epoch": 2.8028776978417267,
      "grad_norm": 0.11067866700964694,
      "learning_rate": 3.853351539124355e-05,
      "loss": 0.3453,
      "step": 1948
    },
    {
      "epoch": 2.80431654676259,
      "grad_norm": 0.11136162558610203,
      "learning_rate": 3.8493356488347345e-05,
      "loss": 0.3383,
      "step": 1949
    },
    {
      "epoch": 2.805755395683453,
      "grad_norm": 0.10376991448915758,
      "learning_rate": 3.845319910618933e-05,
      "loss": 0.3351,
      "step": 1950
    },
    {
      "epoch": 2.8071942446043163,
      "grad_norm": 0.11579359325574481,
      "learning_rate": 3.841304328530254e-05,
      "loss": 0.3407,
      "step": 1951
    },
    {
      "epoch": 2.80863309352518,
      "grad_norm": 0.10374915093741167,
      "learning_rate": 3.83728890662185e-05,
      "loss": 0.3392,
      "step": 1952
    },
    {
      "epoch": 2.810071942446043,
      "grad_norm": 0.10029067431414022,
      "learning_rate": 3.833273648946704e-05,
      "loss": 0.3373,
      "step": 1953
    },
    {
      "epoch": 2.8115107913669064,
      "grad_norm": 0.09402337578438068,
      "learning_rate": 3.829258559557635e-05,
      "loss": 0.3272,
      "step": 1954
    },
    {
      "epoch": 2.81294964028777,
      "grad_norm": 0.10650566276574056,
      "learning_rate": 3.825243642507297e-05,
      "loss": 0.3424,
      "step": 1955
    },
    {
      "epoch": 2.814388489208633,
      "grad_norm": 0.10921757363647926,
      "learning_rate": 3.8212289018481666e-05,
      "loss": 0.3431,
      "step": 1956
    },
    {
      "epoch": 2.8158273381294965,
      "grad_norm": 0.10534874915498878,
      "learning_rate": 3.817214341632539e-05,
      "loss": 0.3329,
      "step": 1957
    },
    {
      "epoch": 2.8172661870503597,
      "grad_norm": 0.1015999510780134,
      "learning_rate": 3.813199965912533e-05,
      "loss": 0.3402,
      "step": 1958
    },
    {
      "epoch": 2.818705035971223,
      "grad_norm": 0.09424198994137044,
      "learning_rate": 3.80918577874008e-05,
      "loss": 0.3428,
      "step": 1959
    },
    {
      "epoch": 2.8201438848920866,
      "grad_norm": 0.10554116905363074,
      "learning_rate": 3.8051717841669196e-05,
      "loss": 0.3487,
      "step": 1960
    },
    {
      "epoch": 2.8215827338129498,
      "grad_norm": 0.11403157042696999,
      "learning_rate": 3.801157986244595e-05,
      "loss": 0.3365,
      "step": 1961
    },
    {
      "epoch": 2.823021582733813,
      "grad_norm": 0.10551621664168667,
      "learning_rate": 3.7971443890244534e-05,
      "loss": 0.3335,
      "step": 1962
    },
    {
      "epoch": 2.824460431654676,
      "grad_norm": 0.10515730375531454,
      "learning_rate": 3.7931309965576426e-05,
      "loss": 0.3381,
      "step": 1963
    },
    {
      "epoch": 2.8258992805755394,
      "grad_norm": 0.09972936999039739,
      "learning_rate": 3.7891178128950975e-05,
      "loss": 0.3359,
      "step": 1964
    },
    {
      "epoch": 2.827338129496403,
      "grad_norm": 0.11430428733733113,
      "learning_rate": 3.785104842087546e-05,
      "loss": 0.3328,
      "step": 1965
    },
    {
      "epoch": 2.8287769784172663,
      "grad_norm": 0.14020430699616854,
      "learning_rate": 3.7810920881855016e-05,
      "loss": 0.3338,
      "step": 1966
    },
    {
      "epoch": 2.8302158273381295,
      "grad_norm": 0.096686224918284,
      "learning_rate": 3.777079555239255e-05,
      "loss": 0.335,
      "step": 1967
    },
    {
      "epoch": 2.8316546762589927,
      "grad_norm": 0.16387851406407417,
      "learning_rate": 3.77306724729888e-05,
      "loss": 0.3459,
      "step": 1968
    },
    {
      "epoch": 2.833093525179856,
      "grad_norm": 0.11753990070533193,
      "learning_rate": 3.769055168414215e-05,
      "loss": 0.3432,
      "step": 1969
    },
    {
      "epoch": 2.8345323741007196,
      "grad_norm": 0.12735840341028878,
      "learning_rate": 3.765043322634877e-05,
      "loss": 0.3423,
      "step": 1970
    },
    {
      "epoch": 2.8359712230215828,
      "grad_norm": 0.1329095366969948,
      "learning_rate": 3.761031714010239e-05,
      "loss": 0.3375,
      "step": 1971
    },
    {
      "epoch": 2.837410071942446,
      "grad_norm": 0.09524702053754186,
      "learning_rate": 3.75702034658944e-05,
      "loss": 0.3357,
      "step": 1972
    },
    {
      "epoch": 2.838848920863309,
      "grad_norm": 0.11737566541841826,
      "learning_rate": 3.753009224421373e-05,
      "loss": 0.3498,
      "step": 1973
    },
    {
      "epoch": 2.8402877697841724,
      "grad_norm": 0.09924741895401636,
      "learning_rate": 3.748998351554684e-05,
      "loss": 0.3335,
      "step": 1974
    },
    {
      "epoch": 2.841726618705036,
      "grad_norm": 0.11945876292369449,
      "learning_rate": 3.74498773203777e-05,
      "loss": 0.3348,
      "step": 1975
    },
    {
      "epoch": 2.8431654676258993,
      "grad_norm": 0.12159139949878082,
      "learning_rate": 3.7409773699187664e-05,
      "loss": 0.3384,
      "step": 1976
    },
    {
      "epoch": 2.8446043165467625,
      "grad_norm": 0.10690729286030871,
      "learning_rate": 3.736967269245555e-05,
      "loss": 0.3345,
      "step": 1977
    },
    {
      "epoch": 2.846043165467626,
      "grad_norm": 0.10519347765975301,
      "learning_rate": 3.732957434065751e-05,
      "loss": 0.3409,
      "step": 1978
    },
    {
      "epoch": 2.8474820143884894,
      "grad_norm": 0.09875549429217262,
      "learning_rate": 3.728947868426701e-05,
      "loss": 0.3333,
      "step": 1979
    },
    {
      "epoch": 2.8489208633093526,
      "grad_norm": 0.0877136411088501,
      "learning_rate": 3.724938576375479e-05,
      "loss": 0.3265,
      "step": 1980
    },
    {
      "epoch": 2.850359712230216,
      "grad_norm": 0.09133539784167514,
      "learning_rate": 3.7209295619588856e-05,
      "loss": 0.3438,
      "step": 1981
    },
    {
      "epoch": 2.851798561151079,
      "grad_norm": 0.09612159565294272,
      "learning_rate": 3.7169208292234395e-05,
      "loss": 0.336,
      "step": 1982
    },
    {
      "epoch": 2.8532374100719426,
      "grad_norm": 0.09685530731482227,
      "learning_rate": 3.7129123822153746e-05,
      "loss": 0.3426,
      "step": 1983
    },
    {
      "epoch": 2.854676258992806,
      "grad_norm": 0.11645804028219138,
      "learning_rate": 3.708904224980636e-05,
      "loss": 0.3433,
      "step": 1984
    },
    {
      "epoch": 2.856115107913669,
      "grad_norm": 0.09789254210421806,
      "learning_rate": 3.704896361564881e-05,
      "loss": 0.3403,
      "step": 1985
    },
    {
      "epoch": 2.8575539568345323,
      "grad_norm": 0.09901140761085257,
      "learning_rate": 3.700888796013462e-05,
      "loss": 0.3358,
      "step": 1986
    },
    {
      "epoch": 2.8589928057553955,
      "grad_norm": 0.11827022633011026,
      "learning_rate": 3.696881532371439e-05,
      "loss": 0.3344,
      "step": 1987
    },
    {
      "epoch": 2.860431654676259,
      "grad_norm": 0.10327626295170113,
      "learning_rate": 3.692874574683562e-05,
      "loss": 0.3421,
      "step": 1988
    },
    {
      "epoch": 2.8618705035971224,
      "grad_norm": 0.11279702701966186,
      "learning_rate": 3.688867926994274e-05,
      "loss": 0.3347,
      "step": 1989
    },
    {
      "epoch": 2.8633093525179856,
      "grad_norm": 0.11634377484743742,
      "learning_rate": 3.684861593347705e-05,
      "loss": 0.3388,
      "step": 1990
    },
    {
      "epoch": 2.864748201438849,
      "grad_norm": 0.09635632525853179,
      "learning_rate": 3.6808555777876673e-05,
      "loss": 0.3468,
      "step": 1991
    },
    {
      "epoch": 2.866187050359712,
      "grad_norm": 0.09886557805436955,
      "learning_rate": 3.676849884357655e-05,
      "loss": 0.3456,
      "step": 1992
    },
    {
      "epoch": 2.8676258992805757,
      "grad_norm": 0.10608790458889766,
      "learning_rate": 3.672844517100833e-05,
      "loss": 0.3444,
      "step": 1993
    },
    {
      "epoch": 2.869064748201439,
      "grad_norm": 0.09498842901617018,
      "learning_rate": 3.66883948006004e-05,
      "loss": 0.3399,
      "step": 1994
    },
    {
      "epoch": 2.870503597122302,
      "grad_norm": 0.14887013631600512,
      "learning_rate": 3.664834777277777e-05,
      "loss": 0.3446,
      "step": 1995
    },
    {
      "epoch": 2.8719424460431657,
      "grad_norm": 0.0959301744637005,
      "learning_rate": 3.6608304127962166e-05,
      "loss": 0.3471,
      "step": 1996
    },
    {
      "epoch": 2.873381294964029,
      "grad_norm": 0.12416508438264422,
      "learning_rate": 3.656826390657179e-05,
      "loss": 0.3422,
      "step": 1997
    },
    {
      "epoch": 2.874820143884892,
      "grad_norm": 0.10944433209308456,
      "learning_rate": 3.6528227149021455e-05,
      "loss": 0.3336,
      "step": 1998
    },
    {
      "epoch": 2.8762589928057554,
      "grad_norm": 0.09549258689455808,
      "learning_rate": 3.648819389572248e-05,
      "loss": 0.3355,
      "step": 1999
    },
    {
      "epoch": 2.8776978417266186,
      "grad_norm": 0.10323613880773544,
      "learning_rate": 3.644816418708261e-05,
      "loss": 0.3396,
      "step": 2000
    },
    {
      "epoch": 2.8791366906474822,
      "grad_norm": 0.11161380918729948,
      "learning_rate": 3.6408138063506057e-05,
      "loss": 0.3422,
      "step": 2001
    },
    {
      "epoch": 2.8805755395683454,
      "grad_norm": 0.09144683037649778,
      "learning_rate": 3.636811556539335e-05,
      "loss": 0.3373,
      "step": 2002
    },
    {
      "epoch": 2.8820143884892087,
      "grad_norm": 0.10156626895683483,
      "learning_rate": 3.6328096733141423e-05,
      "loss": 0.3375,
      "step": 2003
    },
    {
      "epoch": 2.883453237410072,
      "grad_norm": 0.10981190963013603,
      "learning_rate": 3.6288081607143496e-05,
      "loss": 0.3374,
      "step": 2004
    },
    {
      "epoch": 2.884892086330935,
      "grad_norm": 0.09486965282121589,
      "learning_rate": 3.6248070227789034e-05,
      "loss": 0.3335,
      "step": 2005
    },
    {
      "epoch": 2.8863309352517987,
      "grad_norm": 0.12553572365453994,
      "learning_rate": 3.620806263546369e-05,
      "loss": 0.3375,
      "step": 2006
    },
    {
      "epoch": 2.887769784172662,
      "grad_norm": 0.08744989482973632,
      "learning_rate": 3.6168058870549355e-05,
      "loss": 0.3399,
      "step": 2007
    },
    {
      "epoch": 2.889208633093525,
      "grad_norm": 0.10419739270922672,
      "learning_rate": 3.612805897342405e-05,
      "loss": 0.3376,
      "step": 2008
    },
    {
      "epoch": 2.8906474820143884,
      "grad_norm": 0.11601086378233597,
      "learning_rate": 3.608806298446182e-05,
      "loss": 0.3413,
      "step": 2009
    },
    {
      "epoch": 2.8920863309352516,
      "grad_norm": 0.10100418652811571,
      "learning_rate": 3.604807094403286e-05,
      "loss": 0.3369,
      "step": 2010
    },
    {
      "epoch": 2.8935251798561152,
      "grad_norm": 0.09789083730632218,
      "learning_rate": 3.6008082892503325e-05,
      "loss": 0.3339,
      "step": 2011
    },
    {
      "epoch": 2.8949640287769784,
      "grad_norm": 0.113231870060525,
      "learning_rate": 3.596809887023534e-05,
      "loss": 0.3426,
      "step": 2012
    },
    {
      "epoch": 2.8964028776978417,
      "grad_norm": 0.08853598945173508,
      "learning_rate": 3.5928118917587e-05,
      "loss": 0.3417,
      "step": 2013
    },
    {
      "epoch": 2.897841726618705,
      "grad_norm": 0.12189669471820712,
      "learning_rate": 3.588814307491227e-05,
      "loss": 0.3352,
      "step": 2014
    },
    {
      "epoch": 2.899280575539568,
      "grad_norm": 0.12976402273747667,
      "learning_rate": 3.584817138256096e-05,
      "loss": 0.3367,
      "step": 2015
    },
    {
      "epoch": 2.9007194244604317,
      "grad_norm": 0.5517604227527648,
      "learning_rate": 3.580820388087869e-05,
      "loss": 0.3398,
      "step": 2016
    },
    {
      "epoch": 2.902158273381295,
      "grad_norm": 0.11954468702025425,
      "learning_rate": 3.5768240610206855e-05,
      "loss": 0.3519,
      "step": 2017
    },
    {
      "epoch": 2.903597122302158,
      "grad_norm": 0.12000771579204647,
      "learning_rate": 3.572828161088262e-05,
      "loss": 0.3386,
      "step": 2018
    },
    {
      "epoch": 2.905035971223022,
      "grad_norm": 0.1160108192635706,
      "learning_rate": 3.568832692323876e-05,
      "loss": 0.3333,
      "step": 2019
    },
    {
      "epoch": 2.906474820143885,
      "grad_norm": 0.33604009024132314,
      "learning_rate": 3.564837658760376e-05,
      "loss": 0.3488,
      "step": 2020
    },
    {
      "epoch": 2.9079136690647482,
      "grad_norm": 0.13931330547334667,
      "learning_rate": 3.560843064430168e-05,
      "loss": 0.3425,
      "step": 2021
    },
    {
      "epoch": 2.9093525179856115,
      "grad_norm": 0.2274196971791949,
      "learning_rate": 3.556848913365218e-05,
      "loss": 0.3448,
      "step": 2022
    },
    {
      "epoch": 2.9107913669064747,
      "grad_norm": 0.14845749220476806,
      "learning_rate": 3.552855209597039e-05,
      "loss": 0.3501,
      "step": 2023
    },
    {
      "epoch": 2.9122302158273383,
      "grad_norm": 0.14042284150124493,
      "learning_rate": 3.548861957156698e-05,
      "loss": 0.3344,
      "step": 2024
    },
    {
      "epoch": 2.9136690647482015,
      "grad_norm": 0.19669862745580163,
      "learning_rate": 3.544869160074806e-05,
      "loss": 0.3445,
      "step": 2025
    },
    {
      "epoch": 2.9151079136690647,
      "grad_norm": 0.12080075666826642,
      "learning_rate": 3.5408768223815105e-05,
      "loss": 0.3433,
      "step": 2026
    },
    {
      "epoch": 2.916546762589928,
      "grad_norm": 0.11359617833009475,
      "learning_rate": 3.536884948106498e-05,
      "loss": 0.3418,
      "step": 2027
    },
    {
      "epoch": 2.917985611510791,
      "grad_norm": 0.1241496184644772,
      "learning_rate": 3.532893541278986e-05,
      "loss": 0.3449,
      "step": 2028
    },
    {
      "epoch": 2.919424460431655,
      "grad_norm": 0.1252420579703172,
      "learning_rate": 3.528902605927722e-05,
      "loss": 0.3485,
      "step": 2029
    },
    {
      "epoch": 2.920863309352518,
      "grad_norm": 0.12099295193096923,
      "learning_rate": 3.524912146080978e-05,
      "loss": 0.3363,
      "step": 2030
    },
    {
      "epoch": 2.9223021582733812,
      "grad_norm": 0.10313434029543694,
      "learning_rate": 3.5209221657665436e-05,
      "loss": 0.3442,
      "step": 2031
    },
    {
      "epoch": 2.9237410071942445,
      "grad_norm": 0.11331349519355352,
      "learning_rate": 3.516932669011723e-05,
      "loss": 0.3457,
      "step": 2032
    },
    {
      "epoch": 2.9251798561151077,
      "grad_norm": 0.10827193734464523,
      "learning_rate": 3.512943659843337e-05,
      "loss": 0.3446,
      "step": 2033
    },
    {
      "epoch": 2.9266187050359713,
      "grad_norm": 0.13373768241898437,
      "learning_rate": 3.508955142287714e-05,
      "loss": 0.3443,
      "step": 2034
    },
    {
      "epoch": 2.9280575539568345,
      "grad_norm": 0.11755344172389338,
      "learning_rate": 3.50496712037068e-05,
      "loss": 0.3378,
      "step": 2035
    },
    {
      "epoch": 2.9294964028776977,
      "grad_norm": 0.1092987257661601,
      "learning_rate": 3.5009795981175676e-05,
      "loss": 0.3451,
      "step": 2036
    },
    {
      "epoch": 2.9309352517985614,
      "grad_norm": 0.13471333476409209,
      "learning_rate": 3.496992579553203e-05,
      "loss": 0.3414,
      "step": 2037
    },
    {
      "epoch": 2.9323741007194246,
      "grad_norm": 0.09778093965691664,
      "learning_rate": 3.4930060687019015e-05,
      "loss": 0.3398,
      "step": 2038
    },
    {
      "epoch": 2.933812949640288,
      "grad_norm": 0.11407068505788961,
      "learning_rate": 3.489020069587467e-05,
      "loss": 0.3382,
      "step": 2039
    },
    {
      "epoch": 2.935251798561151,
      "grad_norm": 0.10556720511828248,
      "learning_rate": 3.485034586233192e-05,
      "loss": 0.3315,
      "step": 2040
    },
    {
      "epoch": 2.9366906474820142,
      "grad_norm": 0.09627186947222915,
      "learning_rate": 3.4810496226618404e-05,
      "loss": 0.3393,
      "step": 2041
    },
    {
      "epoch": 2.938129496402878,
      "grad_norm": 0.11039089616509497,
      "learning_rate": 3.477065182895656e-05,
      "loss": 0.3369,
      "step": 2042
    },
    {
      "epoch": 2.939568345323741,
      "grad_norm": 0.09607262830034538,
      "learning_rate": 3.473081270956352e-05,
      "loss": 0.3374,
      "step": 2043
    },
    {
      "epoch": 2.9410071942446043,
      "grad_norm": 0.101459526198804,
      "learning_rate": 3.469097890865113e-05,
      "loss": 0.3401,
      "step": 2044
    },
    {
      "epoch": 2.9424460431654675,
      "grad_norm": 0.11182305048557227,
      "learning_rate": 3.465115046642581e-05,
      "loss": 0.3394,
      "step": 2045
    },
    {
      "epoch": 2.9438848920863308,
      "grad_norm": 0.10131498840239124,
      "learning_rate": 3.461132742308859e-05,
      "loss": 0.3389,
      "step": 2046
    },
    {
      "epoch": 2.9453237410071944,
      "grad_norm": 0.0941180917869226,
      "learning_rate": 3.45715098188351e-05,
      "loss": 0.3368,
      "step": 2047
    },
    {
      "epoch": 2.9467625899280576,
      "grad_norm": 0.10370672140159803,
      "learning_rate": 3.453169769385541e-05,
      "loss": 0.3326,
      "step": 2048
    },
    {
      "epoch": 2.948201438848921,
      "grad_norm": 0.09591896159588302,
      "learning_rate": 3.449189108833409e-05,
      "loss": 0.3441,
      "step": 2049
    },
    {
      "epoch": 2.949640287769784,
      "grad_norm": 0.09936462705055035,
      "learning_rate": 3.445209004245012e-05,
      "loss": 0.3385,
      "step": 2050
    },
    {
      "epoch": 2.9510791366906473,
      "grad_norm": 0.09429491483624323,
      "learning_rate": 3.441229459637693e-05,
      "loss": 0.3504,
      "step": 2051
    },
    {
      "epoch": 2.952517985611511,
      "grad_norm": 0.09881581153293392,
      "learning_rate": 3.4372504790282215e-05,
      "loss": 0.3428,
      "step": 2052
    },
    {
      "epoch": 2.953956834532374,
      "grad_norm": 0.09743132407484635,
      "learning_rate": 3.4332720664328034e-05,
      "loss": 0.3378,
      "step": 2053
    },
    {
      "epoch": 2.9553956834532373,
      "grad_norm": 0.09791054902725754,
      "learning_rate": 3.4292942258670675e-05,
      "loss": 0.333,
      "step": 2054
    },
    {
      "epoch": 2.956834532374101,
      "grad_norm": 0.10633424241620104,
      "learning_rate": 3.425316961346069e-05,
      "loss": 0.3512,
      "step": 2055
    },
    {
      "epoch": 2.9582733812949638,
      "grad_norm": 0.09523089276447394,
      "learning_rate": 3.42134027688428e-05,
      "loss": 0.3449,
      "step": 2056
    },
    {
      "epoch": 2.9597122302158274,
      "grad_norm": 0.10504482623409091,
      "learning_rate": 3.417364176495585e-05,
      "loss": 0.337,
      "step": 2057
    },
    {
      "epoch": 2.9611510791366906,
      "grad_norm": 0.0910028613044847,
      "learning_rate": 3.4133886641932834e-05,
      "loss": 0.3352,
      "step": 2058
    },
    {
      "epoch": 2.962589928057554,
      "grad_norm": 0.08738779213644705,
      "learning_rate": 3.409413743990076e-05,
      "loss": 0.3352,
      "step": 2059
    },
    {
      "epoch": 2.9640287769784175,
      "grad_norm": 0.11240302590903321,
      "learning_rate": 3.4054394198980705e-05,
      "loss": 0.3358,
      "step": 2060
    },
    {
      "epoch": 2.9654676258992807,
      "grad_norm": 0.0923897668090509,
      "learning_rate": 3.401465695928768e-05,
      "loss": 0.3525,
      "step": 2061
    },
    {
      "epoch": 2.966906474820144,
      "grad_norm": 0.10130489279016075,
      "learning_rate": 3.3974925760930694e-05,
      "loss": 0.3336,
      "step": 2062
    },
    {
      "epoch": 2.968345323741007,
      "grad_norm": 0.09116925176685219,
      "learning_rate": 3.393520064401264e-05,
      "loss": 0.3437,
      "step": 2063
    },
    {
      "epoch": 2.9697841726618703,
      "grad_norm": 0.09357240837963861,
      "learning_rate": 3.3895481648630234e-05,
      "loss": 0.3397,
      "step": 2064
    },
    {
      "epoch": 2.971223021582734,
      "grad_norm": 0.08921155663052444,
      "learning_rate": 3.385576881487405e-05,
      "loss": 0.3463,
      "step": 2065
    },
    {
      "epoch": 2.972661870503597,
      "grad_norm": 0.18879817973341476,
      "learning_rate": 3.381606218282846e-05,
      "loss": 0.3495,
      "step": 2066
    },
    {
      "epoch": 2.9741007194244604,
      "grad_norm": 0.0828229553796737,
      "learning_rate": 3.377636179257153e-05,
      "loss": 0.34,
      "step": 2067
    },
    {
      "epoch": 2.9755395683453236,
      "grad_norm": 0.0883365063239282,
      "learning_rate": 3.373666768417505e-05,
      "loss": 0.3385,
      "step": 2068
    },
    {
      "epoch": 2.976978417266187,
      "grad_norm": 0.0843524304405967,
      "learning_rate": 3.3696979897704466e-05,
      "loss": 0.3367,
      "step": 2069
    },
    {
      "epoch": 2.9784172661870505,
      "grad_norm": 0.08721692620414641,
      "learning_rate": 3.3657298473218864e-05,
      "loss": 0.3398,
      "step": 2070
    },
    {
      "epoch": 2.9798561151079137,
      "grad_norm": 0.08381150306485105,
      "learning_rate": 3.361762345077087e-05,
      "loss": 0.3337,
      "step": 2071
    },
    {
      "epoch": 2.981294964028777,
      "grad_norm": 0.08431930869069568,
      "learning_rate": 3.3577954870406656e-05,
      "loss": 0.3333,
      "step": 2072
    },
    {
      "epoch": 2.98273381294964,
      "grad_norm": 0.09523256103847537,
      "learning_rate": 3.3538292772165936e-05,
      "loss": 0.3456,
      "step": 2073
    },
    {
      "epoch": 2.9841726618705033,
      "grad_norm": 0.08844468183438381,
      "learning_rate": 3.3498637196081825e-05,
      "loss": 0.3467,
      "step": 2074
    },
    {
      "epoch": 2.985611510791367,
      "grad_norm": 0.11491416674856708,
      "learning_rate": 3.345898818218089e-05,
      "loss": 0.3351,
      "step": 2075
    },
    {
      "epoch": 2.98705035971223,
      "grad_norm": 0.0899026646635497,
      "learning_rate": 3.341934577048304e-05,
      "loss": 0.334,
      "step": 2076
    },
    {
      "epoch": 2.9884892086330934,
      "grad_norm": 0.09680106609239067,
      "learning_rate": 3.337971000100157e-05,
      "loss": 0.3389,
      "step": 2077
    },
    {
      "epoch": 2.989928057553957,
      "grad_norm": 0.09716428675247937,
      "learning_rate": 3.334008091374303e-05,
      "loss": 0.3421,
      "step": 2078
    },
    {
      "epoch": 2.9913669064748203,
      "grad_norm": 0.09397649824466278,
      "learning_rate": 3.3300458548707214e-05,
      "loss": 0.3429,
      "step": 2079
    },
    {
      "epoch": 2.9928057553956835,
      "grad_norm": 0.1032790560661443,
      "learning_rate": 3.326084294588721e-05,
      "loss": 0.3398,
      "step": 2080
    },
    {
      "epoch": 2.9942446043165467,
      "grad_norm": 0.08806441098719499,
      "learning_rate": 3.322123414526917e-05,
      "loss": 0.3374,
      "step": 2081
    },
    {
      "epoch": 2.99568345323741,
      "grad_norm": 0.09040602131734916,
      "learning_rate": 3.3181632186832485e-05,
      "loss": 0.3438,
      "step": 2082
    },
    {
      "epoch": 2.9971223021582736,
      "grad_norm": 0.08450635151958065,
      "learning_rate": 3.3142037110549546e-05,
      "loss": 0.3374,
      "step": 2083
    },
    {
      "epoch": 2.998561151079137,
      "grad_norm": 0.08672084923982686,
      "learning_rate": 3.310244895638587e-05,
      "loss": 0.3392,
      "step": 2084
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.11705616335925603,
      "learning_rate": 3.306286776429995e-05,
      "loss": 0.3239,
      "step": 2085
    },
    {
      "epoch": 3.001438848920863,
      "grad_norm": 0.13793126150799295,
      "learning_rate": 3.302329357424326e-05,
      "loss": 0.3123,
      "step": 2086
    },
    {
      "epoch": 3.0028776978417264,
      "grad_norm": 0.13603474523355813,
      "learning_rate": 3.2983726426160204e-05,
      "loss": 0.3217,
      "step": 2087
    },
    {
      "epoch": 3.00431654676259,
      "grad_norm": 0.1856790529153839,
      "learning_rate": 3.2944166359988083e-05,
      "loss": 0.3162,
      "step": 2088
    },
    {
      "epoch": 3.0057553956834533,
      "grad_norm": 0.1611176916591849,
      "learning_rate": 3.290461341565707e-05,
      "loss": 0.3148,
      "step": 2089
    },
    {
      "epoch": 3.0071942446043165,
      "grad_norm": 0.16726575474877292,
      "learning_rate": 3.286506763309009e-05,
      "loss": 0.3167,
      "step": 2090
    },
    {
      "epoch": 3.0086330935251797,
      "grad_norm": 0.15380789186230392,
      "learning_rate": 3.2825529052202904e-05,
      "loss": 0.3218,
      "step": 2091
    },
    {
      "epoch": 3.0100719424460434,
      "grad_norm": 0.16337070969766343,
      "learning_rate": 3.278599771290397e-05,
      "loss": 0.3158,
      "step": 2092
    },
    {
      "epoch": 3.0115107913669066,
      "grad_norm": 0.14398184962901214,
      "learning_rate": 3.274647365509445e-05,
      "loss": 0.3147,
      "step": 2093
    },
    {
      "epoch": 3.01294964028777,
      "grad_norm": 0.11501375024418786,
      "learning_rate": 3.2706956918668126e-05,
      "loss": 0.309,
      "step": 2094
    },
    {
      "epoch": 3.014388489208633,
      "grad_norm": 0.15656933461861816,
      "learning_rate": 3.266744754351144e-05,
      "loss": 0.3078,
      "step": 2095
    },
    {
      "epoch": 3.015827338129496,
      "grad_norm": 0.11864428376109526,
      "learning_rate": 3.262794556950338e-05,
      "loss": 0.3087,
      "step": 2096
    },
    {
      "epoch": 3.01726618705036,
      "grad_norm": 0.1068241730293408,
      "learning_rate": 3.2588451036515435e-05,
      "loss": 0.3012,
      "step": 2097
    },
    {
      "epoch": 3.018705035971223,
      "grad_norm": 0.11991144173205176,
      "learning_rate": 3.2548963984411623e-05,
      "loss": 0.3064,
      "step": 2098
    },
    {
      "epoch": 3.0201438848920863,
      "grad_norm": 0.11295063162718329,
      "learning_rate": 3.2509484453048413e-05,
      "loss": 0.3069,
      "step": 2099
    },
    {
      "epoch": 3.0215827338129495,
      "grad_norm": 0.10013282575374777,
      "learning_rate": 3.247001248227465e-05,
      "loss": 0.3229,
      "step": 2100
    },
    {
      "epoch": 3.023021582733813,
      "grad_norm": 0.12506616455815658,
      "learning_rate": 3.2430548111931574e-05,
      "loss": 0.3089,
      "step": 2101
    },
    {
      "epoch": 3.0244604316546764,
      "grad_norm": 0.10517337087645212,
      "learning_rate": 3.239109138185275e-05,
      "loss": 0.3069,
      "step": 2102
    },
    {
      "epoch": 3.0258992805755396,
      "grad_norm": 0.10493068657974713,
      "learning_rate": 3.2351642331864024e-05,
      "loss": 0.2984,
      "step": 2103
    },
    {
      "epoch": 3.027338129496403,
      "grad_norm": 0.11380836450829623,
      "learning_rate": 3.2312201001783473e-05,
      "loss": 0.3162,
      "step": 2104
    },
    {
      "epoch": 3.028776978417266,
      "grad_norm": 0.10722577915109091,
      "learning_rate": 3.2272767431421416e-05,
      "loss": 0.31,
      "step": 2105
    },
    {
      "epoch": 3.0302158273381297,
      "grad_norm": 0.12162362941969909,
      "learning_rate": 3.2233341660580335e-05,
      "loss": 0.3083,
      "step": 2106
    },
    {
      "epoch": 3.031654676258993,
      "grad_norm": 0.11168047128844485,
      "learning_rate": 3.219392372905482e-05,
      "loss": 0.3106,
      "step": 2107
    },
    {
      "epoch": 3.033093525179856,
      "grad_norm": 0.12843621459373988,
      "learning_rate": 3.215451367663156e-05,
      "loss": 0.3081,
      "step": 2108
    },
    {
      "epoch": 3.0345323741007193,
      "grad_norm": 0.09702141129244819,
      "learning_rate": 3.211511154308927e-05,
      "loss": 0.3084,
      "step": 2109
    },
    {
      "epoch": 3.0359712230215825,
      "grad_norm": 0.12510881961189294,
      "learning_rate": 3.207571736819873e-05,
      "loss": 0.3097,
      "step": 2110
    },
    {
      "epoch": 3.037410071942446,
      "grad_norm": 0.08843205023499215,
      "learning_rate": 3.203633119172262e-05,
      "loss": 0.313,
      "step": 2111
    },
    {
      "epoch": 3.0388489208633094,
      "grad_norm": 0.10911141103303929,
      "learning_rate": 3.1996953053415575e-05,
      "loss": 0.3062,
      "step": 2112
    },
    {
      "epoch": 3.0402877697841726,
      "grad_norm": 0.11128508018839607,
      "learning_rate": 3.1957582993024135e-05,
      "loss": 0.3032,
      "step": 2113
    },
    {
      "epoch": 3.041726618705036,
      "grad_norm": 0.2589859061373978,
      "learning_rate": 3.191822105028665e-05,
      "loss": 0.3115,
      "step": 2114
    },
    {
      "epoch": 3.0431654676258995,
      "grad_norm": 0.10701641516972452,
      "learning_rate": 3.1878867264933305e-05,
      "loss": 0.3222,
      "step": 2115
    },
    {
      "epoch": 3.0446043165467627,
      "grad_norm": 0.10343551330065237,
      "learning_rate": 3.1839521676686026e-05,
      "loss": 0.3129,
      "step": 2116
    },
    {
      "epoch": 3.046043165467626,
      "grad_norm": 0.09286316473328367,
      "learning_rate": 3.1800184325258494e-05,
      "loss": 0.31,
      "step": 2117
    },
    {
      "epoch": 3.047482014388489,
      "grad_norm": 0.10606551785976652,
      "learning_rate": 3.176085525035607e-05,
      "loss": 0.3084,
      "step": 2118
    },
    {
      "epoch": 3.0489208633093523,
      "grad_norm": 0.09781749171984888,
      "learning_rate": 3.172153449167574e-05,
      "loss": 0.3184,
      "step": 2119
    },
    {
      "epoch": 3.050359712230216,
      "grad_norm": 0.10422034929359225,
      "learning_rate": 3.1682222088906096e-05,
      "loss": 0.3118,
      "step": 2120
    },
    {
      "epoch": 3.051798561151079,
      "grad_norm": 0.08864611099382119,
      "learning_rate": 3.1642918081727327e-05,
      "loss": 0.3197,
      "step": 2121
    },
    {
      "epoch": 3.0532374100719424,
      "grad_norm": 0.36668073828222714,
      "learning_rate": 3.1603622509811144e-05,
      "loss": 0.3119,
      "step": 2122
    },
    {
      "epoch": 3.0546762589928056,
      "grad_norm": 0.10341831271326356,
      "learning_rate": 3.156433541282069e-05,
      "loss": 0.3085,
      "step": 2123
    },
    {
      "epoch": 3.0561151079136692,
      "grad_norm": 0.10780961775187786,
      "learning_rate": 3.152505683041062e-05,
      "loss": 0.3123,
      "step": 2124
    },
    {
      "epoch": 3.0575539568345325,
      "grad_norm": 0.12743920608580703,
      "learning_rate": 3.1485786802226976e-05,
      "loss": 0.3131,
      "step": 2125
    },
    {
      "epoch": 3.0589928057553957,
      "grad_norm": 0.09614525287762177,
      "learning_rate": 3.1446525367907134e-05,
      "loss": 0.3159,
      "step": 2126
    },
    {
      "epoch": 3.060431654676259,
      "grad_norm": 0.1015553385159086,
      "learning_rate": 3.1407272567079834e-05,
      "loss": 0.3121,
      "step": 2127
    },
    {
      "epoch": 3.061870503597122,
      "grad_norm": 0.10752800721482238,
      "learning_rate": 3.136802843936509e-05,
      "loss": 0.312,
      "step": 2128
    },
    {
      "epoch": 3.0633093525179858,
      "grad_norm": 0.19324908126515103,
      "learning_rate": 3.132879302437416e-05,
      "loss": 0.3131,
      "step": 2129
    },
    {
      "epoch": 3.064748201438849,
      "grad_norm": 0.09708957563754647,
      "learning_rate": 3.128956636170949e-05,
      "loss": 0.3101,
      "step": 2130
    },
    {
      "epoch": 3.066187050359712,
      "grad_norm": 0.09301531792483879,
      "learning_rate": 3.125034849096471e-05,
      "loss": 0.3059,
      "step": 2131
    },
    {
      "epoch": 3.0676258992805754,
      "grad_norm": 0.10400184737956308,
      "learning_rate": 3.1211139451724605e-05,
      "loss": 0.3119,
      "step": 2132
    },
    {
      "epoch": 3.069064748201439,
      "grad_norm": 0.10684526359696246,
      "learning_rate": 3.1171939283564986e-05,
      "loss": 0.3095,
      "step": 2133
    },
    {
      "epoch": 3.0705035971223023,
      "grad_norm": 0.10738487120925314,
      "learning_rate": 3.113274802605276e-05,
      "loss": 0.3006,
      "step": 2134
    },
    {
      "epoch": 3.0719424460431655,
      "grad_norm": 0.0810562667527916,
      "learning_rate": 3.109356571874579e-05,
      "loss": 0.3096,
      "step": 2135
    },
    {
      "epoch": 3.0733812949640287,
      "grad_norm": 0.268649406492071,
      "learning_rate": 3.105439240119296e-05,
      "loss": 0.321,
      "step": 2136
    },
    {
      "epoch": 3.074820143884892,
      "grad_norm": 0.10376615835087384,
      "learning_rate": 3.101522811293405e-05,
      "loss": 0.3194,
      "step": 2137
    },
    {
      "epoch": 3.0762589928057555,
      "grad_norm": 0.10463073075381597,
      "learning_rate": 3.0976072893499724e-05,
      "loss": 0.3184,
      "step": 2138
    },
    {
      "epoch": 3.0776978417266188,
      "grad_norm": 0.09668657967871916,
      "learning_rate": 3.093692678241151e-05,
      "loss": 0.3092,
      "step": 2139
    },
    {
      "epoch": 3.079136690647482,
      "grad_norm": 0.10680206670349111,
      "learning_rate": 3.0897789819181715e-05,
      "loss": 0.3103,
      "step": 2140
    },
    {
      "epoch": 3.080575539568345,
      "grad_norm": 0.0996584820848933,
      "learning_rate": 3.0858662043313456e-05,
      "loss": 0.3055,
      "step": 2141
    },
    {
      "epoch": 3.082014388489209,
      "grad_norm": 0.12307212788294597,
      "learning_rate": 3.081954349430051e-05,
      "loss": 0.3091,
      "step": 2142
    },
    {
      "epoch": 3.083453237410072,
      "grad_norm": 0.08558726684483627,
      "learning_rate": 3.0780434211627415e-05,
      "loss": 0.3131,
      "step": 2143
    },
    {
      "epoch": 3.0848920863309353,
      "grad_norm": 0.1172950866722576,
      "learning_rate": 3.074133423476932e-05,
      "loss": 0.3156,
      "step": 2144
    },
    {
      "epoch": 3.0863309352517985,
      "grad_norm": 0.10469453062047866,
      "learning_rate": 3.070224360319197e-05,
      "loss": 0.3218,
      "step": 2145
    },
    {
      "epoch": 3.0877697841726617,
      "grad_norm": 0.0925454967053722,
      "learning_rate": 3.066316235635168e-05,
      "loss": 0.3041,
      "step": 2146
    },
    {
      "epoch": 3.0892086330935253,
      "grad_norm": 0.10145715448704891,
      "learning_rate": 3.0624090533695324e-05,
      "loss": 0.3158,
      "step": 2147
    },
    {
      "epoch": 3.0906474820143885,
      "grad_norm": 0.11844817927214425,
      "learning_rate": 3.0585028174660236e-05,
      "loss": 0.3095,
      "step": 2148
    },
    {
      "epoch": 3.0920863309352518,
      "grad_norm": 0.09077288092441493,
      "learning_rate": 3.054597531867419e-05,
      "loss": 0.3059,
      "step": 2149
    },
    {
      "epoch": 3.093525179856115,
      "grad_norm": 0.11748236066697641,
      "learning_rate": 3.0506932005155407e-05,
      "loss": 0.3081,
      "step": 2150
    },
    {
      "epoch": 3.0949640287769786,
      "grad_norm": 0.11190838616637672,
      "learning_rate": 3.0467898273512446e-05,
      "loss": 0.3197,
      "step": 2151
    },
    {
      "epoch": 3.096402877697842,
      "grad_norm": 0.11697295478129117,
      "learning_rate": 3.042887416314418e-05,
      "loss": 0.3155,
      "step": 2152
    },
    {
      "epoch": 3.097841726618705,
      "grad_norm": 0.10233140792591265,
      "learning_rate": 3.03898597134398e-05,
      "loss": 0.3091,
      "step": 2153
    },
    {
      "epoch": 3.0992805755395683,
      "grad_norm": 0.09267504676657475,
      "learning_rate": 3.0350854963778755e-05,
      "loss": 0.3126,
      "step": 2154
    },
    {
      "epoch": 3.1007194244604315,
      "grad_norm": 0.11576708311909643,
      "learning_rate": 3.0311859953530672e-05,
      "loss": 0.3073,
      "step": 2155
    },
    {
      "epoch": 3.102158273381295,
      "grad_norm": 0.0879816111435866,
      "learning_rate": 3.027287472205535e-05,
      "loss": 0.3064,
      "step": 2156
    },
    {
      "epoch": 3.1035971223021583,
      "grad_norm": 0.09573994059522453,
      "learning_rate": 3.0233899308702722e-05,
      "loss": 0.3138,
      "step": 2157
    },
    {
      "epoch": 3.1050359712230216,
      "grad_norm": 0.11650012965328882,
      "learning_rate": 3.0194933752812853e-05,
      "loss": 0.3111,
      "step": 2158
    },
    {
      "epoch": 3.1064748201438848,
      "grad_norm": 0.0832860465680782,
      "learning_rate": 3.0155978093715787e-05,
      "loss": 0.3195,
      "step": 2159
    },
    {
      "epoch": 3.1079136690647484,
      "grad_norm": 0.10908168360989089,
      "learning_rate": 3.011703237073162e-05,
      "loss": 0.3075,
      "step": 2160
    },
    {
      "epoch": 3.1093525179856116,
      "grad_norm": 0.10251962926405986,
      "learning_rate": 3.0078096623170442e-05,
      "loss": 0.306,
      "step": 2161
    },
    {
      "epoch": 3.110791366906475,
      "grad_norm": 0.08614700299827016,
      "learning_rate": 3.0039170890332214e-05,
      "loss": 0.3146,
      "step": 2162
    },
    {
      "epoch": 3.112230215827338,
      "grad_norm": 0.10245047075815311,
      "learning_rate": 3.0000255211506836e-05,
      "loss": 0.3077,
      "step": 2163
    },
    {
      "epoch": 3.1136690647482013,
      "grad_norm": 0.09054934591074866,
      "learning_rate": 2.9961349625974022e-05,
      "loss": 0.3114,
      "step": 2164
    },
    {
      "epoch": 3.115107913669065,
      "grad_norm": 0.08382115872482102,
      "learning_rate": 2.992245417300335e-05,
      "loss": 0.3032,
      "step": 2165
    },
    {
      "epoch": 3.116546762589928,
      "grad_norm": 0.09618357746864291,
      "learning_rate": 2.9883568891854118e-05,
      "loss": 0.3133,
      "step": 2166
    },
    {
      "epoch": 3.1179856115107913,
      "grad_norm": 0.07157257749831544,
      "learning_rate": 2.9844693821775394e-05,
      "loss": 0.3151,
      "step": 2167
    },
    {
      "epoch": 3.1194244604316546,
      "grad_norm": 0.09815562942238666,
      "learning_rate": 2.9805829002005907e-05,
      "loss": 0.3157,
      "step": 2168
    },
    {
      "epoch": 3.1208633093525178,
      "grad_norm": 0.10227029481953566,
      "learning_rate": 2.9766974471774072e-05,
      "loss": 0.3091,
      "step": 2169
    },
    {
      "epoch": 3.1223021582733814,
      "grad_norm": 0.19938457580233848,
      "learning_rate": 2.9728130270297913e-05,
      "loss": 0.3076,
      "step": 2170
    },
    {
      "epoch": 3.1237410071942446,
      "grad_norm": 0.0938557233143553,
      "learning_rate": 2.968929643678499e-05,
      "loss": 0.3172,
      "step": 2171
    },
    {
      "epoch": 3.125179856115108,
      "grad_norm": 0.08048511224507746,
      "learning_rate": 2.965047301043246e-05,
      "loss": 0.3129,
      "step": 2172
    },
    {
      "epoch": 3.126618705035971,
      "grad_norm": 0.10823148930623065,
      "learning_rate": 2.961166003042692e-05,
      "loss": 0.3188,
      "step": 2173
    },
    {
      "epoch": 3.1280575539568347,
      "grad_norm": 0.07990693174638736,
      "learning_rate": 2.9572857535944473e-05,
      "loss": 0.3151,
      "step": 2174
    },
    {
      "epoch": 3.129496402877698,
      "grad_norm": 0.07803203834549868,
      "learning_rate": 2.9534065566150567e-05,
      "loss": 0.3054,
      "step": 2175
    },
    {
      "epoch": 3.130935251798561,
      "grad_norm": 0.07812299012768215,
      "learning_rate": 2.9495284160200105e-05,
      "loss": 0.3145,
      "step": 2176
    },
    {
      "epoch": 3.1323741007194243,
      "grad_norm": 0.07117217556524316,
      "learning_rate": 2.9456513357237305e-05,
      "loss": 0.31,
      "step": 2177
    },
    {
      "epoch": 3.133812949640288,
      "grad_norm": 0.08353609633978562,
      "learning_rate": 2.9417753196395637e-05,
      "loss": 0.319,
      "step": 2178
    },
    {
      "epoch": 3.135251798561151,
      "grad_norm": 0.09055984034536244,
      "learning_rate": 2.9379003716797877e-05,
      "loss": 0.321,
      "step": 2179
    },
    {
      "epoch": 3.1366906474820144,
      "grad_norm": 0.08081929164201893,
      "learning_rate": 2.9340264957556018e-05,
      "loss": 0.3108,
      "step": 2180
    },
    {
      "epoch": 3.1381294964028776,
      "grad_norm": 0.09007488058816597,
      "learning_rate": 2.9301536957771218e-05,
      "loss": 0.3141,
      "step": 2181
    },
    {
      "epoch": 3.139568345323741,
      "grad_norm": 0.08158773384180579,
      "learning_rate": 2.9262819756533754e-05,
      "loss": 0.303,
      "step": 2182
    },
    {
      "epoch": 3.1410071942446045,
      "grad_norm": 0.0887930878270568,
      "learning_rate": 2.922411339292306e-05,
      "loss": 0.3187,
      "step": 2183
    },
    {
      "epoch": 3.1424460431654677,
      "grad_norm": 0.09076889185748031,
      "learning_rate": 2.9185417906007586e-05,
      "loss": 0.3075,
      "step": 2184
    },
    {
      "epoch": 3.143884892086331,
      "grad_norm": 0.09010997903238532,
      "learning_rate": 2.914673333484481e-05,
      "loss": 0.3153,
      "step": 2185
    },
    {
      "epoch": 3.145323741007194,
      "grad_norm": 0.08601074144589695,
      "learning_rate": 2.9108059718481184e-05,
      "loss": 0.3136,
      "step": 2186
    },
    {
      "epoch": 3.1467625899280574,
      "grad_norm": 0.09473825701510877,
      "learning_rate": 2.906939709595216e-05,
      "loss": 0.3099,
      "step": 2187
    },
    {
      "epoch": 3.148201438848921,
      "grad_norm": 0.07940853642607197,
      "learning_rate": 2.9030745506282017e-05,
      "loss": 0.3081,
      "step": 2188
    },
    {
      "epoch": 3.149640287769784,
      "grad_norm": 0.09176280731150839,
      "learning_rate": 2.8992104988483943e-05,
      "loss": 0.3048,
      "step": 2189
    },
    {
      "epoch": 3.1510791366906474,
      "grad_norm": 0.08137132079708974,
      "learning_rate": 2.895347558155992e-05,
      "loss": 0.3106,
      "step": 2190
    },
    {
      "epoch": 3.1525179856115106,
      "grad_norm": 0.08813337885140969,
      "learning_rate": 2.8914857324500767e-05,
      "loss": 0.3152,
      "step": 2191
    },
    {
      "epoch": 3.1539568345323743,
      "grad_norm": 0.07592244668519561,
      "learning_rate": 2.887625025628599e-05,
      "loss": 0.3064,
      "step": 2192
    },
    {
      "epoch": 3.1553956834532375,
      "grad_norm": 0.10560558598349411,
      "learning_rate": 2.8837654415883817e-05,
      "loss": 0.3116,
      "step": 2193
    },
    {
      "epoch": 3.1568345323741007,
      "grad_norm": 0.0755960624360644,
      "learning_rate": 2.879906984225119e-05,
      "loss": 0.3165,
      "step": 2194
    },
    {
      "epoch": 3.158273381294964,
      "grad_norm": 0.09732919238129134,
      "learning_rate": 2.8760496574333613e-05,
      "loss": 0.3184,
      "step": 2195
    },
    {
      "epoch": 3.159712230215827,
      "grad_norm": 0.07449648470746753,
      "learning_rate": 2.8721934651065227e-05,
      "loss": 0.3145,
      "step": 2196
    },
    {
      "epoch": 3.161151079136691,
      "grad_norm": 0.0846053364879296,
      "learning_rate": 2.8683384111368675e-05,
      "loss": 0.3028,
      "step": 2197
    },
    {
      "epoch": 3.162589928057554,
      "grad_norm": 0.08346396573559035,
      "learning_rate": 2.864484499415517e-05,
      "loss": 0.3074,
      "step": 2198
    },
    {
      "epoch": 3.1640287769784172,
      "grad_norm": 0.09412025107338536,
      "learning_rate": 2.8606317338324347e-05,
      "loss": 0.3094,
      "step": 2199
    },
    {
      "epoch": 3.1654676258992804,
      "grad_norm": 0.07562893334341883,
      "learning_rate": 2.856780118276429e-05,
      "loss": 0.3208,
      "step": 2200
    },
    {
      "epoch": 3.166906474820144,
      "grad_norm": 0.08737397019390829,
      "learning_rate": 2.852929656635146e-05,
      "loss": 0.3107,
      "step": 2201
    },
    {
      "epoch": 3.1683453237410073,
      "grad_norm": 0.08921765424618087,
      "learning_rate": 2.8490803527950706e-05,
      "loss": 0.3095,
      "step": 2202
    },
    {
      "epoch": 3.1697841726618705,
      "grad_norm": 0.07644433162930088,
      "learning_rate": 2.845232210641517e-05,
      "loss": 0.3124,
      "step": 2203
    },
    {
      "epoch": 3.1712230215827337,
      "grad_norm": 0.10046560790987005,
      "learning_rate": 2.841385234058624e-05,
      "loss": 0.3096,
      "step": 2204
    },
    {
      "epoch": 3.172661870503597,
      "grad_norm": 0.09749723028751422,
      "learning_rate": 2.83753942692936e-05,
      "loss": 0.3159,
      "step": 2205
    },
    {
      "epoch": 3.1741007194244606,
      "grad_norm": 0.07485393350887619,
      "learning_rate": 2.8336947931355096e-05,
      "loss": 0.3111,
      "step": 2206
    },
    {
      "epoch": 3.175539568345324,
      "grad_norm": 0.08927179298519106,
      "learning_rate": 2.8298513365576715e-05,
      "loss": 0.3094,
      "step": 2207
    },
    {
      "epoch": 3.176978417266187,
      "grad_norm": 0.3885049074315983,
      "learning_rate": 2.826009061075257e-05,
      "loss": 0.3207,
      "step": 2208
    },
    {
      "epoch": 3.1784172661870502,
      "grad_norm": 0.09307844740534896,
      "learning_rate": 2.822167970566488e-05,
      "loss": 0.3,
      "step": 2209
    },
    {
      "epoch": 3.1798561151079134,
      "grad_norm": 0.09350080081395308,
      "learning_rate": 2.8183280689083895e-05,
      "loss": 0.3101,
      "step": 2210
    },
    {
      "epoch": 3.181294964028777,
      "grad_norm": 0.10753781473299533,
      "learning_rate": 2.8144893599767828e-05,
      "loss": 0.3042,
      "step": 2211
    },
    {
      "epoch": 3.1827338129496403,
      "grad_norm": 0.09175164298439005,
      "learning_rate": 2.8106518476462886e-05,
      "loss": 0.3059,
      "step": 2212
    },
    {
      "epoch": 3.1841726618705035,
      "grad_norm": 0.09216657565504674,
      "learning_rate": 2.806815535790321e-05,
      "loss": 0.3107,
      "step": 2213
    },
    {
      "epoch": 3.1856115107913667,
      "grad_norm": 0.09845501579488738,
      "learning_rate": 2.8029804282810794e-05,
      "loss": 0.3051,
      "step": 2214
    },
    {
      "epoch": 3.1870503597122304,
      "grad_norm": 0.08435034016576407,
      "learning_rate": 2.7991465289895497e-05,
      "loss": 0.3169,
      "step": 2215
    },
    {
      "epoch": 3.1884892086330936,
      "grad_norm": 0.10859317166691841,
      "learning_rate": 2.7953138417854952e-05,
      "loss": 0.3105,
      "step": 2216
    },
    {
      "epoch": 3.189928057553957,
      "grad_norm": 0.08268118205133658,
      "learning_rate": 2.79148237053746e-05,
      "loss": 0.3147,
      "step": 2217
    },
    {
      "epoch": 3.19136690647482,
      "grad_norm": 0.10477342417518032,
      "learning_rate": 2.787652119112758e-05,
      "loss": 0.3103,
      "step": 2218
    },
    {
      "epoch": 3.1928057553956837,
      "grad_norm": 0.0873830025004209,
      "learning_rate": 2.783823091377472e-05,
      "loss": 0.3114,
      "step": 2219
    },
    {
      "epoch": 3.194244604316547,
      "grad_norm": 0.08259462797161037,
      "learning_rate": 2.7799952911964535e-05,
      "loss": 0.3081,
      "step": 2220
    },
    {
      "epoch": 3.19568345323741,
      "grad_norm": 0.09863762634723984,
      "learning_rate": 2.776168722433308e-05,
      "loss": 0.3188,
      "step": 2221
    },
    {
      "epoch": 3.1971223021582733,
      "grad_norm": 0.08376842485912665,
      "learning_rate": 2.7723433889504046e-05,
      "loss": 0.305,
      "step": 2222
    },
    {
      "epoch": 3.1985611510791365,
      "grad_norm": 0.07444689807045426,
      "learning_rate": 2.7685192946088597e-05,
      "loss": 0.3088,
      "step": 2223
    },
    {
      "epoch": 3.2,
      "grad_norm": 0.08472495106391134,
      "learning_rate": 2.7646964432685456e-05,
      "loss": 0.3073,
      "step": 2224
    },
    {
      "epoch": 3.2014388489208634,
      "grad_norm": 0.07662930328782688,
      "learning_rate": 2.7608748387880754e-05,
      "loss": 0.3132,
      "step": 2225
    },
    {
      "epoch": 3.2028776978417266,
      "grad_norm": 0.09687190373515468,
      "learning_rate": 2.7570544850248047e-05,
      "loss": 0.3097,
      "step": 2226
    },
    {
      "epoch": 3.20431654676259,
      "grad_norm": 0.09810302622840962,
      "learning_rate": 2.753235385834824e-05,
      "loss": 0.3174,
      "step": 2227
    },
    {
      "epoch": 3.205755395683453,
      "grad_norm": 0.10861137611650165,
      "learning_rate": 2.749417545072964e-05,
      "loss": 0.3147,
      "step": 2228
    },
    {
      "epoch": 3.2071942446043167,
      "grad_norm": 0.11598410975855965,
      "learning_rate": 2.7456009665927807e-05,
      "loss": 0.3123,
      "step": 2229
    },
    {
      "epoch": 3.20863309352518,
      "grad_norm": 0.0754146028025198,
      "learning_rate": 2.741785654246555e-05,
      "loss": 0.3116,
      "step": 2230
    },
    {
      "epoch": 3.210071942446043,
      "grad_norm": 0.09958175259270793,
      "learning_rate": 2.7379716118852927e-05,
      "loss": 0.319,
      "step": 2231
    },
    {
      "epoch": 3.2115107913669063,
      "grad_norm": 0.08298289356877014,
      "learning_rate": 2.734158843358718e-05,
      "loss": 0.3021,
      "step": 2232
    },
    {
      "epoch": 3.21294964028777,
      "grad_norm": 0.07874592189015894,
      "learning_rate": 2.730347352515266e-05,
      "loss": 0.3162,
      "step": 2233
    },
    {
      "epoch": 3.214388489208633,
      "grad_norm": 0.10664889226983967,
      "learning_rate": 2.7265371432020836e-05,
      "loss": 0.3116,
      "step": 2234
    },
    {
      "epoch": 3.2158273381294964,
      "grad_norm": 0.06975698260073791,
      "learning_rate": 2.7227282192650258e-05,
      "loss": 0.3154,
      "step": 2235
    },
    {
      "epoch": 3.2172661870503596,
      "grad_norm": 0.10253930117577448,
      "learning_rate": 2.7189205845486503e-05,
      "loss": 0.3164,
      "step": 2236
    },
    {
      "epoch": 3.218705035971223,
      "grad_norm": 0.08077061379947138,
      "learning_rate": 2.7151142428962103e-05,
      "loss": 0.3145,
      "step": 2237
    },
    {
      "epoch": 3.2201438848920865,
      "grad_norm": 0.10455540840609091,
      "learning_rate": 2.711309198149655e-05,
      "loss": 0.3078,
      "step": 2238
    },
    {
      "epoch": 3.2215827338129497,
      "grad_norm": 0.10720051510695476,
      "learning_rate": 2.7075054541496296e-05,
      "loss": 0.3096,
      "step": 2239
    },
    {
      "epoch": 3.223021582733813,
      "grad_norm": 0.08730198774992833,
      "learning_rate": 2.7037030147354582e-05,
      "loss": 0.3079,
      "step": 2240
    },
    {
      "epoch": 3.224460431654676,
      "grad_norm": 0.08808245280975563,
      "learning_rate": 2.6999018837451523e-05,
      "loss": 0.3086,
      "step": 2241
    },
    {
      "epoch": 3.2258992805755398,
      "grad_norm": 0.09690990476431383,
      "learning_rate": 2.6961020650154057e-05,
      "loss": 0.3122,
      "step": 2242
    },
    {
      "epoch": 3.227338129496403,
      "grad_norm": 0.07541941606724827,
      "learning_rate": 2.6923035623815824e-05,
      "loss": 0.3149,
      "step": 2243
    },
    {
      "epoch": 3.228776978417266,
      "grad_norm": 0.08953075238235308,
      "learning_rate": 2.6885063796777195e-05,
      "loss": 0.3109,
      "step": 2244
    },
    {
      "epoch": 3.2302158273381294,
      "grad_norm": 0.0756041493121849,
      "learning_rate": 2.6847105207365225e-05,
      "loss": 0.3115,
      "step": 2245
    },
    {
      "epoch": 3.2316546762589926,
      "grad_norm": 0.08150949209268828,
      "learning_rate": 2.6809159893893624e-05,
      "loss": 0.3072,
      "step": 2246
    },
    {
      "epoch": 3.2330935251798563,
      "grad_norm": 0.07674594542794348,
      "learning_rate": 2.6771227894662666e-05,
      "loss": 0.3066,
      "step": 2247
    },
    {
      "epoch": 3.2345323741007195,
      "grad_norm": 0.08437215838706465,
      "learning_rate": 2.6733309247959217e-05,
      "loss": 0.3084,
      "step": 2248
    },
    {
      "epoch": 3.2359712230215827,
      "grad_norm": 0.07488814476800476,
      "learning_rate": 2.669540399205664e-05,
      "loss": 0.3109,
      "step": 2249
    },
    {
      "epoch": 3.237410071942446,
      "grad_norm": 0.08477463075152412,
      "learning_rate": 2.6657512165214806e-05,
      "loss": 0.3081,
      "step": 2250
    },
    {
      "epoch": 3.2388489208633096,
      "grad_norm": 0.08480300227796347,
      "learning_rate": 2.6619633805680028e-05,
      "loss": 0.3151,
      "step": 2251
    },
    {
      "epoch": 3.2402877697841728,
      "grad_norm": 0.08417307835689034,
      "learning_rate": 2.6581768951684992e-05,
      "loss": 0.3126,
      "step": 2252
    },
    {
      "epoch": 3.241726618705036,
      "grad_norm": 0.0869429807242692,
      "learning_rate": 2.6543917641448813e-05,
      "loss": 0.3104,
      "step": 2253
    },
    {
      "epoch": 3.243165467625899,
      "grad_norm": 0.083339481220526,
      "learning_rate": 2.650607991317687e-05,
      "loss": 0.3093,
      "step": 2254
    },
    {
      "epoch": 3.2446043165467624,
      "grad_norm": 0.0841705362687117,
      "learning_rate": 2.6468255805060885e-05,
      "loss": 0.3182,
      "step": 2255
    },
    {
      "epoch": 3.246043165467626,
      "grad_norm": 0.07847401258381193,
      "learning_rate": 2.6430445355278788e-05,
      "loss": 0.3175,
      "step": 2256
    },
    {
      "epoch": 3.2474820143884893,
      "grad_norm": 0.07747532307100845,
      "learning_rate": 2.639264860199477e-05,
      "loss": 0.3108,
      "step": 2257
    },
    {
      "epoch": 3.2489208633093525,
      "grad_norm": 0.07087934641843581,
      "learning_rate": 2.6354865583359175e-05,
      "loss": 0.3105,
      "step": 2258
    },
    {
      "epoch": 3.2503597122302157,
      "grad_norm": 0.08010150707605394,
      "learning_rate": 2.631709633750847e-05,
      "loss": 0.3029,
      "step": 2259
    },
    {
      "epoch": 3.2517985611510793,
      "grad_norm": 0.06538151949892566,
      "learning_rate": 2.6279340902565217e-05,
      "loss": 0.3103,
      "step": 2260
    },
    {
      "epoch": 3.2532374100719426,
      "grad_norm": 0.08170052188335858,
      "learning_rate": 2.6241599316638084e-05,
      "loss": 0.3014,
      "step": 2261
    },
    {
      "epoch": 3.2546762589928058,
      "grad_norm": 0.0817078451653208,
      "learning_rate": 2.6203871617821717e-05,
      "loss": 0.3194,
      "step": 2262
    },
    {
      "epoch": 3.256115107913669,
      "grad_norm": 0.07909170810469295,
      "learning_rate": 2.6166157844196755e-05,
      "loss": 0.3103,
      "step": 2263
    },
    {
      "epoch": 3.257553956834532,
      "grad_norm": 0.08352539220044815,
      "learning_rate": 2.6128458033829792e-05,
      "loss": 0.3139,
      "step": 2264
    },
    {
      "epoch": 3.258992805755396,
      "grad_norm": 0.07913451837672723,
      "learning_rate": 2.609077222477332e-05,
      "loss": 0.3147,
      "step": 2265
    },
    {
      "epoch": 3.260431654676259,
      "grad_norm": 0.08618944966170508,
      "learning_rate": 2.6053100455065693e-05,
      "loss": 0.3123,
      "step": 2266
    },
    {
      "epoch": 3.2618705035971223,
      "grad_norm": 0.07266722689750256,
      "learning_rate": 2.6015442762731095e-05,
      "loss": 0.3152,
      "step": 2267
    },
    {
      "epoch": 3.2633093525179855,
      "grad_norm": 0.09435052405815617,
      "learning_rate": 2.5977799185779534e-05,
      "loss": 0.3127,
      "step": 2268
    },
    {
      "epoch": 3.2647482014388487,
      "grad_norm": 0.07199685293057449,
      "learning_rate": 2.5940169762206722e-05,
      "loss": 0.3027,
      "step": 2269
    },
    {
      "epoch": 3.2661870503597124,
      "grad_norm": 0.08625337724152561,
      "learning_rate": 2.5902554529994105e-05,
      "loss": 0.3144,
      "step": 2270
    },
    {
      "epoch": 3.2676258992805756,
      "grad_norm": 0.08384084708818604,
      "learning_rate": 2.5864953527108805e-05,
      "loss": 0.3091,
      "step": 2271
    },
    {
      "epoch": 3.2690647482014388,
      "grad_norm": 0.09045487677998157,
      "learning_rate": 2.58273667915036e-05,
      "loss": 0.3151,
      "step": 2272
    },
    {
      "epoch": 3.270503597122302,
      "grad_norm": 0.08550608784459429,
      "learning_rate": 2.578979436111684e-05,
      "loss": 0.3069,
      "step": 2273
    },
    {
      "epoch": 3.2719424460431656,
      "grad_norm": 0.07644305283410271,
      "learning_rate": 2.5752236273872432e-05,
      "loss": 0.3089,
      "step": 2274
    },
    {
      "epoch": 3.273381294964029,
      "grad_norm": 0.08672218328853597,
      "learning_rate": 2.5714692567679853e-05,
      "loss": 0.3108,
      "step": 2275
    },
    {
      "epoch": 3.274820143884892,
      "grad_norm": 0.0858839644826738,
      "learning_rate": 2.5677163280433995e-05,
      "loss": 0.3085,
      "step": 2276
    },
    {
      "epoch": 3.2762589928057553,
      "grad_norm": 0.07779820348974278,
      "learning_rate": 2.5639648450015268e-05,
      "loss": 0.3105,
      "step": 2277
    },
    {
      "epoch": 3.277697841726619,
      "grad_norm": 0.09608787847046683,
      "learning_rate": 2.5602148114289415e-05,
      "loss": 0.3106,
      "step": 2278
    },
    {
      "epoch": 3.279136690647482,
      "grad_norm": 0.08810889838597738,
      "learning_rate": 2.556466231110762e-05,
      "loss": 0.3056,
      "step": 2279
    },
    {
      "epoch": 3.2805755395683454,
      "grad_norm": 0.07600342745886936,
      "learning_rate": 2.552719107830635e-05,
      "loss": 0.3064,
      "step": 2280
    },
    {
      "epoch": 3.2820143884892086,
      "grad_norm": 0.0905027839872227,
      "learning_rate": 2.54897344537074e-05,
      "loss": 0.3068,
      "step": 2281
    },
    {
      "epoch": 3.283453237410072,
      "grad_norm": 0.07570693317848128,
      "learning_rate": 2.5452292475117767e-05,
      "loss": 0.3124,
      "step": 2282
    },
    {
      "epoch": 3.2848920863309354,
      "grad_norm": 0.079114700228305,
      "learning_rate": 2.541486518032973e-05,
      "loss": 0.31,
      "step": 2283
    },
    {
      "epoch": 3.2863309352517986,
      "grad_norm": 0.10010897960545055,
      "learning_rate": 2.5377452607120722e-05,
      "loss": 0.3209,
      "step": 2284
    },
    {
      "epoch": 3.287769784172662,
      "grad_norm": 0.07373044643745348,
      "learning_rate": 2.5340054793253276e-05,
      "loss": 0.3154,
      "step": 2285
    },
    {
      "epoch": 3.289208633093525,
      "grad_norm": 0.09561131286368094,
      "learning_rate": 2.5302671776475098e-05,
      "loss": 0.3072,
      "step": 2286
    },
    {
      "epoch": 3.2906474820143883,
      "grad_norm": 0.08481304440195968,
      "learning_rate": 2.526530359451892e-05,
      "loss": 0.3149,
      "step": 2287
    },
    {
      "epoch": 3.292086330935252,
      "grad_norm": 0.07033086586948158,
      "learning_rate": 2.522795028510249e-05,
      "loss": 0.3159,
      "step": 2288
    },
    {
      "epoch": 3.293525179856115,
      "grad_norm": 0.08058728399603567,
      "learning_rate": 2.5190611885928547e-05,
      "loss": 0.3128,
      "step": 2289
    },
    {
      "epoch": 3.2949640287769784,
      "grad_norm": 0.0758475875615919,
      "learning_rate": 2.5153288434684816e-05,
      "loss": 0.3171,
      "step": 2290
    },
    {
      "epoch": 3.2964028776978416,
      "grad_norm": 0.09660252514185105,
      "learning_rate": 2.5115979969043914e-05,
      "loss": 0.3172,
      "step": 2291
    },
    {
      "epoch": 3.2978417266187052,
      "grad_norm": 0.07392542301471512,
      "learning_rate": 2.5078686526663304e-05,
      "loss": 0.3123,
      "step": 2292
    },
    {
      "epoch": 3.2992805755395684,
      "grad_norm": 0.07628148569031089,
      "learning_rate": 2.5041408145185312e-05,
      "loss": 0.3035,
      "step": 2293
    },
    {
      "epoch": 3.3007194244604317,
      "grad_norm": 0.0816470604581458,
      "learning_rate": 2.5004144862237084e-05,
      "loss": 0.3142,
      "step": 2294
    },
    {
      "epoch": 3.302158273381295,
      "grad_norm": 0.078780294760847,
      "learning_rate": 2.4966896715430484e-05,
      "loss": 0.3063,
      "step": 2295
    },
    {
      "epoch": 3.3035971223021585,
      "grad_norm": 0.07937649700673445,
      "learning_rate": 2.4929663742362103e-05,
      "loss": 0.315,
      "step": 2296
    },
    {
      "epoch": 3.3050359712230217,
      "grad_norm": 0.07099358599255484,
      "learning_rate": 2.4892445980613254e-05,
      "loss": 0.3066,
      "step": 2297
    },
    {
      "epoch": 3.306474820143885,
      "grad_norm": 0.08230088318252803,
      "learning_rate": 2.4855243467749865e-05,
      "loss": 0.3166,
      "step": 2298
    },
    {
      "epoch": 3.307913669064748,
      "grad_norm": 0.07839788409142841,
      "learning_rate": 2.481805624132247e-05,
      "loss": 0.3056,
      "step": 2299
    },
    {
      "epoch": 3.3093525179856114,
      "grad_norm": 0.07333852877027737,
      "learning_rate": 2.478088433886618e-05,
      "loss": 0.3048,
      "step": 2300
    },
    {
      "epoch": 3.310791366906475,
      "grad_norm": 0.09456550634859283,
      "learning_rate": 2.4743727797900668e-05,
      "loss": 0.31,
      "step": 2301
    },
    {
      "epoch": 3.3122302158273382,
      "grad_norm": 0.06933779657260399,
      "learning_rate": 2.4706586655930042e-05,
      "loss": 0.3097,
      "step": 2302
    },
    {
      "epoch": 3.3136690647482014,
      "grad_norm": 0.08714194098610613,
      "learning_rate": 2.4669460950442926e-05,
      "loss": 0.3138,
      "step": 2303
    },
    {
      "epoch": 3.3151079136690647,
      "grad_norm": 0.09547255660921417,
      "learning_rate": 2.463235071891231e-05,
      "loss": 0.3132,
      "step": 2304
    },
    {
      "epoch": 3.316546762589928,
      "grad_norm": 0.07190538663347976,
      "learning_rate": 2.4595255998795625e-05,
      "loss": 0.3015,
      "step": 2305
    },
    {
      "epoch": 3.3179856115107915,
      "grad_norm": 0.09186247817974864,
      "learning_rate": 2.4558176827534587e-05,
      "loss": 0.3069,
      "step": 2306
    },
    {
      "epoch": 3.3194244604316547,
      "grad_norm": 0.07892002449578553,
      "learning_rate": 2.452111324255524e-05,
      "loss": 0.3059,
      "step": 2307
    },
    {
      "epoch": 3.320863309352518,
      "grad_norm": 0.08602518467621155,
      "learning_rate": 2.448406528126793e-05,
      "loss": 0.317,
      "step": 2308
    },
    {
      "epoch": 3.322302158273381,
      "grad_norm": 0.09660912181886301,
      "learning_rate": 2.444703298106718e-05,
      "loss": 0.3067,
      "step": 2309
    },
    {
      "epoch": 3.3237410071942444,
      "grad_norm": 0.08146049515388465,
      "learning_rate": 2.441001637933173e-05,
      "loss": 0.313,
      "step": 2310
    },
    {
      "epoch": 3.325179856115108,
      "grad_norm": 0.09560703484537106,
      "learning_rate": 2.437301551342447e-05,
      "loss": 0.3154,
      "step": 2311
    },
    {
      "epoch": 3.3266187050359712,
      "grad_norm": 0.07466795669935042,
      "learning_rate": 2.433603042069242e-05,
      "loss": 0.31,
      "step": 2312
    },
    {
      "epoch": 3.3280575539568344,
      "grad_norm": 0.08325258763084104,
      "learning_rate": 2.4299061138466667e-05,
      "loss": 0.312,
      "step": 2313
    },
    {
      "epoch": 3.3294964028776977,
      "grad_norm": 0.08796488903023479,
      "learning_rate": 2.4262107704062343e-05,
      "loss": 0.3121,
      "step": 2314
    },
    {
      "epoch": 3.3309352517985613,
      "grad_norm": 0.09552304190021914,
      "learning_rate": 2.4225170154778562e-05,
      "loss": 0.298,
      "step": 2315
    },
    {
      "epoch": 3.3323741007194245,
      "grad_norm": 0.08363284586867374,
      "learning_rate": 2.4188248527898446e-05,
      "loss": 0.3157,
      "step": 2316
    },
    {
      "epoch": 3.3338129496402877,
      "grad_norm": 0.09022905284092302,
      "learning_rate": 2.415134286068903e-05,
      "loss": 0.3111,
      "step": 2317
    },
    {
      "epoch": 3.335251798561151,
      "grad_norm": 0.0835668636717724,
      "learning_rate": 2.411445319040121e-05,
      "loss": 0.3079,
      "step": 2318
    },
    {
      "epoch": 3.3366906474820146,
      "grad_norm": 0.0877973742789849,
      "learning_rate": 2.407757955426977e-05,
      "loss": 0.3104,
      "step": 2319
    },
    {
      "epoch": 3.338129496402878,
      "grad_norm": 0.08331668340154326,
      "learning_rate": 2.4040721989513314e-05,
      "loss": 0.3119,
      "step": 2320
    },
    {
      "epoch": 3.339568345323741,
      "grad_norm": 0.08697931951822054,
      "learning_rate": 2.40038805333342e-05,
      "loss": 0.3081,
      "step": 2321
    },
    {
      "epoch": 3.3410071942446042,
      "grad_norm": 0.091564579637049,
      "learning_rate": 2.396705522291852e-05,
      "loss": 0.3151,
      "step": 2322
    },
    {
      "epoch": 3.3424460431654675,
      "grad_norm": 0.08131858824210252,
      "learning_rate": 2.393024609543611e-05,
      "loss": 0.3084,
      "step": 2323
    },
    {
      "epoch": 3.343884892086331,
      "grad_norm": 0.08590487089463734,
      "learning_rate": 2.3893453188040442e-05,
      "loss": 0.3037,
      "step": 2324
    },
    {
      "epoch": 3.3453237410071943,
      "grad_norm": 0.07930906646637645,
      "learning_rate": 2.3856676537868614e-05,
      "loss": 0.3144,
      "step": 2325
    },
    {
      "epoch": 3.3467625899280575,
      "grad_norm": 0.09001420314337577,
      "learning_rate": 2.3819916182041318e-05,
      "loss": 0.3042,
      "step": 2326
    },
    {
      "epoch": 3.3482014388489207,
      "grad_norm": 0.08026328979707366,
      "learning_rate": 2.378317215766283e-05,
      "loss": 0.3042,
      "step": 2327
    },
    {
      "epoch": 3.349640287769784,
      "grad_norm": 0.0733431469374671,
      "learning_rate": 2.3746444501820886e-05,
      "loss": 0.3163,
      "step": 2328
    },
    {
      "epoch": 3.3510791366906476,
      "grad_norm": 0.08584777385276081,
      "learning_rate": 2.370973325158675e-05,
      "loss": 0.3049,
      "step": 2329
    },
    {
      "epoch": 3.352517985611511,
      "grad_norm": 0.06972001281573628,
      "learning_rate": 2.3673038444015087e-05,
      "loss": 0.3141,
      "step": 2330
    },
    {
      "epoch": 3.353956834532374,
      "grad_norm": 0.08060635092830382,
      "learning_rate": 2.363636011614401e-05,
      "loss": 0.3125,
      "step": 2331
    },
    {
      "epoch": 3.3553956834532372,
      "grad_norm": 0.08013278285028885,
      "learning_rate": 2.3599698304994946e-05,
      "loss": 0.3177,
      "step": 2332
    },
    {
      "epoch": 3.356834532374101,
      "grad_norm": 0.08918460812426662,
      "learning_rate": 2.3563053047572683e-05,
      "loss": 0.3154,
      "step": 2333
    },
    {
      "epoch": 3.358273381294964,
      "grad_norm": 0.07615428352899022,
      "learning_rate": 2.352642438086533e-05,
      "loss": 0.2992,
      "step": 2334
    },
    {
      "epoch": 3.3597122302158273,
      "grad_norm": 0.07729121670905142,
      "learning_rate": 2.348981234184417e-05,
      "loss": 0.307,
      "step": 2335
    },
    {
      "epoch": 3.3611510791366905,
      "grad_norm": 0.07141656240900103,
      "learning_rate": 2.3453216967463785e-05,
      "loss": 0.3145,
      "step": 2336
    },
    {
      "epoch": 3.362589928057554,
      "grad_norm": 0.08935058896085595,
      "learning_rate": 2.3416638294661864e-05,
      "loss": 0.3085,
      "step": 2337
    },
    {
      "epoch": 3.3640287769784174,
      "grad_norm": 0.06400882009509953,
      "learning_rate": 2.3380076360359293e-05,
      "loss": 0.3124,
      "step": 2338
    },
    {
      "epoch": 3.3654676258992806,
      "grad_norm": 0.09512916515917591,
      "learning_rate": 2.3343531201460067e-05,
      "loss": 0.3043,
      "step": 2339
    },
    {
      "epoch": 3.366906474820144,
      "grad_norm": 0.06956346991078782,
      "learning_rate": 2.3307002854851188e-05,
      "loss": 0.3122,
      "step": 2340
    },
    {
      "epoch": 3.368345323741007,
      "grad_norm": 0.08138811148963958,
      "learning_rate": 2.3270491357402715e-05,
      "loss": 0.3073,
      "step": 2341
    },
    {
      "epoch": 3.3697841726618707,
      "grad_norm": 0.07568926449823735,
      "learning_rate": 2.3233996745967772e-05,
      "loss": 0.3143,
      "step": 2342
    },
    {
      "epoch": 3.371223021582734,
      "grad_norm": 0.08713346179685631,
      "learning_rate": 2.3197519057382326e-05,
      "loss": 0.3011,
      "step": 2343
    },
    {
      "epoch": 3.372661870503597,
      "grad_norm": 0.06657650013060197,
      "learning_rate": 2.316105832846532e-05,
      "loss": 0.3021,
      "step": 2344
    },
    {
      "epoch": 3.3741007194244603,
      "grad_norm": 0.0909347921883191,
      "learning_rate": 2.3124614596018606e-05,
      "loss": 0.3033,
      "step": 2345
    },
    {
      "epoch": 3.3755395683453235,
      "grad_norm": 0.06633577705016298,
      "learning_rate": 2.308818789682682e-05,
      "loss": 0.2986,
      "step": 2346
    },
    {
      "epoch": 3.376978417266187,
      "grad_norm": 0.08813700890410717,
      "learning_rate": 2.3051778267657436e-05,
      "loss": 0.3099,
      "step": 2347
    },
    {
      "epoch": 3.3784172661870504,
      "grad_norm": 0.08030048225395597,
      "learning_rate": 2.3015385745260704e-05,
      "loss": 0.3121,
      "step": 2348
    },
    {
      "epoch": 3.3798561151079136,
      "grad_norm": 0.08772628266053589,
      "learning_rate": 2.2979010366369595e-05,
      "loss": 0.3138,
      "step": 2349
    },
    {
      "epoch": 3.381294964028777,
      "grad_norm": 0.06915074511307691,
      "learning_rate": 2.294265216769978e-05,
      "loss": 0.3036,
      "step": 2350
    },
    {
      "epoch": 3.38273381294964,
      "grad_norm": 0.0787032873720719,
      "learning_rate": 2.2906311185949605e-05,
      "loss": 0.3159,
      "step": 2351
    },
    {
      "epoch": 3.3841726618705037,
      "grad_norm": 0.06474353356605671,
      "learning_rate": 2.2869987457799977e-05,
      "loss": 0.3013,
      "step": 2352
    },
    {
      "epoch": 3.385611510791367,
      "grad_norm": 0.07936109383021454,
      "learning_rate": 2.283368101991448e-05,
      "loss": 0.3056,
      "step": 2353
    },
    {
      "epoch": 3.38705035971223,
      "grad_norm": 0.06753730195534415,
      "learning_rate": 2.2797391908939196e-05,
      "loss": 0.3138,
      "step": 2354
    },
    {
      "epoch": 3.3884892086330938,
      "grad_norm": 0.07238196048279752,
      "learning_rate": 2.2761120161502674e-05,
      "loss": 0.3043,
      "step": 2355
    },
    {
      "epoch": 3.389928057553957,
      "grad_norm": 0.06924314961884438,
      "learning_rate": 2.2724865814216042e-05,
      "loss": 0.3009,
      "step": 2356
    },
    {
      "epoch": 3.39136690647482,
      "grad_norm": 0.06361823031633611,
      "learning_rate": 2.2688628903672792e-05,
      "loss": 0.3087,
      "step": 2357
    },
    {
      "epoch": 3.3928057553956834,
      "grad_norm": 0.07732323917041639,
      "learning_rate": 2.265240946644881e-05,
      "loss": 0.3157,
      "step": 2358
    },
    {
      "epoch": 3.3942446043165466,
      "grad_norm": 0.06415742640174263,
      "learning_rate": 2.261620753910238e-05,
      "loss": 0.3182,
      "step": 2359
    },
    {
      "epoch": 3.3956834532374103,
      "grad_norm": 0.07267628619293835,
      "learning_rate": 2.25800231581741e-05,
      "loss": 0.3076,
      "step": 2360
    },
    {
      "epoch": 3.3971223021582735,
      "grad_norm": 0.061930859007014914,
      "learning_rate": 2.254385636018686e-05,
      "loss": 0.3084,
      "step": 2361
    },
    {
      "epoch": 3.3985611510791367,
      "grad_norm": 0.07012006894365325,
      "learning_rate": 2.250770718164579e-05,
      "loss": 0.3107,
      "step": 2362
    },
    {
      "epoch": 3.4,
      "grad_norm": 0.07266288046107679,
      "learning_rate": 2.247157565903825e-05,
      "loss": 0.3114,
      "step": 2363
    },
    {
      "epoch": 3.401438848920863,
      "grad_norm": 0.06690274972297683,
      "learning_rate": 2.243546182883377e-05,
      "loss": 0.3102,
      "step": 2364
    },
    {
      "epoch": 3.402877697841727,
      "grad_norm": 0.0692178004843867,
      "learning_rate": 2.2399365727484047e-05,
      "loss": 0.3068,
      "step": 2365
    },
    {
      "epoch": 3.40431654676259,
      "grad_norm": 0.0722367138426255,
      "learning_rate": 2.2363287391422806e-05,
      "loss": 0.3081,
      "step": 2366
    },
    {
      "epoch": 3.405755395683453,
      "grad_norm": 0.06969322455090261,
      "learning_rate": 2.2327226857065954e-05,
      "loss": 0.3126,
      "step": 2367
    },
    {
      "epoch": 3.4071942446043164,
      "grad_norm": 0.0707675931902275,
      "learning_rate": 2.2291184160811374e-05,
      "loss": 0.3148,
      "step": 2368
    },
    {
      "epoch": 3.4086330935251796,
      "grad_norm": 0.06728240197400746,
      "learning_rate": 2.22551593390389e-05,
      "loss": 0.3107,
      "step": 2369
    },
    {
      "epoch": 3.4100719424460433,
      "grad_norm": 0.07436693837659339,
      "learning_rate": 2.2219152428110368e-05,
      "loss": 0.3069,
      "step": 2370
    },
    {
      "epoch": 3.4115107913669065,
      "grad_norm": 0.06117093871196123,
      "learning_rate": 2.218316346436959e-05,
      "loss": 0.306,
      "step": 2371
    },
    {
      "epoch": 3.4129496402877697,
      "grad_norm": 0.07245482528214216,
      "learning_rate": 2.2147192484142154e-05,
      "loss": 0.3062,
      "step": 2372
    },
    {
      "epoch": 3.414388489208633,
      "grad_norm": 0.05973011321086604,
      "learning_rate": 2.2111239523735568e-05,
      "loss": 0.3046,
      "step": 2373
    },
    {
      "epoch": 3.4158273381294966,
      "grad_norm": 0.07457067897114032,
      "learning_rate": 2.2075304619439127e-05,
      "loss": 0.3119,
      "step": 2374
    },
    {
      "epoch": 3.41726618705036,
      "grad_norm": 0.27115162473455456,
      "learning_rate": 2.2039387807523914e-05,
      "loss": 0.3127,
      "step": 2375
    },
    {
      "epoch": 3.418705035971223,
      "grad_norm": 0.08044914751899118,
      "learning_rate": 2.2003489124242742e-05,
      "loss": 0.3103,
      "step": 2376
    },
    {
      "epoch": 3.420143884892086,
      "grad_norm": 0.07157904616628058,
      "learning_rate": 2.1967608605830115e-05,
      "loss": 0.3106,
      "step": 2377
    },
    {
      "epoch": 3.42158273381295,
      "grad_norm": 0.07414897587449552,
      "learning_rate": 2.1931746288502235e-05,
      "loss": 0.3129,
      "step": 2378
    },
    {
      "epoch": 3.423021582733813,
      "grad_norm": 0.07280082346273015,
      "learning_rate": 2.1895902208456903e-05,
      "loss": 0.3127,
      "step": 2379
    },
    {
      "epoch": 3.4244604316546763,
      "grad_norm": 0.07413056477182658,
      "learning_rate": 2.186007640187353e-05,
      "loss": 0.3126,
      "step": 2380
    },
    {
      "epoch": 3.4258992805755395,
      "grad_norm": 0.0712247235301936,
      "learning_rate": 2.1824268904913036e-05,
      "loss": 0.3079,
      "step": 2381
    },
    {
      "epoch": 3.4273381294964027,
      "grad_norm": 0.08602698068407817,
      "learning_rate": 2.1788479753717935e-05,
      "loss": 0.309,
      "step": 2382
    },
    {
      "epoch": 3.4287769784172664,
      "grad_norm": 0.06949678973950535,
      "learning_rate": 2.1752708984412196e-05,
      "loss": 0.305,
      "step": 2383
    },
    {
      "epoch": 3.4302158273381296,
      "grad_norm": 0.0808604625712668,
      "learning_rate": 2.171695663310119e-05,
      "loss": 0.3059,
      "step": 2384
    },
    {
      "epoch": 3.431654676258993,
      "grad_norm": 0.07221729607102836,
      "learning_rate": 2.1681222735871747e-05,
      "loss": 0.3114,
      "step": 2385
    },
    {
      "epoch": 3.433093525179856,
      "grad_norm": 0.09025084305244628,
      "learning_rate": 2.1645507328792058e-05,
      "loss": 0.3136,
      "step": 2386
    },
    {
      "epoch": 3.434532374100719,
      "grad_norm": 0.07081013288539319,
      "learning_rate": 2.1609810447911637e-05,
      "loss": 0.3141,
      "step": 2387
    },
    {
      "epoch": 3.435971223021583,
      "grad_norm": 0.07325114649426695,
      "learning_rate": 2.157413212926133e-05,
      "loss": 0.3055,
      "step": 2388
    },
    {
      "epoch": 3.437410071942446,
      "grad_norm": 0.06552754563676148,
      "learning_rate": 2.1538472408853206e-05,
      "loss": 0.3102,
      "step": 2389
    },
    {
      "epoch": 3.4388489208633093,
      "grad_norm": 0.06958008689182883,
      "learning_rate": 2.1502831322680598e-05,
      "loss": 0.3136,
      "step": 2390
    },
    {
      "epoch": 3.4402877697841725,
      "grad_norm": 0.07411701271075863,
      "learning_rate": 2.1467208906718008e-05,
      "loss": 0.3011,
      "step": 2391
    },
    {
      "epoch": 3.441726618705036,
      "grad_norm": 0.06446652332050883,
      "learning_rate": 2.1431605196921103e-05,
      "loss": 0.3189,
      "step": 2392
    },
    {
      "epoch": 3.4431654676258994,
      "grad_norm": 0.07008631497051318,
      "learning_rate": 2.1396020229226666e-05,
      "loss": 0.3122,
      "step": 2393
    },
    {
      "epoch": 3.4446043165467626,
      "grad_norm": 0.07305952821190029,
      "learning_rate": 2.1360454039552577e-05,
      "loss": 0.3012,
      "step": 2394
    },
    {
      "epoch": 3.446043165467626,
      "grad_norm": 0.06945354549100825,
      "learning_rate": 2.1324906663797718e-05,
      "loss": 0.313,
      "step": 2395
    },
    {
      "epoch": 3.4474820143884894,
      "grad_norm": 0.07681370824552682,
      "learning_rate": 2.1289378137842008e-05,
      "loss": 0.3144,
      "step": 2396
    },
    {
      "epoch": 3.4489208633093527,
      "grad_norm": 0.06996385633447534,
      "learning_rate": 2.125386849754639e-05,
      "loss": 0.3096,
      "step": 2397
    },
    {
      "epoch": 3.450359712230216,
      "grad_norm": 0.07331328439643199,
      "learning_rate": 2.121837777875266e-05,
      "loss": 0.316,
      "step": 2398
    },
    {
      "epoch": 3.451798561151079,
      "grad_norm": 0.07344637977225729,
      "learning_rate": 2.118290601728354e-05,
      "loss": 0.3074,
      "step": 2399
    },
    {
      "epoch": 3.4532374100719423,
      "grad_norm": 0.07668692477154283,
      "learning_rate": 2.1147453248942687e-05,
      "loss": 0.3072,
      "step": 2400
    },
    {
      "epoch": 3.454676258992806,
      "grad_norm": 0.07619582037841803,
      "learning_rate": 2.1112019509514478e-05,
      "loss": 0.3129,
      "step": 2401
    },
    {
      "epoch": 3.456115107913669,
      "grad_norm": 0.07479747846629918,
      "learning_rate": 2.1076604834764154e-05,
      "loss": 0.3099,
      "step": 2402
    },
    {
      "epoch": 3.4575539568345324,
      "grad_norm": 0.090326392764258,
      "learning_rate": 2.1041209260437694e-05,
      "loss": 0.3146,
      "step": 2403
    },
    {
      "epoch": 3.4589928057553956,
      "grad_norm": 0.0777856669266028,
      "learning_rate": 2.10058328222618e-05,
      "loss": 0.3105,
      "step": 2404
    },
    {
      "epoch": 3.460431654676259,
      "grad_norm": 0.10039220659650842,
      "learning_rate": 2.097047555594385e-05,
      "loss": 0.308,
      "step": 2405
    },
    {
      "epoch": 3.4618705035971225,
      "grad_norm": 0.07880503894320168,
      "learning_rate": 2.0935137497171904e-05,
      "loss": 0.3123,
      "step": 2406
    },
    {
      "epoch": 3.4633093525179857,
      "grad_norm": 0.10766337775820017,
      "learning_rate": 2.0899818681614557e-05,
      "loss": 0.3148,
      "step": 2407
    },
    {
      "epoch": 3.464748201438849,
      "grad_norm": 0.07152630066014365,
      "learning_rate": 2.086451914492108e-05,
      "loss": 0.3095,
      "step": 2408
    },
    {
      "epoch": 3.466187050359712,
      "grad_norm": 0.06931007783592509,
      "learning_rate": 2.082923892272124e-05,
      "loss": 0.3094,
      "step": 2409
    },
    {
      "epoch": 3.4676258992805753,
      "grad_norm": 0.08750246007451234,
      "learning_rate": 2.079397805062526e-05,
      "loss": 0.3127,
      "step": 2410
    },
    {
      "epoch": 3.469064748201439,
      "grad_norm": 0.07336793877490577,
      "learning_rate": 2.0758736564223937e-05,
      "loss": 0.3073,
      "step": 2411
    },
    {
      "epoch": 3.470503597122302,
      "grad_norm": 0.06878588227635767,
      "learning_rate": 2.0723514499088388e-05,
      "loss": 0.3151,
      "step": 2412
    },
    {
      "epoch": 3.4719424460431654,
      "grad_norm": 0.07205378099429692,
      "learning_rate": 2.068831189077021e-05,
      "loss": 0.3165,
      "step": 2413
    },
    {
      "epoch": 3.4733812949640286,
      "grad_norm": 0.07254073531532791,
      "learning_rate": 2.065312877480133e-05,
      "loss": 0.2993,
      "step": 2414
    },
    {
      "epoch": 3.4748201438848922,
      "grad_norm": 0.06448071539936977,
      "learning_rate": 2.0617965186694e-05,
      "loss": 0.3061,
      "step": 2415
    },
    {
      "epoch": 3.4762589928057555,
      "grad_norm": 0.07969287347363285,
      "learning_rate": 2.058282116194076e-05,
      "loss": 0.3182,
      "step": 2416
    },
    {
      "epoch": 3.4776978417266187,
      "grad_norm": 0.0703083994859436,
      "learning_rate": 2.0547696736014415e-05,
      "loss": 0.3158,
      "step": 2417
    },
    {
      "epoch": 3.479136690647482,
      "grad_norm": 0.06391401118115385,
      "learning_rate": 2.0512591944367976e-05,
      "loss": 0.3146,
      "step": 2418
    },
    {
      "epoch": 3.4805755395683455,
      "grad_norm": 0.07949909331350384,
      "learning_rate": 2.0477506822434644e-05,
      "loss": 0.3095,
      "step": 2419
    },
    {
      "epoch": 3.4820143884892087,
      "grad_norm": 0.06808962841536517,
      "learning_rate": 2.0442441405627776e-05,
      "loss": 0.3106,
      "step": 2420
    },
    {
      "epoch": 3.483453237410072,
      "grad_norm": 0.08055504695955595,
      "learning_rate": 2.0407395729340792e-05,
      "loss": 0.3074,
      "step": 2421
    },
    {
      "epoch": 3.484892086330935,
      "grad_norm": 0.08062785800145268,
      "learning_rate": 2.037236982894723e-05,
      "loss": 0.3074,
      "step": 2422
    },
    {
      "epoch": 3.4863309352517984,
      "grad_norm": 0.07718758511488966,
      "learning_rate": 2.0337363739800695e-05,
      "loss": 0.3062,
      "step": 2423
    },
    {
      "epoch": 3.487769784172662,
      "grad_norm": 0.07031308809064436,
      "learning_rate": 2.030237749723472e-05,
      "loss": 0.3112,
      "step": 2424
    },
    {
      "epoch": 3.4892086330935252,
      "grad_norm": 0.06298489171916702,
      "learning_rate": 2.026741113656284e-05,
      "loss": 0.3158,
      "step": 2425
    },
    {
      "epoch": 3.4906474820143885,
      "grad_norm": 0.07920017440476117,
      "learning_rate": 2.0232464693078578e-05,
      "loss": 0.3181,
      "step": 2426
    },
    {
      "epoch": 3.4920863309352517,
      "grad_norm": 0.06380879282023667,
      "learning_rate": 2.0197538202055246e-05,
      "loss": 0.3116,
      "step": 2427
    },
    {
      "epoch": 3.493525179856115,
      "grad_norm": 0.06934399087100221,
      "learning_rate": 2.01626316987461e-05,
      "loss": 0.3093,
      "step": 2428
    },
    {
      "epoch": 3.4949640287769785,
      "grad_norm": 0.06646782712492776,
      "learning_rate": 2.0127745218384193e-05,
      "loss": 0.3103,
      "step": 2429
    },
    {
      "epoch": 3.4964028776978417,
      "grad_norm": 0.06950058637538396,
      "learning_rate": 2.009287879618236e-05,
      "loss": 0.3135,
      "step": 2430
    },
    {
      "epoch": 3.497841726618705,
      "grad_norm": 0.06866221637545802,
      "learning_rate": 2.0058032467333204e-05,
      "loss": 0.3169,
      "step": 2431
    },
    {
      "epoch": 3.499280575539568,
      "grad_norm": 0.0645057742473268,
      "learning_rate": 2.0023206267009056e-05,
      "loss": 0.3176,
      "step": 2432
    },
    {
      "epoch": 3.5007194244604314,
      "grad_norm": 0.06589944743996938,
      "learning_rate": 1.9988400230361872e-05,
      "loss": 0.3139,
      "step": 2433
    },
    {
      "epoch": 3.502158273381295,
      "grad_norm": 0.06639231700006921,
      "learning_rate": 1.9953614392523345e-05,
      "loss": 0.3002,
      "step": 2434
    },
    {
      "epoch": 3.5035971223021583,
      "grad_norm": 0.07072191123946665,
      "learning_rate": 1.9918848788604738e-05,
      "loss": 0.3182,
      "step": 2435
    },
    {
      "epoch": 3.5050359712230215,
      "grad_norm": 0.06800598042601348,
      "learning_rate": 1.9884103453696837e-05,
      "loss": 0.3187,
      "step": 2436
    },
    {
      "epoch": 3.506474820143885,
      "grad_norm": 0.06296346756178148,
      "learning_rate": 1.9849378422870082e-05,
      "loss": 0.3143,
      "step": 2437
    },
    {
      "epoch": 3.5079136690647483,
      "grad_norm": 0.06690443343286248,
      "learning_rate": 1.9814673731174315e-05,
      "loss": 0.3164,
      "step": 2438
    },
    {
      "epoch": 3.5093525179856115,
      "grad_norm": 0.06677680553334467,
      "learning_rate": 1.97799894136389e-05,
      "loss": 0.3119,
      "step": 2439
    },
    {
      "epoch": 3.5107913669064748,
      "grad_norm": 0.06444580512826723,
      "learning_rate": 1.9745325505272633e-05,
      "loss": 0.3049,
      "step": 2440
    },
    {
      "epoch": 3.512230215827338,
      "grad_norm": 0.06874289491717574,
      "learning_rate": 1.9710682041063705e-05,
      "loss": 0.3145,
      "step": 2441
    },
    {
      "epoch": 3.5136690647482016,
      "grad_norm": 0.07217181386706867,
      "learning_rate": 1.9676059055979663e-05,
      "loss": 0.3078,
      "step": 2442
    },
    {
      "epoch": 3.515107913669065,
      "grad_norm": 0.06564361893188796,
      "learning_rate": 1.9641456584967392e-05,
      "loss": 0.3032,
      "step": 2443
    },
    {
      "epoch": 3.516546762589928,
      "grad_norm": 0.07487083795793006,
      "learning_rate": 1.9606874662953076e-05,
      "loss": 0.3101,
      "step": 2444
    },
    {
      "epoch": 3.5179856115107913,
      "grad_norm": 0.0618777293904377,
      "learning_rate": 1.9572313324842148e-05,
      "loss": 0.302,
      "step": 2445
    },
    {
      "epoch": 3.5194244604316545,
      "grad_norm": 0.07249629854440197,
      "learning_rate": 1.9537772605519285e-05,
      "loss": 0.3063,
      "step": 2446
    },
    {
      "epoch": 3.520863309352518,
      "grad_norm": 0.06472724303978827,
      "learning_rate": 1.950325253984828e-05,
      "loss": 0.3156,
      "step": 2447
    },
    {
      "epoch": 3.5223021582733813,
      "grad_norm": 0.07258332278787377,
      "learning_rate": 1.946875316267219e-05,
      "loss": 0.3072,
      "step": 2448
    },
    {
      "epoch": 3.5237410071942445,
      "grad_norm": 0.07185391155876489,
      "learning_rate": 1.9434274508813135e-05,
      "loss": 0.3146,
      "step": 2449
    },
    {
      "epoch": 3.5251798561151078,
      "grad_norm": 0.065249610001787,
      "learning_rate": 1.9399816613072287e-05,
      "loss": 0.3059,
      "step": 2450
    },
    {
      "epoch": 3.526618705035971,
      "grad_norm": 0.0720522516278948,
      "learning_rate": 1.9365379510229888e-05,
      "loss": 0.3145,
      "step": 2451
    },
    {
      "epoch": 3.5280575539568346,
      "grad_norm": 0.07040021193745491,
      "learning_rate": 1.9330963235045253e-05,
      "loss": 0.3141,
      "step": 2452
    },
    {
      "epoch": 3.529496402877698,
      "grad_norm": 0.06867038188715696,
      "learning_rate": 1.9296567822256577e-05,
      "loss": 0.3161,
      "step": 2453
    },
    {
      "epoch": 3.530935251798561,
      "grad_norm": 0.06663744178748973,
      "learning_rate": 1.9262193306581052e-05,
      "loss": 0.3059,
      "step": 2454
    },
    {
      "epoch": 3.5323741007194247,
      "grad_norm": 0.0645062042179706,
      "learning_rate": 1.922783972271477e-05,
      "loss": 0.3078,
      "step": 2455
    },
    {
      "epoch": 3.533812949640288,
      "grad_norm": 0.0634394118400146,
      "learning_rate": 1.9193507105332702e-05,
      "loss": 0.2995,
      "step": 2456
    },
    {
      "epoch": 3.535251798561151,
      "grad_norm": 0.06934873745320619,
      "learning_rate": 1.9159195489088636e-05,
      "loss": 0.3049,
      "step": 2457
    },
    {
      "epoch": 3.5366906474820143,
      "grad_norm": 0.0639265233110726,
      "learning_rate": 1.9124904908615178e-05,
      "loss": 0.308,
      "step": 2458
    },
    {
      "epoch": 3.5381294964028775,
      "grad_norm": 0.06375477688785211,
      "learning_rate": 1.9090635398523698e-05,
      "loss": 0.3042,
      "step": 2459
    },
    {
      "epoch": 3.539568345323741,
      "grad_norm": 0.08106792318534015,
      "learning_rate": 1.9056386993404294e-05,
      "loss": 0.312,
      "step": 2460
    },
    {
      "epoch": 3.5410071942446044,
      "grad_norm": 0.06438660808844053,
      "learning_rate": 1.902215972782579e-05,
      "loss": 0.3087,
      "step": 2461
    },
    {
      "epoch": 3.5424460431654676,
      "grad_norm": 0.06804453264424278,
      "learning_rate": 1.8987953636335595e-05,
      "loss": 0.3055,
      "step": 2462
    },
    {
      "epoch": 3.543884892086331,
      "grad_norm": 0.06606195736768096,
      "learning_rate": 1.8953768753459863e-05,
      "loss": 0.3126,
      "step": 2463
    },
    {
      "epoch": 3.545323741007194,
      "grad_norm": 0.07120465726116934,
      "learning_rate": 1.8919605113703227e-05,
      "loss": 0.313,
      "step": 2464
    },
    {
      "epoch": 3.5467625899280577,
      "grad_norm": 0.06464663246904867,
      "learning_rate": 1.888546275154895e-05,
      "loss": 0.3138,
      "step": 2465
    },
    {
      "epoch": 3.548201438848921,
      "grad_norm": 0.06262826257038849,
      "learning_rate": 1.885134170145879e-05,
      "loss": 0.3089,
      "step": 2466
    },
    {
      "epoch": 3.549640287769784,
      "grad_norm": 0.06815666810533745,
      "learning_rate": 1.8817241997873007e-05,
      "loss": 0.2966,
      "step": 2467
    },
    {
      "epoch": 3.5510791366906473,
      "grad_norm": 0.05941656249430085,
      "learning_rate": 1.8783163675210307e-05,
      "loss": 0.3148,
      "step": 2468
    },
    {
      "epoch": 3.5525179856115106,
      "grad_norm": 0.07041289292419853,
      "learning_rate": 1.8749106767867808e-05,
      "loss": 0.3122,
      "step": 2469
    },
    {
      "epoch": 3.553956834532374,
      "grad_norm": 0.07358680810037943,
      "learning_rate": 1.871507131022103e-05,
      "loss": 0.3148,
      "step": 2470
    },
    {
      "epoch": 3.5553956834532374,
      "grad_norm": 0.06256848030022917,
      "learning_rate": 1.8681057336623825e-05,
      "loss": 0.3128,
      "step": 2471
    },
    {
      "epoch": 3.5568345323741006,
      "grad_norm": 0.06645120167514493,
      "learning_rate": 1.864706488140839e-05,
      "loss": 0.308,
      "step": 2472
    },
    {
      "epoch": 3.5582733812949643,
      "grad_norm": 0.07095189238251999,
      "learning_rate": 1.861309397888513e-05,
      "loss": 0.3138,
      "step": 2473
    },
    {
      "epoch": 3.5597122302158275,
      "grad_norm": 0.07511578845905682,
      "learning_rate": 1.857914466334279e-05,
      "loss": 0.3006,
      "step": 2474
    },
    {
      "epoch": 3.5611510791366907,
      "grad_norm": 0.06184718761923469,
      "learning_rate": 1.8545216969048288e-05,
      "loss": 0.3067,
      "step": 2475
    },
    {
      "epoch": 3.562589928057554,
      "grad_norm": 0.0732481003171402,
      "learning_rate": 1.851131093024668e-05,
      "loss": 0.3151,
      "step": 2476
    },
    {
      "epoch": 3.564028776978417,
      "grad_norm": 0.07321066110757826,
      "learning_rate": 1.8477426581161192e-05,
      "loss": 0.3076,
      "step": 2477
    },
    {
      "epoch": 3.565467625899281,
      "grad_norm": 0.06872200415539845,
      "learning_rate": 1.844356395599322e-05,
      "loss": 0.3177,
      "step": 2478
    },
    {
      "epoch": 3.566906474820144,
      "grad_norm": 0.07148169759889017,
      "learning_rate": 1.840972308892211e-05,
      "loss": 0.3085,
      "step": 2479
    },
    {
      "epoch": 3.568345323741007,
      "grad_norm": 0.06651479480795791,
      "learning_rate": 1.837590401410532e-05,
      "loss": 0.3083,
      "step": 2480
    },
    {
      "epoch": 3.5697841726618704,
      "grad_norm": 0.07008142727674253,
      "learning_rate": 1.8342106765678358e-05,
      "loss": 0.3104,
      "step": 2481
    },
    {
      "epoch": 3.5712230215827336,
      "grad_norm": 0.06859727234979068,
      "learning_rate": 1.8308331377754584e-05,
      "loss": 0.3128,
      "step": 2482
    },
    {
      "epoch": 3.5726618705035973,
      "grad_norm": 0.07352856994415677,
      "learning_rate": 1.8274577884425383e-05,
      "loss": 0.3085,
      "step": 2483
    },
    {
      "epoch": 3.5741007194244605,
      "grad_norm": 0.06259710447078544,
      "learning_rate": 1.8240846319760012e-05,
      "loss": 0.3063,
      "step": 2484
    },
    {
      "epoch": 3.5755395683453237,
      "grad_norm": 0.0688473836174342,
      "learning_rate": 1.8207136717805585e-05,
      "loss": 0.297,
      "step": 2485
    },
    {
      "epoch": 3.576978417266187,
      "grad_norm": 0.06606776561775955,
      "learning_rate": 1.8173449112587062e-05,
      "loss": 0.3054,
      "step": 2486
    },
    {
      "epoch": 3.57841726618705,
      "grad_norm": 0.06220494904604853,
      "learning_rate": 1.813978353810722e-05,
      "loss": 0.2996,
      "step": 2487
    },
    {
      "epoch": 3.579856115107914,
      "grad_norm": 0.08421515901945559,
      "learning_rate": 1.8106140028346526e-05,
      "loss": 0.3058,
      "step": 2488
    },
    {
      "epoch": 3.581294964028777,
      "grad_norm": 0.06064233136038843,
      "learning_rate": 1.8072518617263276e-05,
      "loss": 0.3133,
      "step": 2489
    },
    {
      "epoch": 3.58273381294964,
      "grad_norm": 0.07980235943596127,
      "learning_rate": 1.803891933879338e-05,
      "loss": 0.3071,
      "step": 2490
    },
    {
      "epoch": 3.584172661870504,
      "grad_norm": 0.0640938741526328,
      "learning_rate": 1.8005342226850423e-05,
      "loss": 0.3146,
      "step": 2491
    },
    {
      "epoch": 3.5856115107913666,
      "grad_norm": 0.0761611712183301,
      "learning_rate": 1.7971787315325684e-05,
      "loss": 0.316,
      "step": 2492
    },
    {
      "epoch": 3.5870503597122303,
      "grad_norm": 0.06930077652154204,
      "learning_rate": 1.7938254638087946e-05,
      "loss": 0.3079,
      "step": 2493
    },
    {
      "epoch": 3.5884892086330935,
      "grad_norm": 0.0716508617153888,
      "learning_rate": 1.7904744228983585e-05,
      "loss": 0.307,
      "step": 2494
    },
    {
      "epoch": 3.5899280575539567,
      "grad_norm": 0.06378955887830169,
      "learning_rate": 1.7871256121836507e-05,
      "loss": 0.3084,
      "step": 2495
    },
    {
      "epoch": 3.5913669064748204,
      "grad_norm": 0.06898893310679868,
      "learning_rate": 1.7837790350448098e-05,
      "loss": 0.3068,
      "step": 2496
    },
    {
      "epoch": 3.5928057553956836,
      "grad_norm": 0.06604980702701138,
      "learning_rate": 1.7804346948597206e-05,
      "loss": 0.3044,
      "step": 2497
    },
    {
      "epoch": 3.594244604316547,
      "grad_norm": 0.07092813218675978,
      "learning_rate": 1.7770925950040114e-05,
      "loss": 0.301,
      "step": 2498
    },
    {
      "epoch": 3.59568345323741,
      "grad_norm": 0.06175603012900154,
      "learning_rate": 1.773752738851042e-05,
      "loss": 0.3058,
      "step": 2499
    },
    {
      "epoch": 3.597122302158273,
      "grad_norm": 0.0685423524799416,
      "learning_rate": 1.770415129771918e-05,
      "loss": 0.3117,
      "step": 2500
    },
    {
      "epoch": 3.598561151079137,
      "grad_norm": 0.06536157690154143,
      "learning_rate": 1.7670797711354724e-05,
      "loss": 0.3121,
      "step": 2501
    },
    {
      "epoch": 3.6,
      "grad_norm": 0.0710392964305355,
      "learning_rate": 1.763746666308261e-05,
      "loss": 0.3121,
      "step": 2502
    },
    {
      "epoch": 3.6014388489208633,
      "grad_norm": 0.06777863684239872,
      "learning_rate": 1.760415818654574e-05,
      "loss": 0.321,
      "step": 2503
    },
    {
      "epoch": 3.6028776978417265,
      "grad_norm": 0.07728431278232174,
      "learning_rate": 1.75708723153642e-05,
      "loss": 0.3104,
      "step": 2504
    },
    {
      "epoch": 3.6043165467625897,
      "grad_norm": 0.07168939295547448,
      "learning_rate": 1.7537609083135224e-05,
      "loss": 0.3069,
      "step": 2505
    },
    {
      "epoch": 3.6057553956834534,
      "grad_norm": 0.08032230286800195,
      "learning_rate": 1.7504368523433216e-05,
      "loss": 0.3024,
      "step": 2506
    },
    {
      "epoch": 3.6071942446043166,
      "grad_norm": 0.05749273372875255,
      "learning_rate": 1.747115066980974e-05,
      "loss": 0.3017,
      "step": 2507
    },
    {
      "epoch": 3.60863309352518,
      "grad_norm": 0.07140683658749929,
      "learning_rate": 1.7437955555793372e-05,
      "loss": 0.311,
      "step": 2508
    },
    {
      "epoch": 3.610071942446043,
      "grad_norm": 0.0732014880124806,
      "learning_rate": 1.740478321488978e-05,
      "loss": 0.3092,
      "step": 2509
    },
    {
      "epoch": 3.6115107913669062,
      "grad_norm": 0.0662511977338535,
      "learning_rate": 1.737163368058162e-05,
      "loss": 0.3091,
      "step": 2510
    },
    {
      "epoch": 3.61294964028777,
      "grad_norm": 0.06611797050487064,
      "learning_rate": 1.7338506986328552e-05,
      "loss": 0.3054,
      "step": 2511
    },
    {
      "epoch": 3.614388489208633,
      "grad_norm": 0.06937017863617154,
      "learning_rate": 1.730540316556717e-05,
      "loss": 0.307,
      "step": 2512
    },
    {
      "epoch": 3.6158273381294963,
      "grad_norm": 0.06385096341018605,
      "learning_rate": 1.727232225171098e-05,
      "loss": 0.3053,
      "step": 2513
    },
    {
      "epoch": 3.61726618705036,
      "grad_norm": 0.07092946980616527,
      "learning_rate": 1.7239264278150364e-05,
      "loss": 0.317,
      "step": 2514
    },
    {
      "epoch": 3.618705035971223,
      "grad_norm": 0.08718889810295921,
      "learning_rate": 1.7206229278252577e-05,
      "loss": 0.3105,
      "step": 2515
    },
    {
      "epoch": 3.6201438848920864,
      "grad_norm": 0.0719947954139465,
      "learning_rate": 1.717321728536163e-05,
      "loss": 0.3113,
      "step": 2516
    },
    {
      "epoch": 3.6215827338129496,
      "grad_norm": 0.06418350060711016,
      "learning_rate": 1.7140228332798336e-05,
      "loss": 0.3065,
      "step": 2517
    },
    {
      "epoch": 3.623021582733813,
      "grad_norm": 0.07292396382004765,
      "learning_rate": 1.7107262453860308e-05,
      "loss": 0.3162,
      "step": 2518
    },
    {
      "epoch": 3.6244604316546765,
      "grad_norm": 0.07322695486853029,
      "learning_rate": 1.707431968182179e-05,
      "loss": 0.3058,
      "step": 2519
    },
    {
      "epoch": 3.6258992805755397,
      "grad_norm": 0.061263534587851595,
      "learning_rate": 1.7041400049933726e-05,
      "loss": 0.3083,
      "step": 2520
    },
    {
      "epoch": 3.627338129496403,
      "grad_norm": 0.06960676333387338,
      "learning_rate": 1.700850359142373e-05,
      "loss": 0.3056,
      "step": 2521
    },
    {
      "epoch": 3.628776978417266,
      "grad_norm": 0.06683108415097225,
      "learning_rate": 1.6975630339496e-05,
      "loss": 0.3116,
      "step": 2522
    },
    {
      "epoch": 3.6302158273381293,
      "grad_norm": 0.06433834831333483,
      "learning_rate": 1.6942780327331317e-05,
      "loss": 0.3137,
      "step": 2523
    },
    {
      "epoch": 3.631654676258993,
      "grad_norm": 0.0691490163582227,
      "learning_rate": 1.6909953588087024e-05,
      "loss": 0.3068,
      "step": 2524
    },
    {
      "epoch": 3.633093525179856,
      "grad_norm": 0.06750580224774425,
      "learning_rate": 1.687715015489691e-05,
      "loss": 0.3013,
      "step": 2525
    },
    {
      "epoch": 3.6345323741007194,
      "grad_norm": 0.07328497686061296,
      "learning_rate": 1.6844370060871324e-05,
      "loss": 0.3121,
      "step": 2526
    },
    {
      "epoch": 3.6359712230215826,
      "grad_norm": 0.06526965025274721,
      "learning_rate": 1.6811613339097022e-05,
      "loss": 0.3036,
      "step": 2527
    },
    {
      "epoch": 3.637410071942446,
      "grad_norm": 0.07280262414290611,
      "learning_rate": 1.6778880022637123e-05,
      "loss": 0.3101,
      "step": 2528
    },
    {
      "epoch": 3.6388489208633095,
      "grad_norm": 0.06492770967340347,
      "learning_rate": 1.674617014453121e-05,
      "loss": 0.303,
      "step": 2529
    },
    {
      "epoch": 3.6402877697841727,
      "grad_norm": 0.07030555773347764,
      "learning_rate": 1.6713483737795155e-05,
      "loss": 0.3086,
      "step": 2530
    },
    {
      "epoch": 3.641726618705036,
      "grad_norm": 0.07097045668922858,
      "learning_rate": 1.6680820835421124e-05,
      "loss": 0.3136,
      "step": 2531
    },
    {
      "epoch": 3.6431654676258995,
      "grad_norm": 0.06551477015746095,
      "learning_rate": 1.664818147037758e-05,
      "loss": 0.3116,
      "step": 2532
    },
    {
      "epoch": 3.6446043165467623,
      "grad_norm": 0.07000518869280055,
      "learning_rate": 1.6615565675609272e-05,
      "loss": 0.3064,
      "step": 2533
    },
    {
      "epoch": 3.646043165467626,
      "grad_norm": 0.06683324068068973,
      "learning_rate": 1.6582973484037076e-05,
      "loss": 0.3047,
      "step": 2534
    },
    {
      "epoch": 3.647482014388489,
      "grad_norm": 0.06536828321825529,
      "learning_rate": 1.6550404928558094e-05,
      "loss": 0.3101,
      "step": 2535
    },
    {
      "epoch": 3.6489208633093524,
      "grad_norm": 0.06941870589984539,
      "learning_rate": 1.6517860042045564e-05,
      "loss": 0.3107,
      "step": 2536
    },
    {
      "epoch": 3.650359712230216,
      "grad_norm": 0.061878311241111955,
      "learning_rate": 1.6485338857348826e-05,
      "loss": 0.3115,
      "step": 2537
    },
    {
      "epoch": 3.6517985611510793,
      "grad_norm": 0.0727082628618032,
      "learning_rate": 1.6452841407293307e-05,
      "loss": 0.3036,
      "step": 2538
    },
    {
      "epoch": 3.6532374100719425,
      "grad_norm": 0.060659731854837945,
      "learning_rate": 1.642036772468047e-05,
      "loss": 0.3099,
      "step": 2539
    },
    {
      "epoch": 3.6546762589928057,
      "grad_norm": 0.06868866377712568,
      "learning_rate": 1.6387917842287783e-05,
      "loss": 0.3123,
      "step": 2540
    },
    {
      "epoch": 3.656115107913669,
      "grad_norm": 0.07082289976171845,
      "learning_rate": 1.635549179286871e-05,
      "loss": 0.3083,
      "step": 2541
    },
    {
      "epoch": 3.6575539568345325,
      "grad_norm": 0.06037483578222037,
      "learning_rate": 1.6323089609152648e-05,
      "loss": 0.311,
      "step": 2542
    },
    {
      "epoch": 3.6589928057553958,
      "grad_norm": 0.06728793506224287,
      "learning_rate": 1.6290711323844866e-05,
      "loss": 0.3107,
      "step": 2543
    },
    {
      "epoch": 3.660431654676259,
      "grad_norm": 0.06837921156473406,
      "learning_rate": 1.6258356969626614e-05,
      "loss": 0.3121,
      "step": 2544
    },
    {
      "epoch": 3.661870503597122,
      "grad_norm": 0.06371398556508498,
      "learning_rate": 1.622602657915487e-05,
      "loss": 0.3089,
      "step": 2545
    },
    {
      "epoch": 3.6633093525179854,
      "grad_norm": 0.07837110896728085,
      "learning_rate": 1.6193720185062484e-05,
      "loss": 0.3193,
      "step": 2546
    },
    {
      "epoch": 3.664748201438849,
      "grad_norm": 0.07209174598097369,
      "learning_rate": 1.6161437819958087e-05,
      "loss": 0.3161,
      "step": 2547
    },
    {
      "epoch": 3.6661870503597123,
      "grad_norm": 0.07064470115761468,
      "learning_rate": 1.6129179516426048e-05,
      "loss": 0.3111,
      "step": 2548
    },
    {
      "epoch": 3.6676258992805755,
      "grad_norm": 0.08100857076451701,
      "learning_rate": 1.609694530702644e-05,
      "loss": 0.305,
      "step": 2549
    },
    {
      "epoch": 3.6690647482014387,
      "grad_norm": 0.06613161652594873,
      "learning_rate": 1.6064735224295027e-05,
      "loss": 0.3088,
      "step": 2550
    },
    {
      "epoch": 3.670503597122302,
      "grad_norm": 0.0713894278866626,
      "learning_rate": 1.603254930074322e-05,
      "loss": 0.307,
      "step": 2551
    },
    {
      "epoch": 3.6719424460431656,
      "grad_norm": 0.06583824106135408,
      "learning_rate": 1.6000387568858042e-05,
      "loss": 0.3136,
      "step": 2552
    },
    {
      "epoch": 3.6733812949640288,
      "grad_norm": 0.06573697888323053,
      "learning_rate": 1.5968250061102105e-05,
      "loss": 0.3094,
      "step": 2553
    },
    {
      "epoch": 3.674820143884892,
      "grad_norm": 0.06482376965631038,
      "learning_rate": 1.593613680991353e-05,
      "loss": 0.3092,
      "step": 2554
    },
    {
      "epoch": 3.6762589928057556,
      "grad_norm": 0.06347306421022354,
      "learning_rate": 1.590404784770603e-05,
      "loss": 0.3123,
      "step": 2555
    },
    {
      "epoch": 3.677697841726619,
      "grad_norm": 0.06437733500977535,
      "learning_rate": 1.5871983206868756e-05,
      "loss": 0.3204,
      "step": 2556
    },
    {
      "epoch": 3.679136690647482,
      "grad_norm": 0.06822100891443192,
      "learning_rate": 1.583994291976629e-05,
      "loss": 0.3046,
      "step": 2557
    },
    {
      "epoch": 3.6805755395683453,
      "grad_norm": 0.0615780409502821,
      "learning_rate": 1.580792701873865e-05,
      "loss": 0.3131,
      "step": 2558
    },
    {
      "epoch": 3.6820143884892085,
      "grad_norm": 0.060451882456547056,
      "learning_rate": 1.5775935536101296e-05,
      "loss": 0.3077,
      "step": 2559
    },
    {
      "epoch": 3.683453237410072,
      "grad_norm": 0.05949962881040066,
      "learning_rate": 1.5743968504144946e-05,
      "loss": 0.3067,
      "step": 2560
    },
    {
      "epoch": 3.6848920863309353,
      "grad_norm": 0.06496845677062033,
      "learning_rate": 1.57120259551357e-05,
      "loss": 0.3046,
      "step": 2561
    },
    {
      "epoch": 3.6863309352517986,
      "grad_norm": 0.06347335914473873,
      "learning_rate": 1.5680107921314926e-05,
      "loss": 0.3035,
      "step": 2562
    },
    {
      "epoch": 3.6877697841726618,
      "grad_norm": 0.05931661956429885,
      "learning_rate": 1.5648214434899257e-05,
      "loss": 0.3086,
      "step": 2563
    },
    {
      "epoch": 3.689208633093525,
      "grad_norm": 0.057231306144126665,
      "learning_rate": 1.5616345528080537e-05,
      "loss": 0.3024,
      "step": 2564
    },
    {
      "epoch": 3.6906474820143886,
      "grad_norm": 0.05686512028571938,
      "learning_rate": 1.5584501233025813e-05,
      "loss": 0.3125,
      "step": 2565
    },
    {
      "epoch": 3.692086330935252,
      "grad_norm": 0.058433105562290054,
      "learning_rate": 1.555268158187728e-05,
      "loss": 0.3075,
      "step": 2566
    },
    {
      "epoch": 3.693525179856115,
      "grad_norm": 0.055485795231595025,
      "learning_rate": 1.552088660675227e-05,
      "loss": 0.3057,
      "step": 2567
    },
    {
      "epoch": 3.6949640287769783,
      "grad_norm": 0.06043557385741081,
      "learning_rate": 1.54891163397432e-05,
      "loss": 0.3142,
      "step": 2568
    },
    {
      "epoch": 3.6964028776978415,
      "grad_norm": 0.0601382784406409,
      "learning_rate": 1.5457370812917526e-05,
      "loss": 0.3042,
      "step": 2569
    },
    {
      "epoch": 3.697841726618705,
      "grad_norm": 0.05986032978864982,
      "learning_rate": 1.5425650058317795e-05,
      "loss": 0.3104,
      "step": 2570
    },
    {
      "epoch": 3.6992805755395683,
      "grad_norm": 0.05796349899696422,
      "learning_rate": 1.5393954107961467e-05,
      "loss": 0.309,
      "step": 2571
    },
    {
      "epoch": 3.7007194244604316,
      "grad_norm": 0.06231745618425038,
      "learning_rate": 1.536228299384102e-05,
      "loss": 0.3125,
      "step": 2572
    },
    {
      "epoch": 3.702158273381295,
      "grad_norm": 0.06033391802311592,
      "learning_rate": 1.533063674792389e-05,
      "loss": 0.3104,
      "step": 2573
    },
    {
      "epoch": 3.7035971223021584,
      "grad_norm": 0.060051894055961556,
      "learning_rate": 1.529901540215233e-05,
      "loss": 0.3033,
      "step": 2574
    },
    {
      "epoch": 3.7050359712230216,
      "grad_norm": 0.0660384081684206,
      "learning_rate": 1.5267418988443517e-05,
      "loss": 0.3106,
      "step": 2575
    },
    {
      "epoch": 3.706474820143885,
      "grad_norm": 0.06951516962816624,
      "learning_rate": 1.5235847538689452e-05,
      "loss": 0.3085,
      "step": 2576
    },
    {
      "epoch": 3.707913669064748,
      "grad_norm": 0.06775270235025226,
      "learning_rate": 1.5204301084756936e-05,
      "loss": 0.3133,
      "step": 2577
    },
    {
      "epoch": 3.7093525179856117,
      "grad_norm": 0.06216993738083855,
      "learning_rate": 1.5172779658487539e-05,
      "loss": 0.3122,
      "step": 2578
    },
    {
      "epoch": 3.710791366906475,
      "grad_norm": 0.06842070043738754,
      "learning_rate": 1.5141283291697587e-05,
      "loss": 0.3185,
      "step": 2579
    },
    {
      "epoch": 3.712230215827338,
      "grad_norm": 0.0569504302697582,
      "learning_rate": 1.5109812016178053e-05,
      "loss": 0.312,
      "step": 2580
    },
    {
      "epoch": 3.7136690647482014,
      "grad_norm": 0.06461140384892121,
      "learning_rate": 1.5078365863694667e-05,
      "loss": 0.2967,
      "step": 2581
    },
    {
      "epoch": 3.7151079136690646,
      "grad_norm": 0.06872612298563056,
      "learning_rate": 1.5046944865987763e-05,
      "loss": 0.3032,
      "step": 2582
    },
    {
      "epoch": 3.716546762589928,
      "grad_norm": 0.05831441535529986,
      "learning_rate": 1.501554905477224e-05,
      "loss": 0.3159,
      "step": 2583
    },
    {
      "epoch": 3.7179856115107914,
      "grad_norm": 0.06402916303258853,
      "learning_rate": 1.4984178461737663e-05,
      "loss": 0.3135,
      "step": 2584
    },
    {
      "epoch": 3.7194244604316546,
      "grad_norm": 0.06553586847679266,
      "learning_rate": 1.4952833118548094e-05,
      "loss": 0.3127,
      "step": 2585
    },
    {
      "epoch": 3.720863309352518,
      "grad_norm": 0.06150736723057186,
      "learning_rate": 1.492151305684208e-05,
      "loss": 0.3072,
      "step": 2586
    },
    {
      "epoch": 3.722302158273381,
      "grad_norm": 0.06696628246885926,
      "learning_rate": 1.4890218308232704e-05,
      "loss": 0.3099,
      "step": 2587
    },
    {
      "epoch": 3.7237410071942447,
      "grad_norm": 0.06072453891762582,
      "learning_rate": 1.4858948904307476e-05,
      "loss": 0.3116,
      "step": 2588
    },
    {
      "epoch": 3.725179856115108,
      "grad_norm": 0.06592704415232585,
      "learning_rate": 1.4827704876628319e-05,
      "loss": 0.3193,
      "step": 2589
    },
    {
      "epoch": 3.726618705035971,
      "grad_norm": 0.06473227734575612,
      "learning_rate": 1.4796486256731561e-05,
      "loss": 0.3097,
      "step": 2590
    },
    {
      "epoch": 3.728057553956835,
      "grad_norm": 0.05812771491009711,
      "learning_rate": 1.4765293076127862e-05,
      "loss": 0.3141,
      "step": 2591
    },
    {
      "epoch": 3.7294964028776976,
      "grad_norm": 0.0602507431715896,
      "learning_rate": 1.4734125366302224e-05,
      "loss": 0.309,
      "step": 2592
    },
    {
      "epoch": 3.7309352517985612,
      "grad_norm": 0.05787360066411294,
      "learning_rate": 1.470298315871392e-05,
      "loss": 0.3023,
      "step": 2593
    },
    {
      "epoch": 3.7323741007194244,
      "grad_norm": 0.05675735349177474,
      "learning_rate": 1.4671866484796505e-05,
      "loss": 0.3111,
      "step": 2594
    },
    {
      "epoch": 3.7338129496402876,
      "grad_norm": 0.06130772806247975,
      "learning_rate": 1.4640775375957742e-05,
      "loss": 0.3098,
      "step": 2595
    },
    {
      "epoch": 3.7352517985611513,
      "grad_norm": 0.05898101582797852,
      "learning_rate": 1.4609709863579622e-05,
      "loss": 0.3107,
      "step": 2596
    },
    {
      "epoch": 3.7366906474820145,
      "grad_norm": 0.061029615963601236,
      "learning_rate": 1.4578669979018231e-05,
      "loss": 0.3171,
      "step": 2597
    },
    {
      "epoch": 3.7381294964028777,
      "grad_norm": 0.06442740289261426,
      "learning_rate": 1.454765575360385e-05,
      "loss": 0.3149,
      "step": 2598
    },
    {
      "epoch": 3.739568345323741,
      "grad_norm": 0.06792699078247295,
      "learning_rate": 1.4516667218640877e-05,
      "loss": 0.3106,
      "step": 2599
    },
    {
      "epoch": 3.741007194244604,
      "grad_norm": 0.06772812345824679,
      "learning_rate": 1.4485704405407699e-05,
      "loss": 0.3144,
      "step": 2600
    },
    {
      "epoch": 3.742446043165468,
      "grad_norm": 0.06605950317390714,
      "learning_rate": 1.4454767345156806e-05,
      "loss": 0.3065,
      "step": 2601
    },
    {
      "epoch": 3.743884892086331,
      "grad_norm": 0.05788478467738365,
      "learning_rate": 1.4423856069114677e-05,
      "loss": 0.308,
      "step": 2602
    },
    {
      "epoch": 3.7453237410071942,
      "grad_norm": 0.06831851166439465,
      "learning_rate": 1.4392970608481758e-05,
      "loss": 0.3094,
      "step": 2603
    },
    {
      "epoch": 3.7467625899280574,
      "grad_norm": 0.06028037648009121,
      "learning_rate": 1.4362110994432445e-05,
      "loss": 0.3124,
      "step": 2604
    },
    {
      "epoch": 3.7482014388489207,
      "grad_norm": 0.06412255417807958,
      "learning_rate": 1.433127725811505e-05,
      "loss": 0.3081,
      "step": 2605
    },
    {
      "epoch": 3.7496402877697843,
      "grad_norm": 0.05647372134593134,
      "learning_rate": 1.4300469430651754e-05,
      "loss": 0.3042,
      "step": 2606
    },
    {
      "epoch": 3.7510791366906475,
      "grad_norm": 0.09179084647263566,
      "learning_rate": 1.4269687543138594e-05,
      "loss": 0.3141,
      "step": 2607
    },
    {
      "epoch": 3.7525179856115107,
      "grad_norm": 0.06520308960062181,
      "learning_rate": 1.4238931626645434e-05,
      "loss": 0.3088,
      "step": 2608
    },
    {
      "epoch": 3.753956834532374,
      "grad_norm": 0.07436791270658137,
      "learning_rate": 1.4208201712215871e-05,
      "loss": 0.3113,
      "step": 2609
    },
    {
      "epoch": 3.755395683453237,
      "grad_norm": 0.05996695752235399,
      "learning_rate": 1.4177497830867348e-05,
      "loss": 0.3101,
      "step": 2610
    },
    {
      "epoch": 3.756834532374101,
      "grad_norm": 0.05786446268831168,
      "learning_rate": 1.4146820013590973e-05,
      "loss": 0.3077,
      "step": 2611
    },
    {
      "epoch": 3.758273381294964,
      "grad_norm": 0.06983516150573807,
      "learning_rate": 1.411616829135153e-05,
      "loss": 0.2973,
      "step": 2612
    },
    {
      "epoch": 3.7597122302158272,
      "grad_norm": 0.0658565800376622,
      "learning_rate": 1.4085542695087502e-05,
      "loss": 0.3142,
      "step": 2613
    },
    {
      "epoch": 3.761151079136691,
      "grad_norm": 0.16283982642674674,
      "learning_rate": 1.4054943255710987e-05,
      "loss": 0.3197,
      "step": 2614
    },
    {
      "epoch": 3.762589928057554,
      "grad_norm": 0.06299679372917073,
      "learning_rate": 1.4024370004107683e-05,
      "loss": 0.3036,
      "step": 2615
    },
    {
      "epoch": 3.7640287769784173,
      "grad_norm": 0.06638329576126877,
      "learning_rate": 1.3993822971136859e-05,
      "loss": 0.304,
      "step": 2616
    },
    {
      "epoch": 3.7654676258992805,
      "grad_norm": 0.062057024816662185,
      "learning_rate": 1.3963302187631316e-05,
      "loss": 0.3182,
      "step": 2617
    },
    {
      "epoch": 3.7669064748201437,
      "grad_norm": 0.05653817314485878,
      "learning_rate": 1.3932807684397348e-05,
      "loss": 0.3171,
      "step": 2618
    },
    {
      "epoch": 3.7683453237410074,
      "grad_norm": 0.06107756482833042,
      "learning_rate": 1.3902339492214751e-05,
      "loss": 0.317,
      "step": 2619
    },
    {
      "epoch": 3.7697841726618706,
      "grad_norm": 0.057954495172457515,
      "learning_rate": 1.387189764183674e-05,
      "loss": 0.3058,
      "step": 2620
    },
    {
      "epoch": 3.771223021582734,
      "grad_norm": 0.06518604390601693,
      "learning_rate": 1.384148216398995e-05,
      "loss": 0.3053,
      "step": 2621
    },
    {
      "epoch": 3.772661870503597,
      "grad_norm": 0.060156239559592144,
      "learning_rate": 1.381109308937441e-05,
      "loss": 0.3142,
      "step": 2622
    },
    {
      "epoch": 3.7741007194244602,
      "grad_norm": 0.061219184840551555,
      "learning_rate": 1.3780730448663456e-05,
      "loss": 0.3029,
      "step": 2623
    },
    {
      "epoch": 3.775539568345324,
      "grad_norm": 0.06872992961064582,
      "learning_rate": 1.3750394272503775e-05,
      "loss": 0.3038,
      "step": 2624
    },
    {
      "epoch": 3.776978417266187,
      "grad_norm": 0.06207118803048583,
      "learning_rate": 1.3720084591515374e-05,
      "loss": 0.3155,
      "step": 2625
    },
    {
      "epoch": 3.7784172661870503,
      "grad_norm": 0.06276725568796437,
      "learning_rate": 1.3689801436291448e-05,
      "loss": 0.3079,
      "step": 2626
    },
    {
      "epoch": 3.7798561151079135,
      "grad_norm": 0.06301376305449499,
      "learning_rate": 1.365954483739846e-05,
      "loss": 0.3084,
      "step": 2627
    },
    {
      "epoch": 3.7812949640287767,
      "grad_norm": 0.06786189913232177,
      "learning_rate": 1.3629314825376061e-05,
      "loss": 0.3017,
      "step": 2628
    },
    {
      "epoch": 3.7827338129496404,
      "grad_norm": 0.06923629831247584,
      "learning_rate": 1.359911143073707e-05,
      "loss": 0.3126,
      "step": 2629
    },
    {
      "epoch": 3.7841726618705036,
      "grad_norm": 0.05915691173482345,
      "learning_rate": 1.3568934683967427e-05,
      "loss": 0.3135,
      "step": 2630
    },
    {
      "epoch": 3.785611510791367,
      "grad_norm": 0.06480400653976588,
      "learning_rate": 1.3538784615526188e-05,
      "loss": 0.3088,
      "step": 2631
    },
    {
      "epoch": 3.7870503597122305,
      "grad_norm": 0.061066561210690914,
      "learning_rate": 1.3508661255845477e-05,
      "loss": 0.3166,
      "step": 2632
    },
    {
      "epoch": 3.7884892086330937,
      "grad_norm": 0.05631072013232032,
      "learning_rate": 1.3478564635330455e-05,
      "loss": 0.3049,
      "step": 2633
    },
    {
      "epoch": 3.789928057553957,
      "grad_norm": 0.06469049654437241,
      "learning_rate": 1.344849478435931e-05,
      "loss": 0.3153,
      "step": 2634
    },
    {
      "epoch": 3.79136690647482,
      "grad_norm": 0.05619451199322114,
      "learning_rate": 1.3418451733283156e-05,
      "loss": 0.3076,
      "step": 2635
    },
    {
      "epoch": 3.7928057553956833,
      "grad_norm": 0.06318767999957417,
      "learning_rate": 1.3388435512426142e-05,
      "loss": 0.3099,
      "step": 2636
    },
    {
      "epoch": 3.794244604316547,
      "grad_norm": 0.11931093635445984,
      "learning_rate": 1.3358446152085289e-05,
      "loss": 0.3212,
      "step": 2637
    },
    {
      "epoch": 3.79568345323741,
      "grad_norm": 0.05764009252951675,
      "learning_rate": 1.332848368253048e-05,
      "loss": 0.3197,
      "step": 2638
    },
    {
      "epoch": 3.7971223021582734,
      "grad_norm": 0.06625803834027354,
      "learning_rate": 1.3298548134004498e-05,
      "loss": 0.3098,
      "step": 2639
    },
    {
      "epoch": 3.7985611510791366,
      "grad_norm": 0.06722982196993153,
      "learning_rate": 1.326863953672294e-05,
      "loss": 0.3048,
      "step": 2640
    },
    {
      "epoch": 3.8,
      "grad_norm": 0.05956157666694582,
      "learning_rate": 1.3238757920874203e-05,
      "loss": 0.3113,
      "step": 2641
    },
    {
      "epoch": 3.8014388489208635,
      "grad_norm": 0.06408147015237593,
      "learning_rate": 1.3208903316619436e-05,
      "loss": 0.3138,
      "step": 2642
    },
    {
      "epoch": 3.8028776978417267,
      "grad_norm": 0.05836561799046896,
      "learning_rate": 1.317907575409254e-05,
      "loss": 0.3113,
      "step": 2643
    },
    {
      "epoch": 3.80431654676259,
      "grad_norm": 0.06586186700166624,
      "learning_rate": 1.3149275263400116e-05,
      "loss": 0.3068,
      "step": 2644
    },
    {
      "epoch": 3.805755395683453,
      "grad_norm": 0.05574505585504934,
      "learning_rate": 1.3119501874621437e-05,
      "loss": 0.3045,
      "step": 2645
    },
    {
      "epoch": 3.8071942446043163,
      "grad_norm": 0.06191955826077262,
      "learning_rate": 1.3089755617808417e-05,
      "loss": 0.3069,
      "step": 2646
    },
    {
      "epoch": 3.80863309352518,
      "grad_norm": 0.06409880201156713,
      "learning_rate": 1.3060036522985598e-05,
      "loss": 0.3149,
      "step": 2647
    },
    {
      "epoch": 3.810071942446043,
      "grad_norm": 0.05713343345695329,
      "learning_rate": 1.3030344620150105e-05,
      "loss": 0.3054,
      "step": 2648
    },
    {
      "epoch": 3.8115107913669064,
      "grad_norm": 0.07271851070609117,
      "learning_rate": 1.3000679939271588e-05,
      "loss": 0.3102,
      "step": 2649
    },
    {
      "epoch": 3.81294964028777,
      "grad_norm": 0.06882430273590348,
      "learning_rate": 1.2971042510292238e-05,
      "loss": 0.3021,
      "step": 2650
    },
    {
      "epoch": 3.814388489208633,
      "grad_norm": 0.06367380750131976,
      "learning_rate": 1.2941432363126784e-05,
      "loss": 0.3134,
      "step": 2651
    },
    {
      "epoch": 3.8158273381294965,
      "grad_norm": 0.06397499285479019,
      "learning_rate": 1.2911849527662335e-05,
      "loss": 0.3158,
      "step": 2652
    },
    {
      "epoch": 3.8172661870503597,
      "grad_norm": 0.06668572646459466,
      "learning_rate": 1.2882294033758473e-05,
      "loss": 0.3144,
      "step": 2653
    },
    {
      "epoch": 3.818705035971223,
      "grad_norm": 0.06703585684409313,
      "learning_rate": 1.2852765911247227e-05,
      "loss": 0.3043,
      "step": 2654
    },
    {
      "epoch": 3.8201438848920866,
      "grad_norm": 0.06884979624293733,
      "learning_rate": 1.2823265189932914e-05,
      "loss": 0.3147,
      "step": 2655
    },
    {
      "epoch": 3.8215827338129498,
      "grad_norm": 0.059878668782955315,
      "learning_rate": 1.2793791899592254e-05,
      "loss": 0.3191,
      "step": 2656
    },
    {
      "epoch": 3.823021582733813,
      "grad_norm": 0.059485779702165203,
      "learning_rate": 1.2764346069974249e-05,
      "loss": 0.3041,
      "step": 2657
    },
    {
      "epoch": 3.824460431654676,
      "grad_norm": 0.07647157975745708,
      "learning_rate": 1.2734927730800206e-05,
      "loss": 0.3062,
      "step": 2658
    },
    {
      "epoch": 3.8258992805755394,
      "grad_norm": 0.16489538003955,
      "learning_rate": 1.2705536911763665e-05,
      "loss": 0.3165,
      "step": 2659
    },
    {
      "epoch": 3.827338129496403,
      "grad_norm": 0.06318125861693077,
      "learning_rate": 1.2676173642530417e-05,
      "loss": 0.3033,
      "step": 2660
    },
    {
      "epoch": 3.8287769784172663,
      "grad_norm": 0.06567256901298571,
      "learning_rate": 1.2646837952738382e-05,
      "loss": 0.3056,
      "step": 2661
    },
    {
      "epoch": 3.8302158273381295,
      "grad_norm": 0.05518801916037263,
      "learning_rate": 1.2617529871997727e-05,
      "loss": 0.3084,
      "step": 2662
    },
    {
      "epoch": 3.8316546762589927,
      "grad_norm": 0.06215454503091133,
      "learning_rate": 1.2588249429890706e-05,
      "loss": 0.3085,
      "step": 2663
    },
    {
      "epoch": 3.833093525179856,
      "grad_norm": 0.06215747767088541,
      "learning_rate": 1.2558996655971644e-05,
      "loss": 0.3131,
      "step": 2664
    },
    {
      "epoch": 3.8345323741007196,
      "grad_norm": 0.0571081373298139,
      "learning_rate": 1.2529771579767024e-05,
      "loss": 0.3157,
      "step": 2665
    },
    {
      "epoch": 3.8359712230215828,
      "grad_norm": 0.05764282070698008,
      "learning_rate": 1.2500574230775294e-05,
      "loss": 0.3119,
      "step": 2666
    },
    {
      "epoch": 3.837410071942446,
      "grad_norm": 0.061498174350714155,
      "learning_rate": 1.2471404638466949e-05,
      "loss": 0.3065,
      "step": 2667
    },
    {
      "epoch": 3.838848920863309,
      "grad_norm": 0.06444471515107401,
      "learning_rate": 1.2442262832284464e-05,
      "loss": 0.3127,
      "step": 2668
    },
    {
      "epoch": 3.8402877697841724,
      "grad_norm": 0.05249170213553225,
      "learning_rate": 1.2413148841642268e-05,
      "loss": 0.3022,
      "step": 2669
    },
    {
      "epoch": 3.841726618705036,
      "grad_norm": 0.06441398889770678,
      "learning_rate": 1.2384062695926713e-05,
      "loss": 0.3068,
      "step": 2670
    },
    {
      "epoch": 3.8431654676258993,
      "grad_norm": 0.062304092178714035,
      "learning_rate": 1.235500442449605e-05,
      "loss": 0.2992,
      "step": 2671
    },
    {
      "epoch": 3.8446043165467625,
      "grad_norm": 0.060928419468288225,
      "learning_rate": 1.232597405668039e-05,
      "loss": 0.3064,
      "step": 2672
    },
    {
      "epoch": 3.846043165467626,
      "grad_norm": 0.0614376313092846,
      "learning_rate": 1.2296971621781677e-05,
      "loss": 0.302,
      "step": 2673
    },
    {
      "epoch": 3.8474820143884894,
      "grad_norm": 0.07040298267064284,
      "learning_rate": 1.2267997149073679e-05,
      "loss": 0.3107,
      "step": 2674
    },
    {
      "epoch": 3.8489208633093526,
      "grad_norm": 0.06682703801152587,
      "learning_rate": 1.2239050667801885e-05,
      "loss": 0.3143,
      "step": 2675
    },
    {
      "epoch": 3.850359712230216,
      "grad_norm": 0.057759323436254666,
      "learning_rate": 1.2210132207183611e-05,
      "loss": 0.3173,
      "step": 2676
    },
    {
      "epoch": 3.851798561151079,
      "grad_norm": 0.06570271107765477,
      "learning_rate": 1.2181241796407855e-05,
      "loss": 0.3128,
      "step": 2677
    },
    {
      "epoch": 3.8532374100719426,
      "grad_norm": 0.056153261565319953,
      "learning_rate": 1.2152379464635264e-05,
      "loss": 0.3121,
      "step": 2678
    },
    {
      "epoch": 3.854676258992806,
      "grad_norm": 0.07057172176534204,
      "learning_rate": 1.2123545240998182e-05,
      "loss": 0.3091,
      "step": 2679
    },
    {
      "epoch": 3.856115107913669,
      "grad_norm": 0.05974707197694837,
      "learning_rate": 1.2094739154600616e-05,
      "loss": 0.3077,
      "step": 2680
    },
    {
      "epoch": 3.8575539568345323,
      "grad_norm": 0.0585049483720299,
      "learning_rate": 1.2065961234518096e-05,
      "loss": 0.3042,
      "step": 2681
    },
    {
      "epoch": 3.8589928057553955,
      "grad_norm": 0.05401931011856757,
      "learning_rate": 1.2037211509797771e-05,
      "loss": 0.3084,
      "step": 2682
    },
    {
      "epoch": 3.860431654676259,
      "grad_norm": 0.05991698970812978,
      "learning_rate": 1.2008490009458322e-05,
      "loss": 0.3026,
      "step": 2683
    },
    {
      "epoch": 3.8618705035971224,
      "grad_norm": 0.06668618504592472,
      "learning_rate": 1.1979796762489934e-05,
      "loss": 0.3095,
      "step": 2684
    },
    {
      "epoch": 3.8633093525179856,
      "grad_norm": 0.05842009339861893,
      "learning_rate": 1.195113179785429e-05,
      "loss": 0.3123,
      "step": 2685
    },
    {
      "epoch": 3.864748201438849,
      "grad_norm": 0.05566524147630196,
      "learning_rate": 1.1922495144484504e-05,
      "loss": 0.3028,
      "step": 2686
    },
    {
      "epoch": 3.866187050359712,
      "grad_norm": 0.061870388935277666,
      "learning_rate": 1.1893886831285136e-05,
      "loss": 0.3076,
      "step": 2687
    },
    {
      "epoch": 3.8676258992805757,
      "grad_norm": 0.05847049335714899,
      "learning_rate": 1.1865306887132122e-05,
      "loss": 0.3023,
      "step": 2688
    },
    {
      "epoch": 3.869064748201439,
      "grad_norm": 0.0528554059921117,
      "learning_rate": 1.183675534087279e-05,
      "loss": 0.3043,
      "step": 2689
    },
    {
      "epoch": 3.870503597122302,
      "grad_norm": 0.0578540020472865,
      "learning_rate": 1.1808232221325749e-05,
      "loss": 0.301,
      "step": 2690
    },
    {
      "epoch": 3.8719424460431657,
      "grad_norm": 0.057910505762297546,
      "learning_rate": 1.1779737557280985e-05,
      "loss": 0.3087,
      "step": 2691
    },
    {
      "epoch": 3.873381294964029,
      "grad_norm": 0.06273079560410788,
      "learning_rate": 1.1751271377499736e-05,
      "loss": 0.3067,
      "step": 2692
    },
    {
      "epoch": 3.874820143884892,
      "grad_norm": 0.05935430832911722,
      "learning_rate": 1.1722833710714454e-05,
      "loss": 0.3128,
      "step": 2693
    },
    {
      "epoch": 3.8762589928057554,
      "grad_norm": 0.176977809547984,
      "learning_rate": 1.1694424585628861e-05,
      "loss": 0.3207,
      "step": 2694
    },
    {
      "epoch": 3.8776978417266186,
      "grad_norm": 0.06022004430395275,
      "learning_rate": 1.166604403091784e-05,
      "loss": 0.3014,
      "step": 2695
    },
    {
      "epoch": 3.8791366906474822,
      "grad_norm": 0.06223362207654486,
      "learning_rate": 1.1637692075227451e-05,
      "loss": 0.3089,
      "step": 2696
    },
    {
      "epoch": 3.8805755395683454,
      "grad_norm": 0.06874855910589964,
      "learning_rate": 1.1609368747174883e-05,
      "loss": 0.3143,
      "step": 2697
    },
    {
      "epoch": 3.8820143884892087,
      "grad_norm": 0.060370453841061725,
      "learning_rate": 1.1581074075348431e-05,
      "loss": 0.3172,
      "step": 2698
    },
    {
      "epoch": 3.883453237410072,
      "grad_norm": 0.06925325983031541,
      "learning_rate": 1.155280808830746e-05,
      "loss": 0.3071,
      "step": 2699
    },
    {
      "epoch": 3.884892086330935,
      "grad_norm": 0.06409960859469387,
      "learning_rate": 1.15245708145824e-05,
      "loss": 0.309,
      "step": 2700
    },
    {
      "epoch": 3.8863309352517987,
      "grad_norm": 0.062347138259423694,
      "learning_rate": 1.1496362282674647e-05,
      "loss": 0.3066,
      "step": 2701
    },
    {
      "epoch": 3.887769784172662,
      "grad_norm": 0.060137578227819184,
      "learning_rate": 1.1468182521056663e-05,
      "loss": 0.3057,
      "step": 2702
    },
    {
      "epoch": 3.889208633093525,
      "grad_norm": 0.0663260823248241,
      "learning_rate": 1.1440031558171834e-05,
      "loss": 0.3119,
      "step": 2703
    },
    {
      "epoch": 3.8906474820143884,
      "grad_norm": 0.05526910347359479,
      "learning_rate": 1.1411909422434441e-05,
      "loss": 0.3063,
      "step": 2704
    },
    {
      "epoch": 3.8920863309352516,
      "grad_norm": 0.05847051252150774,
      "learning_rate": 1.1383816142229715e-05,
      "loss": 0.2982,
      "step": 2705
    },
    {
      "epoch": 3.8935251798561152,
      "grad_norm": 0.06429140059660536,
      "learning_rate": 1.1355751745913781e-05,
      "loss": 0.3076,
      "step": 2706
    },
    {
      "epoch": 3.8949640287769784,
      "grad_norm": 0.05672662126677904,
      "learning_rate": 1.1327716261813539e-05,
      "loss": 0.3074,
      "step": 2707
    },
    {
      "epoch": 3.8964028776978417,
      "grad_norm": 0.06428826342831176,
      "learning_rate": 1.1299709718226745e-05,
      "loss": 0.3203,
      "step": 2708
    },
    {
      "epoch": 3.897841726618705,
      "grad_norm": 0.06905960502906461,
      "learning_rate": 1.1271732143421992e-05,
      "loss": 0.3145,
      "step": 2709
    },
    {
      "epoch": 3.899280575539568,
      "grad_norm": 0.05555046754317393,
      "learning_rate": 1.1243783565638533e-05,
      "loss": 0.3119,
      "step": 2710
    },
    {
      "epoch": 3.9007194244604317,
      "grad_norm": 0.2581207444305228,
      "learning_rate": 1.121586401308643e-05,
      "loss": 0.3183,
      "step": 2711
    },
    {
      "epoch": 3.902158273381295,
      "grad_norm": 0.05829992626182602,
      "learning_rate": 1.1187973513946417e-05,
      "loss": 0.3074,
      "step": 2712
    },
    {
      "epoch": 3.903597122302158,
      "grad_norm": 0.056376048677371636,
      "learning_rate": 1.1160112096369913e-05,
      "loss": 0.318,
      "step": 2713
    },
    {
      "epoch": 3.905035971223022,
      "grad_norm": 0.059928006320685914,
      "learning_rate": 1.1132279788478977e-05,
      "loss": 0.308,
      "step": 2714
    },
    {
      "epoch": 3.906474820143885,
      "grad_norm": 0.05541920682418896,
      "learning_rate": 1.1104476618366298e-05,
      "loss": 0.3173,
      "step": 2715
    },
    {
      "epoch": 3.9079136690647482,
      "grad_norm": 0.05309312395846593,
      "learning_rate": 1.1076702614095116e-05,
      "loss": 0.2997,
      "step": 2716
    },
    {
      "epoch": 3.9093525179856115,
      "grad_norm": 0.060377652429816915,
      "learning_rate": 1.1048957803699292e-05,
      "loss": 0.3172,
      "step": 2717
    },
    {
      "epoch": 3.9107913669064747,
      "grad_norm": 0.05214388021450585,
      "learning_rate": 1.1021242215183193e-05,
      "loss": 0.308,
      "step": 2718
    },
    {
      "epoch": 3.9122302158273383,
      "grad_norm": 0.054476845725141956,
      "learning_rate": 1.0993555876521658e-05,
      "loss": 0.3024,
      "step": 2719
    },
    {
      "epoch": 3.9136690647482015,
      "grad_norm": 0.05522494995246515,
      "learning_rate": 1.096589881566005e-05,
      "loss": 0.3062,
      "step": 2720
    },
    {
      "epoch": 3.9151079136690647,
      "grad_norm": 0.05330150644521752,
      "learning_rate": 1.0938271060514162e-05,
      "loss": 0.3068,
      "step": 2721
    },
    {
      "epoch": 3.916546762589928,
      "grad_norm": 0.05748988376868886,
      "learning_rate": 1.0910672638970206e-05,
      "loss": 0.3094,
      "step": 2722
    },
    {
      "epoch": 3.917985611510791,
      "grad_norm": 0.0554307762195385,
      "learning_rate": 1.0883103578884784e-05,
      "loss": 0.306,
      "step": 2723
    },
    {
      "epoch": 3.919424460431655,
      "grad_norm": 0.05323938446383709,
      "learning_rate": 1.085556390808487e-05,
      "loss": 0.3193,
      "step": 2724
    },
    {
      "epoch": 3.920863309352518,
      "grad_norm": 0.056347745539254916,
      "learning_rate": 1.082805365436776e-05,
      "loss": 0.3082,
      "step": 2725
    },
    {
      "epoch": 3.9223021582733812,
      "grad_norm": 0.056289500329134816,
      "learning_rate": 1.0800572845501095e-05,
      "loss": 0.3121,
      "step": 2726
    },
    {
      "epoch": 3.9237410071942445,
      "grad_norm": 0.05554719201993376,
      "learning_rate": 1.0773121509222712e-05,
      "loss": 0.3182,
      "step": 2727
    },
    {
      "epoch": 3.9251798561151077,
      "grad_norm": 0.06374598503779488,
      "learning_rate": 1.0745699673240808e-05,
      "loss": 0.3136,
      "step": 2728
    },
    {
      "epoch": 3.9266187050359713,
      "grad_norm": 0.06042213685904112,
      "learning_rate": 1.0718307365233737e-05,
      "loss": 0.3083,
      "step": 2729
    },
    {
      "epoch": 3.9280575539568345,
      "grad_norm": 0.05747646863163404,
      "learning_rate": 1.0690944612850052e-05,
      "loss": 0.3121,
      "step": 2730
    },
    {
      "epoch": 3.9294964028776977,
      "grad_norm": 0.06334409390272748,
      "learning_rate": 1.0663611443708471e-05,
      "loss": 0.3054,
      "step": 2731
    },
    {
      "epoch": 3.9309352517985614,
      "grad_norm": 0.053503891340662454,
      "learning_rate": 1.0636307885397911e-05,
      "loss": 0.3128,
      "step": 2732
    },
    {
      "epoch": 3.9323741007194246,
      "grad_norm": 0.06154617585575193,
      "learning_rate": 1.0609033965477318e-05,
      "loss": 0.3101,
      "step": 2733
    },
    {
      "epoch": 3.933812949640288,
      "grad_norm": 0.0581555444560647,
      "learning_rate": 1.0581789711475752e-05,
      "loss": 0.3088,
      "step": 2734
    },
    {
      "epoch": 3.935251798561151,
      "grad_norm": 0.0592766340142503,
      "learning_rate": 1.0554575150892386e-05,
      "loss": 0.3144,
      "step": 2735
    },
    {
      "epoch": 3.9366906474820142,
      "grad_norm": 0.05803536652402496,
      "learning_rate": 1.0527390311196326e-05,
      "loss": 0.316,
      "step": 2736
    },
    {
      "epoch": 3.938129496402878,
      "grad_norm": 0.12891965481062795,
      "learning_rate": 1.0500235219826748e-05,
      "loss": 0.3216,
      "step": 2737
    },
    {
      "epoch": 3.939568345323741,
      "grad_norm": 0.060135996685002246,
      "learning_rate": 1.0473109904192773e-05,
      "loss": 0.3016,
      "step": 2738
    },
    {
      "epoch": 3.9410071942446043,
      "grad_norm": 0.05936836251326289,
      "learning_rate": 1.0446014391673476e-05,
      "loss": 0.3117,
      "step": 2739
    },
    {
      "epoch": 3.9424460431654675,
      "grad_norm": 0.05606392843941533,
      "learning_rate": 1.0418948709617846e-05,
      "loss": 0.3051,
      "step": 2740
    },
    {
      "epoch": 3.9438848920863308,
      "grad_norm": 0.0615264351109077,
      "learning_rate": 1.0391912885344784e-05,
      "loss": 0.315,
      "step": 2741
    },
    {
      "epoch": 3.9453237410071944,
      "grad_norm": 0.05647955827623214,
      "learning_rate": 1.0364906946142996e-05,
      "loss": 0.3053,
      "step": 2742
    },
    {
      "epoch": 3.9467625899280576,
      "grad_norm": 0.054273776569956106,
      "learning_rate": 1.0337930919271094e-05,
      "loss": 0.308,
      "step": 2743
    },
    {
      "epoch": 3.948201438848921,
      "grad_norm": 0.10330088413233861,
      "learning_rate": 1.0310984831957471e-05,
      "loss": 0.3156,
      "step": 2744
    },
    {
      "epoch": 3.949640287769784,
      "grad_norm": 0.05528293581596261,
      "learning_rate": 1.0284068711400254e-05,
      "loss": 0.3031,
      "step": 2745
    },
    {
      "epoch": 3.9510791366906473,
      "grad_norm": 0.05894655309137916,
      "learning_rate": 1.0257182584767423e-05,
      "loss": 0.3075,
      "step": 2746
    },
    {
      "epoch": 3.952517985611511,
      "grad_norm": 0.05607153633072281,
      "learning_rate": 1.0230326479196573e-05,
      "loss": 0.3065,
      "step": 2747
    },
    {
      "epoch": 3.953956834532374,
      "grad_norm": 0.055195443443561164,
      "learning_rate": 1.0203500421795075e-05,
      "loss": 0.3152,
      "step": 2748
    },
    {
      "epoch": 3.9553956834532373,
      "grad_norm": 0.06149427347933062,
      "learning_rate": 1.017670443963994e-05,
      "loss": 0.3079,
      "step": 2749
    },
    {
      "epoch": 3.956834532374101,
      "grad_norm": 0.06094647156893332,
      "learning_rate": 1.0149938559777825e-05,
      "loss": 0.309,
      "step": 2750
    },
    {
      "epoch": 3.9582733812949638,
      "grad_norm": 0.05664212483371998,
      "learning_rate": 1.0123202809225009e-05,
      "loss": 0.3117,
      "step": 2751
    },
    {
      "epoch": 3.9597122302158274,
      "grad_norm": 0.06329360387733363,
      "learning_rate": 1.0096497214967349e-05,
      "loss": 0.3036,
      "step": 2752
    },
    {
      "epoch": 3.9611510791366906,
      "grad_norm": 0.058119174435414415,
      "learning_rate": 1.0069821803960277e-05,
      "loss": 0.3131,
      "step": 2753
    },
    {
      "epoch": 3.962589928057554,
      "grad_norm": 0.05595191987767366,
      "learning_rate": 1.0043176603128755e-05,
      "loss": 0.3103,
      "step": 2754
    },
    {
      "epoch": 3.9640287769784175,
      "grad_norm": 0.05502864366856433,
      "learning_rate": 1.0016561639367253e-05,
      "loss": 0.3097,
      "step": 2755
    },
    {
      "epoch": 3.9654676258992807,
      "grad_norm": 0.052737631100398905,
      "learning_rate": 9.989976939539687e-06,
      "loss": 0.3118,
      "step": 2756
    },
    {
      "epoch": 3.966906474820144,
      "grad_norm": 0.05642274570543027,
      "learning_rate": 9.963422530479496e-06,
      "loss": 0.3046,
      "step": 2757
    },
    {
      "epoch": 3.968345323741007,
      "grad_norm": 0.053250375328443185,
      "learning_rate": 9.936898438989507e-06,
      "loss": 0.3016,
      "step": 2758
    },
    {
      "epoch": 3.9697841726618703,
      "grad_norm": 0.05711820644911219,
      "learning_rate": 9.910404691841915e-06,
      "loss": 0.3117,
      "step": 2759
    },
    {
      "epoch": 3.971223021582734,
      "grad_norm": 0.05621493149689219,
      "learning_rate": 9.883941315778319e-06,
      "loss": 0.3079,
      "step": 2760
    },
    {
      "epoch": 3.972661870503597,
      "grad_norm": 0.05627457631006604,
      "learning_rate": 9.857508337509692e-06,
      "loss": 0.3143,
      "step": 2761
    },
    {
      "epoch": 3.9741007194244604,
      "grad_norm": 0.0528755363079526,
      "learning_rate": 9.831105783716266e-06,
      "loss": 0.3088,
      "step": 2762
    },
    {
      "epoch": 3.9755395683453236,
      "grad_norm": 0.05672953310062539,
      "learning_rate": 9.8047336810476e-06,
      "loss": 0.3107,
      "step": 2763
    },
    {
      "epoch": 3.976978417266187,
      "grad_norm": 0.05792026261999815,
      "learning_rate": 9.778392056122503e-06,
      "loss": 0.3116,
      "step": 2764
    },
    {
      "epoch": 3.9784172661870505,
      "grad_norm": 0.05859681948245761,
      "learning_rate": 9.752080935529037e-06,
      "loss": 0.3099,
      "step": 2765
    },
    {
      "epoch": 3.9798561151079137,
      "grad_norm": 0.062351998100664094,
      "learning_rate": 9.725800345824453e-06,
      "loss": 0.3054,
      "step": 2766
    },
    {
      "epoch": 3.981294964028777,
      "grad_norm": 0.05538121828358899,
      "learning_rate": 9.699550313535196e-06,
      "loss": 0.3097,
      "step": 2767
    },
    {
      "epoch": 3.98273381294964,
      "grad_norm": 0.054767113499204136,
      "learning_rate": 9.673330865156875e-06,
      "loss": 0.3009,
      "step": 2768
    },
    {
      "epoch": 3.9841726618705033,
      "grad_norm": 0.05460313364278765,
      "learning_rate": 9.647142027154222e-06,
      "loss": 0.3121,
      "step": 2769
    },
    {
      "epoch": 3.985611510791367,
      "grad_norm": 0.057503736480089745,
      "learning_rate": 9.620983825961078e-06,
      "loss": 0.3081,
      "step": 2770
    },
    {
      "epoch": 3.98705035971223,
      "grad_norm": 0.05405237866099866,
      "learning_rate": 9.594856287980323e-06,
      "loss": 0.3043,
      "step": 2771
    },
    {
      "epoch": 3.9884892086330934,
      "grad_norm": 0.05758398953886903,
      "learning_rate": 9.56875943958396e-06,
      "loss": 0.305,
      "step": 2772
    },
    {
      "epoch": 3.989928057553957,
      "grad_norm": 0.059091244036538114,
      "learning_rate": 9.542693307112949e-06,
      "loss": 0.3065,
      "step": 2773
    },
    {
      "epoch": 3.9913669064748203,
      "grad_norm": 0.05447280953122306,
      "learning_rate": 9.516657916877272e-06,
      "loss": 0.3101,
      "step": 2774
    },
    {
      "epoch": 3.9928057553956835,
      "grad_norm": 0.052766134508733596,
      "learning_rate": 9.490653295155891e-06,
      "loss": 0.3047,
      "step": 2775
    },
    {
      "epoch": 3.9942446043165467,
      "grad_norm": 0.05510053829467285,
      "learning_rate": 9.464679468196696e-06,
      "loss": 0.3141,
      "step": 2776
    },
    {
      "epoch": 3.99568345323741,
      "grad_norm": 0.05623421629121918,
      "learning_rate": 9.438736462216496e-06,
      "loss": 0.3088,
      "step": 2777
    },
    {
      "epoch": 3.9971223021582736,
      "grad_norm": 0.05780685261699166,
      "learning_rate": 9.412824303401003e-06,
      "loss": 0.3022,
      "step": 2778
    },
    {
      "epoch": 3.998561151079137,
      "grad_norm": 0.05594874455328415,
      "learning_rate": 9.38694301790478e-06,
      "loss": 0.3117,
      "step": 2779
    },
    {
      "epoch": 4.0,
      "grad_norm": 0.05307323279651753,
      "learning_rate": 9.361092631851228e-06,
      "loss": 0.3,
      "step": 2780
    },
    {
      "epoch": 4.001438848920864,
      "grad_norm": 0.1043212830115124,
      "learning_rate": 9.335273171332581e-06,
      "loss": 0.2854,
      "step": 2781
    },
    {
      "epoch": 4.002877697841726,
      "grad_norm": 0.07730198498278092,
      "learning_rate": 9.30948466240981e-06,
      "loss": 0.2831,
      "step": 2782
    },
    {
      "epoch": 4.00431654676259,
      "grad_norm": 0.08892434575632624,
      "learning_rate": 9.2837271311127e-06,
      "loss": 0.2884,
      "step": 2783
    },
    {
      "epoch": 4.005755395683453,
      "grad_norm": 0.09610014701185844,
      "learning_rate": 9.25800060343975e-06,
      "loss": 0.2853,
      "step": 2784
    },
    {
      "epoch": 4.0071942446043165,
      "grad_norm": 0.09566633437440125,
      "learning_rate": 9.232305105358139e-06,
      "loss": 0.2764,
      "step": 2785
    },
    {
      "epoch": 4.00863309352518,
      "grad_norm": 0.09732839889203922,
      "learning_rate": 9.206640662803746e-06,
      "loss": 0.2841,
      "step": 2786
    },
    {
      "epoch": 4.010071942446043,
      "grad_norm": 0.08107566020332119,
      "learning_rate": 9.181007301681135e-06,
      "loss": 0.2827,
      "step": 2787
    },
    {
      "epoch": 4.011510791366907,
      "grad_norm": 0.09798451170105568,
      "learning_rate": 9.155405047863439e-06,
      "loss": 0.2901,
      "step": 2788
    },
    {
      "epoch": 4.012949640287769,
      "grad_norm": 0.09339308886157084,
      "learning_rate": 9.12983392719243e-06,
      "loss": 0.2843,
      "step": 2789
    },
    {
      "epoch": 4.014388489208633,
      "grad_norm": 0.07831082980908396,
      "learning_rate": 9.104293965478446e-06,
      "loss": 0.2975,
      "step": 2790
    },
    {
      "epoch": 4.015827338129497,
      "grad_norm": 0.09726756244872185,
      "learning_rate": 9.078785188500378e-06,
      "loss": 0.2786,
      "step": 2791
    },
    {
      "epoch": 4.017266187050359,
      "grad_norm": 0.0839026602171746,
      "learning_rate": 9.053307622005639e-06,
      "loss": 0.2891,
      "step": 2792
    },
    {
      "epoch": 4.018705035971223,
      "grad_norm": 0.2569418270484783,
      "learning_rate": 9.02786129171013e-06,
      "loss": 0.2872,
      "step": 2793
    },
    {
      "epoch": 4.020143884892087,
      "grad_norm": 0.09036941895856587,
      "learning_rate": 9.002446223298244e-06,
      "loss": 0.2899,
      "step": 2794
    },
    {
      "epoch": 4.0215827338129495,
      "grad_norm": 0.07823056260316286,
      "learning_rate": 8.977062442422796e-06,
      "loss": 0.2825,
      "step": 2795
    },
    {
      "epoch": 4.023021582733813,
      "grad_norm": 0.06939967074653287,
      "learning_rate": 8.951709974705057e-06,
      "loss": 0.2839,
      "step": 2796
    },
    {
      "epoch": 4.024460431654676,
      "grad_norm": 0.07775161937964527,
      "learning_rate": 8.926388845734624e-06,
      "loss": 0.2867,
      "step": 2797
    },
    {
      "epoch": 4.02589928057554,
      "grad_norm": 0.08344607559720935,
      "learning_rate": 8.901099081069553e-06,
      "loss": 0.2841,
      "step": 2798
    },
    {
      "epoch": 4.027338129496403,
      "grad_norm": 0.0697908363212248,
      "learning_rate": 8.875840706236163e-06,
      "loss": 0.2797,
      "step": 2799
    },
    {
      "epoch": 4.028776978417266,
      "grad_norm": 0.0702787799782245,
      "learning_rate": 8.850613746729117e-06,
      "loss": 0.2824,
      "step": 2800
    },
    {
      "epoch": 4.03021582733813,
      "grad_norm": 0.0784344071210075,
      "learning_rate": 8.825418228011413e-06,
      "loss": 0.2858,
      "step": 2801
    },
    {
      "epoch": 4.031654676258992,
      "grad_norm": 0.06910497452870903,
      "learning_rate": 8.80025417551424e-06,
      "loss": 0.2845,
      "step": 2802
    },
    {
      "epoch": 4.033093525179856,
      "grad_norm": 0.0703298040829297,
      "learning_rate": 8.775121614637064e-06,
      "loss": 0.2899,
      "step": 2803
    },
    {
      "epoch": 4.03453237410072,
      "grad_norm": 0.07461618628151065,
      "learning_rate": 8.750020570747568e-06,
      "loss": 0.2925,
      "step": 2804
    },
    {
      "epoch": 4.0359712230215825,
      "grad_norm": 0.060595293834618015,
      "learning_rate": 8.724951069181617e-06,
      "loss": 0.2826,
      "step": 2805
    },
    {
      "epoch": 4.037410071942446,
      "grad_norm": 0.05947476428356925,
      "learning_rate": 8.699913135243237e-06,
      "loss": 0.2854,
      "step": 2806
    },
    {
      "epoch": 4.038848920863309,
      "grad_norm": 0.06970235397677162,
      "learning_rate": 8.6749067942046e-06,
      "loss": 0.289,
      "step": 2807
    },
    {
      "epoch": 4.040287769784173,
      "grad_norm": 0.057161855218202745,
      "learning_rate": 8.649932071305952e-06,
      "loss": 0.2812,
      "step": 2808
    },
    {
      "epoch": 4.041726618705036,
      "grad_norm": 0.056968773294844594,
      "learning_rate": 8.624988991755687e-06,
      "loss": 0.2892,
      "step": 2809
    },
    {
      "epoch": 4.043165467625899,
      "grad_norm": 0.06523034947999166,
      "learning_rate": 8.60007758073023e-06,
      "loss": 0.2918,
      "step": 2810
    },
    {
      "epoch": 4.044604316546763,
      "grad_norm": 0.05586819145618191,
      "learning_rate": 8.575197863374006e-06,
      "loss": 0.2866,
      "step": 2811
    },
    {
      "epoch": 4.046043165467626,
      "grad_norm": 0.05690920989007164,
      "learning_rate": 8.550349864799505e-06,
      "loss": 0.2859,
      "step": 2812
    },
    {
      "epoch": 4.047482014388489,
      "grad_norm": 0.06543859621756212,
      "learning_rate": 8.525533610087193e-06,
      "loss": 0.2774,
      "step": 2813
    },
    {
      "epoch": 4.048920863309353,
      "grad_norm": 0.05824744569514629,
      "learning_rate": 8.500749124285455e-06,
      "loss": 0.285,
      "step": 2814
    },
    {
      "epoch": 4.0503597122302155,
      "grad_norm": 0.05628826449906327,
      "learning_rate": 8.475996432410642e-06,
      "loss": 0.2843,
      "step": 2815
    },
    {
      "epoch": 4.051798561151079,
      "grad_norm": 0.0619226003891041,
      "learning_rate": 8.451275559447011e-06,
      "loss": 0.2814,
      "step": 2816
    },
    {
      "epoch": 4.053237410071943,
      "grad_norm": 0.062451691131779286,
      "learning_rate": 8.426586530346705e-06,
      "loss": 0.2873,
      "step": 2817
    },
    {
      "epoch": 4.054676258992806,
      "grad_norm": 0.05631344243537601,
      "learning_rate": 8.401929370029708e-06,
      "loss": 0.2905,
      "step": 2818
    },
    {
      "epoch": 4.056115107913669,
      "grad_norm": 0.0676675896500192,
      "learning_rate": 8.377304103383857e-06,
      "loss": 0.2824,
      "step": 2819
    },
    {
      "epoch": 4.057553956834532,
      "grad_norm": 0.05470175927050552,
      "learning_rate": 8.352710755264786e-06,
      "loss": 0.2843,
      "step": 2820
    },
    {
      "epoch": 4.058992805755396,
      "grad_norm": 0.05634470720435986,
      "learning_rate": 8.328149350495916e-06,
      "loss": 0.2859,
      "step": 2821
    },
    {
      "epoch": 4.060431654676259,
      "grad_norm": 0.062579593233253,
      "learning_rate": 8.303619913868427e-06,
      "loss": 0.29,
      "step": 2822
    },
    {
      "epoch": 4.061870503597122,
      "grad_norm": 0.054386224661288236,
      "learning_rate": 8.279122470141208e-06,
      "loss": 0.2776,
      "step": 2823
    },
    {
      "epoch": 4.063309352517986,
      "grad_norm": 0.062338182507232595,
      "learning_rate": 8.254657044040914e-06,
      "loss": 0.2892,
      "step": 2824
    },
    {
      "epoch": 4.0647482014388485,
      "grad_norm": 0.052951289601991906,
      "learning_rate": 8.230223660261814e-06,
      "loss": 0.2882,
      "step": 2825
    },
    {
      "epoch": 4.066187050359712,
      "grad_norm": 0.06809591160216252,
      "learning_rate": 8.205822343465865e-06,
      "loss": 0.2906,
      "step": 2826
    },
    {
      "epoch": 4.067625899280576,
      "grad_norm": 0.05262875248953221,
      "learning_rate": 8.181453118282694e-06,
      "loss": 0.282,
      "step": 2827
    },
    {
      "epoch": 4.069064748201439,
      "grad_norm": 0.05940219092006468,
      "learning_rate": 8.157116009309467e-06,
      "loss": 0.2852,
      "step": 2828
    },
    {
      "epoch": 4.070503597122302,
      "grad_norm": 0.057024370076768506,
      "learning_rate": 8.132811041110976e-06,
      "loss": 0.2928,
      "step": 2829
    },
    {
      "epoch": 4.071942446043165,
      "grad_norm": 0.05326639214772234,
      "learning_rate": 8.108538238219564e-06,
      "loss": 0.2848,
      "step": 2830
    },
    {
      "epoch": 4.073381294964029,
      "grad_norm": 0.053437726714684115,
      "learning_rate": 8.084297625135104e-06,
      "loss": 0.2895,
      "step": 2831
    },
    {
      "epoch": 4.074820143884892,
      "grad_norm": 0.058604609107049246,
      "learning_rate": 8.060089226324987e-06,
      "loss": 0.2856,
      "step": 2832
    },
    {
      "epoch": 4.076258992805755,
      "grad_norm": 0.053144560045417705,
      "learning_rate": 8.035913066224088e-06,
      "loss": 0.2802,
      "step": 2833
    },
    {
      "epoch": 4.077697841726619,
      "grad_norm": 0.05393640789482759,
      "learning_rate": 8.0117691692347e-06,
      "loss": 0.2861,
      "step": 2834
    },
    {
      "epoch": 4.079136690647482,
      "grad_norm": 0.0554108586723657,
      "learning_rate": 7.987657559726628e-06,
      "loss": 0.2834,
      "step": 2835
    },
    {
      "epoch": 4.080575539568345,
      "grad_norm": 0.052701212312735835,
      "learning_rate": 7.963578262037038e-06,
      "loss": 0.2852,
      "step": 2836
    },
    {
      "epoch": 4.082014388489209,
      "grad_norm": 0.05505255808236752,
      "learning_rate": 7.939531300470458e-06,
      "loss": 0.2876,
      "step": 2837
    },
    {
      "epoch": 4.083453237410072,
      "grad_norm": 0.05295422349610247,
      "learning_rate": 7.915516699298847e-06,
      "loss": 0.296,
      "step": 2838
    },
    {
      "epoch": 4.084892086330935,
      "grad_norm": 0.051567097545618855,
      "learning_rate": 7.891534482761463e-06,
      "loss": 0.2833,
      "step": 2839
    },
    {
      "epoch": 4.086330935251799,
      "grad_norm": 0.05219702522484231,
      "learning_rate": 7.867584675064846e-06,
      "loss": 0.2864,
      "step": 2840
    },
    {
      "epoch": 4.087769784172662,
      "grad_norm": 0.05179815914083395,
      "learning_rate": 7.843667300382863e-06,
      "loss": 0.2806,
      "step": 2841
    },
    {
      "epoch": 4.089208633093525,
      "grad_norm": 0.05368160641813598,
      "learning_rate": 7.81978238285667e-06,
      "loss": 0.2848,
      "step": 2842
    },
    {
      "epoch": 4.090647482014388,
      "grad_norm": 0.05810459252023169,
      "learning_rate": 7.795929946594584e-06,
      "loss": 0.2789,
      "step": 2843
    },
    {
      "epoch": 4.092086330935252,
      "grad_norm": 0.050098165455327956,
      "learning_rate": 7.772110015672209e-06,
      "loss": 0.2793,
      "step": 2844
    },
    {
      "epoch": 4.093525179856115,
      "grad_norm": 0.050937414080107296,
      "learning_rate": 7.748322614132297e-06,
      "loss": 0.2897,
      "step": 2845
    },
    {
      "epoch": 4.094964028776978,
      "grad_norm": 0.05238091268905972,
      "learning_rate": 7.72456776598479e-06,
      "loss": 0.2835,
      "step": 2846
    },
    {
      "epoch": 4.096402877697842,
      "grad_norm": 0.05434773894332223,
      "learning_rate": 7.70084549520676e-06,
      "loss": 0.2825,
      "step": 2847
    },
    {
      "epoch": 4.097841726618705,
      "grad_norm": 0.05174764145307135,
      "learning_rate": 7.6771558257424e-06,
      "loss": 0.2854,
      "step": 2848
    },
    {
      "epoch": 4.099280575539568,
      "grad_norm": 0.057151568563367755,
      "learning_rate": 7.653498781502997e-06,
      "loss": 0.2813,
      "step": 2849
    },
    {
      "epoch": 4.100719424460432,
      "grad_norm": 0.04875085068136776,
      "learning_rate": 7.629874386366918e-06,
      "loss": 0.2777,
      "step": 2850
    },
    {
      "epoch": 4.102158273381295,
      "grad_norm": 0.05195638644729961,
      "learning_rate": 7.606282664179545e-06,
      "loss": 0.2883,
      "step": 2851
    },
    {
      "epoch": 4.103597122302158,
      "grad_norm": 0.058528895470629484,
      "learning_rate": 7.5827236387532976e-06,
      "loss": 0.2874,
      "step": 2852
    },
    {
      "epoch": 4.105035971223022,
      "grad_norm": 0.05478092147502408,
      "learning_rate": 7.559197333867629e-06,
      "loss": 0.274,
      "step": 2853
    },
    {
      "epoch": 4.106474820143885,
      "grad_norm": 0.05885883121797831,
      "learning_rate": 7.53570377326891e-06,
      "loss": 0.2886,
      "step": 2854
    },
    {
      "epoch": 4.107913669064748,
      "grad_norm": 0.05378570314666407,
      "learning_rate": 7.512242980670481e-06,
      "loss": 0.2856,
      "step": 2855
    },
    {
      "epoch": 4.109352517985611,
      "grad_norm": 0.05047021641202359,
      "learning_rate": 7.488814979752615e-06,
      "loss": 0.2835,
      "step": 2856
    },
    {
      "epoch": 4.110791366906475,
      "grad_norm": 0.050547742980969305,
      "learning_rate": 7.465419794162487e-06,
      "loss": 0.2867,
      "step": 2857
    },
    {
      "epoch": 4.1122302158273385,
      "grad_norm": 0.0490690485972627,
      "learning_rate": 7.442057447514144e-06,
      "loss": 0.2797,
      "step": 2858
    },
    {
      "epoch": 4.113669064748201,
      "grad_norm": 0.05206071276401406,
      "learning_rate": 7.418727963388481e-06,
      "loss": 0.2942,
      "step": 2859
    },
    {
      "epoch": 4.115107913669065,
      "grad_norm": 0.04900725135791448,
      "learning_rate": 7.395431365333241e-06,
      "loss": 0.2897,
      "step": 2860
    },
    {
      "epoch": 4.116546762589928,
      "grad_norm": 0.04918959977849764,
      "learning_rate": 7.372167676862952e-06,
      "loss": 0.2752,
      "step": 2861
    },
    {
      "epoch": 4.117985611510791,
      "grad_norm": 0.05385482651396476,
      "learning_rate": 7.348936921458949e-06,
      "loss": 0.2832,
      "step": 2862
    },
    {
      "epoch": 4.119424460431655,
      "grad_norm": 0.05256305439467217,
      "learning_rate": 7.325739122569282e-06,
      "loss": 0.2877,
      "step": 2863
    },
    {
      "epoch": 4.120863309352518,
      "grad_norm": 0.05078066127205468,
      "learning_rate": 7.302574303608794e-06,
      "loss": 0.2825,
      "step": 2864
    },
    {
      "epoch": 4.122302158273381,
      "grad_norm": 0.05145114814377242,
      "learning_rate": 7.279442487959012e-06,
      "loss": 0.2796,
      "step": 2865
    },
    {
      "epoch": 4.123741007194244,
      "grad_norm": 0.05472436663389367,
      "learning_rate": 7.256343698968131e-06,
      "loss": 0.295,
      "step": 2866
    },
    {
      "epoch": 4.125179856115108,
      "grad_norm": 0.054104706784023536,
      "learning_rate": 7.233277959951026e-06,
      "loss": 0.2896,
      "step": 2867
    },
    {
      "epoch": 4.1266187050359715,
      "grad_norm": 0.05674997857150882,
      "learning_rate": 7.210245294189251e-06,
      "loss": 0.2823,
      "step": 2868
    },
    {
      "epoch": 4.128057553956834,
      "grad_norm": 0.04874500387689634,
      "learning_rate": 7.187245724930911e-06,
      "loss": 0.2918,
      "step": 2869
    },
    {
      "epoch": 4.129496402877698,
      "grad_norm": 0.04619627546770947,
      "learning_rate": 7.164279275390749e-06,
      "loss": 0.282,
      "step": 2870
    },
    {
      "epoch": 4.130935251798562,
      "grad_norm": 0.049751766918390714,
      "learning_rate": 7.14134596875006e-06,
      "loss": 0.2848,
      "step": 2871
    },
    {
      "epoch": 4.132374100719424,
      "grad_norm": 0.05350054049527425,
      "learning_rate": 7.118445828156697e-06,
      "loss": 0.282,
      "step": 2872
    },
    {
      "epoch": 4.133812949640288,
      "grad_norm": 0.04690832284051088,
      "learning_rate": 7.0955788767250334e-06,
      "loss": 0.2863,
      "step": 2873
    },
    {
      "epoch": 4.135251798561151,
      "grad_norm": 0.04754333086734813,
      "learning_rate": 7.0727451375359345e-06,
      "loss": 0.2879,
      "step": 2874
    },
    {
      "epoch": 4.136690647482014,
      "grad_norm": 0.05222521043696767,
      "learning_rate": 7.049944633636756e-06,
      "loss": 0.2887,
      "step": 2875
    },
    {
      "epoch": 4.138129496402878,
      "grad_norm": 0.049879256026004834,
      "learning_rate": 7.027177388041311e-06,
      "loss": 0.2852,
      "step": 2876
    },
    {
      "epoch": 4.139568345323741,
      "grad_norm": 0.049328534052848386,
      "learning_rate": 7.004443423729808e-06,
      "loss": 0.2886,
      "step": 2877
    },
    {
      "epoch": 4.1410071942446045,
      "grad_norm": 0.05023564759363154,
      "learning_rate": 6.981742763648891e-06,
      "loss": 0.2866,
      "step": 2878
    },
    {
      "epoch": 4.142446043165467,
      "grad_norm": 0.20239413390092711,
      "learning_rate": 6.959075430711614e-06,
      "loss": 0.2971,
      "step": 2879
    },
    {
      "epoch": 4.143884892086331,
      "grad_norm": 0.05076010444591421,
      "learning_rate": 6.936441447797335e-06,
      "loss": 0.2866,
      "step": 2880
    },
    {
      "epoch": 4.145323741007195,
      "grad_norm": 0.049379185923188657,
      "learning_rate": 6.913840837751778e-06,
      "loss": 0.2833,
      "step": 2881
    },
    {
      "epoch": 4.146762589928057,
      "grad_norm": 0.0529015164113446,
      "learning_rate": 6.8912736233870095e-06,
      "loss": 0.2891,
      "step": 2882
    },
    {
      "epoch": 4.148201438848921,
      "grad_norm": 0.04926536197741727,
      "learning_rate": 6.868739827481335e-06,
      "loss": 0.2782,
      "step": 2883
    },
    {
      "epoch": 4.149640287769784,
      "grad_norm": 0.04723571998825495,
      "learning_rate": 6.846239472779359e-06,
      "loss": 0.2848,
      "step": 2884
    },
    {
      "epoch": 4.151079136690647,
      "grad_norm": 0.05075196787398027,
      "learning_rate": 6.82377258199193e-06,
      "loss": 0.2809,
      "step": 2885
    },
    {
      "epoch": 4.152517985611511,
      "grad_norm": 0.05010532301754759,
      "learning_rate": 6.80133917779612e-06,
      "loss": 0.2842,
      "step": 2886
    },
    {
      "epoch": 4.153956834532374,
      "grad_norm": 0.05281873962011146,
      "learning_rate": 6.778939282835195e-06,
      "loss": 0.2861,
      "step": 2887
    },
    {
      "epoch": 4.1553956834532375,
      "grad_norm": 0.049782188980158876,
      "learning_rate": 6.756572919718611e-06,
      "loss": 0.2875,
      "step": 2888
    },
    {
      "epoch": 4.1568345323741,
      "grad_norm": 0.055433584840673934,
      "learning_rate": 6.734240111021937e-06,
      "loss": 0.2891,
      "step": 2889
    },
    {
      "epoch": 4.158273381294964,
      "grad_norm": 0.04956071896209945,
      "learning_rate": 6.711940879286944e-06,
      "loss": 0.2807,
      "step": 2890
    },
    {
      "epoch": 4.159712230215828,
      "grad_norm": 0.05422120966942815,
      "learning_rate": 6.689675247021461e-06,
      "loss": 0.29,
      "step": 2891
    },
    {
      "epoch": 4.16115107913669,
      "grad_norm": 0.05157553339280195,
      "learning_rate": 6.667443236699398e-06,
      "loss": 0.2847,
      "step": 2892
    },
    {
      "epoch": 4.162589928057554,
      "grad_norm": 0.04944057862866461,
      "learning_rate": 6.64524487076077e-06,
      "loss": 0.293,
      "step": 2893
    },
    {
      "epoch": 4.164028776978418,
      "grad_norm": 0.05023221501561263,
      "learning_rate": 6.623080171611605e-06,
      "loss": 0.2794,
      "step": 2894
    },
    {
      "epoch": 4.16546762589928,
      "grad_norm": 0.05555620939833201,
      "learning_rate": 6.600949161623939e-06,
      "loss": 0.2769,
      "step": 2895
    },
    {
      "epoch": 4.166906474820144,
      "grad_norm": 0.053803815152891114,
      "learning_rate": 6.578851863135831e-06,
      "loss": 0.2832,
      "step": 2896
    },
    {
      "epoch": 4.168345323741007,
      "grad_norm": 0.047906703153829215,
      "learning_rate": 6.556788298451291e-06,
      "loss": 0.2848,
      "step": 2897
    },
    {
      "epoch": 4.1697841726618705,
      "grad_norm": 0.056442437064043754,
      "learning_rate": 6.534758489840296e-06,
      "loss": 0.2852,
      "step": 2898
    },
    {
      "epoch": 4.171223021582734,
      "grad_norm": 0.053616468579709003,
      "learning_rate": 6.512762459538744e-06,
      "loss": 0.2905,
      "step": 2899
    },
    {
      "epoch": 4.172661870503597,
      "grad_norm": 0.04887273018888442,
      "learning_rate": 6.49080022974843e-06,
      "loss": 0.2821,
      "step": 2900
    },
    {
      "epoch": 4.174100719424461,
      "grad_norm": 0.05351889400146498,
      "learning_rate": 6.468871822637051e-06,
      "loss": 0.2819,
      "step": 2901
    },
    {
      "epoch": 4.175539568345323,
      "grad_norm": 0.05547256928044035,
      "learning_rate": 6.446977260338152e-06,
      "loss": 0.2887,
      "step": 2902
    },
    {
      "epoch": 4.176978417266187,
      "grad_norm": 0.05084436131281766,
      "learning_rate": 6.425116564951115e-06,
      "loss": 0.2866,
      "step": 2903
    },
    {
      "epoch": 4.178417266187051,
      "grad_norm": 0.052107981016798956,
      "learning_rate": 6.403289758541143e-06,
      "loss": 0.2891,
      "step": 2904
    },
    {
      "epoch": 4.179856115107913,
      "grad_norm": 0.05355343441284484,
      "learning_rate": 6.381496863139247e-06,
      "loss": 0.2835,
      "step": 2905
    },
    {
      "epoch": 4.181294964028777,
      "grad_norm": 0.049893232291485824,
      "learning_rate": 6.3597379007421755e-06,
      "loss": 0.2878,
      "step": 2906
    },
    {
      "epoch": 4.18273381294964,
      "grad_norm": 0.07239619223915701,
      "learning_rate": 6.338012893312444e-06,
      "loss": 0.2949,
      "step": 2907
    },
    {
      "epoch": 4.1841726618705035,
      "grad_norm": 0.04993222977179514,
      "learning_rate": 6.31632186277833e-06,
      "loss": 0.2844,
      "step": 2908
    },
    {
      "epoch": 4.185611510791367,
      "grad_norm": 0.05044234362148215,
      "learning_rate": 6.294664831033746e-06,
      "loss": 0.2951,
      "step": 2909
    },
    {
      "epoch": 4.18705035971223,
      "grad_norm": 0.048300204379515224,
      "learning_rate": 6.273041819938343e-06,
      "loss": 0.2876,
      "step": 2910
    },
    {
      "epoch": 4.188489208633094,
      "grad_norm": 0.047482360054201984,
      "learning_rate": 6.251452851317421e-06,
      "loss": 0.2958,
      "step": 2911
    },
    {
      "epoch": 4.189928057553957,
      "grad_norm": 0.047487943147196676,
      "learning_rate": 6.229897946961903e-06,
      "loss": 0.2908,
      "step": 2912
    },
    {
      "epoch": 4.19136690647482,
      "grad_norm": 0.050600528349371905,
      "learning_rate": 6.20837712862834e-06,
      "loss": 0.2834,
      "step": 2913
    },
    {
      "epoch": 4.192805755395684,
      "grad_norm": 0.05127597982053406,
      "learning_rate": 6.186890418038887e-06,
      "loss": 0.2929,
      "step": 2914
    },
    {
      "epoch": 4.194244604316546,
      "grad_norm": 0.050312325645171053,
      "learning_rate": 6.165437836881256e-06,
      "loss": 0.2868,
      "step": 2915
    },
    {
      "epoch": 4.19568345323741,
      "grad_norm": 0.049959117057170654,
      "learning_rate": 6.144019406808724e-06,
      "loss": 0.2888,
      "step": 2916
    },
    {
      "epoch": 4.197122302158274,
      "grad_norm": 0.05182875516760443,
      "learning_rate": 6.122635149440093e-06,
      "loss": 0.2814,
      "step": 2917
    },
    {
      "epoch": 4.1985611510791365,
      "grad_norm": 0.05272581281709548,
      "learning_rate": 6.101285086359645e-06,
      "loss": 0.2988,
      "step": 2918
    },
    {
      "epoch": 4.2,
      "grad_norm": 0.048477762613489274,
      "learning_rate": 6.079969239117201e-06,
      "loss": 0.286,
      "step": 2919
    },
    {
      "epoch": 4.201438848920863,
      "grad_norm": 0.048996323200934035,
      "learning_rate": 6.05868762922802e-06,
      "loss": 0.2876,
      "step": 2920
    },
    {
      "epoch": 4.202877697841727,
      "grad_norm": 0.04875957798976624,
      "learning_rate": 6.037440278172782e-06,
      "loss": 0.289,
      "step": 2921
    },
    {
      "epoch": 4.20431654676259,
      "grad_norm": 0.04765756332215801,
      "learning_rate": 6.016227207397616e-06,
      "loss": 0.2815,
      "step": 2922
    },
    {
      "epoch": 4.205755395683453,
      "grad_norm": 0.0497126707077397,
      "learning_rate": 5.995048438314044e-06,
      "loss": 0.2791,
      "step": 2923
    },
    {
      "epoch": 4.207194244604317,
      "grad_norm": 0.04653946342028209,
      "learning_rate": 5.973903992298962e-06,
      "loss": 0.2875,
      "step": 2924
    },
    {
      "epoch": 4.2086330935251794,
      "grad_norm": 0.04924042551777451,
      "learning_rate": 5.952793890694617e-06,
      "loss": 0.2862,
      "step": 2925
    },
    {
      "epoch": 4.210071942446043,
      "grad_norm": 0.051758404134836025,
      "learning_rate": 5.9317181548086055e-06,
      "loss": 0.2816,
      "step": 2926
    },
    {
      "epoch": 4.211510791366907,
      "grad_norm": 0.047704734972427824,
      "learning_rate": 5.910676805913822e-06,
      "loss": 0.294,
      "step": 2927
    },
    {
      "epoch": 4.2129496402877695,
      "grad_norm": 0.05089276802228076,
      "learning_rate": 5.889669865248455e-06,
      "loss": 0.2869,
      "step": 2928
    },
    {
      "epoch": 4.214388489208633,
      "grad_norm": 0.047597601617219244,
      "learning_rate": 5.8686973540159706e-06,
      "loss": 0.2882,
      "step": 2929
    },
    {
      "epoch": 4.215827338129497,
      "grad_norm": 0.04924280137176742,
      "learning_rate": 5.847759293385075e-06,
      "loss": 0.2842,
      "step": 2930
    },
    {
      "epoch": 4.21726618705036,
      "grad_norm": 0.04903403474162036,
      "learning_rate": 5.8268557044897175e-06,
      "loss": 0.2924,
      "step": 2931
    },
    {
      "epoch": 4.218705035971223,
      "grad_norm": 0.04818439389958206,
      "learning_rate": 5.805986608429019e-06,
      "loss": 0.2897,
      "step": 2932
    },
    {
      "epoch": 4.220143884892086,
      "grad_norm": 0.04581373055337451,
      "learning_rate": 5.785152026267309e-06,
      "loss": 0.2798,
      "step": 2933
    },
    {
      "epoch": 4.22158273381295,
      "grad_norm": 0.048126025527469155,
      "learning_rate": 5.764351979034102e-06,
      "loss": 0.2807,
      "step": 2934
    },
    {
      "epoch": 4.223021582733813,
      "grad_norm": 0.04746825529325539,
      "learning_rate": 5.743586487724e-06,
      "loss": 0.2832,
      "step": 2935
    },
    {
      "epoch": 4.224460431654676,
      "grad_norm": 0.05151018633435883,
      "learning_rate": 5.722855573296775e-06,
      "loss": 0.2869,
      "step": 2936
    },
    {
      "epoch": 4.22589928057554,
      "grad_norm": 0.04761838167446043,
      "learning_rate": 5.702159256677266e-06,
      "loss": 0.28,
      "step": 2937
    },
    {
      "epoch": 4.2273381294964025,
      "grad_norm": 0.05200047832651735,
      "learning_rate": 5.681497558755417e-06,
      "loss": 0.2941,
      "step": 2938
    },
    {
      "epoch": 4.228776978417266,
      "grad_norm": 0.04740391127957364,
      "learning_rate": 5.6608705003862085e-06,
      "loss": 0.2791,
      "step": 2939
    },
    {
      "epoch": 4.23021582733813,
      "grad_norm": 0.048150554546777156,
      "learning_rate": 5.6402781023896695e-06,
      "loss": 0.2857,
      "step": 2940
    },
    {
      "epoch": 4.231654676258993,
      "grad_norm": 0.046799705680378056,
      "learning_rate": 5.619720385550835e-06,
      "loss": 0.2853,
      "step": 2941
    },
    {
      "epoch": 4.233093525179856,
      "grad_norm": 0.04725239128908153,
      "learning_rate": 5.5991973706197445e-06,
      "loss": 0.2836,
      "step": 2942
    },
    {
      "epoch": 4.234532374100719,
      "grad_norm": 0.044872887371123056,
      "learning_rate": 5.578709078311417e-06,
      "loss": 0.2863,
      "step": 2943
    },
    {
      "epoch": 4.235971223021583,
      "grad_norm": 0.05015078020635212,
      "learning_rate": 5.558255529305779e-06,
      "loss": 0.2843,
      "step": 2944
    },
    {
      "epoch": 4.237410071942446,
      "grad_norm": 0.051963637365920454,
      "learning_rate": 5.537836744247753e-06,
      "loss": 0.2859,
      "step": 2945
    },
    {
      "epoch": 4.238848920863309,
      "grad_norm": 0.04704737505765592,
      "learning_rate": 5.517452743747145e-06,
      "loss": 0.2848,
      "step": 2946
    },
    {
      "epoch": 4.240287769784173,
      "grad_norm": 0.04987546625046638,
      "learning_rate": 5.497103548378628e-06,
      "loss": 0.2858,
      "step": 2947
    },
    {
      "epoch": 4.2417266187050355,
      "grad_norm": 0.04864707037982135,
      "learning_rate": 5.476789178681769e-06,
      "loss": 0.2813,
      "step": 2948
    },
    {
      "epoch": 4.243165467625899,
      "grad_norm": 0.05300509050782678,
      "learning_rate": 5.456509655160989e-06,
      "loss": 0.284,
      "step": 2949
    },
    {
      "epoch": 4.244604316546763,
      "grad_norm": 0.04704427554950865,
      "learning_rate": 5.436264998285516e-06,
      "loss": 0.2902,
      "step": 2950
    },
    {
      "epoch": 4.246043165467626,
      "grad_norm": 0.04890309795539629,
      "learning_rate": 5.4160552284894075e-06,
      "loss": 0.2958,
      "step": 2951
    },
    {
      "epoch": 4.247482014388489,
      "grad_norm": 0.049436994603225766,
      "learning_rate": 5.3958803661714865e-06,
      "loss": 0.2792,
      "step": 2952
    },
    {
      "epoch": 4.248920863309353,
      "grad_norm": 0.05064437091392666,
      "learning_rate": 5.375740431695353e-06,
      "loss": 0.2879,
      "step": 2953
    },
    {
      "epoch": 4.250359712230216,
      "grad_norm": 0.24718137686965794,
      "learning_rate": 5.355635445389355e-06,
      "loss": 0.3011,
      "step": 2954
    },
    {
      "epoch": 4.251798561151079,
      "grad_norm": 0.051772752251279835,
      "learning_rate": 5.3355654275465584e-06,
      "loss": 0.2924,
      "step": 2955
    },
    {
      "epoch": 4.253237410071942,
      "grad_norm": 0.05156888608226264,
      "learning_rate": 5.315530398424735e-06,
      "loss": 0.2843,
      "step": 2956
    },
    {
      "epoch": 4.254676258992806,
      "grad_norm": 0.049485260469823086,
      "learning_rate": 5.295530378246354e-06,
      "loss": 0.2913,
      "step": 2957
    },
    {
      "epoch": 4.256115107913669,
      "grad_norm": 0.04677394192798888,
      "learning_rate": 5.27556538719852e-06,
      "loss": 0.2944,
      "step": 2958
    },
    {
      "epoch": 4.257553956834532,
      "grad_norm": 0.05233186413123022,
      "learning_rate": 5.2556354454329895e-06,
      "loss": 0.2794,
      "step": 2959
    },
    {
      "epoch": 4.258992805755396,
      "grad_norm": 0.04930931732994099,
      "learning_rate": 5.235740573066186e-06,
      "loss": 0.2779,
      "step": 2960
    },
    {
      "epoch": 4.260431654676259,
      "grad_norm": 0.049213575603995675,
      "learning_rate": 5.21588079017906e-06,
      "loss": 0.2845,
      "step": 2961
    },
    {
      "epoch": 4.261870503597122,
      "grad_norm": 0.04659787031115534,
      "learning_rate": 5.196056116817194e-06,
      "loss": 0.2837,
      "step": 2962
    },
    {
      "epoch": 4.263309352517986,
      "grad_norm": 0.05293959590454975,
      "learning_rate": 5.1762665729907424e-06,
      "loss": 0.2851,
      "step": 2963
    },
    {
      "epoch": 4.264748201438849,
      "grad_norm": 0.04998528391845439,
      "learning_rate": 5.156512178674358e-06,
      "loss": 0.2863,
      "step": 2964
    },
    {
      "epoch": 4.266187050359712,
      "grad_norm": 0.047380819754018616,
      "learning_rate": 5.136792953807242e-06,
      "loss": 0.2786,
      "step": 2965
    },
    {
      "epoch": 4.267625899280576,
      "grad_norm": 0.05463844211593878,
      "learning_rate": 5.117108918293095e-06,
      "loss": 0.282,
      "step": 2966
    },
    {
      "epoch": 4.269064748201439,
      "grad_norm": 0.047928910779058316,
      "learning_rate": 5.097460092000095e-06,
      "loss": 0.2849,
      "step": 2967
    },
    {
      "epoch": 4.270503597122302,
      "grad_norm": 0.047804611394234274,
      "learning_rate": 5.07784649476089e-06,
      "loss": 0.2804,
      "step": 2968
    },
    {
      "epoch": 4.271942446043165,
      "grad_norm": 0.0529496731240823,
      "learning_rate": 5.058268146372562e-06,
      "loss": 0.2933,
      "step": 2969
    },
    {
      "epoch": 4.273381294964029,
      "grad_norm": 0.048064374075540754,
      "learning_rate": 5.038725066596595e-06,
      "loss": 0.2856,
      "step": 2970
    },
    {
      "epoch": 4.274820143884892,
      "grad_norm": 0.054688345166117924,
      "learning_rate": 5.019217275158923e-06,
      "loss": 0.2903,
      "step": 2971
    },
    {
      "epoch": 4.276258992805755,
      "grad_norm": 0.05497703483626272,
      "learning_rate": 4.9997447917498276e-06,
      "loss": 0.2852,
      "step": 2972
    },
    {
      "epoch": 4.277697841726619,
      "grad_norm": 0.050839990022528606,
      "learning_rate": 4.9803076360239335e-06,
      "loss": 0.2816,
      "step": 2973
    },
    {
      "epoch": 4.279136690647482,
      "grad_norm": 0.048344516305128385,
      "learning_rate": 4.960905827600266e-06,
      "loss": 0.2862,
      "step": 2974
    },
    {
      "epoch": 4.280575539568345,
      "grad_norm": 0.05405339556587912,
      "learning_rate": 4.941539386062113e-06,
      "loss": 0.2904,
      "step": 2975
    },
    {
      "epoch": 4.282014388489209,
      "grad_norm": 0.051578164450635594,
      "learning_rate": 4.922208330957094e-06,
      "loss": 0.2865,
      "step": 2976
    },
    {
      "epoch": 4.283453237410072,
      "grad_norm": 0.04809405998317956,
      "learning_rate": 4.902912681797114e-06,
      "loss": 0.2876,
      "step": 2977
    },
    {
      "epoch": 4.284892086330935,
      "grad_norm": 0.04874372234845439,
      "learning_rate": 4.88365245805833e-06,
      "loss": 0.2881,
      "step": 2978
    },
    {
      "epoch": 4.286330935251798,
      "grad_norm": 0.04939751644300752,
      "learning_rate": 4.864427679181143e-06,
      "loss": 0.2755,
      "step": 2979
    },
    {
      "epoch": 4.287769784172662,
      "grad_norm": 0.05277997570760775,
      "learning_rate": 4.8452383645701815e-06,
      "loss": 0.2828,
      "step": 2980
    },
    {
      "epoch": 4.2892086330935255,
      "grad_norm": 0.04706576694360547,
      "learning_rate": 4.826084533594277e-06,
      "loss": 0.2818,
      "step": 2981
    },
    {
      "epoch": 4.290647482014388,
      "grad_norm": 0.048706280989824775,
      "learning_rate": 4.806966205586441e-06,
      "loss": 0.2825,
      "step": 2982
    },
    {
      "epoch": 4.292086330935252,
      "grad_norm": 0.04804532513842935,
      "learning_rate": 4.787883399843871e-06,
      "loss": 0.291,
      "step": 2983
    },
    {
      "epoch": 4.293525179856115,
      "grad_norm": 0.045865371855854924,
      "learning_rate": 4.768836135627859e-06,
      "loss": 0.2872,
      "step": 2984
    },
    {
      "epoch": 4.294964028776978,
      "grad_norm": 0.047470561617020374,
      "learning_rate": 4.749824432163888e-06,
      "loss": 0.2849,
      "step": 2985
    },
    {
      "epoch": 4.296402877697842,
      "grad_norm": 0.04793577633768564,
      "learning_rate": 4.730848308641509e-06,
      "loss": 0.2793,
      "step": 2986
    },
    {
      "epoch": 4.297841726618705,
      "grad_norm": 0.045949851344600036,
      "learning_rate": 4.711907784214358e-06,
      "loss": 0.2862,
      "step": 2987
    },
    {
      "epoch": 4.299280575539568,
      "grad_norm": 0.046652218656282315,
      "learning_rate": 4.693002878000146e-06,
      "loss": 0.2819,
      "step": 2988
    },
    {
      "epoch": 4.300719424460432,
      "grad_norm": 0.04750890426157474,
      "learning_rate": 4.674133609080658e-06,
      "loss": 0.2912,
      "step": 2989
    },
    {
      "epoch": 4.302158273381295,
      "grad_norm": 0.045998792865143925,
      "learning_rate": 4.6552999965016634e-06,
      "loss": 0.2843,
      "step": 2990
    },
    {
      "epoch": 4.3035971223021585,
      "grad_norm": 0.045687884736637555,
      "learning_rate": 4.6365020592729694e-06,
      "loss": 0.2889,
      "step": 2991
    },
    {
      "epoch": 4.305035971223021,
      "grad_norm": 0.05166995591409614,
      "learning_rate": 4.617739816368367e-06,
      "loss": 0.2857,
      "step": 2992
    },
    {
      "epoch": 4.306474820143885,
      "grad_norm": 0.048667639183769144,
      "learning_rate": 4.599013286725624e-06,
      "loss": 0.2878,
      "step": 2993
    },
    {
      "epoch": 4.307913669064749,
      "grad_norm": 0.04707175291891023,
      "learning_rate": 4.580322489246456e-06,
      "loss": 0.2795,
      "step": 2994
    },
    {
      "epoch": 4.309352517985611,
      "grad_norm": 0.04526119084528856,
      "learning_rate": 4.5616674427965135e-06,
      "loss": 0.286,
      "step": 2995
    },
    {
      "epoch": 4.310791366906475,
      "grad_norm": 0.047858303369376694,
      "learning_rate": 4.543048166205357e-06,
      "loss": 0.289,
      "step": 2996
    },
    {
      "epoch": 4.312230215827338,
      "grad_norm": 0.04558739027643893,
      "learning_rate": 4.524464678266452e-06,
      "loss": 0.2821,
      "step": 2997
    },
    {
      "epoch": 4.313669064748201,
      "grad_norm": 0.0481016114455754,
      "learning_rate": 4.505916997737143e-06,
      "loss": 0.2831,
      "step": 2998
    },
    {
      "epoch": 4.315107913669065,
      "grad_norm": 0.045722359491594936,
      "learning_rate": 4.487405143338599e-06,
      "loss": 0.2859,
      "step": 2999
    },
    {
      "epoch": 4.316546762589928,
      "grad_norm": 0.0470161732124256,
      "learning_rate": 4.468929133755881e-06,
      "loss": 0.2812,
      "step": 3000
    },
    {
      "epoch": 4.3179856115107915,
      "grad_norm": 0.04591954625816037,
      "learning_rate": 4.450488987637824e-06,
      "loss": 0.2886,
      "step": 3001
    },
    {
      "epoch": 4.319424460431654,
      "grad_norm": 0.04657325914607652,
      "learning_rate": 4.43208472359709e-06,
      "loss": 0.2914,
      "step": 3002
    },
    {
      "epoch": 4.320863309352518,
      "grad_norm": 0.04707613298814469,
      "learning_rate": 4.4137163602101114e-06,
      "loss": 0.293,
      "step": 3003
    },
    {
      "epoch": 4.322302158273382,
      "grad_norm": 0.04785380336076747,
      "learning_rate": 4.3953839160170906e-06,
      "loss": 0.2857,
      "step": 3004
    },
    {
      "epoch": 4.323741007194244,
      "grad_norm": 0.04516417550977605,
      "learning_rate": 4.377087409521972e-06,
      "loss": 0.2886,
      "step": 3005
    },
    {
      "epoch": 4.325179856115108,
      "grad_norm": 0.05188756071690509,
      "learning_rate": 4.358826859192422e-06,
      "loss": 0.2795,
      "step": 3006
    },
    {
      "epoch": 4.326618705035971,
      "grad_norm": 0.05010143541911294,
      "learning_rate": 4.340602283459827e-06,
      "loss": 0.2902,
      "step": 3007
    },
    {
      "epoch": 4.3280575539568344,
      "grad_norm": 0.045632732949339,
      "learning_rate": 4.322413700719246e-06,
      "loss": 0.2881,
      "step": 3008
    },
    {
      "epoch": 4.329496402877698,
      "grad_norm": 0.047000573332118295,
      "learning_rate": 4.3042611293294276e-06,
      "loss": 0.2845,
      "step": 3009
    },
    {
      "epoch": 4.330935251798561,
      "grad_norm": 0.04966870119308216,
      "learning_rate": 4.28614458761274e-06,
      "loss": 0.2879,
      "step": 3010
    },
    {
      "epoch": 4.3323741007194245,
      "grad_norm": 0.047705707552179094,
      "learning_rate": 4.2680640938552245e-06,
      "loss": 0.2788,
      "step": 3011
    },
    {
      "epoch": 4.333812949640288,
      "grad_norm": 0.045191298550149736,
      "learning_rate": 4.250019666306515e-06,
      "loss": 0.2784,
      "step": 3012
    },
    {
      "epoch": 4.335251798561151,
      "grad_norm": 0.049970789129198244,
      "learning_rate": 4.232011323179839e-06,
      "loss": 0.2856,
      "step": 3013
    },
    {
      "epoch": 4.336690647482015,
      "grad_norm": 0.04438060522753159,
      "learning_rate": 4.214039082652002e-06,
      "loss": 0.29,
      "step": 3014
    },
    {
      "epoch": 4.338129496402877,
      "grad_norm": 0.044969327942639095,
      "learning_rate": 4.1961029628634e-06,
      "loss": 0.2829,
      "step": 3015
    },
    {
      "epoch": 4.339568345323741,
      "grad_norm": 0.04651094800999467,
      "learning_rate": 4.17820298191792e-06,
      "loss": 0.2822,
      "step": 3016
    },
    {
      "epoch": 4.341007194244605,
      "grad_norm": 0.0448637402914183,
      "learning_rate": 4.160339157883e-06,
      "loss": 0.2852,
      "step": 3017
    },
    {
      "epoch": 4.3424460431654675,
      "grad_norm": 0.045239343753276685,
      "learning_rate": 4.142511508789606e-06,
      "loss": 0.2851,
      "step": 3018
    },
    {
      "epoch": 4.343884892086331,
      "grad_norm": 0.04649656395277292,
      "learning_rate": 4.1247200526321364e-06,
      "loss": 0.2913,
      "step": 3019
    },
    {
      "epoch": 4.345323741007194,
      "grad_norm": 0.04437618628613109,
      "learning_rate": 4.106964807368496e-06,
      "loss": 0.2888,
      "step": 3020
    },
    {
      "epoch": 4.3467625899280575,
      "grad_norm": 0.04763457548306963,
      "learning_rate": 4.089245790920031e-06,
      "loss": 0.2854,
      "step": 3021
    },
    {
      "epoch": 4.348201438848921,
      "grad_norm": 0.047633259428891156,
      "learning_rate": 4.071563021171523e-06,
      "loss": 0.284,
      "step": 3022
    },
    {
      "epoch": 4.349640287769784,
      "grad_norm": 0.04789058302376588,
      "learning_rate": 4.0539165159711615e-06,
      "loss": 0.2882,
      "step": 3023
    },
    {
      "epoch": 4.351079136690648,
      "grad_norm": 0.04800378524550687,
      "learning_rate": 4.036306293130543e-06,
      "loss": 0.2865,
      "step": 3024
    },
    {
      "epoch": 4.35251798561151,
      "grad_norm": 0.04766128596285838,
      "learning_rate": 4.01873237042461e-06,
      "loss": 0.2825,
      "step": 3025
    },
    {
      "epoch": 4.353956834532374,
      "grad_norm": 0.04679739443368158,
      "learning_rate": 4.001194765591723e-06,
      "loss": 0.2801,
      "step": 3026
    },
    {
      "epoch": 4.355395683453238,
      "grad_norm": 0.04681997545874503,
      "learning_rate": 3.983693496333522e-06,
      "loss": 0.2831,
      "step": 3027
    },
    {
      "epoch": 4.3568345323741005,
      "grad_norm": 0.04428659452277041,
      "learning_rate": 3.966228580315017e-06,
      "loss": 0.2835,
      "step": 3028
    },
    {
      "epoch": 4.358273381294964,
      "grad_norm": 0.046913444364728246,
      "learning_rate": 3.9488000351645036e-06,
      "loss": 0.2817,
      "step": 3029
    },
    {
      "epoch": 4.359712230215827,
      "grad_norm": 0.04959811180884263,
      "learning_rate": 3.931407878473575e-06,
      "loss": 0.2864,
      "step": 3030
    },
    {
      "epoch": 4.3611510791366905,
      "grad_norm": 0.04734009750172826,
      "learning_rate": 3.914052127797088e-06,
      "loss": 0.2829,
      "step": 3031
    },
    {
      "epoch": 4.362589928057554,
      "grad_norm": 0.045639018353351817,
      "learning_rate": 3.8967328006531605e-06,
      "loss": 0.2849,
      "step": 3032
    },
    {
      "epoch": 4.364028776978417,
      "grad_norm": 0.04457688635649878,
      "learning_rate": 3.879449914523137e-06,
      "loss": 0.284,
      "step": 3033
    },
    {
      "epoch": 4.365467625899281,
      "grad_norm": 0.04532307133138978,
      "learning_rate": 3.862203486851588e-06,
      "loss": 0.2829,
      "step": 3034
    },
    {
      "epoch": 4.366906474820144,
      "grad_norm": 0.04462611668844534,
      "learning_rate": 3.844993535046291e-06,
      "loss": 0.2845,
      "step": 3035
    },
    {
      "epoch": 4.368345323741007,
      "grad_norm": 0.04572426626114917,
      "learning_rate": 3.8278200764781725e-06,
      "loss": 0.2863,
      "step": 3036
    },
    {
      "epoch": 4.369784172661871,
      "grad_norm": 0.04710485572015182,
      "learning_rate": 3.8106831284813718e-06,
      "loss": 0.2775,
      "step": 3037
    },
    {
      "epoch": 4.3712230215827335,
      "grad_norm": 0.04511969348995074,
      "learning_rate": 3.7935827083531585e-06,
      "loss": 0.2946,
      "step": 3038
    },
    {
      "epoch": 4.372661870503597,
      "grad_norm": 0.046330465569408265,
      "learning_rate": 3.7765188333539037e-06,
      "loss": 0.2906,
      "step": 3039
    },
    {
      "epoch": 4.374100719424461,
      "grad_norm": 0.045991357799606274,
      "learning_rate": 3.759491520707119e-06,
      "loss": 0.2858,
      "step": 3040
    },
    {
      "epoch": 4.3755395683453235,
      "grad_norm": 0.04775861216740306,
      "learning_rate": 3.74250078759943e-06,
      "loss": 0.2827,
      "step": 3041
    },
    {
      "epoch": 4.376978417266187,
      "grad_norm": 0.047533453521402605,
      "learning_rate": 3.7255466511805007e-06,
      "loss": 0.2917,
      "step": 3042
    },
    {
      "epoch": 4.37841726618705,
      "grad_norm": 0.045122859269612174,
      "learning_rate": 3.7086291285630683e-06,
      "loss": 0.2796,
      "step": 3043
    },
    {
      "epoch": 4.379856115107914,
      "grad_norm": 0.047364012253357936,
      "learning_rate": 3.6917482368229406e-06,
      "loss": 0.2852,
      "step": 3044
    },
    {
      "epoch": 4.381294964028777,
      "grad_norm": 0.04676052940699597,
      "learning_rate": 3.674903992998915e-06,
      "loss": 0.275,
      "step": 3045
    },
    {
      "epoch": 4.38273381294964,
      "grad_norm": 0.046548802436815834,
      "learning_rate": 3.6580964140928133e-06,
      "loss": 0.2871,
      "step": 3046
    },
    {
      "epoch": 4.384172661870504,
      "grad_norm": 0.04508602407063141,
      "learning_rate": 3.6413255170694515e-06,
      "loss": 0.2895,
      "step": 3047
    },
    {
      "epoch": 4.385611510791367,
      "grad_norm": 0.0488251074647898,
      "learning_rate": 3.6245913188566227e-06,
      "loss": 0.2774,
      "step": 3048
    },
    {
      "epoch": 4.38705035971223,
      "grad_norm": 0.046903569771117315,
      "learning_rate": 3.607893836345069e-06,
      "loss": 0.2806,
      "step": 3049
    },
    {
      "epoch": 4.388489208633094,
      "grad_norm": 0.04365027073587901,
      "learning_rate": 3.5912330863884904e-06,
      "loss": 0.2892,
      "step": 3050
    },
    {
      "epoch": 4.3899280575539565,
      "grad_norm": 0.045772150095634324,
      "learning_rate": 3.574609085803471e-06,
      "loss": 0.282,
      "step": 3051
    },
    {
      "epoch": 4.39136690647482,
      "grad_norm": 0.0457368953801093,
      "learning_rate": 3.5580218513695573e-06,
      "loss": 0.2892,
      "step": 3052
    },
    {
      "epoch": 4.392805755395684,
      "grad_norm": 0.04792704812052965,
      "learning_rate": 3.5414713998291483e-06,
      "loss": 0.2943,
      "step": 3053
    },
    {
      "epoch": 4.394244604316547,
      "grad_norm": 0.04752511450125093,
      "learning_rate": 3.524957747887512e-06,
      "loss": 0.2805,
      "step": 3054
    },
    {
      "epoch": 4.39568345323741,
      "grad_norm": 0.04881368473854953,
      "learning_rate": 3.5084809122128125e-06,
      "loss": 0.2807,
      "step": 3055
    },
    {
      "epoch": 4.397122302158273,
      "grad_norm": 0.0470367816327077,
      "learning_rate": 3.4920409094360054e-06,
      "loss": 0.2783,
      "step": 3056
    },
    {
      "epoch": 4.398561151079137,
      "grad_norm": 0.0465763916174093,
      "learning_rate": 3.475637756150896e-06,
      "loss": 0.2917,
      "step": 3057
    },
    {
      "epoch": 4.4,
      "grad_norm": 0.0451566593898251,
      "learning_rate": 3.4592714689140895e-06,
      "loss": 0.2828,
      "step": 3058
    },
    {
      "epoch": 4.401438848920863,
      "grad_norm": 0.04725631766489214,
      "learning_rate": 3.442942064244981e-06,
      "loss": 0.2901,
      "step": 3059
    },
    {
      "epoch": 4.402877697841727,
      "grad_norm": 0.045376106227395,
      "learning_rate": 3.426649558625732e-06,
      "loss": 0.2899,
      "step": 3060
    },
    {
      "epoch": 4.4043165467625895,
      "grad_norm": 0.04467265942320661,
      "learning_rate": 3.4103939685012823e-06,
      "loss": 0.2853,
      "step": 3061
    },
    {
      "epoch": 4.405755395683453,
      "grad_norm": 0.047030313069995736,
      "learning_rate": 3.3941753102792617e-06,
      "loss": 0.284,
      "step": 3062
    },
    {
      "epoch": 4.407194244604317,
      "grad_norm": 0.04592765541447238,
      "learning_rate": 3.377993600330083e-06,
      "loss": 0.2819,
      "step": 3063
    },
    {
      "epoch": 4.40863309352518,
      "grad_norm": 0.044707990629991354,
      "learning_rate": 3.361848854986831e-06,
      "loss": 0.2891,
      "step": 3064
    },
    {
      "epoch": 4.410071942446043,
      "grad_norm": 0.0461525481123277,
      "learning_rate": 3.3457410905452624e-06,
      "loss": 0.2793,
      "step": 3065
    },
    {
      "epoch": 4.411510791366906,
      "grad_norm": 0.04469960961460827,
      "learning_rate": 3.3296703232638606e-06,
      "loss": 0.2867,
      "step": 3066
    },
    {
      "epoch": 4.41294964028777,
      "grad_norm": 0.045028596250381685,
      "learning_rate": 3.3136365693637294e-06,
      "loss": 0.2847,
      "step": 3067
    },
    {
      "epoch": 4.414388489208633,
      "grad_norm": 0.0451846137511454,
      "learning_rate": 3.297639845028604e-06,
      "loss": 0.2938,
      "step": 3068
    },
    {
      "epoch": 4.415827338129496,
      "grad_norm": 0.04576617100329387,
      "learning_rate": 3.281680166404857e-06,
      "loss": 0.2866,
      "step": 3069
    },
    {
      "epoch": 4.41726618705036,
      "grad_norm": 0.048420259568492184,
      "learning_rate": 3.265757549601496e-06,
      "loss": 0.2819,
      "step": 3070
    },
    {
      "epoch": 4.418705035971223,
      "grad_norm": 0.047462371508189154,
      "learning_rate": 3.249872010690074e-06,
      "loss": 0.2861,
      "step": 3071
    },
    {
      "epoch": 4.420143884892086,
      "grad_norm": 0.04572972380824012,
      "learning_rate": 3.234023565704738e-06,
      "loss": 0.2836,
      "step": 3072
    },
    {
      "epoch": 4.42158273381295,
      "grad_norm": 0.04665768998623574,
      "learning_rate": 3.2182122306422035e-06,
      "loss": 0.2818,
      "step": 3073
    },
    {
      "epoch": 4.423021582733813,
      "grad_norm": 0.046315803895643375,
      "learning_rate": 3.2024380214617136e-06,
      "loss": 0.2803,
      "step": 3074
    },
    {
      "epoch": 4.424460431654676,
      "grad_norm": 0.04494539815332072,
      "learning_rate": 3.186700954085056e-06,
      "loss": 0.2934,
      "step": 3075
    },
    {
      "epoch": 4.42589928057554,
      "grad_norm": 0.04586469090336672,
      "learning_rate": 3.1710010443965065e-06,
      "loss": 0.2816,
      "step": 3076
    },
    {
      "epoch": 4.427338129496403,
      "grad_norm": 0.044106793033919324,
      "learning_rate": 3.1553383082428568e-06,
      "loss": 0.2807,
      "step": 3077
    },
    {
      "epoch": 4.428776978417266,
      "grad_norm": 0.046162670508771264,
      "learning_rate": 3.139712761433367e-06,
      "loss": 0.2807,
      "step": 3078
    },
    {
      "epoch": 4.430215827338129,
      "grad_norm": 0.04590486676601341,
      "learning_rate": 3.1241244197397626e-06,
      "loss": 0.2835,
      "step": 3079
    },
    {
      "epoch": 4.431654676258993,
      "grad_norm": 0.04625623474456893,
      "learning_rate": 3.1085732988962003e-06,
      "loss": 0.2827,
      "step": 3080
    },
    {
      "epoch": 4.433093525179856,
      "grad_norm": 0.04544268712327201,
      "learning_rate": 3.0930594145993063e-06,
      "loss": 0.2845,
      "step": 3081
    },
    {
      "epoch": 4.434532374100719,
      "grad_norm": 0.04629058843461137,
      "learning_rate": 3.077582782508075e-06,
      "loss": 0.2857,
      "step": 3082
    },
    {
      "epoch": 4.435971223021583,
      "grad_norm": 0.05036946924362949,
      "learning_rate": 3.0621434182439345e-06,
      "loss": 0.2908,
      "step": 3083
    },
    {
      "epoch": 4.437410071942446,
      "grad_norm": 0.04580338282889522,
      "learning_rate": 3.0467413373906773e-06,
      "loss": 0.2932,
      "step": 3084
    },
    {
      "epoch": 4.438848920863309,
      "grad_norm": 0.04563365127109482,
      "learning_rate": 3.0313765554944806e-06,
      "loss": 0.2924,
      "step": 3085
    },
    {
      "epoch": 4.440287769784173,
      "grad_norm": 0.04691341222883385,
      "learning_rate": 3.0160490880638593e-06,
      "loss": 0.2903,
      "step": 3086
    },
    {
      "epoch": 4.441726618705036,
      "grad_norm": 0.04352245865742581,
      "learning_rate": 3.0007589505696645e-06,
      "loss": 0.2887,
      "step": 3087
    },
    {
      "epoch": 4.443165467625899,
      "grad_norm": 0.04569527256131873,
      "learning_rate": 2.9855061584450795e-06,
      "loss": 0.2877,
      "step": 3088
    },
    {
      "epoch": 4.444604316546762,
      "grad_norm": 0.04539684907823774,
      "learning_rate": 2.97029072708559e-06,
      "loss": 0.2878,
      "step": 3089
    },
    {
      "epoch": 4.446043165467626,
      "grad_norm": 0.04655124115211836,
      "learning_rate": 2.955112671848963e-06,
      "loss": 0.277,
      "step": 3090
    },
    {
      "epoch": 4.4474820143884894,
      "grad_norm": 0.04452158634531101,
      "learning_rate": 2.9399720080552383e-06,
      "loss": 0.2839,
      "step": 3091
    },
    {
      "epoch": 4.448920863309352,
      "grad_norm": 0.04487552335110992,
      "learning_rate": 2.924868750986729e-06,
      "loss": 0.2836,
      "step": 3092
    },
    {
      "epoch": 4.450359712230216,
      "grad_norm": 0.045344931430501334,
      "learning_rate": 2.9098029158879914e-06,
      "loss": 0.288,
      "step": 3093
    },
    {
      "epoch": 4.4517985611510795,
      "grad_norm": 0.04471278511391404,
      "learning_rate": 2.8947745179657815e-06,
      "loss": 0.2819,
      "step": 3094
    },
    {
      "epoch": 4.453237410071942,
      "grad_norm": 0.047830060036277836,
      "learning_rate": 2.8797835723890944e-06,
      "loss": 0.2802,
      "step": 3095
    },
    {
      "epoch": 4.454676258992806,
      "grad_norm": 0.04946417523984009,
      "learning_rate": 2.864830094289137e-06,
      "loss": 0.296,
      "step": 3096
    },
    {
      "epoch": 4.456115107913669,
      "grad_norm": 0.04623580196593618,
      "learning_rate": 2.84991409875925e-06,
      "loss": 0.2866,
      "step": 3097
    },
    {
      "epoch": 4.457553956834532,
      "grad_norm": 0.04510489404083211,
      "learning_rate": 2.8350356008549806e-06,
      "loss": 0.2859,
      "step": 3098
    },
    {
      "epoch": 4.458992805755396,
      "grad_norm": 0.045666606052583214,
      "learning_rate": 2.8201946155940142e-06,
      "loss": 0.2808,
      "step": 3099
    },
    {
      "epoch": 4.460431654676259,
      "grad_norm": 0.04983704496186958,
      "learning_rate": 2.8053911579561764e-06,
      "loss": 0.2934,
      "step": 3100
    },
    {
      "epoch": 4.4618705035971225,
      "grad_norm": 0.048865598009512075,
      "learning_rate": 2.7906252428834044e-06,
      "loss": 0.2844,
      "step": 3101
    },
    {
      "epoch": 4.463309352517985,
      "grad_norm": 0.045393191516494076,
      "learning_rate": 2.7758968852797542e-06,
      "loss": 0.2822,
      "step": 3102
    },
    {
      "epoch": 4.464748201438849,
      "grad_norm": 0.044241543153539384,
      "learning_rate": 2.761206100011369e-06,
      "loss": 0.2838,
      "step": 3103
    },
    {
      "epoch": 4.4661870503597125,
      "grad_norm": 0.048380967966845406,
      "learning_rate": 2.746552901906463e-06,
      "loss": 0.2856,
      "step": 3104
    },
    {
      "epoch": 4.467625899280575,
      "grad_norm": 0.04589060710974179,
      "learning_rate": 2.731937305755321e-06,
      "loss": 0.2898,
      "step": 3105
    },
    {
      "epoch": 4.469064748201439,
      "grad_norm": 0.04799604377711833,
      "learning_rate": 2.717359326310249e-06,
      "loss": 0.2804,
      "step": 3106
    },
    {
      "epoch": 4.470503597122303,
      "grad_norm": 0.04399166756583154,
      "learning_rate": 2.702818978285633e-06,
      "loss": 0.2802,
      "step": 3107
    },
    {
      "epoch": 4.471942446043165,
      "grad_norm": 0.045173616948004265,
      "learning_rate": 2.688316276357825e-06,
      "loss": 0.2852,
      "step": 3108
    },
    {
      "epoch": 4.473381294964029,
      "grad_norm": 0.04880876466760763,
      "learning_rate": 2.6738512351652012e-06,
      "loss": 0.2824,
      "step": 3109
    },
    {
      "epoch": 4.474820143884892,
      "grad_norm": 0.04937337994587513,
      "learning_rate": 2.65942386930814e-06,
      "loss": 0.2874,
      "step": 3110
    },
    {
      "epoch": 4.4762589928057555,
      "grad_norm": 0.04798329139322227,
      "learning_rate": 2.645034193348961e-06,
      "loss": 0.2918,
      "step": 3111
    },
    {
      "epoch": 4.477697841726619,
      "grad_norm": 0.043177094305288106,
      "learning_rate": 2.6306822218119533e-06,
      "loss": 0.2825,
      "step": 3112
    },
    {
      "epoch": 4.479136690647482,
      "grad_norm": 0.04726294683534518,
      "learning_rate": 2.61636796918336e-06,
      "loss": 0.2918,
      "step": 3113
    },
    {
      "epoch": 4.4805755395683455,
      "grad_norm": 0.0506165230183232,
      "learning_rate": 2.6020914499113438e-06,
      "loss": 0.2893,
      "step": 3114
    },
    {
      "epoch": 4.482014388489208,
      "grad_norm": 0.04698891076583604,
      "learning_rate": 2.587852678405973e-06,
      "loss": 0.2871,
      "step": 3115
    },
    {
      "epoch": 4.483453237410072,
      "grad_norm": 0.045791826692126864,
      "learning_rate": 2.5736516690392366e-06,
      "loss": 0.2823,
      "step": 3116
    },
    {
      "epoch": 4.484892086330936,
      "grad_norm": 0.045936359862766014,
      "learning_rate": 2.5594884361449746e-06,
      "loss": 0.2845,
      "step": 3117
    },
    {
      "epoch": 4.486330935251798,
      "grad_norm": 0.049870589317582585,
      "learning_rate": 2.5453629940189338e-06,
      "loss": 0.29,
      "step": 3118
    },
    {
      "epoch": 4.487769784172662,
      "grad_norm": 0.044952325310770015,
      "learning_rate": 2.531275356918701e-06,
      "loss": 0.2792,
      "step": 3119
    },
    {
      "epoch": 4.489208633093525,
      "grad_norm": 0.04474980272543277,
      "learning_rate": 2.5172255390636878e-06,
      "loss": 0.282,
      "step": 3120
    },
    {
      "epoch": 4.4906474820143885,
      "grad_norm": 0.04619342703112689,
      "learning_rate": 2.5032135546351644e-06,
      "loss": 0.2855,
      "step": 3121
    },
    {
      "epoch": 4.492086330935252,
      "grad_norm": 0.04904437438371335,
      "learning_rate": 2.4892394177761947e-06,
      "loss": 0.2975,
      "step": 3122
    },
    {
      "epoch": 4.493525179856115,
      "grad_norm": 0.04481478449252868,
      "learning_rate": 2.475303142591634e-06,
      "loss": 0.287,
      "step": 3123
    },
    {
      "epoch": 4.4949640287769785,
      "grad_norm": 0.04430038256500023,
      "learning_rate": 2.461404743148141e-06,
      "loss": 0.2838,
      "step": 3124
    },
    {
      "epoch": 4.496402877697841,
      "grad_norm": 0.04863974155020966,
      "learning_rate": 2.4475442334741306e-06,
      "loss": 0.289,
      "step": 3125
    },
    {
      "epoch": 4.497841726618705,
      "grad_norm": 0.047389061537361936,
      "learning_rate": 2.43372162755978e-06,
      "loss": 0.2872,
      "step": 3126
    },
    {
      "epoch": 4.499280575539569,
      "grad_norm": 0.04717389258461824,
      "learning_rate": 2.419936939357004e-06,
      "loss": 0.2805,
      "step": 3127
    },
    {
      "epoch": 4.500719424460431,
      "grad_norm": 0.04648293926121682,
      "learning_rate": 2.4061901827794466e-06,
      "loss": 0.2762,
      "step": 3128
    },
    {
      "epoch": 4.502158273381295,
      "grad_norm": 0.04495179640002291,
      "learning_rate": 2.3924813717024663e-06,
      "loss": 0.2798,
      "step": 3129
    },
    {
      "epoch": 4.503597122302159,
      "grad_norm": 0.04786332178584631,
      "learning_rate": 2.378810519963124e-06,
      "loss": 0.2884,
      "step": 3130
    },
    {
      "epoch": 4.5050359712230215,
      "grad_norm": 0.046038653109148474,
      "learning_rate": 2.3651776413601634e-06,
      "loss": 0.2879,
      "step": 3131
    },
    {
      "epoch": 4.506474820143885,
      "grad_norm": 0.04552477765442219,
      "learning_rate": 2.3515827496539823e-06,
      "loss": 0.2891,
      "step": 3132
    },
    {
      "epoch": 4.507913669064748,
      "grad_norm": 0.04590163299711689,
      "learning_rate": 2.3380258585666793e-06,
      "loss": 0.2835,
      "step": 3133
    },
    {
      "epoch": 4.5093525179856115,
      "grad_norm": 0.04224172604796208,
      "learning_rate": 2.324506981781949e-06,
      "loss": 0.2789,
      "step": 3134
    },
    {
      "epoch": 4.510791366906475,
      "grad_norm": 0.04482762947979673,
      "learning_rate": 2.311026132945138e-06,
      "loss": 0.2871,
      "step": 3135
    },
    {
      "epoch": 4.512230215827338,
      "grad_norm": 0.04499861476986286,
      "learning_rate": 2.297583325663233e-06,
      "loss": 0.282,
      "step": 3136
    },
    {
      "epoch": 4.513669064748202,
      "grad_norm": 0.042959657151869356,
      "learning_rate": 2.2841785735047717e-06,
      "loss": 0.2818,
      "step": 3137
    },
    {
      "epoch": 4.515107913669064,
      "grad_norm": 0.04649272008206621,
      "learning_rate": 2.2708118899999175e-06,
      "loss": 0.2856,
      "step": 3138
    },
    {
      "epoch": 4.516546762589928,
      "grad_norm": 0.0460211142434342,
      "learning_rate": 2.2574832886403988e-06,
      "loss": 0.2901,
      "step": 3139
    },
    {
      "epoch": 4.517985611510792,
      "grad_norm": 0.0451331665609917,
      "learning_rate": 2.2441927828795106e-06,
      "loss": 0.2844,
      "step": 3140
    },
    {
      "epoch": 4.5194244604316545,
      "grad_norm": 0.044479318953797725,
      "learning_rate": 2.230940386132088e-06,
      "loss": 0.2845,
      "step": 3141
    },
    {
      "epoch": 4.520863309352518,
      "grad_norm": 0.044434221175025886,
      "learning_rate": 2.21772611177451e-06,
      "loss": 0.2871,
      "step": 3142
    },
    {
      "epoch": 4.522302158273382,
      "grad_norm": 0.04497800404610502,
      "learning_rate": 2.204549973144654e-06,
      "loss": 0.2907,
      "step": 3143
    },
    {
      "epoch": 4.5237410071942445,
      "grad_norm": 0.04434053092042553,
      "learning_rate": 2.1914119835419358e-06,
      "loss": 0.289,
      "step": 3144
    },
    {
      "epoch": 4.525179856115108,
      "grad_norm": 0.044472727823722906,
      "learning_rate": 2.178312156227258e-06,
      "loss": 0.2903,
      "step": 3145
    },
    {
      "epoch": 4.526618705035971,
      "grad_norm": 0.04352294261035754,
      "learning_rate": 2.1652505044229734e-06,
      "loss": 0.287,
      "step": 3146
    },
    {
      "epoch": 4.528057553956835,
      "grad_norm": 0.047867061889741554,
      "learning_rate": 2.1522270413129444e-06,
      "loss": 0.2931,
      "step": 3147
    },
    {
      "epoch": 4.529496402877697,
      "grad_norm": 0.04381400605651353,
      "learning_rate": 2.1392417800424738e-06,
      "loss": 0.2869,
      "step": 3148
    },
    {
      "epoch": 4.530935251798561,
      "grad_norm": 0.04521821920349319,
      "learning_rate": 2.1262947337182815e-06,
      "loss": 0.2893,
      "step": 3149
    },
    {
      "epoch": 4.532374100719425,
      "grad_norm": 0.04553689821434878,
      "learning_rate": 2.113385915408546e-06,
      "loss": 0.2836,
      "step": 3150
    },
    {
      "epoch": 4.5338129496402875,
      "grad_norm": 0.04379746931920125,
      "learning_rate": 2.100515338142839e-06,
      "loss": 0.2811,
      "step": 3151
    },
    {
      "epoch": 4.535251798561151,
      "grad_norm": 0.04413680616493023,
      "learning_rate": 2.087683014912152e-06,
      "loss": 0.279,
      "step": 3152
    },
    {
      "epoch": 4.536690647482015,
      "grad_norm": 0.046387639158312635,
      "learning_rate": 2.0748889586688526e-06,
      "loss": 0.286,
      "step": 3153
    },
    {
      "epoch": 4.5381294964028775,
      "grad_norm": 0.04508938254342468,
      "learning_rate": 2.0621331823266777e-06,
      "loss": 0.2795,
      "step": 3154
    },
    {
      "epoch": 4.539568345323741,
      "grad_norm": 0.0468802883650182,
      "learning_rate": 2.049415698760746e-06,
      "loss": 0.2865,
      "step": 3155
    },
    {
      "epoch": 4.541007194244604,
      "grad_norm": 0.04624928272944986,
      "learning_rate": 2.036736520807505e-06,
      "loss": 0.2889,
      "step": 3156
    },
    {
      "epoch": 4.542446043165468,
      "grad_norm": 0.04474152738166633,
      "learning_rate": 2.0240956612647487e-06,
      "loss": 0.2876,
      "step": 3157
    },
    {
      "epoch": 4.543884892086331,
      "grad_norm": 0.04216196285366899,
      "learning_rate": 2.011493132891591e-06,
      "loss": 0.2785,
      "step": 3158
    },
    {
      "epoch": 4.545323741007194,
      "grad_norm": 0.04497742131776572,
      "learning_rate": 1.998928948408465e-06,
      "loss": 0.2885,
      "step": 3159
    },
    {
      "epoch": 4.546762589928058,
      "grad_norm": 0.04569780342971736,
      "learning_rate": 1.9864031204970847e-06,
      "loss": 0.2855,
      "step": 3160
    },
    {
      "epoch": 4.5482014388489205,
      "grad_norm": 0.04775506469401655,
      "learning_rate": 1.973915661800452e-06,
      "loss": 0.2822,
      "step": 3161
    },
    {
      "epoch": 4.549640287769784,
      "grad_norm": 0.04382785070841345,
      "learning_rate": 1.9614665849228666e-06,
      "loss": 0.2893,
      "step": 3162
    },
    {
      "epoch": 4.551079136690648,
      "grad_norm": 0.04308916245297393,
      "learning_rate": 1.949055902429846e-06,
      "loss": 0.2979,
      "step": 3163
    },
    {
      "epoch": 4.5525179856115106,
      "grad_norm": 0.045927616822700745,
      "learning_rate": 1.936683626848179e-06,
      "loss": 0.2961,
      "step": 3164
    },
    {
      "epoch": 4.553956834532374,
      "grad_norm": 0.04318819506247855,
      "learning_rate": 1.9243497706658944e-06,
      "loss": 0.2817,
      "step": 3165
    },
    {
      "epoch": 4.555395683453238,
      "grad_norm": 0.04395040323865869,
      "learning_rate": 1.9120543463322238e-06,
      "loss": 0.2878,
      "step": 3166
    },
    {
      "epoch": 4.556834532374101,
      "grad_norm": 0.04870584182566934,
      "learning_rate": 1.899797366257614e-06,
      "loss": 0.2904,
      "step": 3167
    },
    {
      "epoch": 4.558273381294964,
      "grad_norm": 0.042490273184344104,
      "learning_rate": 1.887578842813711e-06,
      "loss": 0.2747,
      "step": 3168
    },
    {
      "epoch": 4.559712230215827,
      "grad_norm": 0.044974677146473956,
      "learning_rate": 1.875398788333347e-06,
      "loss": 0.2916,
      "step": 3169
    },
    {
      "epoch": 4.561151079136691,
      "grad_norm": 0.04218317060094236,
      "learning_rate": 1.8632572151105189e-06,
      "loss": 0.2782,
      "step": 3170
    },
    {
      "epoch": 4.5625899280575535,
      "grad_norm": 0.04113782679463585,
      "learning_rate": 1.8511541354003882e-06,
      "loss": 0.2765,
      "step": 3171
    },
    {
      "epoch": 4.564028776978417,
      "grad_norm": 0.04285203911801668,
      "learning_rate": 1.8390895614192405e-06,
      "loss": 0.2899,
      "step": 3172
    },
    {
      "epoch": 4.565467625899281,
      "grad_norm": 0.04210684058326767,
      "learning_rate": 1.8270635053445352e-06,
      "loss": 0.2863,
      "step": 3173
    },
    {
      "epoch": 4.566906474820144,
      "grad_norm": 0.04274374266127899,
      "learning_rate": 1.8150759793148332e-06,
      "loss": 0.289,
      "step": 3174
    },
    {
      "epoch": 4.568345323741007,
      "grad_norm": 0.04433194830899028,
      "learning_rate": 1.803126995429789e-06,
      "loss": 0.2898,
      "step": 3175
    },
    {
      "epoch": 4.569784172661871,
      "grad_norm": 0.04368810837599763,
      "learning_rate": 1.7912165657501779e-06,
      "loss": 0.2907,
      "step": 3176
    },
    {
      "epoch": 4.571223021582734,
      "grad_norm": 0.04298745026737905,
      "learning_rate": 1.779344702297845e-06,
      "loss": 0.2894,
      "step": 3177
    },
    {
      "epoch": 4.572661870503597,
      "grad_norm": 0.04218022078052001,
      "learning_rate": 1.767511417055725e-06,
      "loss": 0.2862,
      "step": 3178
    },
    {
      "epoch": 4.57410071942446,
      "grad_norm": 0.04265609620970078,
      "learning_rate": 1.7557167219678018e-06,
      "loss": 0.277,
      "step": 3179
    },
    {
      "epoch": 4.575539568345324,
      "grad_norm": 0.0439300849590657,
      "learning_rate": 1.7439606289391032e-06,
      "loss": 0.2821,
      "step": 3180
    },
    {
      "epoch": 4.576978417266187,
      "grad_norm": 0.04227905131910428,
      "learning_rate": 1.7322431498357063e-06,
      "loss": 0.2809,
      "step": 3181
    },
    {
      "epoch": 4.57841726618705,
      "grad_norm": 0.04472282355744868,
      "learning_rate": 1.7205642964847103e-06,
      "loss": 0.2885,
      "step": 3182
    },
    {
      "epoch": 4.579856115107914,
      "grad_norm": 0.044805856192444427,
      "learning_rate": 1.7089240806742147e-06,
      "loss": 0.279,
      "step": 3183
    },
    {
      "epoch": 4.581294964028777,
      "grad_norm": 0.046279114883507784,
      "learning_rate": 1.697322514153341e-06,
      "loss": 0.2905,
      "step": 3184
    },
    {
      "epoch": 4.58273381294964,
      "grad_norm": 0.042683798448461145,
      "learning_rate": 1.6857596086321848e-06,
      "loss": 0.2815,
      "step": 3185
    },
    {
      "epoch": 4.584172661870504,
      "grad_norm": 0.04562810926048852,
      "learning_rate": 1.6742353757818187e-06,
      "loss": 0.2852,
      "step": 3186
    },
    {
      "epoch": 4.585611510791367,
      "grad_norm": 0.04768178060373971,
      "learning_rate": 1.6627498272342802e-06,
      "loss": 0.2831,
      "step": 3187
    },
    {
      "epoch": 4.58705035971223,
      "grad_norm": 0.04487900169719795,
      "learning_rate": 1.6513029745825803e-06,
      "loss": 0.286,
      "step": 3188
    },
    {
      "epoch": 4.588489208633094,
      "grad_norm": 0.042795918038530806,
      "learning_rate": 1.6398948293806504e-06,
      "loss": 0.2842,
      "step": 3189
    },
    {
      "epoch": 4.589928057553957,
      "grad_norm": 0.04303209927416349,
      "learning_rate": 1.6285254031433462e-06,
      "loss": 0.2828,
      "step": 3190
    },
    {
      "epoch": 4.59136690647482,
      "grad_norm": 0.042562349055923286,
      "learning_rate": 1.6171947073464834e-06,
      "loss": 0.2758,
      "step": 3191
    },
    {
      "epoch": 4.592805755395683,
      "grad_norm": 0.044191634516535336,
      "learning_rate": 1.6059027534267313e-06,
      "loss": 0.2851,
      "step": 3192
    },
    {
      "epoch": 4.594244604316547,
      "grad_norm": 0.042741946610148566,
      "learning_rate": 1.594649552781693e-06,
      "loss": 0.2836,
      "step": 3193
    },
    {
      "epoch": 4.5956834532374105,
      "grad_norm": 0.04473222357427111,
      "learning_rate": 1.5834351167698336e-06,
      "loss": 0.2888,
      "step": 3194
    },
    {
      "epoch": 4.597122302158273,
      "grad_norm": 0.04678881538800095,
      "learning_rate": 1.572259456710512e-06,
      "loss": 0.2889,
      "step": 3195
    },
    {
      "epoch": 4.598561151079137,
      "grad_norm": 0.0445047082886457,
      "learning_rate": 1.5611225838839272e-06,
      "loss": 0.2866,
      "step": 3196
    },
    {
      "epoch": 4.6,
      "grad_norm": 0.04198922277969195,
      "learning_rate": 1.550024509531145e-06,
      "loss": 0.2832,
      "step": 3197
    },
    {
      "epoch": 4.601438848920863,
      "grad_norm": 0.04780944683412724,
      "learning_rate": 1.5389652448540537e-06,
      "loss": 0.2825,
      "step": 3198
    },
    {
      "epoch": 4.602877697841727,
      "grad_norm": 0.047847664878454765,
      "learning_rate": 1.527944801015382e-06,
      "loss": 0.2883,
      "step": 3199
    },
    {
      "epoch": 4.60431654676259,
      "grad_norm": 0.0458686980032547,
      "learning_rate": 1.5169631891386805e-06,
      "loss": 0.2896,
      "step": 3200
    },
    {
      "epoch": 4.605755395683453,
      "grad_norm": 0.04451970044204277,
      "learning_rate": 1.506020420308274e-06,
      "loss": 0.2809,
      "step": 3201
    },
    {
      "epoch": 4.607194244604317,
      "grad_norm": 0.047936867529953305,
      "learning_rate": 1.495116505569314e-06,
      "loss": 0.2807,
      "step": 3202
    },
    {
      "epoch": 4.60863309352518,
      "grad_norm": 0.04600991740189068,
      "learning_rate": 1.4842514559277254e-06,
      "loss": 0.2827,
      "step": 3203
    },
    {
      "epoch": 4.6100719424460435,
      "grad_norm": 0.042638124793315746,
      "learning_rate": 1.4734252823501894e-06,
      "loss": 0.2845,
      "step": 3204
    },
    {
      "epoch": 4.611510791366906,
      "grad_norm": 0.04414564436388609,
      "learning_rate": 1.4626379957641646e-06,
      "loss": 0.285,
      "step": 3205
    },
    {
      "epoch": 4.61294964028777,
      "grad_norm": 0.0429683620965709,
      "learning_rate": 1.451889607057848e-06,
      "loss": 0.2841,
      "step": 3206
    },
    {
      "epoch": 4.614388489208633,
      "grad_norm": 0.043993994658264805,
      "learning_rate": 1.4411801270801885e-06,
      "loss": 0.2918,
      "step": 3207
    },
    {
      "epoch": 4.615827338129496,
      "grad_norm": 0.044701654054872725,
      "learning_rate": 1.4305095666408453e-06,
      "loss": 0.2775,
      "step": 3208
    },
    {
      "epoch": 4.61726618705036,
      "grad_norm": 0.0485189917496429,
      "learning_rate": 1.4198779365102077e-06,
      "loss": 0.2769,
      "step": 3209
    },
    {
      "epoch": 4.618705035971223,
      "grad_norm": 0.04409380806889366,
      "learning_rate": 1.409285247419363e-06,
      "loss": 0.2815,
      "step": 3210
    },
    {
      "epoch": 4.620143884892086,
      "grad_norm": 0.043161989577127885,
      "learning_rate": 1.3987315100600961e-06,
      "loss": 0.2882,
      "step": 3211
    },
    {
      "epoch": 4.62158273381295,
      "grad_norm": 0.042634187481623204,
      "learning_rate": 1.3882167350848686e-06,
      "loss": 0.2779,
      "step": 3212
    },
    {
      "epoch": 4.623021582733813,
      "grad_norm": 0.04483985224792961,
      "learning_rate": 1.3777409331068258e-06,
      "loss": 0.2838,
      "step": 3213
    },
    {
      "epoch": 4.6244604316546765,
      "grad_norm": 0.04550005592664216,
      "learning_rate": 1.3673041146997768e-06,
      "loss": 0.2885,
      "step": 3214
    },
    {
      "epoch": 4.625899280575539,
      "grad_norm": 0.044129693064556864,
      "learning_rate": 1.35690629039817e-06,
      "loss": 0.2885,
      "step": 3215
    },
    {
      "epoch": 4.627338129496403,
      "grad_norm": 0.044326368086113545,
      "learning_rate": 1.346547470697095e-06,
      "loss": 0.2887,
      "step": 3216
    },
    {
      "epoch": 4.6287769784172665,
      "grad_norm": 0.04446788991957044,
      "learning_rate": 1.3362276660522943e-06,
      "loss": 0.2831,
      "step": 3217
    },
    {
      "epoch": 4.630215827338129,
      "grad_norm": 0.043743410105604154,
      "learning_rate": 1.325946886880103e-06,
      "loss": 0.2779,
      "step": 3218
    },
    {
      "epoch": 4.631654676258993,
      "grad_norm": 0.04217257440269431,
      "learning_rate": 1.315705143557482e-06,
      "loss": 0.2919,
      "step": 3219
    },
    {
      "epoch": 4.633093525179856,
      "grad_norm": 0.04389969404816933,
      "learning_rate": 1.3055024464219846e-06,
      "loss": 0.2857,
      "step": 3220
    },
    {
      "epoch": 4.634532374100719,
      "grad_norm": 0.04109587558951776,
      "learning_rate": 1.295338805771751e-06,
      "loss": 0.2914,
      "step": 3221
    },
    {
      "epoch": 4.635971223021583,
      "grad_norm": 0.04231590117042742,
      "learning_rate": 1.285214231865508e-06,
      "loss": 0.2856,
      "step": 3222
    },
    {
      "epoch": 4.637410071942446,
      "grad_norm": 0.0443269152627286,
      "learning_rate": 1.2751287349225484e-06,
      "loss": 0.2855,
      "step": 3223
    },
    {
      "epoch": 4.6388489208633095,
      "grad_norm": 0.0462980954449299,
      "learning_rate": 1.2650823251227062e-06,
      "loss": 0.2884,
      "step": 3224
    },
    {
      "epoch": 4.640287769784173,
      "grad_norm": 0.040429547658139595,
      "learning_rate": 1.255075012606386e-06,
      "loss": 0.2832,
      "step": 3225
    },
    {
      "epoch": 4.641726618705036,
      "grad_norm": 0.04486184187050472,
      "learning_rate": 1.2451068074745254e-06,
      "loss": 0.2843,
      "step": 3226
    },
    {
      "epoch": 4.6431654676258995,
      "grad_norm": 0.042249295960709955,
      "learning_rate": 1.2351777197885606e-06,
      "loss": 0.2881,
      "step": 3227
    },
    {
      "epoch": 4.644604316546762,
      "grad_norm": 0.04458340484085948,
      "learning_rate": 1.2252877595704838e-06,
      "loss": 0.2837,
      "step": 3228
    },
    {
      "epoch": 4.646043165467626,
      "grad_norm": 0.04208468376135168,
      "learning_rate": 1.2154369368027763e-06,
      "loss": 0.2875,
      "step": 3229
    },
    {
      "epoch": 4.647482014388489,
      "grad_norm": 0.04374783965571623,
      "learning_rate": 1.2056252614284047e-06,
      "loss": 0.2819,
      "step": 3230
    },
    {
      "epoch": 4.648920863309352,
      "grad_norm": 0.04525843062003917,
      "learning_rate": 1.1958527433508381e-06,
      "loss": 0.2836,
      "step": 3231
    },
    {
      "epoch": 4.650359712230216,
      "grad_norm": 0.04466294146865429,
      "learning_rate": 1.1861193924340176e-06,
      "loss": 0.2917,
      "step": 3232
    },
    {
      "epoch": 4.651798561151079,
      "grad_norm": 0.04071085015830011,
      "learning_rate": 1.176425218502346e-06,
      "loss": 0.283,
      "step": 3233
    },
    {
      "epoch": 4.6532374100719425,
      "grad_norm": 0.04326246127923558,
      "learning_rate": 1.1667702313406903e-06,
      "loss": 0.2778,
      "step": 3234
    },
    {
      "epoch": 4.654676258992806,
      "grad_norm": 0.041761485943218456,
      "learning_rate": 1.1571544406943614e-06,
      "loss": 0.2802,
      "step": 3235
    },
    {
      "epoch": 4.656115107913669,
      "grad_norm": 0.04394084699989742,
      "learning_rate": 1.147577856269102e-06,
      "loss": 0.2851,
      "step": 3236
    },
    {
      "epoch": 4.6575539568345325,
      "grad_norm": 0.04384334299676405,
      "learning_rate": 1.1380404877310957e-06,
      "loss": 0.2799,
      "step": 3237
    },
    {
      "epoch": 4.658992805755395,
      "grad_norm": 0.044116322355897984,
      "learning_rate": 1.1285423447069133e-06,
      "loss": 0.2828,
      "step": 3238
    },
    {
      "epoch": 4.660431654676259,
      "grad_norm": 0.04552590103355667,
      "learning_rate": 1.1190834367835701e-06,
      "loss": 0.286,
      "step": 3239
    },
    {
      "epoch": 4.661870503597123,
      "grad_norm": 0.0465443435010611,
      "learning_rate": 1.1096637735084602e-06,
      "loss": 0.2814,
      "step": 3240
    },
    {
      "epoch": 4.663309352517985,
      "grad_norm": 0.04360432862627896,
      "learning_rate": 1.1002833643893606e-06,
      "loss": 0.2806,
      "step": 3241
    },
    {
      "epoch": 4.664748201438849,
      "grad_norm": 0.041014423211236745,
      "learning_rate": 1.0909422188944308e-06,
      "loss": 0.288,
      "step": 3242
    },
    {
      "epoch": 4.666187050359712,
      "grad_norm": 0.043487675209465046,
      "learning_rate": 1.0816403464522262e-06,
      "loss": 0.2781,
      "step": 3243
    },
    {
      "epoch": 4.6676258992805755,
      "grad_norm": 0.0434688738922224,
      "learning_rate": 1.0723777564516148e-06,
      "loss": 0.283,
      "step": 3244
    },
    {
      "epoch": 4.669064748201439,
      "grad_norm": 0.0436478159108985,
      "learning_rate": 1.0631544582418463e-06,
      "loss": 0.2798,
      "step": 3245
    },
    {
      "epoch": 4.670503597122302,
      "grad_norm": 0.04698488687018127,
      "learning_rate": 1.0539704611325008e-06,
      "loss": 0.2891,
      "step": 3246
    },
    {
      "epoch": 4.6719424460431656,
      "grad_norm": 0.04252447518711718,
      "learning_rate": 1.0448257743934964e-06,
      "loss": 0.2794,
      "step": 3247
    },
    {
      "epoch": 4.673381294964029,
      "grad_norm": 0.04133669991529513,
      "learning_rate": 1.0357204072550676e-06,
      "loss": 0.2791,
      "step": 3248
    },
    {
      "epoch": 4.674820143884892,
      "grad_norm": 0.04590539275118129,
      "learning_rate": 1.0266543689077602e-06,
      "loss": 0.2867,
      "step": 3249
    },
    {
      "epoch": 4.676258992805756,
      "grad_norm": 0.04123553797223722,
      "learning_rate": 1.0176276685024233e-06,
      "loss": 0.2832,
      "step": 3250
    },
    {
      "epoch": 4.677697841726618,
      "grad_norm": 0.04289768657332813,
      "learning_rate": 1.0086403151502088e-06,
      "loss": 0.2893,
      "step": 3251
    },
    {
      "epoch": 4.679136690647482,
      "grad_norm": 0.042216007316176335,
      "learning_rate": 9.996923179225448e-07,
      "loss": 0.2832,
      "step": 3252
    },
    {
      "epoch": 4.680575539568346,
      "grad_norm": 0.04522822430226165,
      "learning_rate": 9.90783685851131e-07,
      "loss": 0.2889,
      "step": 3253
    },
    {
      "epoch": 4.6820143884892085,
      "grad_norm": 0.04486886217148146,
      "learning_rate": 9.81914427927948e-07,
      "loss": 0.2807,
      "step": 3254
    },
    {
      "epoch": 4.683453237410072,
      "grad_norm": 0.04206957973117205,
      "learning_rate": 9.730845531052214e-07,
      "loss": 0.285,
      "step": 3255
    },
    {
      "epoch": 4.684892086330935,
      "grad_norm": 0.04268891368382537,
      "learning_rate": 9.642940702954306e-07,
      "loss": 0.2899,
      "step": 3256
    },
    {
      "epoch": 4.686330935251799,
      "grad_norm": 0.04223922193649368,
      "learning_rate": 9.555429883712963e-07,
      "loss": 0.2839,
      "step": 3257
    },
    {
      "epoch": 4.687769784172662,
      "grad_norm": 0.04116027131343593,
      "learning_rate": 9.468313161657617e-07,
      "loss": 0.288,
      "step": 3258
    },
    {
      "epoch": 4.689208633093525,
      "grad_norm": 0.042695974314423804,
      "learning_rate": 9.381590624719972e-07,
      "loss": 0.2844,
      "step": 3259
    },
    {
      "epoch": 4.690647482014389,
      "grad_norm": 0.04190157933084132,
      "learning_rate": 9.295262360433921e-07,
      "loss": 0.2799,
      "step": 3260
    },
    {
      "epoch": 4.692086330935252,
      "grad_norm": 0.04509694273732758,
      "learning_rate": 9.209328455935274e-07,
      "loss": 0.2827,
      "step": 3261
    },
    {
      "epoch": 4.693525179856115,
      "grad_norm": 0.04320569445563929,
      "learning_rate": 9.123788997961847e-07,
      "loss": 0.2849,
      "step": 3262
    },
    {
      "epoch": 4.694964028776979,
      "grad_norm": 0.04161867791634296,
      "learning_rate": 9.038644072853331e-07,
      "loss": 0.2789,
      "step": 3263
    },
    {
      "epoch": 4.6964028776978415,
      "grad_norm": 0.042314134927405225,
      "learning_rate": 8.953893766551203e-07,
      "loss": 0.2861,
      "step": 3264
    },
    {
      "epoch": 4.697841726618705,
      "grad_norm": 0.041629546653669705,
      "learning_rate": 8.86953816459859e-07,
      "loss": 0.2879,
      "step": 3265
    },
    {
      "epoch": 4.699280575539568,
      "grad_norm": 0.041723942617953015,
      "learning_rate": 8.785577352140317e-07,
      "loss": 0.2806,
      "step": 3266
    },
    {
      "epoch": 4.700719424460432,
      "grad_norm": 0.04454818248431817,
      "learning_rate": 8.702011413922506e-07,
      "loss": 0.2862,
      "step": 3267
    },
    {
      "epoch": 4.702158273381295,
      "grad_norm": 0.04355139769352831,
      "learning_rate": 8.61884043429293e-07,
      "loss": 0.2818,
      "step": 3268
    },
    {
      "epoch": 4.703597122302158,
      "grad_norm": 0.04204134591185999,
      "learning_rate": 8.536064497200702e-07,
      "loss": 0.2805,
      "step": 3269
    },
    {
      "epoch": 4.705035971223022,
      "grad_norm": 0.04270053767629897,
      "learning_rate": 8.453683686196012e-07,
      "loss": 0.2837,
      "step": 3270
    },
    {
      "epoch": 4.706474820143885,
      "grad_norm": 0.04240735124788586,
      "learning_rate": 8.371698084430346e-07,
      "loss": 0.2822,
      "step": 3271
    },
    {
      "epoch": 4.707913669064748,
      "grad_norm": 0.04245460034719818,
      "learning_rate": 8.290107774656441e-07,
      "loss": 0.2882,
      "step": 3272
    },
    {
      "epoch": 4.709352517985612,
      "grad_norm": 0.04200433994331035,
      "learning_rate": 8.208912839227712e-07,
      "loss": 0.2759,
      "step": 3273
    },
    {
      "epoch": 4.7107913669064745,
      "grad_norm": 0.04154456441805117,
      "learning_rate": 8.128113360098777e-07,
      "loss": 0.2841,
      "step": 3274
    },
    {
      "epoch": 4.712230215827338,
      "grad_norm": 0.043042952807340926,
      "learning_rate": 8.047709418824934e-07,
      "loss": 0.2871,
      "step": 3275
    },
    {
      "epoch": 4.713669064748202,
      "grad_norm": 0.043440337986098886,
      "learning_rate": 7.96770109656233e-07,
      "loss": 0.2901,
      "step": 3276
    },
    {
      "epoch": 4.715107913669065,
      "grad_norm": 0.04303030245868966,
      "learning_rate": 7.88808847406779e-07,
      "loss": 0.286,
      "step": 3277
    },
    {
      "epoch": 4.716546762589928,
      "grad_norm": 0.041327486619416834,
      "learning_rate": 7.808871631698723e-07,
      "loss": 0.2942,
      "step": 3278
    },
    {
      "epoch": 4.717985611510791,
      "grad_norm": 0.0407580233165075,
      "learning_rate": 7.730050649412946e-07,
      "loss": 0.2797,
      "step": 3279
    },
    {
      "epoch": 4.719424460431655,
      "grad_norm": 0.045055105516116564,
      "learning_rate": 7.651625606768908e-07,
      "loss": 0.2856,
      "step": 3280
    },
    {
      "epoch": 4.720863309352518,
      "grad_norm": 0.04163709593874125,
      "learning_rate": 7.573596582925291e-07,
      "loss": 0.277,
      "step": 3281
    },
    {
      "epoch": 4.722302158273381,
      "grad_norm": 0.04259533465764146,
      "learning_rate": 7.495963656641048e-07,
      "loss": 0.2947,
      "step": 3282
    },
    {
      "epoch": 4.723741007194245,
      "grad_norm": 0.043496681746895215,
      "learning_rate": 7.418726906275497e-07,
      "loss": 0.2899,
      "step": 3283
    },
    {
      "epoch": 4.725179856115108,
      "grad_norm": 0.04103629481090085,
      "learning_rate": 7.341886409787746e-07,
      "loss": 0.2759,
      "step": 3284
    },
    {
      "epoch": 4.726618705035971,
      "grad_norm": 0.042188130402698934,
      "learning_rate": 7.265442244737264e-07,
      "loss": 0.2772,
      "step": 3285
    },
    {
      "epoch": 4.728057553956835,
      "grad_norm": 0.04149586786309318,
      "learning_rate": 7.189394488283307e-07,
      "loss": 0.2861,
      "step": 3286
    },
    {
      "epoch": 4.729496402877698,
      "grad_norm": 0.041391815410889245,
      "learning_rate": 7.113743217185099e-07,
      "loss": 0.2805,
      "step": 3287
    },
    {
      "epoch": 4.730935251798561,
      "grad_norm": 0.04115458244017699,
      "learning_rate": 7.0384885078016e-07,
      "loss": 0.293,
      "step": 3288
    },
    {
      "epoch": 4.732374100719424,
      "grad_norm": 0.04172485851828882,
      "learning_rate": 6.963630436091518e-07,
      "loss": 0.2848,
      "step": 3289
    },
    {
      "epoch": 4.733812949640288,
      "grad_norm": 0.040586990215047185,
      "learning_rate": 6.889169077613212e-07,
      "loss": 0.2868,
      "step": 3290
    },
    {
      "epoch": 4.735251798561151,
      "grad_norm": 0.04112353475003874,
      "learning_rate": 6.815104507524695e-07,
      "loss": 0.2801,
      "step": 3291
    },
    {
      "epoch": 4.736690647482014,
      "grad_norm": 0.04157853046899118,
      "learning_rate": 6.741436800583367e-07,
      "loss": 0.2927,
      "step": 3292
    },
    {
      "epoch": 4.738129496402878,
      "grad_norm": 0.0438786444494133,
      "learning_rate": 6.668166031146062e-07,
      "loss": 0.2866,
      "step": 3293
    },
    {
      "epoch": 4.739568345323741,
      "grad_norm": 0.042734560172589184,
      "learning_rate": 6.595292273169041e-07,
      "loss": 0.2839,
      "step": 3294
    },
    {
      "epoch": 4.741007194244604,
      "grad_norm": 0.042515843125377976,
      "learning_rate": 6.522815600207866e-07,
      "loss": 0.2772,
      "step": 3295
    },
    {
      "epoch": 4.742446043165468,
      "grad_norm": 0.0402613039250005,
      "learning_rate": 6.450736085417086e-07,
      "loss": 0.279,
      "step": 3296
    },
    {
      "epoch": 4.743884892086331,
      "grad_norm": 0.04109263705843827,
      "learning_rate": 6.379053801550594e-07,
      "loss": 0.2822,
      "step": 3297
    },
    {
      "epoch": 4.745323741007194,
      "grad_norm": 0.09293908614636029,
      "learning_rate": 6.307768820961269e-07,
      "loss": 0.2841,
      "step": 3298
    },
    {
      "epoch": 4.746762589928058,
      "grad_norm": 0.04041588774309918,
      "learning_rate": 6.236881215600976e-07,
      "loss": 0.2862,
      "step": 3299
    },
    {
      "epoch": 4.748201438848921,
      "grad_norm": 0.04211380249832902,
      "learning_rate": 6.166391057020438e-07,
      "loss": 0.2841,
      "step": 3300
    },
    {
      "epoch": 4.749640287769784,
      "grad_norm": 0.04119439798710801,
      "learning_rate": 6.096298416369273e-07,
      "loss": 0.2942,
      "step": 3301
    },
    {
      "epoch": 4.751079136690647,
      "grad_norm": 0.04190903520399297,
      "learning_rate": 6.026603364395867e-07,
      "loss": 0.2875,
      "step": 3302
    },
    {
      "epoch": 4.752517985611511,
      "grad_norm": 0.046643300434686695,
      "learning_rate": 5.957305971447192e-07,
      "loss": 0.2799,
      "step": 3303
    },
    {
      "epoch": 4.753956834532374,
      "grad_norm": 0.04302145218360165,
      "learning_rate": 5.888406307468986e-07,
      "loss": 0.2889,
      "step": 3304
    },
    {
      "epoch": 4.755395683453237,
      "grad_norm": 0.04132944990889099,
      "learning_rate": 5.819904442005442e-07,
      "loss": 0.2838,
      "step": 3305
    },
    {
      "epoch": 4.756834532374101,
      "grad_norm": 0.04161240484067499,
      "learning_rate": 5.751800444199295e-07,
      "loss": 0.2844,
      "step": 3306
    },
    {
      "epoch": 4.7582733812949645,
      "grad_norm": 0.04256756179905745,
      "learning_rate": 5.684094382791605e-07,
      "loss": 0.2956,
      "step": 3307
    },
    {
      "epoch": 4.759712230215827,
      "grad_norm": 0.043508014652496096,
      "learning_rate": 5.616786326121837e-07,
      "loss": 0.2856,
      "step": 3308
    },
    {
      "epoch": 4.761151079136691,
      "grad_norm": 0.040238845233041366,
      "learning_rate": 5.54987634212778e-07,
      "loss": 0.2831,
      "step": 3309
    },
    {
      "epoch": 4.762589928057554,
      "grad_norm": 0.04020307345574689,
      "learning_rate": 5.483364498345279e-07,
      "loss": 0.2814,
      "step": 3310
    },
    {
      "epoch": 4.764028776978417,
      "grad_norm": 0.04346406741583254,
      "learning_rate": 5.417250861908452e-07,
      "loss": 0.2861,
      "step": 3311
    },
    {
      "epoch": 4.76546762589928,
      "grad_norm": 0.04217980322246722,
      "learning_rate": 5.351535499549387e-07,
      "loss": 0.2864,
      "step": 3312
    },
    {
      "epoch": 4.766906474820144,
      "grad_norm": 0.0431790541198276,
      "learning_rate": 5.286218477598226e-07,
      "loss": 0.2881,
      "step": 3313
    },
    {
      "epoch": 4.768345323741007,
      "grad_norm": 0.040920642398328155,
      "learning_rate": 5.221299861983075e-07,
      "loss": 0.283,
      "step": 3314
    },
    {
      "epoch": 4.76978417266187,
      "grad_norm": 0.04113680865948008,
      "learning_rate": 5.156779718229787e-07,
      "loss": 0.283,
      "step": 3315
    },
    {
      "epoch": 4.771223021582734,
      "grad_norm": 0.04206869377518095,
      "learning_rate": 5.092658111462179e-07,
      "loss": 0.2819,
      "step": 3316
    },
    {
      "epoch": 4.7726618705035975,
      "grad_norm": 0.04145270335602184,
      "learning_rate": 5.028935106401678e-07,
      "loss": 0.2862,
      "step": 3317
    },
    {
      "epoch": 4.77410071942446,
      "grad_norm": 0.041916908198382855,
      "learning_rate": 4.965610767367413e-07,
      "loss": 0.284,
      "step": 3318
    },
    {
      "epoch": 4.775539568345324,
      "grad_norm": 0.04318371796542412,
      "learning_rate": 4.902685158276078e-07,
      "loss": 0.2872,
      "step": 3319
    },
    {
      "epoch": 4.7769784172661875,
      "grad_norm": 0.04269750788765892,
      "learning_rate": 4.840158342642021e-07,
      "loss": 0.2844,
      "step": 3320
    },
    {
      "epoch": 4.77841726618705,
      "grad_norm": 0.04167765037493412,
      "learning_rate": 4.778030383577026e-07,
      "loss": 0.2817,
      "step": 3321
    },
    {
      "epoch": 4.779856115107914,
      "grad_norm": 0.04676371457990791,
      "learning_rate": 4.716301343790175e-07,
      "loss": 0.2924,
      "step": 3322
    },
    {
      "epoch": 4.781294964028777,
      "grad_norm": 0.04279536578991645,
      "learning_rate": 4.6549712855879837e-07,
      "loss": 0.2751,
      "step": 3323
    },
    {
      "epoch": 4.78273381294964,
      "grad_norm": 0.040420197504620795,
      "learning_rate": 4.5940402708744005e-07,
      "loss": 0.2852,
      "step": 3324
    },
    {
      "epoch": 4.784172661870503,
      "grad_norm": 0.04245397455825739,
      "learning_rate": 4.5335083611502293e-07,
      "loss": 0.2924,
      "step": 3325
    },
    {
      "epoch": 4.785611510791367,
      "grad_norm": 0.042732378110714474,
      "learning_rate": 4.473375617513842e-07,
      "loss": 0.2889,
      "step": 3326
    },
    {
      "epoch": 4.7870503597122305,
      "grad_norm": 0.04075680836884234,
      "learning_rate": 4.41364210066042e-07,
      "loss": 0.2857,
      "step": 3327
    },
    {
      "epoch": 4.788489208633093,
      "grad_norm": 0.041591069620682895,
      "learning_rate": 4.3543078708823126e-07,
      "loss": 0.2886,
      "step": 3328
    },
    {
      "epoch": 4.789928057553957,
      "grad_norm": 0.041438570320909945,
      "learning_rate": 4.295372988068813e-07,
      "loss": 0.2901,
      "step": 3329
    },
    {
      "epoch": 4.7913669064748206,
      "grad_norm": 0.04146670829980007,
      "learning_rate": 4.2368375117062043e-07,
      "loss": 0.285,
      "step": 3330
    },
    {
      "epoch": 4.792805755395683,
      "grad_norm": 0.0413497691397624,
      "learning_rate": 4.178701500877491e-07,
      "loss": 0.2769,
      "step": 3331
    },
    {
      "epoch": 4.794244604316547,
      "grad_norm": 0.041040800163841984,
      "learning_rate": 4.120965014262579e-07,
      "loss": 0.2882,
      "step": 3332
    },
    {
      "epoch": 4.79568345323741,
      "grad_norm": 0.04189017618203409,
      "learning_rate": 4.063628110138096e-07,
      "loss": 0.2904,
      "step": 3333
    },
    {
      "epoch": 4.797122302158273,
      "grad_norm": 0.04061719266311827,
      "learning_rate": 4.0066908463772593e-07,
      "loss": 0.2803,
      "step": 3334
    },
    {
      "epoch": 4.798561151079137,
      "grad_norm": 0.042636097032954345,
      "learning_rate": 3.9501532804500974e-07,
      "loss": 0.2904,
      "step": 3335
    },
    {
      "epoch": 4.8,
      "grad_norm": 0.042510369470737885,
      "learning_rate": 3.894015469423007e-07,
      "loss": 0.2862,
      "step": 3336
    },
    {
      "epoch": 4.8014388489208635,
      "grad_norm": 0.04042658066542344,
      "learning_rate": 3.838277469958973e-07,
      "loss": 0.2839,
      "step": 3337
    },
    {
      "epoch": 4.802877697841726,
      "grad_norm": 0.04304923392502034,
      "learning_rate": 3.7829393383174375e-07,
      "loss": 0.2915,
      "step": 3338
    },
    {
      "epoch": 4.80431654676259,
      "grad_norm": 0.04140898979796202,
      "learning_rate": 3.7280011303542084e-07,
      "loss": 0.2866,
      "step": 3339
    },
    {
      "epoch": 4.805755395683454,
      "grad_norm": 0.041749156567567884,
      "learning_rate": 3.673462901521463e-07,
      "loss": 0.2822,
      "step": 3340
    },
    {
      "epoch": 4.807194244604316,
      "grad_norm": 0.041097810617877185,
      "learning_rate": 3.619324706867655e-07,
      "loss": 0.2845,
      "step": 3341
    },
    {
      "epoch": 4.80863309352518,
      "grad_norm": 0.04357761380107742,
      "learning_rate": 3.5655866010373853e-07,
      "loss": 0.2825,
      "step": 3342
    },
    {
      "epoch": 4.810071942446044,
      "grad_norm": 0.04202729999373872,
      "learning_rate": 3.5122486382715314e-07,
      "loss": 0.287,
      "step": 3343
    },
    {
      "epoch": 4.811510791366906,
      "grad_norm": 0.04187911764346768,
      "learning_rate": 3.459310872407029e-07,
      "loss": 0.2838,
      "step": 3344
    },
    {
      "epoch": 4.81294964028777,
      "grad_norm": 0.04103324603536389,
      "learning_rate": 3.4067733568768246e-07,
      "loss": 0.2916,
      "step": 3345
    },
    {
      "epoch": 4.814388489208633,
      "grad_norm": 0.041161310147595144,
      "learning_rate": 3.3546361447099664e-07,
      "loss": 0.2868,
      "step": 3346
    },
    {
      "epoch": 4.8158273381294965,
      "grad_norm": 0.04100603431956255,
      "learning_rate": 3.3028992885314247e-07,
      "loss": 0.2885,
      "step": 3347
    },
    {
      "epoch": 4.817266187050359,
      "grad_norm": 0.04072316462824549,
      "learning_rate": 3.2515628405620503e-07,
      "loss": 0.2834,
      "step": 3348
    },
    {
      "epoch": 4.818705035971223,
      "grad_norm": 0.041417910096654374,
      "learning_rate": 3.2006268526184824e-07,
      "loss": 0.2861,
      "step": 3349
    },
    {
      "epoch": 4.820143884892087,
      "grad_norm": 0.039557375878430695,
      "learning_rate": 3.150091376113329e-07,
      "loss": 0.2802,
      "step": 3350
    },
    {
      "epoch": 4.821582733812949,
      "grad_norm": 0.04192453172556301,
      "learning_rate": 3.0999564620547207e-07,
      "loss": 0.2848,
      "step": 3351
    },
    {
      "epoch": 4.823021582733813,
      "grad_norm": 0.04212047586899518,
      "learning_rate": 3.0502221610465786e-07,
      "loss": 0.2944,
      "step": 3352
    },
    {
      "epoch": 4.824460431654677,
      "grad_norm": 0.04134459177446905,
      "learning_rate": 3.0008885232886144e-07,
      "loss": 0.2884,
      "step": 3353
    },
    {
      "epoch": 4.825899280575539,
      "grad_norm": 0.04105737532473463,
      "learning_rate": 2.95195559857584e-07,
      "loss": 0.2785,
      "step": 3354
    },
    {
      "epoch": 4.827338129496403,
      "grad_norm": 0.04026592035207497,
      "learning_rate": 2.9034234362989687e-07,
      "loss": 0.2819,
      "step": 3355
    },
    {
      "epoch": 4.828776978417266,
      "grad_norm": 0.042417527147535355,
      "learning_rate": 2.855292085444239e-07,
      "loss": 0.2875,
      "step": 3356
    },
    {
      "epoch": 4.8302158273381295,
      "grad_norm": 0.04173478745196841,
      "learning_rate": 2.8075615945932333e-07,
      "loss": 0.2861,
      "step": 3357
    },
    {
      "epoch": 4.831654676258993,
      "grad_norm": 0.04172308960992333,
      "learning_rate": 2.7602320119229254e-07,
      "loss": 0.2822,
      "step": 3358
    },
    {
      "epoch": 4.833093525179856,
      "grad_norm": 0.04299243006807228,
      "learning_rate": 2.7133033852057675e-07,
      "loss": 0.2888,
      "step": 3359
    },
    {
      "epoch": 4.83453237410072,
      "grad_norm": 0.04150364052335503,
      "learning_rate": 2.666775761809337e-07,
      "loss": 0.2869,
      "step": 3360
    },
    {
      "epoch": 4.835971223021582,
      "grad_norm": 0.04033984987840341,
      "learning_rate": 2.620649188696511e-07,
      "loss": 0.2857,
      "step": 3361
    },
    {
      "epoch": 4.837410071942446,
      "grad_norm": 0.040675359063735185,
      "learning_rate": 2.574923712425426e-07,
      "loss": 0.2923,
      "step": 3362
    },
    {
      "epoch": 4.83884892086331,
      "grad_norm": 0.04233903725882189,
      "learning_rate": 2.52959937914925e-07,
      "loss": 0.2911,
      "step": 3363
    },
    {
      "epoch": 4.840287769784172,
      "grad_norm": 0.04076547458038119,
      "learning_rate": 2.484676234616412e-07,
      "loss": 0.2899,
      "step": 3364
    },
    {
      "epoch": 4.841726618705036,
      "grad_norm": 0.041433913736535356,
      "learning_rate": 2.440154324170285e-07,
      "loss": 0.2884,
      "step": 3365
    },
    {
      "epoch": 4.8431654676259,
      "grad_norm": 0.041269377675926526,
      "learning_rate": 2.3960336927492333e-07,
      "loss": 0.287,
      "step": 3366
    },
    {
      "epoch": 4.8446043165467625,
      "grad_norm": 0.041061994988089365,
      "learning_rate": 2.3523143848867003e-07,
      "loss": 0.2827,
      "step": 3367
    },
    {
      "epoch": 4.846043165467626,
      "grad_norm": 0.04187010556305427,
      "learning_rate": 2.3089964447109425e-07,
      "loss": 0.2809,
      "step": 3368
    },
    {
      "epoch": 4.847482014388489,
      "grad_norm": 0.04303825153494671,
      "learning_rate": 2.2660799159451629e-07,
      "loss": 0.2815,
      "step": 3369
    },
    {
      "epoch": 4.848920863309353,
      "grad_norm": 0.041983880590083476,
      "learning_rate": 2.2235648419073773e-07,
      "loss": 0.2796,
      "step": 3370
    },
    {
      "epoch": 4.850359712230215,
      "grad_norm": 0.04145071337212364,
      "learning_rate": 2.1814512655103703e-07,
      "loss": 0.2846,
      "step": 3371
    },
    {
      "epoch": 4.851798561151079,
      "grad_norm": 0.04035087733302524,
      "learning_rate": 2.1397392292617392e-07,
      "loss": 0.2809,
      "step": 3372
    },
    {
      "epoch": 4.853237410071943,
      "grad_norm": 0.04175320728399013,
      "learning_rate": 2.0984287752636722e-07,
      "loss": 0.2871,
      "step": 3373
    },
    {
      "epoch": 4.854676258992805,
      "grad_norm": 0.040747681149666495,
      "learning_rate": 2.0575199452131268e-07,
      "loss": 0.2835,
      "step": 3374
    },
    {
      "epoch": 4.856115107913669,
      "grad_norm": 0.04010326090661748,
      "learning_rate": 2.017012780401606e-07,
      "loss": 0.2862,
      "step": 3375
    },
    {
      "epoch": 4.857553956834533,
      "grad_norm": 0.040912678927391144,
      "learning_rate": 1.9769073217152933e-07,
      "loss": 0.2812,
      "step": 3376
    },
    {
      "epoch": 4.8589928057553955,
      "grad_norm": 0.040760288220981,
      "learning_rate": 1.9372036096347414e-07,
      "loss": 0.2959,
      "step": 3377
    },
    {
      "epoch": 4.860431654676259,
      "grad_norm": 0.041949715345728174,
      "learning_rate": 1.8979016842350928e-07,
      "loss": 0.2813,
      "step": 3378
    },
    {
      "epoch": 4.861870503597122,
      "grad_norm": 0.04341096348456281,
      "learning_rate": 1.8590015851860376e-07,
      "loss": 0.2807,
      "step": 3379
    },
    {
      "epoch": 4.863309352517986,
      "grad_norm": 0.043875483322259254,
      "learning_rate": 1.8205033517515015e-07,
      "loss": 0.2886,
      "step": 3380
    },
    {
      "epoch": 4.864748201438849,
      "grad_norm": 0.04046833701881062,
      "learning_rate": 1.7824070227899115e-07,
      "loss": 0.2873,
      "step": 3381
    },
    {
      "epoch": 4.866187050359712,
      "grad_norm": 0.04098147062761545,
      "learning_rate": 1.7447126367539313e-07,
      "loss": 0.2895,
      "step": 3382
    },
    {
      "epoch": 4.867625899280576,
      "grad_norm": 0.04041993174822328,
      "learning_rate": 1.7074202316906374e-07,
      "loss": 0.2901,
      "step": 3383
    },
    {
      "epoch": 4.869064748201438,
      "grad_norm": 0.040792225360416115,
      "learning_rate": 1.6705298452412978e-07,
      "loss": 0.2842,
      "step": 3384
    },
    {
      "epoch": 4.870503597122302,
      "grad_norm": 0.04172365959670488,
      "learning_rate": 1.6340415146414157e-07,
      "loss": 0.2865,
      "step": 3385
    },
    {
      "epoch": 4.871942446043166,
      "grad_norm": 0.04127806179352284,
      "learning_rate": 1.597955276720642e-07,
      "loss": 0.2777,
      "step": 3386
    },
    {
      "epoch": 4.8733812949640285,
      "grad_norm": 0.041416591174507904,
      "learning_rate": 1.562271167902818e-07,
      "loss": 0.2827,
      "step": 3387
    },
    {
      "epoch": 4.874820143884892,
      "grad_norm": 0.040091316222597644,
      "learning_rate": 1.526989224205888e-07,
      "loss": 0.2871,
      "step": 3388
    },
    {
      "epoch": 4.876258992805756,
      "grad_norm": 0.04107944987078495,
      "learning_rate": 1.4921094812418103e-07,
      "loss": 0.2883,
      "step": 3389
    },
    {
      "epoch": 4.877697841726619,
      "grad_norm": 0.041313202214486346,
      "learning_rate": 1.457631974216689e-07,
      "loss": 0.2885,
      "step": 3390
    },
    {
      "epoch": 4.879136690647482,
      "grad_norm": 0.04079308145614227,
      "learning_rate": 1.4235567379305536e-07,
      "loss": 0.2856,
      "step": 3391
    },
    {
      "epoch": 4.880575539568345,
      "grad_norm": 0.0415866997071899,
      "learning_rate": 1.389883806777359e-07,
      "loss": 0.2879,
      "step": 3392
    },
    {
      "epoch": 4.882014388489209,
      "grad_norm": 0.04234944058835651,
      "learning_rate": 1.356613214745117e-07,
      "loss": 0.2876,
      "step": 3393
    },
    {
      "epoch": 4.883453237410072,
      "grad_norm": 0.041253715501214684,
      "learning_rate": 1.3237449954156767e-07,
      "loss": 0.2869,
      "step": 3394
    },
    {
      "epoch": 4.884892086330935,
      "grad_norm": 0.04213450458140177,
      "learning_rate": 1.2912791819646774e-07,
      "loss": 0.2905,
      "step": 3395
    },
    {
      "epoch": 4.886330935251799,
      "grad_norm": 0.04155949050167922,
      "learning_rate": 1.2592158071616844e-07,
      "loss": 0.2832,
      "step": 3396
    },
    {
      "epoch": 4.8877697841726615,
      "grad_norm": 0.041056507695268085,
      "learning_rate": 1.2275549033700097e-07,
      "loss": 0.2785,
      "step": 3397
    },
    {
      "epoch": 4.889208633093525,
      "grad_norm": 0.04080476316594233,
      "learning_rate": 1.1962965025467564e-07,
      "loss": 0.2853,
      "step": 3398
    },
    {
      "epoch": 4.890647482014389,
      "grad_norm": 0.0416474307419702,
      "learning_rate": 1.1654406362427762e-07,
      "loss": 0.2871,
      "step": 3399
    },
    {
      "epoch": 4.892086330935252,
      "grad_norm": 0.04127363787913204,
      "learning_rate": 1.1349873356025332e-07,
      "loss": 0.286,
      "step": 3400
    },
    {
      "epoch": 4.893525179856115,
      "grad_norm": 0.041206338684211516,
      "learning_rate": 1.1049366313642395e-07,
      "loss": 0.2896,
      "step": 3401
    },
    {
      "epoch": 4.894964028776979,
      "grad_norm": 0.03976419154372853,
      "learning_rate": 1.0752885538598102e-07,
      "loss": 0.2836,
      "step": 3402
    },
    {
      "epoch": 4.896402877697842,
      "grad_norm": 0.04048593648075003,
      "learning_rate": 1.0460431330145515e-07,
      "loss": 0.2758,
      "step": 3403
    },
    {
      "epoch": 4.897841726618705,
      "grad_norm": 0.09191720051199531,
      "learning_rate": 1.0172003983475176e-07,
      "loss": 0.2869,
      "step": 3404
    },
    {
      "epoch": 4.899280575539568,
      "grad_norm": 0.04114152568043639,
      "learning_rate": 9.887603789712875e-08,
      "loss": 0.278,
      "step": 3405
    },
    {
      "epoch": 4.900719424460432,
      "grad_norm": 0.04285267729633756,
      "learning_rate": 9.607231035919651e-08,
      "loss": 0.2994,
      "step": 3406
    },
    {
      "epoch": 4.9021582733812945,
      "grad_norm": 0.04083695716288777,
      "learning_rate": 9.330886005090467e-08,
      "loss": 0.2801,
      "step": 3407
    },
    {
      "epoch": 4.903597122302158,
      "grad_norm": 0.0393357487579918,
      "learning_rate": 9.058568976155979e-08,
      "loss": 0.2851,
      "step": 3408
    },
    {
      "epoch": 4.905035971223022,
      "grad_norm": 0.040494727324840725,
      "learning_rate": 8.790280223980763e-08,
      "loss": 0.2884,
      "step": 3409
    },
    {
      "epoch": 4.906474820143885,
      "grad_norm": 0.041520914283993134,
      "learning_rate": 8.526020019363313e-08,
      "loss": 0.2866,
      "step": 3410
    },
    {
      "epoch": 4.907913669064748,
      "grad_norm": 0.04080158727412204,
      "learning_rate": 8.265788629036043e-08,
      "loss": 0.2807,
      "step": 3411
    },
    {
      "epoch": 4.909352517985612,
      "grad_norm": 0.03972572760435433,
      "learning_rate": 8.009586315664842e-08,
      "loss": 0.2847,
      "step": 3412
    },
    {
      "epoch": 4.910791366906475,
      "grad_norm": 0.04125019779174967,
      "learning_rate": 7.757413337848629e-08,
      "loss": 0.2875,
      "step": 3413
    },
    {
      "epoch": 4.912230215827338,
      "grad_norm": 0.04035270128054002,
      "learning_rate": 7.509269950119358e-08,
      "loss": 0.2852,
      "step": 3414
    },
    {
      "epoch": 4.913669064748201,
      "grad_norm": 0.04091786942262446,
      "learning_rate": 7.265156402942452e-08,
      "loss": 0.2889,
      "step": 3415
    },
    {
      "epoch": 4.915107913669065,
      "grad_norm": 0.04040951017184305,
      "learning_rate": 7.025072942714595e-08,
      "loss": 0.2865,
      "step": 3416
    },
    {
      "epoch": 4.916546762589928,
      "grad_norm": 0.040690841064591504,
      "learning_rate": 6.789019811765052e-08,
      "loss": 0.2852,
      "step": 3417
    },
    {
      "epoch": 4.917985611510791,
      "grad_norm": 0.04147303639496497,
      "learning_rate": 6.556997248355679e-08,
      "loss": 0.2905,
      "step": 3418
    },
    {
      "epoch": 4.919424460431655,
      "grad_norm": 0.04404809316123878,
      "learning_rate": 6.329005486679584e-08,
      "loss": 0.2929,
      "step": 3419
    },
    {
      "epoch": 4.920863309352518,
      "grad_norm": 0.04059398598689888,
      "learning_rate": 6.105044756861134e-08,
      "loss": 0.2824,
      "step": 3420
    },
    {
      "epoch": 4.922302158273381,
      "grad_norm": 0.04216826638341303,
      "learning_rate": 5.8851152849563886e-08,
      "loss": 0.2819,
      "step": 3421
    },
    {
      "epoch": 4.923741007194245,
      "grad_norm": 0.0415649921570707,
      "learning_rate": 5.669217292952223e-08,
      "loss": 0.2816,
      "step": 3422
    },
    {
      "epoch": 4.925179856115108,
      "grad_norm": 0.04255796731782151,
      "learning_rate": 5.4573509987663196e-08,
      "loss": 0.2847,
      "step": 3423
    },
    {
      "epoch": 4.926618705035971,
      "grad_norm": 0.03971819662193374,
      "learning_rate": 5.2495166162471747e-08,
      "loss": 0.2772,
      "step": 3424
    },
    {
      "epoch": 4.928057553956835,
      "grad_norm": 0.04116670684744748,
      "learning_rate": 5.045714355173203e-08,
      "loss": 0.2943,
      "step": 3425
    },
    {
      "epoch": 4.929496402877698,
      "grad_norm": 0.04207188074099609,
      "learning_rate": 4.845944421253634e-08,
      "loss": 0.2915,
      "step": 3426
    },
    {
      "epoch": 4.930935251798561,
      "grad_norm": 0.040857963264732865,
      "learning_rate": 4.650207016126729e-08,
      "loss": 0.2871,
      "step": 3427
    },
    {
      "epoch": 4.932374100719424,
      "grad_norm": 0.040268768737739886,
      "learning_rate": 4.458502337361115e-08,
      "loss": 0.2837,
      "step": 3428
    },
    {
      "epoch": 4.933812949640288,
      "grad_norm": 0.07856901734688387,
      "learning_rate": 4.270830578455343e-08,
      "loss": 0.2801,
      "step": 3429
    },
    {
      "epoch": 4.935251798561151,
      "grad_norm": 0.04017694922339272,
      "learning_rate": 4.087191928836554e-08,
      "loss": 0.2778,
      "step": 3430
    },
    {
      "epoch": 4.936690647482014,
      "grad_norm": 0.041636434158801784,
      "learning_rate": 3.907586573860922e-08,
      "loss": 0.2875,
      "step": 3431
    },
    {
      "epoch": 4.938129496402878,
      "grad_norm": 0.04151392621853386,
      "learning_rate": 3.7320146948149894e-08,
      "loss": 0.2801,
      "step": 3432
    },
    {
      "epoch": 4.939568345323741,
      "grad_norm": 0.04170858249943328,
      "learning_rate": 3.560476468912111e-08,
      "loss": 0.2859,
      "step": 3433
    },
    {
      "epoch": 4.941007194244604,
      "grad_norm": 0.04144939942650809,
      "learning_rate": 3.392972069295564e-08,
      "loss": 0.2896,
      "step": 3434
    },
    {
      "epoch": 4.942446043165468,
      "grad_norm": 0.04072381817838615,
      "learning_rate": 3.229501665037216e-08,
      "loss": 0.2849,
      "step": 3435
    },
    {
      "epoch": 4.943884892086331,
      "grad_norm": 0.03980670385989213,
      "learning_rate": 3.0700654211361925e-08,
      "loss": 0.2881,
      "step": 3436
    },
    {
      "epoch": 4.945323741007194,
      "grad_norm": 0.040739092121564686,
      "learning_rate": 2.9146634985206535e-08,
      "loss": 0.2806,
      "step": 3437
    },
    {
      "epoch": 4.946762589928057,
      "grad_norm": 0.04188578120567552,
      "learning_rate": 2.7632960540460162e-08,
      "loss": 0.2904,
      "step": 3438
    },
    {
      "epoch": 4.948201438848921,
      "grad_norm": 0.04031944356055068,
      "learning_rate": 2.6159632404958447e-08,
      "loss": 0.2948,
      "step": 3439
    },
    {
      "epoch": 4.9496402877697845,
      "grad_norm": 0.03966164742709178,
      "learning_rate": 2.472665206581404e-08,
      "loss": 0.2852,
      "step": 3440
    },
    {
      "epoch": 4.951079136690647,
      "grad_norm": 0.04129213051444172,
      "learning_rate": 2.3334020969407733e-08,
      "loss": 0.2842,
      "step": 3441
    },
    {
      "epoch": 4.952517985611511,
      "grad_norm": 0.04107420396143875,
      "learning_rate": 2.1981740521406226e-08,
      "loss": 0.2846,
      "step": 3442
    },
    {
      "epoch": 4.953956834532374,
      "grad_norm": 0.041070165648897115,
      "learning_rate": 2.0669812086735464e-08,
      "loss": 0.2912,
      "step": 3443
    },
    {
      "epoch": 4.955395683453237,
      "grad_norm": 0.040347565351461646,
      "learning_rate": 1.9398236989598418e-08,
      "loss": 0.2825,
      "step": 3444
    },
    {
      "epoch": 4.956834532374101,
      "grad_norm": 0.039675938953722636,
      "learning_rate": 1.8167016513470636e-08,
      "loss": 0.2843,
      "step": 3445
    },
    {
      "epoch": 4.958273381294964,
      "grad_norm": 0.04016568107289864,
      "learning_rate": 1.697615190107804e-08,
      "loss": 0.2842,
      "step": 3446
    },
    {
      "epoch": 4.959712230215827,
      "grad_norm": 0.040123864608462276,
      "learning_rate": 1.582564435444134e-08,
      "loss": 0.2822,
      "step": 3447
    },
    {
      "epoch": 4.961151079136691,
      "grad_norm": 0.04190621407744655,
      "learning_rate": 1.4715495034818284e-08,
      "loss": 0.2897,
      "step": 3448
    },
    {
      "epoch": 4.962589928057554,
      "grad_norm": 0.0410268233631543,
      "learning_rate": 1.3645705062748094e-08,
      "loss": 0.2897,
      "step": 3449
    },
    {
      "epoch": 4.9640287769784175,
      "grad_norm": 0.04235511323058934,
      "learning_rate": 1.2616275518033683e-08,
      "loss": 0.2884,
      "step": 3450
    },
    {
      "epoch": 4.96546762589928,
      "grad_norm": 0.03967760133464778,
      "learning_rate": 1.1627207439728339e-08,
      "loss": 0.2842,
      "step": 3451
    },
    {
      "epoch": 4.966906474820144,
      "grad_norm": 0.04083234181194213,
      "learning_rate": 1.0678501826153486e-08,
      "loss": 0.288,
      "step": 3452
    },
    {
      "epoch": 4.968345323741008,
      "grad_norm": 0.04245635744108195,
      "learning_rate": 9.770159634894249e-09,
      "loss": 0.2869,
      "step": 3453
    },
    {
      "epoch": 4.96978417266187,
      "grad_norm": 0.0413453937775347,
      "learning_rate": 8.902181782786124e-09,
      "loss": 0.2834,
      "step": 3454
    },
    {
      "epoch": 4.971223021582734,
      "grad_norm": 0.04127801612306726,
      "learning_rate": 8.074569145928301e-09,
      "loss": 0.2856,
      "step": 3455
    },
    {
      "epoch": 4.972661870503597,
      "grad_norm": 0.040774570467288315,
      "learning_rate": 7.287322559679233e-09,
      "loss": 0.2851,
      "step": 3456
    },
    {
      "epoch": 4.97410071942446,
      "grad_norm": 0.041267924692496766,
      "learning_rate": 6.5404428186433e-09,
      "loss": 0.2932,
      "step": 3457
    },
    {
      "epoch": 4.975539568345324,
      "grad_norm": 0.04191204714568303,
      "learning_rate": 5.833930676693023e-09,
      "loss": 0.2872,
      "step": 3458
    },
    {
      "epoch": 4.976978417266187,
      "grad_norm": 0.04124089367530566,
      "learning_rate": 5.167786846946854e-09,
      "loss": 0.2859,
      "step": 3459
    },
    {
      "epoch": 4.9784172661870505,
      "grad_norm": 0.0400216176741452,
      "learning_rate": 4.542012001778062e-09,
      "loss": 0.286,
      "step": 3460
    },
    {
      "epoch": 4.979856115107914,
      "grad_norm": 0.04047404384354867,
      "learning_rate": 3.956606772823613e-09,
      "loss": 0.2883,
      "step": 3461
    },
    {
      "epoch": 4.981294964028777,
      "grad_norm": 0.040197633392815,
      "learning_rate": 3.4115717509619616e-09,
      "loss": 0.285,
      "step": 3462
    },
    {
      "epoch": 4.982733812949641,
      "grad_norm": 0.04129146054521192,
      "learning_rate": 2.9069074863219414e-09,
      "loss": 0.2841,
      "step": 3463
    },
    {
      "epoch": 4.984172661870503,
      "grad_norm": 0.04042323219819034,
      "learning_rate": 2.4426144882916392e-09,
      "loss": 0.2853,
      "step": 3464
    },
    {
      "epoch": 4.985611510791367,
      "grad_norm": 0.04112481051522228,
      "learning_rate": 2.018693225509516e-09,
      "loss": 0.2896,
      "step": 3465
    },
    {
      "epoch": 4.98705035971223,
      "grad_norm": 0.04102996245279678,
      "learning_rate": 1.6351441258644073e-09,
      "loss": 0.2834,
      "step": 3466
    },
    {
      "epoch": 4.988489208633093,
      "grad_norm": 0.042034830155993054,
      "learning_rate": 1.2919675764910823e-09,
      "loss": 0.2753,
      "step": 3467
    },
    {
      "epoch": 4.989928057553957,
      "grad_norm": 0.04012243583380766,
      "learning_rate": 9.89163923770242e-10,
      "loss": 0.2795,
      "step": 3468
    },
    {
      "epoch": 4.99136690647482,
      "grad_norm": 0.04165166345300335,
      "learning_rate": 7.26733473350727e-10,
      "loss": 0.2903,
      "step": 3469
    },
    {
      "epoch": 4.9928057553956835,
      "grad_norm": 0.04109840813348842,
      "learning_rate": 5.046764901095457e-10,
      "loss": 0.2836,
      "step": 3470
    },
    {
      "epoch": 4.994244604316547,
      "grad_norm": 0.042065480515719435,
      "learning_rate": 3.2299319817852283e-10,
      "loss": 0.2871,
      "step": 3471
    },
    {
      "epoch": 4.99568345323741,
      "grad_norm": 0.04048558566819859,
      "learning_rate": 1.816837809487382e-10,
      "loss": 0.2906,
      "step": 3472
    },
    {
      "epoch": 4.997122302158274,
      "grad_norm": 0.040932974287600316,
      "learning_rate": 8.074838104832338e-11,
      "loss": 0.2845,
      "step": 3473
    },
    {
      "epoch": 4.998561151079136,
      "grad_norm": 0.04198094321209325,
      "learning_rate": 2.0187100355784085e-11,
      "loss": 0.2817,
      "step": 3474
    },
    {
      "epoch": 5.0,
      "grad_norm": 0.040263433455370916,
      "learning_rate": 0.0,
      "loss": 0.277,
      "step": 3475
    },
    {
      "epoch": 5.0,
      "step": 3475,
      "total_flos": 1.957564489138176e+17,
      "train_loss": 0.0,
      "train_runtime": 5.6508,
      "train_samples_per_second": 314804.431,
      "train_steps_per_second": 614.958
    }
  ],
  "logging_steps": 1,
  "max_steps": 3475,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.957564489138176e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}