{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0,
  "eval_steps": 500,
  "global_step": 1124,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0017793594306049821,
      "grad_norm": 3.659052439038323,
      "learning_rate": 9.99998046979289e-06,
      "loss": 0.1395,
      "step": 1
    },
    {
      "epoch": 0.0035587188612099642,
      "grad_norm": 3.156711114499553,
      "learning_rate": 9.999921879324127e-06,
      "loss": 0.127,
      "step": 2
    },
    {
      "epoch": 0.005338078291814947,
      "grad_norm": 3.930350494055212,
      "learning_rate": 9.999824229051425e-06,
      "loss": 0.148,
      "step": 3
    },
    {
      "epoch": 0.0071174377224199285,
      "grad_norm": 4.801867648555069,
      "learning_rate": 9.999687519737639e-06,
      "loss": 0.1847,
      "step": 4
    },
    {
      "epoch": 0.008896797153024912,
      "grad_norm": 4.0035033214905,
      "learning_rate": 9.99951175245075e-06,
      "loss": 0.1283,
      "step": 5
    },
    {
      "epoch": 0.010676156583629894,
      "grad_norm": 4.940444416887165,
      "learning_rate": 9.999296928563868e-06,
      "loss": 0.1844,
      "step": 6
    },
    {
      "epoch": 0.012455516014234875,
      "grad_norm": 5.1085362963384044,
      "learning_rate": 9.999043049755216e-06,
      "loss": 0.1776,
      "step": 7
    },
    {
      "epoch": 0.014234875444839857,
      "grad_norm": 3.968188071034837,
      "learning_rate": 9.998750118008117e-06,
      "loss": 0.1126,
      "step": 8
    },
    {
      "epoch": 0.01601423487544484,
      "grad_norm": 4.270505103398457,
      "learning_rate": 9.998418135610974e-06,
      "loss": 0.1284,
      "step": 9
    },
    {
      "epoch": 0.017793594306049824,
      "grad_norm": 5.262056525757216,
      "learning_rate": 9.998047105157265e-06,
      "loss": 0.2256,
      "step": 10
    },
    {
      "epoch": 0.019572953736654804,
      "grad_norm": 3.892181050424344,
      "learning_rate": 9.997637029545509e-06,
      "loss": 0.1172,
      "step": 11
    },
    {
      "epoch": 0.021352313167259787,
      "grad_norm": 5.176536871669479,
      "learning_rate": 9.997187911979252e-06,
      "loss": 0.1609,
      "step": 12
    },
    {
      "epoch": 0.023131672597864767,
      "grad_norm": 4.6071020502541735,
      "learning_rate": 9.996699755967035e-06,
      "loss": 0.1785,
      "step": 13
    },
    {
      "epoch": 0.02491103202846975,
      "grad_norm": 4.2253441862620384,
      "learning_rate": 9.996172565322375e-06,
      "loss": 0.154,
      "step": 14
    },
    {
      "epoch": 0.026690391459074734,
      "grad_norm": 3.9913727746663663,
      "learning_rate": 9.995606344163728e-06,
      "loss": 0.1451,
      "step": 15
    },
    {
      "epoch": 0.028469750889679714,
      "grad_norm": 3.9074159184180113,
      "learning_rate": 9.995001096914462e-06,
      "loss": 0.1517,
      "step": 16
    },
    {
      "epoch": 0.030249110320284697,
      "grad_norm": 3.9154224170375116,
      "learning_rate": 9.994356828302818e-06,
      "loss": 0.1518,
      "step": 17
    },
    {
      "epoch": 0.03202846975088968,
      "grad_norm": 3.3347796920928547,
      "learning_rate": 9.993673543361874e-06,
      "loss": 0.1304,
      "step": 18
    },
    {
      "epoch": 0.033807829181494664,
      "grad_norm": 5.521081822691506,
      "learning_rate": 9.992951247429512e-06,
      "loss": 0.2166,
      "step": 19
    },
    {
      "epoch": 0.03558718861209965,
      "grad_norm": 3.8539113680908903,
      "learning_rate": 9.992189946148366e-06,
      "loss": 0.1376,
      "step": 20
    },
    {
      "epoch": 0.037366548042704624,
      "grad_norm": 3.6184281112346075,
      "learning_rate": 9.991389645465786e-06,
      "loss": 0.1143,
      "step": 21
    },
    {
      "epoch": 0.03914590747330961,
      "grad_norm": 3.961588411589733,
      "learning_rate": 9.990550351633784e-06,
      "loss": 0.1706,
      "step": 22
    },
    {
      "epoch": 0.04092526690391459,
      "grad_norm": 3.7061466090244646,
      "learning_rate": 9.989672071208993e-06,
      "loss": 0.1807,
      "step": 23
    },
    {
      "epoch": 0.042704626334519574,
      "grad_norm": 3.9713068677113523,
      "learning_rate": 9.988754811052616e-06,
      "loss": 0.2095,
      "step": 24
    },
    {
      "epoch": 0.04448398576512456,
      "grad_norm": 4.53608729930377,
      "learning_rate": 9.987798578330365e-06,
      "loss": 0.2067,
      "step": 25
    },
    {
      "epoch": 0.046263345195729534,
      "grad_norm": 4.061149026316582,
      "learning_rate": 9.986803380512406e-06,
      "loss": 0.1721,
      "step": 26
    },
    {
      "epoch": 0.04804270462633452,
      "grad_norm": 3.554504043593526,
      "learning_rate": 9.98576922537331e-06,
      "loss": 0.185,
      "step": 27
    },
    {
      "epoch": 0.0498220640569395,
      "grad_norm": 3.9306221234350383,
      "learning_rate": 9.984696120991979e-06,
      "loss": 0.1872,
      "step": 28
    },
    {
      "epoch": 0.051601423487544484,
      "grad_norm": 3.5987380444505024,
      "learning_rate": 9.983584075751598e-06,
      "loss": 0.1269,
      "step": 29
    },
    {
      "epoch": 0.05338078291814947,
      "grad_norm": 3.4170135810678812,
      "learning_rate": 9.982433098339553e-06,
      "loss": 0.1599,
      "step": 30
    },
    {
      "epoch": 0.05516014234875445,
      "grad_norm": 4.577397109635813,
      "learning_rate": 9.981243197747375e-06,
      "loss": 0.2054,
      "step": 31
    },
    {
      "epoch": 0.05693950177935943,
      "grad_norm": 5.442783928083816,
      "learning_rate": 9.980014383270668e-06,
      "loss": 0.2276,
      "step": 32
    },
    {
      "epoch": 0.05871886120996441,
      "grad_norm": 4.181377492600741,
      "learning_rate": 9.978746664509032e-06,
      "loss": 0.1682,
      "step": 33
    },
    {
      "epoch": 0.060498220640569395,
      "grad_norm": 3.5320993396989495,
      "learning_rate": 9.97744005136599e-06,
      "loss": 0.1395,
      "step": 34
    },
    {
      "epoch": 0.06227758007117438,
      "grad_norm": 4.012129883139223,
      "learning_rate": 9.976094554048912e-06,
      "loss": 0.177,
      "step": 35
    },
    {
      "epoch": 0.06405693950177936,
      "grad_norm": 4.395681687867508,
      "learning_rate": 9.974710183068935e-06,
      "loss": 0.15,
      "step": 36
    },
    {
      "epoch": 0.06583629893238434,
      "grad_norm": 3.5625834777936776,
      "learning_rate": 9.97328694924088e-06,
      "loss": 0.1306,
      "step": 37
    },
    {
      "epoch": 0.06761565836298933,
      "grad_norm": 3.5550987486875063,
      "learning_rate": 9.971824863683168e-06,
      "loss": 0.1798,
      "step": 38
    },
    {
      "epoch": 0.0693950177935943,
      "grad_norm": 4.418061273114013,
      "learning_rate": 9.970323937817732e-06,
      "loss": 0.1533,
      "step": 39
    },
    {
      "epoch": 0.0711743772241993,
      "grad_norm": 4.037961914302329,
      "learning_rate": 9.968784183369929e-06,
      "loss": 0.1321,
      "step": 40
    },
    {
      "epoch": 0.07295373665480427,
      "grad_norm": 4.154779423227439,
      "learning_rate": 9.96720561236845e-06,
      "loss": 0.1672,
      "step": 41
    },
    {
      "epoch": 0.07473309608540925,
      "grad_norm": 4.908346755461008,
      "learning_rate": 9.965588237145219e-06,
      "loss": 0.1914,
      "step": 42
    },
    {
      "epoch": 0.07651245551601424,
      "grad_norm": 3.992151127697039,
      "learning_rate": 9.963932070335307e-06,
      "loss": 0.1504,
      "step": 43
    },
    {
      "epoch": 0.07829181494661921,
      "grad_norm": 4.242733466418419,
      "learning_rate": 9.962237124876828e-06,
      "loss": 0.171,
      "step": 44
    },
    {
      "epoch": 0.0800711743772242,
      "grad_norm": 3.5848511138833006,
      "learning_rate": 9.960503414010833e-06,
      "loss": 0.1608,
      "step": 45
    },
    {
      "epoch": 0.08185053380782918,
      "grad_norm": 3.9562949864572534,
      "learning_rate": 9.958730951281218e-06,
      "loss": 0.1526,
      "step": 46
    },
    {
      "epoch": 0.08362989323843416,
      "grad_norm": 4.18917702446439,
      "learning_rate": 9.956919750534607e-06,
      "loss": 0.1754,
      "step": 47
    },
    {
      "epoch": 0.08540925266903915,
      "grad_norm": 4.111553491872384,
      "learning_rate": 9.955069825920249e-06,
      "loss": 0.1982,
      "step": 48
    },
    {
      "epoch": 0.08718861209964412,
      "grad_norm": 3.727474294093174,
      "learning_rate": 9.953181191889913e-06,
      "loss": 0.1771,
      "step": 49
    },
    {
      "epoch": 0.08896797153024912,
      "grad_norm": 3.7563018909451507,
      "learning_rate": 9.95125386319776e-06,
      "loss": 0.1333,
      "step": 50
    },
    {
      "epoch": 0.09074733096085409,
      "grad_norm": 4.909566466886204,
      "learning_rate": 9.949287854900243e-06,
      "loss": 0.1927,
      "step": 51
    },
    {
      "epoch": 0.09252669039145907,
      "grad_norm": 4.41606204052901,
      "learning_rate": 9.947283182355982e-06,
      "loss": 0.1688,
      "step": 52
    },
    {
      "epoch": 0.09430604982206406,
      "grad_norm": 4.578611514610308,
      "learning_rate": 9.945239861225644e-06,
      "loss": 0.1622,
      "step": 53
    },
    {
      "epoch": 0.09608540925266904,
      "grad_norm": 3.574252192836812,
      "learning_rate": 9.943157907471825e-06,
      "loss": 0.1579,
      "step": 54
    },
    {
      "epoch": 0.09786476868327403,
      "grad_norm": 3.3029452048900074,
      "learning_rate": 9.941037337358918e-06,
      "loss": 0.1229,
      "step": 55
    },
    {
      "epoch": 0.099644128113879,
      "grad_norm": 4.536603474008181,
      "learning_rate": 9.938878167452991e-06,
      "loss": 0.1872,
      "step": 56
    },
    {
      "epoch": 0.10142348754448399,
      "grad_norm": 4.777651054375178,
      "learning_rate": 9.936680414621663e-06,
      "loss": 0.184,
      "step": 57
    },
    {
      "epoch": 0.10320284697508897,
      "grad_norm": 3.521518078817995,
      "learning_rate": 9.934444096033958e-06,
      "loss": 0.1386,
      "step": 58
    },
    {
      "epoch": 0.10498220640569395,
      "grad_norm": 3.8841235167417585,
      "learning_rate": 9.932169229160183e-06,
      "loss": 0.1843,
      "step": 59
    },
    {
      "epoch": 0.10676156583629894,
      "grad_norm": 4.047249043142044,
      "learning_rate": 9.929855831771787e-06,
      "loss": 0.1466,
      "step": 60
    },
    {
      "epoch": 0.10854092526690391,
      "grad_norm": 4.052309043583553,
      "learning_rate": 9.927503921941218e-06,
      "loss": 0.1719,
      "step": 61
    },
    {
      "epoch": 0.1103202846975089,
      "grad_norm": 4.087293673316626,
      "learning_rate": 9.925113518041796e-06,
      "loss": 0.1947,
      "step": 62
    },
    {
      "epoch": 0.11209964412811388,
      "grad_norm": 4.728650656417226,
      "learning_rate": 9.922684638747551e-06,
      "loss": 0.2133,
      "step": 63
    },
    {
      "epoch": 0.11387900355871886,
      "grad_norm": 3.5348328493850545,
      "learning_rate": 9.920217303033091e-06,
      "loss": 0.1444,
      "step": 64
    },
    {
      "epoch": 0.11565836298932385,
      "grad_norm": 4.398687733465476,
      "learning_rate": 9.917711530173444e-06,
      "loss": 0.1763,
      "step": 65
    },
    {
      "epoch": 0.11743772241992882,
      "grad_norm": 4.797177913349625,
      "learning_rate": 9.91516733974392e-06,
      "loss": 0.1925,
      "step": 66
    },
    {
      "epoch": 0.11921708185053381,
      "grad_norm": 3.5287715103923554,
      "learning_rate": 9.912584751619943e-06,
      "loss": 0.1675,
      "step": 67
    },
    {
      "epoch": 0.12099644128113879,
      "grad_norm": 3.2542842238681873,
      "learning_rate": 9.909963785976902e-06,
      "loss": 0.1519,
      "step": 68
    },
    {
      "epoch": 0.12277580071174377,
      "grad_norm": 3.7432006322982745,
      "learning_rate": 9.907304463290004e-06,
      "loss": 0.1721,
      "step": 69
    },
    {
      "epoch": 0.12455516014234876,
      "grad_norm": 3.4138912401341432,
      "learning_rate": 9.904606804334094e-06,
      "loss": 0.1346,
      "step": 70
    },
    {
      "epoch": 0.12633451957295375,
      "grad_norm": 3.6298052626218715,
      "learning_rate": 9.901870830183506e-06,
      "loss": 0.1478,
      "step": 71
    },
    {
      "epoch": 0.12811387900355872,
      "grad_norm": 3.5868557899174607,
      "learning_rate": 9.899096562211902e-06,
      "loss": 0.157,
      "step": 72
    },
    {
      "epoch": 0.1298932384341637,
      "grad_norm": 3.9509423727957333,
      "learning_rate": 9.896284022092088e-06,
      "loss": 0.1952,
      "step": 73
    },
    {
      "epoch": 0.13167259786476868,
      "grad_norm": 4.558671095802092,
      "learning_rate": 9.893433231795864e-06,
      "loss": 0.2317,
      "step": 74
    },
    {
      "epoch": 0.13345195729537365,
      "grad_norm": 4.700403845296946,
      "learning_rate": 9.890544213593838e-06,
      "loss": 0.209,
      "step": 75
    },
    {
      "epoch": 0.13523131672597866,
      "grad_norm": 3.627872066349445,
      "learning_rate": 9.887616990055262e-06,
      "loss": 0.1767,
      "step": 76
    },
    {
      "epoch": 0.13701067615658363,
      "grad_norm": 4.151947277093625,
      "learning_rate": 9.884651584047845e-06,
      "loss": 0.1773,
      "step": 77
    },
    {
      "epoch": 0.1387900355871886,
      "grad_norm": 3.942127140232508,
      "learning_rate": 9.881648018737587e-06,
      "loss": 0.1893,
      "step": 78
    },
    {
      "epoch": 0.14056939501779359,
      "grad_norm": 3.8110333750937504,
      "learning_rate": 9.878606317588588e-06,
      "loss": 0.1375,
      "step": 79
    },
    {
      "epoch": 0.1423487544483986,
      "grad_norm": 4.867445068236545,
      "learning_rate": 9.875526504362868e-06,
      "loss": 0.2156,
      "step": 80
    },
    {
      "epoch": 0.14412811387900357,
      "grad_norm": 4.305335815555553,
      "learning_rate": 9.872408603120187e-06,
      "loss": 0.2112,
      "step": 81
    },
    {
      "epoch": 0.14590747330960854,
      "grad_norm": 4.0546557970083645,
      "learning_rate": 9.869252638217846e-06,
      "loss": 0.1511,
      "step": 82
    },
    {
      "epoch": 0.14768683274021352,
      "grad_norm": 3.9386939528966685,
      "learning_rate": 9.866058634310503e-06,
      "loss": 0.1848,
      "step": 83
    },
    {
      "epoch": 0.1494661921708185,
      "grad_norm": 3.8741256658287204,
      "learning_rate": 9.862826616349981e-06,
      "loss": 0.174,
      "step": 84
    },
    {
      "epoch": 0.1512455516014235,
      "grad_norm": 4.433973189246578,
      "learning_rate": 9.859556609585075e-06,
      "loss": 0.18,
      "step": 85
    },
    {
      "epoch": 0.15302491103202848,
      "grad_norm": 3.793165996553822,
      "learning_rate": 9.856248639561346e-06,
      "loss": 0.1485,
      "step": 86
    },
    {
      "epoch": 0.15480427046263345,
      "grad_norm": 4.287227414713171,
      "learning_rate": 9.85290273212093e-06,
      "loss": 0.1511,
      "step": 87
    },
    {
      "epoch": 0.15658362989323843,
      "grad_norm": 3.9467717443082395,
      "learning_rate": 9.849518913402334e-06,
      "loss": 0.1704,
      "step": 88
    },
    {
      "epoch": 0.1583629893238434,
      "grad_norm": 4.100496614891011,
      "learning_rate": 9.84609720984023e-06,
      "loss": 0.1702,
      "step": 89
    },
    {
      "epoch": 0.1601423487544484,
      "grad_norm": 4.0978584262126505,
      "learning_rate": 9.84263764816525e-06,
      "loss": 0.1704,
      "step": 90
    },
    {
      "epoch": 0.1619217081850534,
      "grad_norm": 3.593604061443781,
      "learning_rate": 9.839140255403776e-06,
      "loss": 0.1427,
      "step": 91
    },
    {
      "epoch": 0.16370106761565836,
      "grad_norm": 3.808029517347225,
      "learning_rate": 9.83560505887773e-06,
      "loss": 0.1754,
      "step": 92
    },
    {
      "epoch": 0.16548042704626334,
      "grad_norm": 3.308728113109198,
      "learning_rate": 9.83203208620436e-06,
      "loss": 0.1609,
      "step": 93
    },
    {
      "epoch": 0.16725978647686832,
      "grad_norm": 3.8252995794626887,
      "learning_rate": 9.828421365296023e-06,
      "loss": 0.1499,
      "step": 94
    },
    {
      "epoch": 0.16903914590747332,
      "grad_norm": 4.582910332194453,
      "learning_rate": 9.824772924359974e-06,
      "loss": 0.1875,
      "step": 95
    },
    {
      "epoch": 0.1708185053380783,
      "grad_norm": 3.687995356758649,
      "learning_rate": 9.821086791898133e-06,
      "loss": 0.1596,
      "step": 96
    },
    {
      "epoch": 0.17259786476868327,
      "grad_norm": 3.1911287290104853,
      "learning_rate": 9.817362996706872e-06,
      "loss": 0.1677,
      "step": 97
    },
    {
      "epoch": 0.17437722419928825,
      "grad_norm": 3.203499629091107,
      "learning_rate": 9.81360156787679e-06,
      "loss": 0.1344,
      "step": 98
    },
    {
      "epoch": 0.17615658362989323,
      "grad_norm": 3.513310500572757,
      "learning_rate": 9.809802534792477e-06,
      "loss": 0.16,
      "step": 99
    },
    {
      "epoch": 0.17793594306049823,
      "grad_norm": 3.975942212632882,
      "learning_rate": 9.805965927132294e-06,
      "loss": 0.1623,
      "step": 100
    },
    {
      "epoch": 0.1797153024911032,
      "grad_norm": 3.5457919046585613,
      "learning_rate": 9.802091774868143e-06,
      "loss": 0.1584,
      "step": 101
    },
    {
      "epoch": 0.18149466192170818,
      "grad_norm": 3.3421220165769894,
      "learning_rate": 9.798180108265218e-06,
      "loss": 0.1729,
      "step": 102
    },
    {
      "epoch": 0.18327402135231316,
      "grad_norm": 3.293975007203058,
      "learning_rate": 9.794230957881785e-06,
      "loss": 0.1245,
      "step": 103
    },
    {
      "epoch": 0.18505338078291814,
      "grad_norm": 4.125548461025262,
      "learning_rate": 9.79024435456893e-06,
      "loss": 0.1583,
      "step": 104
    },
    {
      "epoch": 0.18683274021352314,
      "grad_norm": 4.498953599221692,
      "learning_rate": 9.786220329470334e-06,
      "loss": 0.1648,
      "step": 105
    },
    {
      "epoch": 0.18861209964412812,
      "grad_norm": 3.616502808564312,
      "learning_rate": 9.782158914022011e-06,
      "loss": 0.1513,
      "step": 106
    },
    {
      "epoch": 0.1903914590747331,
      "grad_norm": 3.9457816847275144,
      "learning_rate": 9.778060139952075e-06,
      "loss": 0.2045,
      "step": 107
    },
    {
      "epoch": 0.19217081850533807,
      "grad_norm": 4.001094543504865,
      "learning_rate": 9.773924039280488e-06,
      "loss": 0.1593,
      "step": 108
    },
    {
      "epoch": 0.19395017793594305,
      "grad_norm": 3.086146381640191,
      "learning_rate": 9.769750644318814e-06,
      "loss": 0.1149,
      "step": 109
    },
    {
      "epoch": 0.19572953736654805,
      "grad_norm": 4.267149736315256,
      "learning_rate": 9.765539987669956e-06,
      "loss": 0.1614,
      "step": 110
    },
    {
      "epoch": 0.19750889679715303,
      "grad_norm": 4.169799931459572,
      "learning_rate": 9.761292102227917e-06,
      "loss": 0.2129,
      "step": 111
    },
    {
      "epoch": 0.199288256227758,
      "grad_norm": 4.91528431021248,
      "learning_rate": 9.757007021177529e-06,
      "loss": 0.2125,
      "step": 112
    },
    {
      "epoch": 0.20106761565836298,
      "grad_norm": 3.043554193107207,
      "learning_rate": 9.752684777994197e-06,
      "loss": 0.1368,
      "step": 113
    },
    {
      "epoch": 0.20284697508896798,
      "grad_norm": 4.126023218016504,
      "learning_rate": 9.748325406443647e-06,
      "loss": 0.1724,
      "step": 114
    },
    {
      "epoch": 0.20462633451957296,
      "grad_norm": 4.302837464486555,
      "learning_rate": 9.743928940581646e-06,
      "loss": 0.1938,
      "step": 115
    },
    {
      "epoch": 0.20640569395017794,
      "grad_norm": 5.636305845984065,
      "learning_rate": 9.739495414753754e-06,
      "loss": 0.2155,
      "step": 116
    },
    {
      "epoch": 0.20818505338078291,
      "grad_norm": 3.8112884504775697,
      "learning_rate": 9.73502486359504e-06,
      "loss": 0.1608,
      "step": 117
    },
    {
      "epoch": 0.2099644128113879,
      "grad_norm": 4.461914127700233,
      "learning_rate": 9.73051732202982e-06,
      "loss": 0.1679,
      "step": 118
    },
    {
      "epoch": 0.2117437722419929,
      "grad_norm": 3.680580973224424,
      "learning_rate": 9.725972825271381e-06,
      "loss": 0.1511,
      "step": 119
    },
    {
      "epoch": 0.21352313167259787,
      "grad_norm": 3.794951381555653,
      "learning_rate": 9.721391408821713e-06,
      "loss": 0.1671,
      "step": 120
    },
    {
      "epoch": 0.21530249110320285,
      "grad_norm": 3.6930449358039117,
      "learning_rate": 9.716773108471213e-06,
      "loss": 0.1924,
      "step": 121
    },
    {
      "epoch": 0.21708185053380782,
      "grad_norm": 3.4713630701138696,
      "learning_rate": 9.712117960298433e-06,
      "loss": 0.1687,
      "step": 122
    },
    {
      "epoch": 0.2188612099644128,
      "grad_norm": 4.0364688447877795,
      "learning_rate": 9.707426000669773e-06,
      "loss": 0.1619,
      "step": 123
    },
    {
      "epoch": 0.2206405693950178,
      "grad_norm": 3.8784844057350614,
      "learning_rate": 9.702697266239211e-06,
      "loss": 0.1671,
      "step": 124
    },
    {
      "epoch": 0.22241992882562278,
      "grad_norm": 4.256677926035706,
      "learning_rate": 9.697931793948012e-06,
      "loss": 0.2094,
      "step": 125
    },
    {
      "epoch": 0.22419928825622776,
      "grad_norm": 3.2930225361958083,
      "learning_rate": 9.693129621024441e-06,
      "loss": 0.1599,
      "step": 126
    },
    {
      "epoch": 0.22597864768683273,
      "grad_norm": 3.3618066362558023,
      "learning_rate": 9.68829078498347e-06,
      "loss": 0.1545,
      "step": 127
    },
    {
      "epoch": 0.2277580071174377,
      "grad_norm": 4.340818349134813,
      "learning_rate": 9.683415323626487e-06,
      "loss": 0.1643,
      "step": 128
    },
    {
      "epoch": 0.22953736654804271,
      "grad_norm": 5.0198581689280415,
      "learning_rate": 9.678503275040997e-06,
      "loss": 0.2027,
      "step": 129
    },
    {
      "epoch": 0.2313167259786477,
      "grad_norm": 3.914040767380737,
      "learning_rate": 9.673554677600336e-06,
      "loss": 0.1679,
      "step": 130
    },
    {
      "epoch": 0.23309608540925267,
      "grad_norm": 3.4827884526902504,
      "learning_rate": 9.668569569963355e-06,
      "loss": 0.1775,
      "step": 131
    },
    {
      "epoch": 0.23487544483985764,
      "grad_norm": 2.8129613867967227,
      "learning_rate": 9.663547991074129e-06,
      "loss": 0.1203,
      "step": 132
    },
    {
      "epoch": 0.23665480427046262,
      "grad_norm": 4.172629971974784,
      "learning_rate": 9.658489980161643e-06,
      "loss": 0.1673,
      "step": 133
    },
    {
      "epoch": 0.23843416370106763,
      "grad_norm": 3.824248891189838,
      "learning_rate": 9.653395576739504e-06,
      "loss": 0.1821,
      "step": 134
    },
    {
      "epoch": 0.2402135231316726,
      "grad_norm": 4.417500481336389,
      "learning_rate": 9.648264820605611e-06,
      "loss": 0.1892,
      "step": 135
    },
    {
      "epoch": 0.24199288256227758,
      "grad_norm": 3.6227047407521726,
      "learning_rate": 9.643097751841854e-06,
      "loss": 0.2354,
      "step": 136
    },
    {
      "epoch": 0.24377224199288255,
      "grad_norm": 3.54295751103917,
      "learning_rate": 9.637894410813803e-06,
      "loss": 0.1838,
      "step": 137
    },
    {
      "epoch": 0.24555160142348753,
      "grad_norm": 4.335552255087848,
      "learning_rate": 9.632654838170393e-06,
      "loss": 0.1967,
      "step": 138
    },
    {
      "epoch": 0.24733096085409254,
      "grad_norm": 3.9710169233858874,
      "learning_rate": 9.627379074843595e-06,
      "loss": 0.1863,
      "step": 139
    },
    {
      "epoch": 0.2491103202846975,
      "grad_norm": 4.022741500947156,
      "learning_rate": 9.622067162048111e-06,
      "loss": 0.2014,
      "step": 140
    },
    {
      "epoch": 0.2508896797153025,
      "grad_norm": 3.140561056840528,
      "learning_rate": 9.616719141281044e-06,
      "loss": 0.1776,
      "step": 141
    },
    {
      "epoch": 0.2526690391459075,
      "grad_norm": 4.293850861529394,
      "learning_rate": 9.611335054321576e-06,
      "loss": 0.2285,
      "step": 142
    },
    {
      "epoch": 0.25444839857651247,
      "grad_norm": 4.057362498464448,
      "learning_rate": 9.605914943230637e-06,
      "loss": 0.1742,
      "step": 143
    },
    {
      "epoch": 0.25622775800711745,
      "grad_norm": 3.3759874584472582,
      "learning_rate": 9.600458850350588e-06,
      "loss": 0.1586,
      "step": 144
    },
    {
      "epoch": 0.2580071174377224,
      "grad_norm": 3.9841832631602707,
      "learning_rate": 9.594966818304875e-06,
      "loss": 0.1827,
      "step": 145
    },
    {
      "epoch": 0.2597864768683274,
      "grad_norm": 3.660802460402094,
      "learning_rate": 9.589438889997712e-06,
      "loss": 0.1404,
      "step": 146
    },
    {
      "epoch": 0.2615658362989324,
      "grad_norm": 3.374959737383286,
      "learning_rate": 9.583875108613727e-06,
      "loss": 0.1389,
      "step": 147
    },
    {
      "epoch": 0.26334519572953735,
      "grad_norm": 3.696022886243752,
      "learning_rate": 9.578275517617646e-06,
      "loss": 0.1737,
      "step": 148
    },
    {
      "epoch": 0.26512455516014233,
      "grad_norm": 3.74295849426895,
      "learning_rate": 9.572640160753936e-06,
      "loss": 0.1634,
      "step": 149
    },
    {
      "epoch": 0.2669039145907473,
      "grad_norm": 3.763773616497655,
      "learning_rate": 9.566969082046471e-06,
      "loss": 0.1791,
      "step": 150
    },
    {
      "epoch": 0.26868327402135234,
      "grad_norm": 2.700019269266813,
      "learning_rate": 9.561262325798188e-06,
      "loss": 0.132,
      "step": 151
    },
    {
      "epoch": 0.2704626334519573,
      "grad_norm": 3.2717199294998243,
      "learning_rate": 9.555519936590739e-06,
      "loss": 0.143,
      "step": 152
    },
    {
      "epoch": 0.2722419928825623,
      "grad_norm": 3.333714288905169,
      "learning_rate": 9.549741959284147e-06,
      "loss": 0.1414,
      "step": 153
    },
    {
      "epoch": 0.27402135231316727,
      "grad_norm": 4.233864534892011,
      "learning_rate": 9.543928439016445e-06,
      "loss": 0.1688,
      "step": 154
    },
    {
      "epoch": 0.27580071174377224,
      "grad_norm": 3.671147620590123,
      "learning_rate": 9.538079421203339e-06,
      "loss": 0.1669,
      "step": 155
    },
    {
      "epoch": 0.2775800711743772,
      "grad_norm": 3.2549808511581064,
      "learning_rate": 9.532194951537838e-06,
      "loss": 0.145,
      "step": 156
    },
    {
      "epoch": 0.2793594306049822,
      "grad_norm": 3.987585254370317,
      "learning_rate": 9.52627507598991e-06,
      "loss": 0.1628,
      "step": 157
    },
    {
      "epoch": 0.28113879003558717,
      "grad_norm": 3.2298657362857637,
      "learning_rate": 9.52031984080611e-06,
      "loss": 0.14,
      "step": 158
    },
    {
      "epoch": 0.28291814946619215,
      "grad_norm": 3.7023777000508336,
      "learning_rate": 9.514329292509227e-06,
      "loss": 0.1436,
      "step": 159
    },
    {
      "epoch": 0.2846975088967972,
      "grad_norm": 3.597341029393787,
      "learning_rate": 9.508303477897925e-06,
      "loss": 0.1538,
      "step": 160
    },
    {
      "epoch": 0.28647686832740216,
      "grad_norm": 4.004735854877035,
      "learning_rate": 9.502242444046365e-06,
      "loss": 0.1647,
      "step": 161
    },
    {
      "epoch": 0.28825622775800713,
      "grad_norm": 4.0348294023927105,
      "learning_rate": 9.496146238303846e-06,
      "loss": 0.199,
      "step": 162
    },
    {
      "epoch": 0.2900355871886121,
      "grad_norm": 4.140681868395475,
      "learning_rate": 9.49001490829443e-06,
      "loss": 0.1691,
      "step": 163
    },
    {
      "epoch": 0.2918149466192171,
      "grad_norm": 4.4249376276217625,
      "learning_rate": 9.483848501916578e-06,
      "loss": 0.2474,
      "step": 164
    },
    {
      "epoch": 0.29359430604982206,
      "grad_norm": 3.663900826422634,
      "learning_rate": 9.477647067342766e-06,
      "loss": 0.1824,
      "step": 165
    },
    {
      "epoch": 0.29537366548042704,
      "grad_norm": 3.952257574972633,
      "learning_rate": 9.471410653019115e-06,
      "loss": 0.1801,
      "step": 166
    },
    {
      "epoch": 0.297153024911032,
      "grad_norm": 3.9694081027604873,
      "learning_rate": 9.46513930766501e-06,
      "loss": 0.1957,
      "step": 167
    },
    {
      "epoch": 0.298932384341637,
      "grad_norm": 4.0088155669308465,
      "learning_rate": 9.458833080272723e-06,
      "loss": 0.1659,
      "step": 168
    },
    {
      "epoch": 0.30071174377224197,
      "grad_norm": 3.9228366445742906,
      "learning_rate": 9.45249202010702e-06,
      "loss": 0.1631,
      "step": 169
    },
    {
      "epoch": 0.302491103202847,
      "grad_norm": 3.736197203647476,
      "learning_rate": 9.446116176704791e-06,
      "loss": 0.1614,
      "step": 170
    },
    {
      "epoch": 0.304270462633452,
      "grad_norm": 3.9995109577405836,
      "learning_rate": 9.439705599874653e-06,
      "loss": 0.1786,
      "step": 171
    },
    {
      "epoch": 0.30604982206405695,
      "grad_norm": 4.118868954439227,
      "learning_rate": 9.433260339696564e-06,
      "loss": 0.2157,
      "step": 172
    },
    {
      "epoch": 0.30782918149466193,
      "grad_norm": 3.844433269829606,
      "learning_rate": 9.426780446521429e-06,
      "loss": 0.2011,
      "step": 173
    },
    {
      "epoch": 0.3096085409252669,
      "grad_norm": 3.42744859461912,
      "learning_rate": 9.42026597097071e-06,
      "loss": 0.1855,
      "step": 174
    },
    {
      "epoch": 0.3113879003558719,
      "grad_norm": 3.1067491004161436,
      "learning_rate": 9.413716963936033e-06,
      "loss": 0.1453,
      "step": 175
    },
    {
      "epoch": 0.31316725978647686,
      "grad_norm": 3.2722150805677543,
      "learning_rate": 9.407133476578778e-06,
      "loss": 0.1648,
      "step": 176
    },
    {
      "epoch": 0.31494661921708184,
      "grad_norm": 4.249523858220171,
      "learning_rate": 9.400515560329698e-06,
      "loss": 0.2154,
      "step": 177
    },
    {
      "epoch": 0.3167259786476868,
      "grad_norm": 3.55759405515388,
      "learning_rate": 9.393863266888501e-06,
      "loss": 0.1582,
      "step": 178
    },
    {
      "epoch": 0.3185053380782918,
      "grad_norm": 4.000867106344722,
      "learning_rate": 9.387176648223457e-06,
      "loss": 0.1808,
      "step": 179
    },
    {
      "epoch": 0.3202846975088968,
      "grad_norm": 4.177732769746734,
      "learning_rate": 9.38045575657098e-06,
      "loss": 0.1855,
      "step": 180
    },
    {
      "epoch": 0.3220640569395018,
      "grad_norm": 3.8140222648879334,
      "learning_rate": 9.37370064443524e-06,
      "loss": 0.173,
      "step": 181
    },
    {
      "epoch": 0.3238434163701068,
      "grad_norm": 3.0606081710512156,
      "learning_rate": 9.366911364587726e-06,
      "loss": 0.1485,
      "step": 182
    },
    {
      "epoch": 0.32562277580071175,
      "grad_norm": 3.315736641345634,
      "learning_rate": 9.360087970066854e-06,
      "loss": 0.1481,
      "step": 183
    },
    {
      "epoch": 0.3274021352313167,
      "grad_norm": 3.6743859642969103,
      "learning_rate": 9.353230514177553e-06,
      "loss": 0.1408,
      "step": 184
    },
    {
      "epoch": 0.3291814946619217,
      "grad_norm": 3.6259486838756847,
      "learning_rate": 9.346339050490832e-06,
      "loss": 0.1435,
      "step": 185
    },
    {
      "epoch": 0.3309608540925267,
      "grad_norm": 4.4770908051119385,
      "learning_rate": 9.33941363284338e-06,
      "loss": 0.164,
      "step": 186
    },
    {
      "epoch": 0.33274021352313166,
      "grad_norm": 3.626597452238082,
      "learning_rate": 9.332454315337129e-06,
      "loss": 0.1519,
      "step": 187
    },
    {
      "epoch": 0.33451957295373663,
      "grad_norm": 3.9533588393691463,
      "learning_rate": 9.325461152338846e-06,
      "loss": 0.1735,
      "step": 188
    },
    {
      "epoch": 0.33629893238434166,
      "grad_norm": 3.9911050465867857,
      "learning_rate": 9.3184341984797e-06,
      "loss": 0.1616,
      "step": 189
    },
    {
      "epoch": 0.33807829181494664,
      "grad_norm": 4.430623771429184,
      "learning_rate": 9.311373508654838e-06,
      "loss": 0.1723,
      "step": 190
    },
    {
      "epoch": 0.3398576512455516,
      "grad_norm": 4.941250956984447,
      "learning_rate": 9.30427913802295e-06,
      "loss": 0.2389,
      "step": 191
    },
    {
      "epoch": 0.3416370106761566,
      "grad_norm": 4.2278781357586706,
      "learning_rate": 9.297151142005852e-06,
      "loss": 0.1677,
      "step": 192
    },
    {
      "epoch": 0.34341637010676157,
      "grad_norm": 3.7126756606225078,
      "learning_rate": 9.289989576288035e-06,
      "loss": 0.159,
      "step": 193
    },
    {
      "epoch": 0.34519572953736655,
      "grad_norm": 3.9897114128702387,
      "learning_rate": 9.282794496816244e-06,
      "loss": 0.189,
      "step": 194
    },
    {
      "epoch": 0.3469750889679715,
      "grad_norm": 4.282792579104738,
      "learning_rate": 9.27556595979904e-06,
      "loss": 0.1546,
      "step": 195
    },
    {
      "epoch": 0.3487544483985765,
      "grad_norm": 3.696261001510593,
      "learning_rate": 9.26830402170635e-06,
      "loss": 0.1615,
      "step": 196
    },
    {
      "epoch": 0.3505338078291815,
      "grad_norm": 3.7602490866549503,
      "learning_rate": 9.261008739269035e-06,
      "loss": 0.1479,
      "step": 197
    },
    {
      "epoch": 0.35231316725978645,
      "grad_norm": 4.120543259746711,
      "learning_rate": 9.253680169478448e-06,
      "loss": 0.1703,
      "step": 198
    },
    {
      "epoch": 0.3540925266903915,
      "grad_norm": 3.9172558657585514,
      "learning_rate": 9.246318369585983e-06,
      "loss": 0.1569,
      "step": 199
    },
    {
      "epoch": 0.35587188612099646,
      "grad_norm": 4.634059498170498,
      "learning_rate": 9.238923397102629e-06,
      "loss": 0.1969,
      "step": 200
    },
    {
      "epoch": 0.35765124555160144,
      "grad_norm": 3.3615947609632455,
      "learning_rate": 9.231495309798525e-06,
      "loss": 0.1378,
      "step": 201
    },
    {
      "epoch": 0.3594306049822064,
      "grad_norm": 3.927986834711984,
      "learning_rate": 9.224034165702506e-06,
      "loss": 0.1706,
      "step": 202
    },
    {
      "epoch": 0.3612099644128114,
      "grad_norm": 2.992300817138368,
      "learning_rate": 9.216540023101646e-06,
      "loss": 0.1443,
      "step": 203
    },
    {
      "epoch": 0.36298932384341637,
      "grad_norm": 3.851869967241809,
      "learning_rate": 9.209012940540806e-06,
      "loss": 0.1505,
      "step": 204
    },
    {
      "epoch": 0.36476868327402134,
      "grad_norm": 3.1520200711410333,
      "learning_rate": 9.20145297682218e-06,
      "loss": 0.1658,
      "step": 205
    },
    {
      "epoch": 0.3665480427046263,
      "grad_norm": 4.491137316292099,
      "learning_rate": 9.193860191004833e-06,
      "loss": 0.2251,
      "step": 206
    },
    {
      "epoch": 0.3683274021352313,
      "grad_norm": 3.7374495943847768,
      "learning_rate": 9.186234642404234e-06,
      "loss": 0.2004,
      "step": 207
    },
    {
      "epoch": 0.3701067615658363,
      "grad_norm": 4.419219501392304,
      "learning_rate": 9.178576390591803e-06,
      "loss": 0.1813,
      "step": 208
    },
    {
      "epoch": 0.3718861209964413,
      "grad_norm": 3.172109535453968,
      "learning_rate": 9.170885495394435e-06,
      "loss": 0.1396,
      "step": 209
    },
    {
      "epoch": 0.3736654804270463,
      "grad_norm": 4.367274863636535,
      "learning_rate": 9.16316201689404e-06,
      "loss": 0.1768,
      "step": 210
    },
    {
      "epoch": 0.37544483985765126,
      "grad_norm": 5.576343360549105,
      "learning_rate": 9.155406015427076e-06,
      "loss": 0.2042,
      "step": 211
    },
    {
      "epoch": 0.37722419928825623,
      "grad_norm": 4.291672329718272,
      "learning_rate": 9.147617551584066e-06,
      "loss": 0.1777,
      "step": 212
    },
    {
      "epoch": 0.3790035587188612,
      "grad_norm": 3.482564857299368,
      "learning_rate": 9.139796686209135e-06,
      "loss": 0.2029,
      "step": 213
    },
    {
      "epoch": 0.3807829181494662,
      "grad_norm": 3.5875612241397383,
      "learning_rate": 9.131943480399531e-06,
      "loss": 0.1532,
      "step": 214
    },
    {
      "epoch": 0.38256227758007116,
      "grad_norm": 3.649911450784079,
      "learning_rate": 9.124057995505148e-06,
      "loss": 0.157,
      "step": 215
    },
    {
      "epoch": 0.38434163701067614,
      "grad_norm": 4.9588610853184605,
      "learning_rate": 9.11614029312805e-06,
      "loss": 0.261,
      "step": 216
    },
    {
      "epoch": 0.3861209964412811,
      "grad_norm": 4.5055163965441025,
      "learning_rate": 9.108190435121982e-06,
      "loss": 0.1831,
      "step": 217
    },
    {
      "epoch": 0.3879003558718861,
      "grad_norm": 4.056825305977216,
      "learning_rate": 9.100208483591892e-06,
      "loss": 0.1615,
      "step": 218
    },
    {
      "epoch": 0.3896797153024911,
      "grad_norm": 5.090064402874391,
      "learning_rate": 9.092194500893448e-06,
      "loss": 0.2379,
      "step": 219
    },
    {
      "epoch": 0.3914590747330961,
      "grad_norm": 3.811134729933426,
      "learning_rate": 9.084148549632547e-06,
      "loss": 0.1619,
      "step": 220
    },
    {
      "epoch": 0.3932384341637011,
      "grad_norm": 3.152200682014577,
      "learning_rate": 9.076070692664827e-06,
      "loss": 0.1665,
      "step": 221
    },
    {
      "epoch": 0.39501779359430605,
      "grad_norm": 3.845770926880344,
      "learning_rate": 9.067960993095176e-06,
      "loss": 0.1789,
      "step": 222
    },
    {
      "epoch": 0.39679715302491103,
      "grad_norm": 3.8979319410623536,
      "learning_rate": 9.059819514277238e-06,
      "loss": 0.1664,
      "step": 223
    },
    {
      "epoch": 0.398576512455516,
      "grad_norm": 3.294838440921431,
      "learning_rate": 9.05164631981292e-06,
      "loss": 0.1572,
      "step": 224
    },
    {
      "epoch": 0.400355871886121,
      "grad_norm": 3.6030395831947213,
      "learning_rate": 9.043441473551893e-06,
      "loss": 0.1853,
      "step": 225
    },
    {
      "epoch": 0.40213523131672596,
      "grad_norm": 3.8703911138287026,
      "learning_rate": 9.035205039591099e-06,
      "loss": 0.1838,
      "step": 226
    },
    {
      "epoch": 0.40391459074733094,
      "grad_norm": 4.243788587748394,
      "learning_rate": 9.02693708227424e-06,
      "loss": 0.1772,
      "step": 227
    },
    {
      "epoch": 0.40569395017793597,
      "grad_norm": 3.2300470339992478,
      "learning_rate": 9.018637666191284e-06,
      "loss": 0.1989,
      "step": 228
    },
    {
      "epoch": 0.40747330960854095,
      "grad_norm": 3.7612800636313555,
      "learning_rate": 9.010306856177958e-06,
      "loss": 0.2153,
      "step": 229
    },
    {
      "epoch": 0.4092526690391459,
      "grad_norm": 4.246203172354286,
      "learning_rate": 9.001944717315236e-06,
      "loss": 0.2066,
      "step": 230
    },
    {
      "epoch": 0.4110320284697509,
      "grad_norm": 4.153583038587477,
      "learning_rate": 8.993551314928846e-06,
      "loss": 0.1861,
      "step": 231
    },
    {
      "epoch": 0.4128113879003559,
      "grad_norm": 3.3897759335432425,
      "learning_rate": 8.985126714588739e-06,
      "loss": 0.1223,
      "step": 232
    },
    {
      "epoch": 0.41459074733096085,
      "grad_norm": 3.8062875787900303,
      "learning_rate": 8.976670982108591e-06,
      "loss": 0.1744,
      "step": 233
    },
    {
      "epoch": 0.41637010676156583,
      "grad_norm": 4.5714894286310335,
      "learning_rate": 8.968184183545285e-06,
      "loss": 0.2338,
      "step": 234
    },
    {
      "epoch": 0.4181494661921708,
      "grad_norm": 3.9247405721746644,
      "learning_rate": 8.959666385198396e-06,
      "loss": 0.1825,
      "step": 235
    },
    {
      "epoch": 0.4199288256227758,
      "grad_norm": 3.596833149115134,
      "learning_rate": 8.951117653609666e-06,
      "loss": 0.1509,
      "step": 236
    },
    {
      "epoch": 0.42170818505338076,
      "grad_norm": 4.407699223008337,
      "learning_rate": 8.9425380555625e-06,
      "loss": 0.1726,
      "step": 237
    },
    {
      "epoch": 0.4234875444839858,
      "grad_norm": 2.699659825334132,
      "learning_rate": 8.933927658081423e-06,
      "loss": 0.1195,
      "step": 238
    },
    {
      "epoch": 0.42526690391459077,
      "grad_norm": 3.7180731802850513,
      "learning_rate": 8.925286528431578e-06,
      "loss": 0.1806,
      "step": 239
    },
    {
      "epoch": 0.42704626334519574,
      "grad_norm": 3.7345479731652125,
      "learning_rate": 8.916614734118184e-06,
      "loss": 0.1346,
      "step": 240
    },
    {
      "epoch": 0.4288256227758007,
      "grad_norm": 3.222714246810284,
      "learning_rate": 8.907912342886016e-06,
      "loss": 0.1575,
      "step": 241
    },
    {
      "epoch": 0.4306049822064057,
      "grad_norm": 3.6716349219590203,
      "learning_rate": 8.899179422718877e-06,
      "loss": 0.1649,
      "step": 242
    },
    {
      "epoch": 0.43238434163701067,
      "grad_norm": 3.9126172153774634,
      "learning_rate": 8.890416041839061e-06,
      "loss": 0.159,
      "step": 243
    },
    {
      "epoch": 0.43416370106761565,
      "grad_norm": 3.145707067973985,
      "learning_rate": 8.881622268706825e-06,
      "loss": 0.1645,
      "step": 244
    },
    {
      "epoch": 0.4359430604982206,
      "grad_norm": 3.4799536681880796,
      "learning_rate": 8.872798172019856e-06,
      "loss": 0.1556,
      "step": 245
    },
    {
      "epoch": 0.4377224199288256,
      "grad_norm": 3.8146105456725623,
      "learning_rate": 8.863943820712726e-06,
      "loss": 0.1824,
      "step": 246
    },
    {
      "epoch": 0.4395017793594306,
      "grad_norm": 3.6765735138154816,
      "learning_rate": 8.855059283956363e-06,
      "loss": 0.2119,
      "step": 247
    },
    {
      "epoch": 0.4412811387900356,
      "grad_norm": 3.725967372830946,
      "learning_rate": 8.8461446311575e-06,
      "loss": 0.1543,
      "step": 248
    },
    {
      "epoch": 0.4430604982206406,
      "grad_norm": 3.5457301212147763,
      "learning_rate": 8.837199931958147e-06,
      "loss": 0.14,
      "step": 249
    },
    {
      "epoch": 0.44483985765124556,
      "grad_norm": 3.715677742509787,
      "learning_rate": 8.828225256235035e-06,
      "loss": 0.1868,
      "step": 250
    },
    {
      "epoch": 0.44661921708185054,
      "grad_norm": 4.012496709372657,
      "learning_rate": 8.819220674099074e-06,
      "loss": 0.176,
      "step": 251
    },
    {
      "epoch": 0.4483985765124555,
      "grad_norm": 3.406268766358841,
      "learning_rate": 8.810186255894804e-06,
      "loss": 0.1597,
      "step": 252
    },
    {
      "epoch": 0.4501779359430605,
      "grad_norm": 3.876923167071431,
      "learning_rate": 8.801122072199848e-06,
      "loss": 0.1507,
      "step": 253
    },
    {
      "epoch": 0.45195729537366547,
      "grad_norm": 3.201628355349819,
      "learning_rate": 8.792028193824364e-06,
      "loss": 0.1263,
      "step": 254
    },
    {
      "epoch": 0.45373665480427045,
      "grad_norm": 3.09797484042757,
      "learning_rate": 8.782904691810478e-06,
      "loss": 0.1279,
      "step": 255
    },
    {
      "epoch": 0.4555160142348754,
      "grad_norm": 3.381965722739208,
      "learning_rate": 8.77375163743175e-06,
      "loss": 0.1541,
      "step": 256
    },
    {
      "epoch": 0.45729537366548045,
      "grad_norm": 3.521361727070609,
      "learning_rate": 8.764569102192593e-06,
      "loss": 0.1745,
      "step": 257
    },
    {
      "epoch": 0.45907473309608543,
      "grad_norm": 3.2039820230080185,
      "learning_rate": 8.755357157827735e-06,
      "loss": 0.1165,
      "step": 258
    },
    {
      "epoch": 0.4608540925266904,
      "grad_norm": 4.414527453831133,
      "learning_rate": 8.746115876301651e-06,
      "loss": 0.1632,
      "step": 259
    },
    {
      "epoch": 0.4626334519572954,
      "grad_norm": 3.304162299804278,
      "learning_rate": 8.736845329807994e-06,
      "loss": 0.1569,
      "step": 260
    },
    {
      "epoch": 0.46441281138790036,
      "grad_norm": 4.23920622419111,
      "learning_rate": 8.727545590769044e-06,
      "loss": 0.166,
      "step": 261
    },
    {
      "epoch": 0.46619217081850534,
      "grad_norm": 3.8010665544626185,
      "learning_rate": 8.718216731835131e-06,
      "loss": 0.1715,
      "step": 262
    },
    {
      "epoch": 0.4679715302491103,
      "grad_norm": 3.6913346357148082,
      "learning_rate": 8.708858825884075e-06,
      "loss": 0.1895,
      "step": 263
    },
    {
      "epoch": 0.4697508896797153,
      "grad_norm": 2.806760110644972,
      "learning_rate": 8.699471946020612e-06,
      "loss": 0.1402,
      "step": 264
    },
    {
      "epoch": 0.47153024911032027,
      "grad_norm": 4.15922880607463,
      "learning_rate": 8.690056165575825e-06,
      "loss": 0.1683,
      "step": 265
    },
    {
      "epoch": 0.47330960854092524,
      "grad_norm": 3.577351711864184,
      "learning_rate": 8.680611558106571e-06,
      "loss": 0.1648,
      "step": 266
    },
    {
      "epoch": 0.4750889679715303,
      "grad_norm": 3.1810835924233163,
      "learning_rate": 8.671138197394907e-06,
      "loss": 0.1272,
      "step": 267
    },
    {
      "epoch": 0.47686832740213525,
      "grad_norm": 2.8516403305725486,
      "learning_rate": 8.661636157447511e-06,
      "loss": 0.1504,
      "step": 268
    },
    {
      "epoch": 0.4786476868327402,
      "grad_norm": 4.906429564744661,
      "learning_rate": 8.652105512495106e-06,
      "loss": 0.1973,
      "step": 269
    },
    {
      "epoch": 0.4804270462633452,
      "grad_norm": 3.718384704413059,
      "learning_rate": 8.64254633699188e-06,
      "loss": 0.1494,
      "step": 270
    },
    {
      "epoch": 0.4822064056939502,
      "grad_norm": 3.7081417502020786,
      "learning_rate": 8.632958705614905e-06,
      "loss": 0.1671,
      "step": 271
    },
    {
      "epoch": 0.48398576512455516,
      "grad_norm": 3.889245861240765,
      "learning_rate": 8.623342693263549e-06,
      "loss": 0.1419,
      "step": 272
    },
    {
      "epoch": 0.48576512455516013,
      "grad_norm": 3.943208525425288,
      "learning_rate": 8.6136983750589e-06,
      "loss": 0.1645,
      "step": 273
    },
    {
      "epoch": 0.4875444839857651,
      "grad_norm": 3.804671003672085,
      "learning_rate": 8.604025826343167e-06,
      "loss": 0.1546,
      "step": 274
    },
    {
      "epoch": 0.4893238434163701,
      "grad_norm": 4.62726008857445,
      "learning_rate": 8.594325122679107e-06,
      "loss": 0.2093,
      "step": 275
    },
    {
      "epoch": 0.49110320284697506,
      "grad_norm": 3.5396602726734345,
      "learning_rate": 8.584596339849419e-06,
      "loss": 0.1901,
      "step": 276
    },
    {
      "epoch": 0.4928825622775801,
      "grad_norm": 3.336325882393596,
      "learning_rate": 8.574839553856157e-06,
      "loss": 0.1413,
      "step": 277
    },
    {
      "epoch": 0.49466192170818507,
      "grad_norm": 4.271673176454281,
      "learning_rate": 8.565054840920145e-06,
      "loss": 0.21,
      "step": 278
    },
    {
      "epoch": 0.49644128113879005,
      "grad_norm": 3.2698211625585447,
      "learning_rate": 8.55524227748037e-06,
      "loss": 0.1372,
      "step": 279
    },
    {
      "epoch": 0.498220640569395,
      "grad_norm": 3.4159093797069113,
      "learning_rate": 8.545401940193392e-06,
      "loss": 0.129,
      "step": 280
    },
    {
      "epoch": 0.5,
      "grad_norm": 3.487406068409208,
      "learning_rate": 8.535533905932739e-06,
      "loss": 0.1539,
      "step": 281
    },
    {
      "epoch": 0.501779359430605,
      "grad_norm": 4.60545181081035,
      "learning_rate": 8.525638251788312e-06,
      "loss": 0.2257,
      "step": 282
    },
    {
      "epoch": 0.50355871886121,
      "grad_norm": 3.6796861620222074,
      "learning_rate": 8.515715055065783e-06,
      "loss": 0.1828,
      "step": 283
    },
    {
      "epoch": 0.505338078291815,
      "grad_norm": 4.086405516579162,
      "learning_rate": 8.505764393285985e-06,
      "loss": 0.2079,
      "step": 284
    },
    {
      "epoch": 0.5071174377224199,
      "grad_norm": 3.6920946988483476,
      "learning_rate": 8.495786344184314e-06,
      "loss": 0.1538,
      "step": 285
    },
    {
      "epoch": 0.5088967971530249,
      "grad_norm": 4.52615756214946,
      "learning_rate": 8.485780985710113e-06,
      "loss": 0.1826,
      "step": 286
    },
    {
      "epoch": 0.5106761565836299,
      "grad_norm": 3.4348189134351905,
      "learning_rate": 8.475748396026074e-06,
      "loss": 0.1586,
      "step": 287
    },
    {
      "epoch": 0.5124555160142349,
      "grad_norm": 3.73348979909646,
      "learning_rate": 8.46568865350762e-06,
      "loss": 0.1608,
      "step": 288
    },
    {
      "epoch": 0.5142348754448398,
      "grad_norm": 4.503445601744546,
      "learning_rate": 8.45560183674229e-06,
      "loss": 0.2025,
      "step": 289
    },
    {
      "epoch": 0.5160142348754448,
      "grad_norm": 4.306875298605766,
      "learning_rate": 8.445488024529133e-06,
      "loss": 0.1905,
      "step": 290
    },
    {
      "epoch": 0.5177935943060499,
      "grad_norm": 3.193806087958825,
      "learning_rate": 8.435347295878087e-06,
      "loss": 0.1259,
      "step": 291
    },
    {
      "epoch": 0.5195729537366548,
      "grad_norm": 3.286850274646684,
      "learning_rate": 8.425179730009368e-06,
      "loss": 0.1317,
      "step": 292
    },
    {
      "epoch": 0.5213523131672598,
      "grad_norm": 3.9812673984482405,
      "learning_rate": 8.41498540635284e-06,
      "loss": 0.1659,
      "step": 293
    },
    {
      "epoch": 0.5231316725978647,
      "grad_norm": 4.389895436895122,
      "learning_rate": 8.404764404547404e-06,
      "loss": 0.171,
      "step": 294
    },
    {
      "epoch": 0.5249110320284698,
      "grad_norm": 4.522445446395583,
      "learning_rate": 8.394516804440374e-06,
      "loss": 0.1841,
      "step": 295
    },
    {
      "epoch": 0.5266903914590747,
      "grad_norm": 4.737777564015061,
      "learning_rate": 8.384242686086848e-06,
      "loss": 0.1788,
      "step": 296
    },
    {
      "epoch": 0.5284697508896797,
      "grad_norm": 3.7868636935237605,
      "learning_rate": 8.373942129749094e-06,
      "loss": 0.2074,
      "step": 297
    },
    {
      "epoch": 0.5302491103202847,
      "grad_norm": 2.952714261066169,
      "learning_rate": 8.363615215895908e-06,
      "loss": 0.1511,
      "step": 298
    },
    {
      "epoch": 0.5320284697508897,
      "grad_norm": 4.41372395000726,
      "learning_rate": 8.353262025202e-06,
      "loss": 0.189,
      "step": 299
    },
    {
      "epoch": 0.5338078291814946,
      "grad_norm": 3.2848197175143485,
      "learning_rate": 8.342882638547351e-06,
      "loss": 0.1455,
      "step": 300
    },
    {
      "epoch": 0.5355871886120996,
      "grad_norm": 3.5499905025180185,
      "learning_rate": 8.332477137016587e-06,
      "loss": 0.1509,
      "step": 301
    },
    {
      "epoch": 0.5373665480427047,
      "grad_norm": 3.6466305665097036,
      "learning_rate": 8.322045601898354e-06,
      "loss": 0.1567,
      "step": 302
    },
    {
      "epoch": 0.5391459074733096,
      "grad_norm": 3.3416674915627107,
      "learning_rate": 8.311588114684665e-06,
      "loss": 0.1454,
      "step": 303
    },
    {
      "epoch": 0.5409252669039146,
      "grad_norm": 5.515460204266922,
      "learning_rate": 8.301104757070276e-06,
      "loss": 0.2548,
      "step": 304
    },
    {
      "epoch": 0.5427046263345195,
      "grad_norm": 4.502270215684508,
      "learning_rate": 8.290595610952045e-06,
      "loss": 0.1646,
      "step": 305
    },
    {
      "epoch": 0.5444839857651246,
      "grad_norm": 4.396693225419102,
      "learning_rate": 8.280060758428294e-06,
      "loss": 0.2272,
      "step": 306
    },
    {
      "epoch": 0.5462633451957295,
      "grad_norm": 2.790290137485554,
      "learning_rate": 8.269500281798164e-06,
      "loss": 0.1216,
      "step": 307
    },
    {
      "epoch": 0.5480427046263345,
      "grad_norm": 3.813171808061726,
      "learning_rate": 8.258914263560971e-06,
      "loss": 0.1667,
      "step": 308
    },
    {
      "epoch": 0.5498220640569395,
      "grad_norm": 3.30689873052669,
      "learning_rate": 8.248302786415567e-06,
      "loss": 0.1567,
      "step": 309
    },
    {
      "epoch": 0.5516014234875445,
      "grad_norm": 3.9829463650054326,
      "learning_rate": 8.237665933259693e-06,
      "loss": 0.1753,
      "step": 310
    },
    {
      "epoch": 0.5533807829181495,
      "grad_norm": 3.7642040444435505,
      "learning_rate": 8.227003787189323e-06,
      "loss": 0.1535,
      "step": 311
    },
    {
      "epoch": 0.5551601423487544,
      "grad_norm": 2.9592466446799146,
      "learning_rate": 8.216316431498028e-06,
      "loss": 0.1458,
      "step": 312
    },
    {
      "epoch": 0.5569395017793595,
      "grad_norm": 3.398467093300974,
      "learning_rate": 8.205603949676317e-06,
      "loss": 0.1886,
      "step": 313
    },
    {
      "epoch": 0.5587188612099644,
      "grad_norm": 3.4590281040383344,
      "learning_rate": 8.194866425410984e-06,
      "loss": 0.1694,
      "step": 314
    },
    {
      "epoch": 0.5604982206405694,
      "grad_norm": 3.922675781929667,
      "learning_rate": 8.184103942584456e-06,
      "loss": 0.1715,
      "step": 315
    },
    {
      "epoch": 0.5622775800711743,
      "grad_norm": 3.6305083821032826,
      "learning_rate": 8.173316585274144e-06,
      "loss": 0.1577,
      "step": 316
    },
    {
      "epoch": 0.5640569395017794,
      "grad_norm": 3.462956987379063,
      "learning_rate": 8.162504437751775e-06,
      "loss": 0.1531,
      "step": 317
    },
    {
      "epoch": 0.5658362989323843,
      "grad_norm": 4.408451080880068,
      "learning_rate": 8.151667584482742e-06,
      "loss": 0.1574,
      "step": 318
    },
    {
      "epoch": 0.5676156583629893,
      "grad_norm": 4.3589508034400675,
      "learning_rate": 8.140806110125442e-06,
      "loss": 0.1874,
      "step": 319
    },
    {
      "epoch": 0.5693950177935944,
      "grad_norm": 3.6455916027318445,
      "learning_rate": 8.129920099530608e-06,
      "loss": 0.1871,
      "step": 320
    },
    {
      "epoch": 0.5711743772241993,
      "grad_norm": 3.0738889743986153,
      "learning_rate": 8.119009637740663e-06,
      "loss": 0.1008,
      "step": 321
    },
    {
      "epoch": 0.5729537366548043,
      "grad_norm": 3.4144469764758485,
      "learning_rate": 8.108074809989032e-06,
      "loss": 0.1597,
      "step": 322
    },
    {
      "epoch": 0.5747330960854092,
      "grad_norm": 3.1224999828020326,
      "learning_rate": 8.097115701699498e-06,
      "loss": 0.1147,
      "step": 323
    },
    {
      "epoch": 0.5765124555160143,
      "grad_norm": 3.8063708206274014,
      "learning_rate": 8.086132398485525e-06,
      "loss": 0.1834,
      "step": 324
    },
    {
      "epoch": 0.5782918149466192,
      "grad_norm": 3.3124905212499924,
      "learning_rate": 8.075124986149583e-06,
      "loss": 0.137,
      "step": 325
    },
    {
      "epoch": 0.5800711743772242,
      "grad_norm": 3.561597181329593,
      "learning_rate": 8.064093550682494e-06,
      "loss": 0.1494,
      "step": 326
    },
    {
      "epoch": 0.5818505338078291,
      "grad_norm": 3.7132406457426756,
      "learning_rate": 8.053038178262742e-06,
      "loss": 0.1309,
      "step": 327
    },
    {
      "epoch": 0.5836298932384342,
      "grad_norm": 3.6592869576152065,
      "learning_rate": 8.041958955255815e-06,
      "loss": 0.1721,
      "step": 328
    },
    {
      "epoch": 0.5854092526690391,
      "grad_norm": 3.4384332348186795,
      "learning_rate": 8.030855968213518e-06,
      "loss": 0.15,
      "step": 329
    },
    {
      "epoch": 0.5871886120996441,
      "grad_norm": 4.196078169433854,
      "learning_rate": 8.019729303873307e-06,
      "loss": 0.1772,
      "step": 330
    },
    {
      "epoch": 0.5889679715302492,
      "grad_norm": 3.3459586902454834,
      "learning_rate": 8.008579049157607e-06,
      "loss": 0.1326,
      "step": 331
    },
    {
      "epoch": 0.5907473309608541,
      "grad_norm": 2.8377876655203877,
      "learning_rate": 7.99740529117313e-06,
      "loss": 0.1392,
      "step": 332
    },
    {
      "epoch": 0.5925266903914591,
      "grad_norm": 3.887093450531788,
      "learning_rate": 7.986208117210198e-06,
      "loss": 0.1581,
      "step": 333
    },
    {
      "epoch": 0.594306049822064,
      "grad_norm": 4.29349555061408,
      "learning_rate": 7.974987614742066e-06,
      "loss": 0.1825,
      "step": 334
    },
    {
      "epoch": 0.5960854092526691,
      "grad_norm": 4.434799050880575,
      "learning_rate": 7.963743871424224e-06,
      "loss": 0.1949,
      "step": 335
    },
    {
      "epoch": 0.597864768683274,
      "grad_norm": 3.5654606851671673,
      "learning_rate": 7.952476975093729e-06,
      "loss": 0.1462,
      "step": 336
    },
    {
      "epoch": 0.599644128113879,
      "grad_norm": 4.01873806625503,
      "learning_rate": 7.941187013768508e-06,
      "loss": 0.1968,
      "step": 337
    },
    {
      "epoch": 0.6014234875444839,
      "grad_norm": 3.8061607099494195,
      "learning_rate": 7.929874075646673e-06,
      "loss": 0.1634,
      "step": 338
    },
    {
      "epoch": 0.603202846975089,
      "grad_norm": 4.275647551074874,
      "learning_rate": 7.918538249105835e-06,
      "loss": 0.1565,
      "step": 339
    },
    {
      "epoch": 0.604982206405694,
      "grad_norm": 4.098294246949144,
      "learning_rate": 7.907179622702409e-06,
      "loss": 0.1704,
      "step": 340
    },
    {
      "epoch": 0.6067615658362989,
      "grad_norm": 3.542080796046305,
      "learning_rate": 7.895798285170927e-06,
      "loss": 0.1537,
      "step": 341
    },
    {
      "epoch": 0.608540925266904,
      "grad_norm": 3.307312598893261,
      "learning_rate": 7.88439432542334e-06,
      "loss": 0.1328,
      "step": 342
    },
    {
      "epoch": 0.6103202846975089,
      "grad_norm": 3.749173752554754,
      "learning_rate": 7.872967832548327e-06,
      "loss": 0.188,
      "step": 343
    },
    {
      "epoch": 0.6120996441281139,
      "grad_norm": 3.4378394484820625,
      "learning_rate": 7.861518895810597e-06,
      "loss": 0.1957,
      "step": 344
    },
    {
      "epoch": 0.6138790035587188,
      "grad_norm": 3.47810513578558,
      "learning_rate": 7.850047604650188e-06,
      "loss": 0.1819,
      "step": 345
    },
    {
      "epoch": 0.6156583629893239,
      "grad_norm": 3.9794789948639764,
      "learning_rate": 7.838554048681783e-06,
      "loss": 0.1905,
      "step": 346
    },
    {
      "epoch": 0.6174377224199288,
      "grad_norm": 3.5604565295656445,
      "learning_rate": 7.827038317693988e-06,
      "loss": 0.2414,
      "step": 347
    },
    {
      "epoch": 0.6192170818505338,
      "grad_norm": 3.4776494437850483,
      "learning_rate": 7.815500501648654e-06,
      "loss": 0.1476,
      "step": 348
    },
    {
      "epoch": 0.6209964412811388,
      "grad_norm": 3.7626223902586706,
      "learning_rate": 7.80394069068015e-06,
      "loss": 0.1905,
      "step": 349
    },
    {
      "epoch": 0.6227758007117438,
      "grad_norm": 3.6193663632783646,
      "learning_rate": 7.79235897509468e-06,
      "loss": 0.1507,
      "step": 350
    },
    {
      "epoch": 0.6245551601423488,
      "grad_norm": 4.087591267185595,
      "learning_rate": 7.780755445369563e-06,
      "loss": 0.1787,
      "step": 351
    },
    {
      "epoch": 0.6263345195729537,
      "grad_norm": 4.160111479069727,
      "learning_rate": 7.769130192152538e-06,
      "loss": 0.206,
      "step": 352
    },
    {
      "epoch": 0.6281138790035588,
      "grad_norm": 3.592982230181124,
      "learning_rate": 7.757483306261042e-06,
      "loss": 0.1929,
      "step": 353
    },
    {
      "epoch": 0.6298932384341637,
      "grad_norm": 4.127850515506907,
      "learning_rate": 7.745814878681516e-06,
      "loss": 0.1842,
      "step": 354
    },
    {
      "epoch": 0.6316725978647687,
      "grad_norm": 3.3706487807966203,
      "learning_rate": 7.734125000568684e-06,
      "loss": 0.157,
      "step": 355
    },
    {
      "epoch": 0.6334519572953736,
      "grad_norm": 3.5926424937673507,
      "learning_rate": 7.722413763244837e-06,
      "loss": 0.1488,
      "step": 356
    },
    {
      "epoch": 0.6352313167259787,
      "grad_norm": 2.772267351848365,
      "learning_rate": 7.710681258199136e-06,
      "loss": 0.1251,
      "step": 357
    },
    {
      "epoch": 0.6370106761565836,
      "grad_norm": 3.8577889929925835,
      "learning_rate": 7.69892757708688e-06,
      "loss": 0.1552,
      "step": 358
    },
    {
      "epoch": 0.6387900355871886,
      "grad_norm": 3.761030067500859,
      "learning_rate": 7.687152811728799e-06,
      "loss": 0.1627,
      "step": 359
    },
    {
      "epoch": 0.6405693950177936,
      "grad_norm": 3.8409367400554544,
      "learning_rate": 7.675357054110337e-06,
      "loss": 0.1789,
      "step": 360
    },
    {
      "epoch": 0.6423487544483986,
      "grad_norm": 3.5755053604624143,
      "learning_rate": 7.663540396380931e-06,
      "loss": 0.1465,
      "step": 361
    },
    {
      "epoch": 0.6441281138790036,
      "grad_norm": 3.4194967858780525,
      "learning_rate": 7.651702930853287e-06,
      "loss": 0.1505,
      "step": 362
    },
    {
      "epoch": 0.6459074733096085,
      "grad_norm": 3.8424830336725595,
      "learning_rate": 7.639844750002668e-06,
      "loss": 0.1807,
      "step": 363
    },
    {
      "epoch": 0.6476868327402135,
      "grad_norm": 3.1618885674092185,
      "learning_rate": 7.627965946466167e-06,
      "loss": 0.1767,
      "step": 364
    },
    {
      "epoch": 0.6494661921708185,
      "grad_norm": 3.184280718284312,
      "learning_rate": 7.616066613041977e-06,
      "loss": 0.1516,
      "step": 365
    },
    {
      "epoch": 0.6512455516014235,
      "grad_norm": 3.5113548048605536,
      "learning_rate": 7.6041468426886785e-06,
      "loss": 0.1595,
      "step": 366
    },
    {
      "epoch": 0.6530249110320284,
      "grad_norm": 2.5102036210293868,
      "learning_rate": 7.592206728524507e-06,
      "loss": 0.1032,
      "step": 367
    },
    {
      "epoch": 0.6548042704626335,
      "grad_norm": 3.446005439403226,
      "learning_rate": 7.580246363826621e-06,
      "loss": 0.1487,
      "step": 368
    },
    {
      "epoch": 0.6565836298932385,
      "grad_norm": 4.108496676770851,
      "learning_rate": 7.568265842030381e-06,
      "loss": 0.1851,
      "step": 369
    },
    {
      "epoch": 0.6583629893238434,
      "grad_norm": 3.6813653483624456,
      "learning_rate": 7.556265256728618e-06,
      "loss": 0.1856,
      "step": 370
    },
    {
      "epoch": 0.6601423487544484,
      "grad_norm": 3.14090247397153,
      "learning_rate": 7.544244701670894e-06,
      "loss": 0.1621,
      "step": 371
    },
    {
      "epoch": 0.6619217081850534,
      "grad_norm": 3.492585903527491,
      "learning_rate": 7.532204270762786e-06,
      "loss": 0.1577,
      "step": 372
    },
    {
      "epoch": 0.6637010676156584,
      "grad_norm": 3.4283433223069952,
      "learning_rate": 7.520144058065133e-06,
      "loss": 0.144,
      "step": 373
    },
    {
      "epoch": 0.6654804270462633,
      "grad_norm": 3.7863016730138384,
      "learning_rate": 7.50806415779332e-06,
      "loss": 0.1477,
      "step": 374
    },
    {
      "epoch": 0.6672597864768683,
      "grad_norm": 4.052791973431194,
      "learning_rate": 7.495964664316525e-06,
      "loss": 0.1698,
      "step": 375
    },
    {
      "epoch": 0.6690391459074733,
      "grad_norm": 3.623948491994843,
      "learning_rate": 7.4838456721569975e-06,
      "loss": 0.1621,
      "step": 376
    },
    {
      "epoch": 0.6708185053380783,
      "grad_norm": 3.275215281775822,
      "learning_rate": 7.471707275989304e-06,
      "loss": 0.1522,
      "step": 377
    },
    {
      "epoch": 0.6725978647686833,
      "grad_norm": 3.3004264546674484,
      "learning_rate": 7.459549570639602e-06,
      "loss": 0.1456,
      "step": 378
    },
    {
      "epoch": 0.6743772241992882,
      "grad_norm": 3.862576441258142,
      "learning_rate": 7.447372651084896e-06,
      "loss": 0.1587,
      "step": 379
    },
    {
      "epoch": 0.6761565836298933,
      "grad_norm": 3.4296974267778766,
      "learning_rate": 7.435176612452286e-06,
      "loss": 0.1375,
      "step": 380
    },
    {
      "epoch": 0.6779359430604982,
      "grad_norm": 3.5922328969054376,
      "learning_rate": 7.4229615500182396e-06,
      "loss": 0.1662,
      "step": 381
    },
    {
      "epoch": 0.6797153024911032,
      "grad_norm": 2.9642790255597458,
      "learning_rate": 7.4107275592078345e-06,
      "loss": 0.1293,
      "step": 382
    },
    {
      "epoch": 0.6814946619217082,
      "grad_norm": 3.0828737266251487,
      "learning_rate": 7.398474735594022e-06,
      "loss": 0.1318,
      "step": 383
    },
    {
      "epoch": 0.6832740213523132,
      "grad_norm": 3.8011161021043813,
      "learning_rate": 7.386203174896872e-06,
      "loss": 0.1565,
      "step": 384
    },
    {
      "epoch": 0.6850533807829181,
      "grad_norm": 3.507538314793557,
      "learning_rate": 7.373912972982838e-06,
      "loss": 0.147,
      "step": 385
    },
    {
      "epoch": 0.6868327402135231,
      "grad_norm": 3.318006880872963,
      "learning_rate": 7.361604225863992e-06,
      "loss": 0.1516,
      "step": 386
    },
    {
      "epoch": 0.6886120996441281,
      "grad_norm": 4.001100606324702,
      "learning_rate": 7.349277029697287e-06,
      "loss": 0.1565,
      "step": 387
    },
    {
      "epoch": 0.6903914590747331,
      "grad_norm": 3.34550522974217,
      "learning_rate": 7.336931480783801e-06,
      "loss": 0.1385,
      "step": 388
    },
    {
      "epoch": 0.6921708185053381,
      "grad_norm": 4.412165386308153,
      "learning_rate": 7.3245676755679854e-06,
      "loss": 0.183,
      "step": 389
    },
    {
      "epoch": 0.693950177935943,
      "grad_norm": 4.043462693583462,
      "learning_rate": 7.312185710636911e-06,
      "loss": 0.1735,
      "step": 390
    },
    {
      "epoch": 0.6957295373665481,
      "grad_norm": 3.423115519061462,
      "learning_rate": 7.299785682719512e-06,
      "loss": 0.115,
      "step": 391
    },
    {
      "epoch": 0.697508896797153,
      "grad_norm": 3.5195687313060984,
      "learning_rate": 7.287367688685835e-06,
      "loss": 0.158,
      "step": 392
    },
    {
      "epoch": 0.699288256227758,
      "grad_norm": 4.911917856870993,
      "learning_rate": 7.274931825546279e-06,
      "loss": 0.1792,
      "step": 393
    },
    {
      "epoch": 0.701067615658363,
      "grad_norm": 3.469619740549378,
      "learning_rate": 7.262478190450834e-06,
      "loss": 0.1353,
      "step": 394
    },
    {
      "epoch": 0.702846975088968,
      "grad_norm": 4.118793316960568,
      "learning_rate": 7.250006880688332e-06,
      "loss": 0.1879,
      "step": 395
    },
    {
      "epoch": 0.7046263345195729,
      "grad_norm": 3.7046757500270266,
      "learning_rate": 7.2375179936856775e-06,
      "loss": 0.1523,
      "step": 396
    },
    {
      "epoch": 0.7064056939501779,
      "grad_norm": 3.228987162069876,
      "learning_rate": 7.22501162700709e-06,
      "loss": 0.1553,
      "step": 397
    },
    {
      "epoch": 0.708185053380783,
      "grad_norm": 3.556040422055483,
      "learning_rate": 7.21248787835334e-06,
      "loss": 0.1348,
      "step": 398
    },
    {
      "epoch": 0.7099644128113879,
      "grad_norm": 3.853488506475087,
      "learning_rate": 7.199946845560994e-06,
      "loss": 0.1538,
      "step": 399
    },
    {
      "epoch": 0.7117437722419929,
      "grad_norm": 4.543072702978003,
      "learning_rate": 7.1873886266016365e-06,
      "loss": 0.2353,
      "step": 400
    },
    {
      "epoch": 0.7135231316725978,
      "grad_norm": 4.4755393615804495,
      "learning_rate": 7.174813319581115e-06,
      "loss": 0.1761,
      "step": 401
    },
    {
      "epoch": 0.7153024911032029,
      "grad_norm": 5.215853290886184,
      "learning_rate": 7.162221022738768e-06,
      "loss": 0.1726,
      "step": 402
    },
    {
      "epoch": 0.7170818505338078,
      "grad_norm": 3.8386235895787033,
      "learning_rate": 7.149611834446664e-06,
      "loss": 0.2003,
      "step": 403
    },
    {
      "epoch": 0.7188612099644128,
      "grad_norm": 3.91087098371206,
      "learning_rate": 7.136985853208824e-06,
      "loss": 0.1832,
      "step": 404
    },
    {
      "epoch": 0.7206405693950177,
      "grad_norm": 3.3841765221588624,
      "learning_rate": 7.124343177660462e-06,
      "loss": 0.1601,
      "step": 405
    },
    {
      "epoch": 0.7224199288256228,
      "grad_norm": 4.552812449564969,
      "learning_rate": 7.111683906567206e-06,
      "loss": 0.1843,
      "step": 406
    },
    {
      "epoch": 0.7241992882562278,
      "grad_norm": 3.7011586441530433,
      "learning_rate": 7.099008138824329e-06,
      "loss": 0.1437,
      "step": 407
    },
    {
      "epoch": 0.7259786476868327,
      "grad_norm": 3.1540986156424067,
      "learning_rate": 7.086315973455982e-06,
      "loss": 0.16,
      "step": 408
    },
    {
      "epoch": 0.7277580071174378,
      "grad_norm": 3.718370033260771,
      "learning_rate": 7.0736075096144084e-06,
      "loss": 0.1718,
      "step": 409
    },
    {
      "epoch": 0.7295373665480427,
      "grad_norm": 4.034738043016495,
      "learning_rate": 7.060882846579182e-06,
      "loss": 0.1472,
      "step": 410
    },
    {
      "epoch": 0.7313167259786477,
      "grad_norm": 3.73550376670331,
      "learning_rate": 7.048142083756427e-06,
      "loss": 0.1664,
      "step": 411
    },
    {
      "epoch": 0.7330960854092526,
      "grad_norm": 3.9924451006301234,
      "learning_rate": 7.035385320678035e-06,
      "loss": 0.1837,
      "step": 412
    },
    {
      "epoch": 0.7348754448398577,
      "grad_norm": 3.513138957376053,
      "learning_rate": 7.022612657000898e-06,
      "loss": 0.1676,
      "step": 413
    },
    {
      "epoch": 0.7366548042704626,
      "grad_norm": 3.3230754680176267,
      "learning_rate": 7.0098241925061215e-06,
      "loss": 0.1773,
      "step": 414
    },
    {
      "epoch": 0.7384341637010676,
      "grad_norm": 4.190730640315547,
      "learning_rate": 6.997020027098249e-06,
      "loss": 0.1569,
      "step": 415
    },
    {
      "epoch": 0.7402135231316725,
      "grad_norm": 3.740932222600797,
      "learning_rate": 6.9842002608044844e-06,
      "loss": 0.1712,
      "step": 416
    },
    {
      "epoch": 0.7419928825622776,
      "grad_norm": 3.42920271306724,
      "learning_rate": 6.971364993773901e-06,
      "loss": 0.1636,
      "step": 417
    },
    {
      "epoch": 0.7437722419928826,
      "grad_norm": 3.4051914804506245,
      "learning_rate": 6.958514326276669e-06,
      "loss": 0.1528,
      "step": 418
    },
    {
      "epoch": 0.7455516014234875,
      "grad_norm": 3.653218483567315,
      "learning_rate": 6.945648358703269e-06,
      "loss": 0.156,
      "step": 419
    },
    {
      "epoch": 0.7473309608540926,
      "grad_norm": 3.1608825623324344,
      "learning_rate": 6.932767191563703e-06,
      "loss": 0.1767,
      "step": 420
    },
    {
      "epoch": 0.7491103202846975,
      "grad_norm": 3.1876183665033238,
      "learning_rate": 6.919870925486718e-06,
      "loss": 0.1274,
      "step": 421
    },
    {
      "epoch": 0.7508896797153025,
      "grad_norm": 3.0083014965140698,
      "learning_rate": 6.906959661219011e-06,
      "loss": 0.1431,
      "step": 422
    },
    {
      "epoch": 0.7526690391459074,
      "grad_norm": 3.1996078943592914,
      "learning_rate": 6.8940334996244505e-06,
      "loss": 0.1462,
      "step": 423
    },
    {
      "epoch": 0.7544483985765125,
      "grad_norm": 3.5344911943782904,
      "learning_rate": 6.881092541683279e-06,
      "loss": 0.1376,
      "step": 424
    },
    {
      "epoch": 0.7562277580071174,
      "grad_norm": 3.1938993122275785,
      "learning_rate": 6.8681368884913345e-06,
      "loss": 0.1249,
      "step": 425
    },
    {
      "epoch": 0.7580071174377224,
      "grad_norm": 2.8723731241423014,
      "learning_rate": 6.855166641259252e-06,
      "loss": 0.1465,
      "step": 426
    },
    {
      "epoch": 0.7597864768683275,
      "grad_norm": 3.3524876076503203,
      "learning_rate": 6.8421819013116766e-06,
      "loss": 0.1558,
      "step": 427
    },
    {
      "epoch": 0.7615658362989324,
      "grad_norm": 3.4308669678014634,
      "learning_rate": 6.829182770086474e-06,
      "loss": 0.1727,
      "step": 428
    },
    {
      "epoch": 0.7633451957295374,
      "grad_norm": 3.5793116688453064,
      "learning_rate": 6.816169349133934e-06,
      "loss": 0.1636,
      "step": 429
    },
    {
      "epoch": 0.7651245551601423,
      "grad_norm": 3.257744853573949,
      "learning_rate": 6.803141740115979e-06,
      "loss": 0.1127,
      "step": 430
    },
    {
      "epoch": 0.7669039145907474,
      "grad_norm": 5.210065531984686,
      "learning_rate": 6.7901000448053676e-06,
      "loss": 0.222,
      "step": 431
    },
    {
      "epoch": 0.7686832740213523,
      "grad_norm": 3.434066344724471,
      "learning_rate": 6.777044365084907e-06,
      "loss": 0.1361,
      "step": 432
    },
    {
      "epoch": 0.7704626334519573,
      "grad_norm": 3.3583468771088154,
      "learning_rate": 6.763974802946649e-06,
      "loss": 0.1431,
      "step": 433
    },
    {
      "epoch": 0.7722419928825622,
      "grad_norm": 3.8272010782080166,
      "learning_rate": 6.750891460491093e-06,
      "loss": 0.2122,
      "step": 434
    },
    {
      "epoch": 0.7740213523131673,
      "grad_norm": 4.120373033919094,
      "learning_rate": 6.737794439926395e-06,
      "loss": 0.2068,
      "step": 435
    },
    {
      "epoch": 0.7758007117437722,
      "grad_norm": 3.534476898655974,
      "learning_rate": 6.724683843567567e-06,
      "loss": 0.1654,
      "step": 436
    },
    {
      "epoch": 0.7775800711743772,
      "grad_norm": 3.7814829269008547,
      "learning_rate": 6.711559773835672e-06,
      "loss": 0.1615,
      "step": 437
    },
    {
      "epoch": 0.7793594306049823,
      "grad_norm": 3.6882512683636657,
      "learning_rate": 6.69842233325703e-06,
      "loss": 0.1721,
      "step": 438
    },
    {
      "epoch": 0.7811387900355872,
      "grad_norm": 4.282001790062711,
      "learning_rate": 6.685271624462416e-06,
      "loss": 0.2067,
      "step": 439
    },
    {
      "epoch": 0.7829181494661922,
      "grad_norm": 3.175666254653991,
      "learning_rate": 6.672107750186255e-06,
      "loss": 0.1478,
      "step": 440
    },
    {
      "epoch": 0.7846975088967971,
      "grad_norm": 2.557537634918939,
      "learning_rate": 6.658930813265825e-06,
      "loss": 0.1249,
      "step": 441
    },
    {
      "epoch": 0.7864768683274022,
      "grad_norm": 3.3109369826436628,
      "learning_rate": 6.645740916640449e-06,
      "loss": 0.1576,
      "step": 442
    },
    {
      "epoch": 0.7882562277580071,
      "grad_norm": 2.610005111770994,
      "learning_rate": 6.63253816335069e-06,
      "loss": 0.1184,
      "step": 443
    },
    {
      "epoch": 0.7900355871886121,
      "grad_norm": 3.1793778156768515,
      "learning_rate": 6.619322656537552e-06,
      "loss": 0.1609,
      "step": 444
    },
    {
      "epoch": 0.791814946619217,
      "grad_norm": 3.182830733127857,
      "learning_rate": 6.606094499441671e-06,
      "loss": 0.148,
      "step": 445
    },
    {
      "epoch": 0.7935943060498221,
      "grad_norm": 3.4962711482477036,
      "learning_rate": 6.592853795402502e-06,
      "loss": 0.1517,
      "step": 446
    },
    {
      "epoch": 0.7953736654804271,
      "grad_norm": 3.793398772006309,
      "learning_rate": 6.579600647857525e-06,
      "loss": 0.1546,
      "step": 447
    },
    {
      "epoch": 0.797153024911032,
      "grad_norm": 3.9639455628853426,
      "learning_rate": 6.566335160341425e-06,
      "loss": 0.1869,
      "step": 448
    },
    {
      "epoch": 0.798932384341637,
      "grad_norm": 2.92716196045127,
      "learning_rate": 6.553057436485289e-06,
      "loss": 0.1501,
      "step": 449
    },
    {
      "epoch": 0.800711743772242,
      "grad_norm": 3.0268004914559765,
      "learning_rate": 6.539767580015799e-06,
      "loss": 0.1858,
      "step": 450
    },
    {
      "epoch": 0.802491103202847,
      "grad_norm": 3.9989507102210853,
      "learning_rate": 6.52646569475441e-06,
      "loss": 0.1479,
      "step": 451
    },
    {
      "epoch": 0.8042704626334519,
      "grad_norm": 4.083655131675356,
      "learning_rate": 6.513151884616556e-06,
      "loss": 0.2112,
      "step": 452
    },
    {
      "epoch": 0.806049822064057,
      "grad_norm": 3.8321381225260174,
      "learning_rate": 6.499826253610823e-06,
      "loss": 0.1533,
      "step": 453
    },
    {
      "epoch": 0.8078291814946619,
      "grad_norm": 3.4945132900548757,
      "learning_rate": 6.486488905838143e-06,
      "loss": 0.1294,
      "step": 454
    },
    {
      "epoch": 0.8096085409252669,
      "grad_norm": 3.1650557757811204,
      "learning_rate": 6.473139945490984e-06,
      "loss": 0.1194,
      "step": 455
    },
    {
      "epoch": 0.8113879003558719,
      "grad_norm": 3.7701166013743457,
      "learning_rate": 6.459779476852528e-06,
      "loss": 0.1573,
      "step": 456
    },
    {
      "epoch": 0.8131672597864769,
      "grad_norm": 3.6531146031370447,
      "learning_rate": 6.446407604295863e-06,
      "loss": 0.1768,
      "step": 457
    },
    {
      "epoch": 0.8149466192170819,
      "grad_norm": 3.586736966225412,
      "learning_rate": 6.433024432283169e-06,
      "loss": 0.1332,
      "step": 458
    },
    {
      "epoch": 0.8167259786476868,
      "grad_norm": 3.6575846628516264,
      "learning_rate": 6.41963006536489e-06,
      "loss": 0.1502,
      "step": 459
    },
    {
      "epoch": 0.8185053380782918,
      "grad_norm": 3.4574890363178485,
      "learning_rate": 6.4062246081789316e-06,
      "loss": 0.1517,
      "step": 460
    },
    {
      "epoch": 0.8202846975088968,
      "grad_norm": 3.111596665714797,
      "learning_rate": 6.392808165449836e-06,
      "loss": 0.1122,
      "step": 461
    },
    {
      "epoch": 0.8220640569395018,
      "grad_norm": 3.2132156099349416,
      "learning_rate": 6.379380841987965e-06,
      "loss": 0.136,
      "step": 462
    },
    {
      "epoch": 0.8238434163701067,
      "grad_norm": 3.526710266124468,
      "learning_rate": 6.365942742688684e-06,
      "loss": 0.1245,
      "step": 463
    },
    {
      "epoch": 0.8256227758007118,
      "grad_norm": 4.622676163501748,
      "learning_rate": 6.352493972531535e-06,
      "loss": 0.2088,
      "step": 464
    },
    {
      "epoch": 0.8274021352313167,
      "grad_norm": 4.133995492491854,
      "learning_rate": 6.339034636579425e-06,
      "loss": 0.168,
      "step": 465
    },
    {
      "epoch": 0.8291814946619217,
      "grad_norm": 3.3077970166539528,
      "learning_rate": 6.325564839977802e-06,
      "loss": 0.1242,
      "step": 466
    },
    {
      "epoch": 0.8309608540925267,
      "grad_norm": 2.92178056452928,
      "learning_rate": 6.312084687953835e-06,
      "loss": 0.1143,
      "step": 467
    },
    {
      "epoch": 0.8327402135231317,
      "grad_norm": 4.235103148793485,
      "learning_rate": 6.298594285815585e-06,
      "loss": 0.1856,
      "step": 468
    },
    {
      "epoch": 0.8345195729537367,
      "grad_norm": 4.209464185852595,
      "learning_rate": 6.2850937389511936e-06,
      "loss": 0.1962,
      "step": 469
    },
    {
      "epoch": 0.8362989323843416,
      "grad_norm": 3.4009915058766724,
      "learning_rate": 6.271583152828049e-06,
      "loss": 0.1192,
      "step": 470
    },
    {
      "epoch": 0.8380782918149466,
      "grad_norm": 3.3831076370359017,
      "learning_rate": 6.258062632991972e-06,
      "loss": 0.115,
      "step": 471
    },
    {
      "epoch": 0.8398576512455516,
      "grad_norm": 3.5242026972893274,
      "learning_rate": 6.244532285066382e-06,
      "loss": 0.1499,
      "step": 472
    },
    {
      "epoch": 0.8416370106761566,
      "grad_norm": 3.369789672850011,
      "learning_rate": 6.2309922147514775e-06,
      "loss": 0.1491,
      "step": 473
    },
    {
      "epoch": 0.8434163701067615,
      "grad_norm": 3.2909922062235304,
      "learning_rate": 6.2174425278234115e-06,
      "loss": 0.1835,
      "step": 474
    },
    {
      "epoch": 0.8451957295373665,
      "grad_norm": 4.029244342069236,
      "learning_rate": 6.20388333013346e-06,
      "loss": 0.1339,
      "step": 475
    },
    {
      "epoch": 0.8469750889679716,
      "grad_norm": 3.35838548183512,
      "learning_rate": 6.190314727607196e-06,
      "loss": 0.1782,
      "step": 476
    },
    {
      "epoch": 0.8487544483985765,
      "grad_norm": 3.68710962530028,
      "learning_rate": 6.176736826243671e-06,
      "loss": 0.1669,
      "step": 477
    },
    {
      "epoch": 0.8505338078291815,
      "grad_norm": 3.5489421927802542,
      "learning_rate": 6.163149732114571e-06,
      "loss": 0.1551,
      "step": 478
    },
    {
      "epoch": 0.8523131672597865,
      "grad_norm": 4.014283055481644,
      "learning_rate": 6.149553551363404e-06,
      "loss": 0.1567,
      "step": 479
    },
    {
      "epoch": 0.8540925266903915,
      "grad_norm": 3.568721730946564,
      "learning_rate": 6.1359483902046605e-06,
      "loss": 0.1703,
      "step": 480
    },
    {
      "epoch": 0.8558718861209964,
      "grad_norm": 3.4411103471391966,
      "learning_rate": 6.122334354922984e-06,
      "loss": 0.1503,
      "step": 481
    },
    {
      "epoch": 0.8576512455516014,
      "grad_norm": 3.1618196275103445,
      "learning_rate": 6.108711551872347e-06,
      "loss": 0.1526,
      "step": 482
    },
    {
      "epoch": 0.8594306049822064,
      "grad_norm": 3.906009704602092,
      "learning_rate": 6.095080087475218e-06,
      "loss": 0.1542,
      "step": 483
    },
    {
      "epoch": 0.8612099644128114,
      "grad_norm": 3.2864076417528625,
      "learning_rate": 6.0814400682217236e-06,
      "loss": 0.169,
      "step": 484
    },
    {
      "epoch": 0.8629893238434164,
      "grad_norm": 3.3623374204337235,
      "learning_rate": 6.067791600668823e-06,
      "loss": 0.1173,
      "step": 485
    },
    {
      "epoch": 0.8647686832740213,
      "grad_norm": 2.9633545351774697,
      "learning_rate": 6.054134791439479e-06,
      "loss": 0.1108,
      "step": 486
    },
    {
      "epoch": 0.8665480427046264,
      "grad_norm": 3.5148162824423177,
      "learning_rate": 6.040469747221815e-06,
      "loss": 0.1287,
      "step": 487
    },
    {
      "epoch": 0.8683274021352313,
      "grad_norm": 3.8201827955567627,
      "learning_rate": 6.026796574768288e-06,
      "loss": 0.146,
      "step": 488
    },
    {
      "epoch": 0.8701067615658363,
      "grad_norm": 3.746707457256004,
      "learning_rate": 6.013115380894854e-06,
      "loss": 0.1459,
      "step": 489
    },
    {
      "epoch": 0.8718861209964412,
      "grad_norm": 2.962986156808653,
      "learning_rate": 5.999426272480133e-06,
      "loss": 0.1097,
      "step": 490
    },
    {
      "epoch": 0.8736654804270463,
      "grad_norm": 3.858955560868093,
      "learning_rate": 5.985729356464575e-06,
      "loss": 0.1641,
      "step": 491
    },
    {
      "epoch": 0.8754448398576512,
      "grad_norm": 3.578267397186503,
      "learning_rate": 5.972024739849622e-06,
      "loss": 0.1501,
      "step": 492
    },
    {
      "epoch": 0.8772241992882562,
      "grad_norm": 2.5574074259264328,
      "learning_rate": 5.958312529696874e-06,
      "loss": 0.104,
      "step": 493
    },
    {
      "epoch": 0.8790035587188612,
      "grad_norm": 3.8842253957645365,
      "learning_rate": 5.944592833127253e-06,
      "loss": 0.199,
      "step": 494
    },
    {
      "epoch": 0.8807829181494662,
      "grad_norm": 3.462253138250446,
      "learning_rate": 5.9308657573201645e-06,
      "loss": 0.1788,
      "step": 495
    },
    {
      "epoch": 0.8825622775800712,
      "grad_norm": 3.490027013378324,
      "learning_rate": 5.917131409512663e-06,
      "loss": 0.1409,
      "step": 496
    },
    {
      "epoch": 0.8843416370106761,
      "grad_norm": 3.2625397840291597,
      "learning_rate": 5.903389896998611e-06,
      "loss": 0.1446,
      "step": 497
    },
    {
      "epoch": 0.8861209964412812,
      "grad_norm": 3.6164387792997674,
      "learning_rate": 5.889641327127843e-06,
      "loss": 0.1553,
      "step": 498
    },
    {
      "epoch": 0.8879003558718861,
      "grad_norm": 3.5773938281248645,
      "learning_rate": 5.875885807305326e-06,
      "loss": 0.1664,
      "step": 499
    },
    {
      "epoch": 0.8896797153024911,
      "grad_norm": 2.9745473834794645,
      "learning_rate": 5.862123444990319e-06,
      "loss": 0.1327,
      "step": 500
    },
    {
      "epoch": 0.8896797153024911,
      "eval_loss": 0.1705450415611267,
      "eval_runtime": 1.56,
      "eval_samples_per_second": 29.488,
      "eval_steps_per_second": 7.692,
      "step": 500
    },
    {
      "epoch": 0.891459074733096,
      "grad_norm": 4.1920693297660385,
      "learning_rate": 5.848354347695537e-06,
      "loss": 0.2311,
      "step": 501
    },
    {
      "epoch": 0.8932384341637011,
      "grad_norm": 3.481170304681233,
      "learning_rate": 5.83457862298631e-06,
      "loss": 0.1715,
      "step": 502
    },
    {
      "epoch": 0.895017793594306,
      "grad_norm": 4.068887670554145,
      "learning_rate": 5.8207963784797396e-06,
      "loss": 0.1669,
      "step": 503
    },
    {
      "epoch": 0.896797153024911,
      "grad_norm": 3.439840115170194,
      "learning_rate": 5.807007721843862e-06,
      "loss": 0.1662,
      "step": 504
    },
    {
      "epoch": 0.8985765124555161,
      "grad_norm": 4.256985163563125,
      "learning_rate": 5.793212760796804e-06,
      "loss": 0.2127,
      "step": 505
    },
    {
      "epoch": 0.900355871886121,
      "grad_norm": 3.492771107302106,
      "learning_rate": 5.779411603105947e-06,
      "loss": 0.1831,
      "step": 506
    },
    {
      "epoch": 0.902135231316726,
      "grad_norm": 3.2645072452193387,
      "learning_rate": 5.765604356587076e-06,
      "loss": 0.1495,
      "step": 507
    },
    {
      "epoch": 0.9039145907473309,
      "grad_norm": 3.587518130968404,
      "learning_rate": 5.751791129103545e-06,
      "loss": 0.155,
      "step": 508
    },
    {
      "epoch": 0.905693950177936,
      "grad_norm": 3.6632566621866216,
      "learning_rate": 5.737972028565431e-06,
      "loss": 0.1772,
      "step": 509
    },
    {
      "epoch": 0.9074733096085409,
      "grad_norm": 3.687239473432287,
      "learning_rate": 5.7241471629286934e-06,
      "loss": 0.1602,
      "step": 510
    },
    {
      "epoch": 0.9092526690391459,
      "grad_norm": 3.90800860559227,
      "learning_rate": 5.7103166401943276e-06,
      "loss": 0.1724,
      "step": 511
    },
    {
      "epoch": 0.9110320284697508,
      "grad_norm": 3.7565374005164913,
      "learning_rate": 5.696480568407523e-06,
      "loss": 0.1444,
      "step": 512
    },
    {
      "epoch": 0.9128113879003559,
      "grad_norm": 3.911662556944032,
      "learning_rate": 5.682639055656817e-06,
      "loss": 0.1823,
      "step": 513
    },
    {
      "epoch": 0.9145907473309609,
      "grad_norm": 4.338822362873238,
      "learning_rate": 5.668792210073255e-06,
      "loss": 0.2015,
      "step": 514
    },
    {
      "epoch": 0.9163701067615658,
      "grad_norm": 3.810545471776136,
      "learning_rate": 5.654940139829544e-06,
      "loss": 0.1576,
      "step": 515
    },
    {
      "epoch": 0.9181494661921709,
      "grad_norm": 3.4512180752005563,
      "learning_rate": 5.641082953139201e-06,
      "loss": 0.1382,
      "step": 516
    },
    {
      "epoch": 0.9199288256227758,
      "grad_norm": 3.1358034850283443,
      "learning_rate": 5.6272207582557195e-06,
      "loss": 0.1166,
      "step": 517
    },
    {
      "epoch": 0.9217081850533808,
      "grad_norm": 3.254105658788418,
      "learning_rate": 5.61335366347171e-06,
      "loss": 0.1212,
      "step": 518
    },
    {
      "epoch": 0.9234875444839857,
      "grad_norm": 3.5585651670615923,
      "learning_rate": 5.599481777118071e-06,
      "loss": 0.1688,
      "step": 519
    },
    {
      "epoch": 0.9252669039145908,
      "grad_norm": 3.824721937195978,
      "learning_rate": 5.585605207563124e-06,
      "loss": 0.1265,
      "step": 520
    },
    {
      "epoch": 0.9270462633451957,
      "grad_norm": 3.2627478599030444,
      "learning_rate": 5.571724063211782e-06,
      "loss": 0.1379,
      "step": 521
    },
    {
      "epoch": 0.9288256227758007,
      "grad_norm": 3.239149755416829,
      "learning_rate": 5.557838452504692e-06,
      "loss": 0.1083,
      "step": 522
    },
    {
      "epoch": 0.9306049822064056,
      "grad_norm": 4.133270014167604,
      "learning_rate": 5.5439484839173996e-06,
      "loss": 0.1708,
      "step": 523
    },
    {
      "epoch": 0.9323843416370107,
      "grad_norm": 3.693919492585582,
      "learning_rate": 5.530054265959486e-06,
      "loss": 0.1454,
      "step": 524
    },
    {
      "epoch": 0.9341637010676157,
      "grad_norm": 3.5555941387932632,
      "learning_rate": 5.516155907173735e-06,
      "loss": 0.1736,
      "step": 525
    },
    {
      "epoch": 0.9359430604982206,
      "grad_norm": 2.998891359424065,
      "learning_rate": 5.5022535161352764e-06,
      "loss": 0.1358,
      "step": 526
    },
    {
      "epoch": 0.9377224199288257,
      "grad_norm": 4.3195155515396415,
      "learning_rate": 5.488347201450741e-06,
      "loss": 0.1519,
      "step": 527
    },
    {
      "epoch": 0.9395017793594306,
      "grad_norm": 3.0991300433328632,
      "learning_rate": 5.47443707175741e-06,
      "loss": 0.1489,
      "step": 528
    },
    {
      "epoch": 0.9412811387900356,
      "grad_norm": 3.3398044876279043,
      "learning_rate": 5.46052323572237e-06,
      "loss": 0.1348,
      "step": 529
    },
    {
      "epoch": 0.9430604982206405,
      "grad_norm": 2.8922913936014134,
      "learning_rate": 5.446605802041662e-06,
      "loss": 0.1145,
      "step": 530
    },
    {
      "epoch": 0.9448398576512456,
      "grad_norm": 2.764927680546361,
      "learning_rate": 5.432684879439428e-06,
      "loss": 0.1223,
      "step": 531
    },
    {
      "epoch": 0.9466192170818505,
      "grad_norm": 3.030472111047228,
      "learning_rate": 5.418760576667071e-06,
      "loss": 0.1022,
      "step": 532
    },
    {
      "epoch": 0.9483985765124555,
      "grad_norm": 3.6427565854970543,
      "learning_rate": 5.404833002502398e-06,
      "loss": 0.1926,
      "step": 533
    },
    {
      "epoch": 0.9501779359430605,
      "grad_norm": 4.195710217852611,
      "learning_rate": 5.39090226574877e-06,
      "loss": 0.1644,
      "step": 534
    },
    {
      "epoch": 0.9519572953736655,
      "grad_norm": 4.363267396256809,
      "learning_rate": 5.376968475234258e-06,
      "loss": 0.202,
      "step": 535
    },
    {
      "epoch": 0.9537366548042705,
      "grad_norm": 2.7821088387473947,
      "learning_rate": 5.363031739810787e-06,
      "loss": 0.1141,
      "step": 536
    },
    {
      "epoch": 0.9555160142348754,
      "grad_norm": 3.6867524621327292,
      "learning_rate": 5.349092168353291e-06,
      "loss": 0.1664,
      "step": 537
    },
    {
      "epoch": 0.9572953736654805,
      "grad_norm": 3.125382210169913,
      "learning_rate": 5.335149869758855e-06,
      "loss": 0.1266,
      "step": 538
    },
    {
      "epoch": 0.9590747330960854,
      "grad_norm": 3.9339155616663413,
      "learning_rate": 5.32120495294587e-06,
      "loss": 0.1538,
      "step": 539
    },
    {
      "epoch": 0.9608540925266904,
      "grad_norm": 3.2222430592689277,
      "learning_rate": 5.3072575268531835e-06,
      "loss": 0.1598,
      "step": 540
    },
    {
      "epoch": 0.9626334519572953,
      "grad_norm": 3.7576395107766323,
      "learning_rate": 5.293307700439242e-06,
      "loss": 0.1677,
      "step": 541
    },
    {
      "epoch": 0.9644128113879004,
      "grad_norm": 4.658503930442991,
      "learning_rate": 5.2793555826812456e-06,
      "loss": 0.1792,
      "step": 542
    },
    {
      "epoch": 0.9661921708185054,
      "grad_norm": 3.449898452228417,
      "learning_rate": 5.265401282574294e-06,
      "loss": 0.1194,
      "step": 543
    },
    {
      "epoch": 0.9679715302491103,
      "grad_norm": 3.780378336702093,
      "learning_rate": 5.2514449091305375e-06,
      "loss": 0.1486,
      "step": 544
    },
    {
      "epoch": 0.9697508896797153,
      "grad_norm": 4.052822533750717,
      "learning_rate": 5.237486571378317e-06,
      "loss": 0.1594,
      "step": 545
    },
    {
      "epoch": 0.9715302491103203,
      "grad_norm": 3.4526419318159847,
      "learning_rate": 5.22352637836133e-06,
      "loss": 0.1929,
      "step": 546
    },
    {
      "epoch": 0.9733096085409253,
      "grad_norm": 3.4355419286827162,
      "learning_rate": 5.209564439137755e-06,
      "loss": 0.1401,
      "step": 547
    },
    {
      "epoch": 0.9750889679715302,
      "grad_norm": 3.6361729830059604,
      "learning_rate": 5.195600862779421e-06,
      "loss": 0.1989,
      "step": 548
    },
    {
      "epoch": 0.9768683274021353,
      "grad_norm": 2.628883993664625,
      "learning_rate": 5.181635758370942e-06,
      "loss": 0.099,
      "step": 549
    },
    {
      "epoch": 0.9786476868327402,
      "grad_norm": 3.823926080803227,
      "learning_rate": 5.167669235008871e-06,
      "loss": 0.1809,
      "step": 550
    },
    {
      "epoch": 0.9804270462633452,
      "grad_norm": 4.384226851641799,
      "learning_rate": 5.153701401800845e-06,
      "loss": 0.1999,
      "step": 551
    },
    {
      "epoch": 0.9822064056939501,
      "grad_norm": 3.6395346152019403,
      "learning_rate": 5.139732367864736e-06,
      "loss": 0.1617,
      "step": 552
    },
    {
      "epoch": 0.9839857651245552,
      "grad_norm": 3.572063023035529,
      "learning_rate": 5.1257622423277934e-06,
      "loss": 0.1432,
      "step": 553
    },
    {
      "epoch": 0.9857651245551602,
      "grad_norm": 3.150287807537195,
      "learning_rate": 5.111791134325793e-06,
      "loss": 0.1433,
      "step": 554
    },
    {
      "epoch": 0.9875444839857651,
      "grad_norm": 2.841048418910797,
      "learning_rate": 5.097819153002192e-06,
      "loss": 0.1299,
      "step": 555
    },
    {
      "epoch": 0.9893238434163701,
      "grad_norm": 3.0463697903034554,
      "learning_rate": 5.083846407507263e-06,
      "loss": 0.1251,
      "step": 556
    },
    {
      "epoch": 0.9911032028469751,
      "grad_norm": 4.0997746699318,
      "learning_rate": 5.0698730069972535e-06,
      "loss": 0.2014,
      "step": 557
    },
    {
      "epoch": 0.9928825622775801,
      "grad_norm": 3.1493757472338215,
      "learning_rate": 5.055899060633524e-06,
      "loss": 0.1188,
      "step": 558
    },
    {
      "epoch": 0.994661921708185,
      "grad_norm": 3.465724398859495,
      "learning_rate": 5.041924677581702e-06,
      "loss": 0.1628,
      "step": 559
    },
    {
      "epoch": 0.99644128113879,
      "grad_norm": 3.4419781973237473,
      "learning_rate": 5.0279499670108245e-06,
      "loss": 0.1569,
      "step": 560
    },
    {
      "epoch": 0.998220640569395,
      "grad_norm": 3.1311392413331784,
      "learning_rate": 5.013975038092491e-06,
      "loss": 0.138,
      "step": 561
    },
    {
      "epoch": 1.0,
      "grad_norm": 3.0067449939210547,
      "learning_rate": 5e-06,
      "loss": 0.1114,
      "step": 562
    },
    {
      "epoch": 1.001779359430605,
      "grad_norm": 2.5428196016266096,
      "learning_rate": 4.98602496190751e-06,
      "loss": 0.082,
      "step": 563
    },
    {
      "epoch": 1.00355871886121,
      "grad_norm": 2.070523921419281,
      "learning_rate": 4.9720500329891755e-06,
      "loss": 0.0663,
      "step": 564
    },
    {
      "epoch": 1.0053380782918149,
      "grad_norm": 3.2074185839641105,
      "learning_rate": 4.9580753224183005e-06,
      "loss": 0.1032,
      "step": 565
    },
    {
      "epoch": 1.00711743772242,
      "grad_norm": 2.5673028042218387,
      "learning_rate": 4.944100939366478e-06,
      "loss": 0.0767,
      "step": 566
    },
    {
      "epoch": 1.008896797153025,
      "grad_norm": 2.0130260935723396,
      "learning_rate": 4.930126993002748e-06,
      "loss": 0.0673,
      "step": 567
    },
    {
      "epoch": 1.01067615658363,
      "grad_norm": 2.4696439550185425,
      "learning_rate": 4.9161535924927375e-06,
      "loss": 0.0757,
      "step": 568
    },
    {
      "epoch": 1.0124555160142348,
      "grad_norm": 2.4302284676813923,
      "learning_rate": 4.90218084699781e-06,
      "loss": 0.0651,
      "step": 569
    },
    {
      "epoch": 1.0142348754448398,
      "grad_norm": 2.9208576105489352,
      "learning_rate": 4.888208865674208e-06,
      "loss": 0.0956,
      "step": 570
    },
    {
      "epoch": 1.0160142348754448,
      "grad_norm": 2.5863537127322824,
      "learning_rate": 4.874237757672209e-06,
      "loss": 0.0809,
      "step": 571
    },
    {
      "epoch": 1.0177935943060499,
      "grad_norm": 3.4849484393831114,
      "learning_rate": 4.8602676321352646e-06,
      "loss": 0.112,
      "step": 572
    },
    {
      "epoch": 1.019572953736655,
      "grad_norm": 3.023643189876397,
      "learning_rate": 4.846298598199155e-06,
      "loss": 0.0749,
      "step": 573
    },
    {
      "epoch": 1.0213523131672597,
      "grad_norm": 3.007607244394418,
      "learning_rate": 4.832330764991131e-06,
      "loss": 0.0725,
      "step": 574
    },
    {
      "epoch": 1.0231316725978647,
      "grad_norm": 3.527010108523154,
      "learning_rate": 4.81836424162906e-06,
      "loss": 0.0979,
      "step": 575
    },
    {
      "epoch": 1.0249110320284698,
      "grad_norm": 3.299696949614294,
      "learning_rate": 4.80439913722058e-06,
      "loss": 0.0609,
      "step": 576
    },
    {
      "epoch": 1.0266903914590748,
      "grad_norm": 3.2208500754421516,
      "learning_rate": 4.790435560862247e-06,
      "loss": 0.0868,
      "step": 577
    },
    {
      "epoch": 1.0284697508896796,
      "grad_norm": 3.6669111789081774,
      "learning_rate": 4.776473621638673e-06,
      "loss": 0.0688,
      "step": 578
    },
    {
      "epoch": 1.0302491103202847,
      "grad_norm": 3.6936612623811347,
      "learning_rate": 4.762513428621684e-06,
      "loss": 0.0831,
      "step": 579
    },
    {
      "epoch": 1.0320284697508897,
      "grad_norm": 3.715419134232282,
      "learning_rate": 4.748555090869464e-06,
      "loss": 0.0662,
      "step": 580
    },
    {
      "epoch": 1.0338078291814947,
      "grad_norm": 3.4702614508448915,
      "learning_rate": 4.734598717425706e-06,
      "loss": 0.0868,
      "step": 581
    },
    {
      "epoch": 1.0355871886120998,
      "grad_norm": 4.405013855673697,
      "learning_rate": 4.720644417318755e-06,
      "loss": 0.1062,
      "step": 582
    },
    {
      "epoch": 1.0373665480427046,
      "grad_norm": 4.932211058513835,
      "learning_rate": 4.70669229956076e-06,
      "loss": 0.0859,
      "step": 583
    },
    {
      "epoch": 1.0391459074733096,
      "grad_norm": 3.389056538199424,
      "learning_rate": 4.692742473146818e-06,
      "loss": 0.0756,
      "step": 584
    },
    {
      "epoch": 1.0409252669039146,
      "grad_norm": 3.1103074476478656,
      "learning_rate": 4.678795047054131e-06,
      "loss": 0.0849,
      "step": 585
    },
    {
      "epoch": 1.0427046263345197,
      "grad_norm": 3.56012940771853,
      "learning_rate": 4.664850130241146e-06,
      "loss": 0.0732,
      "step": 586
    },
    {
      "epoch": 1.0444839857651245,
      "grad_norm": 3.051217886420107,
      "learning_rate": 4.650907831646711e-06,
      "loss": 0.0649,
      "step": 587
    },
    {
      "epoch": 1.0462633451957295,
      "grad_norm": 2.923919045285875,
      "learning_rate": 4.636968260189214e-06,
      "loss": 0.1006,
      "step": 588
    },
    {
      "epoch": 1.0480427046263345,
      "grad_norm": 3.3479153559580768,
      "learning_rate": 4.623031524765744e-06,
      "loss": 0.0659,
      "step": 589
    },
    {
      "epoch": 1.0498220640569396,
      "grad_norm": 3.0358339355107256,
      "learning_rate": 4.609097734251231e-06,
      "loss": 0.0759,
      "step": 590
    },
    {
      "epoch": 1.0516014234875444,
      "grad_norm": 3.559919778427983,
      "learning_rate": 4.595166997497605e-06,
      "loss": 0.0641,
      "step": 591
    },
    {
      "epoch": 1.0533807829181494,
      "grad_norm": 2.8598289054393438,
      "learning_rate": 4.58123942333293e-06,
      "loss": 0.0749,
      "step": 592
    },
    {
      "epoch": 1.0551601423487544,
      "grad_norm": 2.8320235628710178,
      "learning_rate": 4.567315120560573e-06,
      "loss": 0.0675,
      "step": 593
    },
    {
      "epoch": 1.0569395017793595,
      "grad_norm": 3.024996554343902,
      "learning_rate": 4.553394197958339e-06,
      "loss": 0.0718,
      "step": 594
    },
    {
      "epoch": 1.0587188612099645,
      "grad_norm": 2.7652556255632184,
      "learning_rate": 4.539476764277631e-06,
      "loss": 0.0595,
      "step": 595
    },
    {
      "epoch": 1.0604982206405693,
      "grad_norm": 2.801693096700755,
      "learning_rate": 4.525562928242592e-06,
      "loss": 0.0558,
      "step": 596
    },
    {
      "epoch": 1.0622775800711743,
      "grad_norm": 3.49182241268502,
      "learning_rate": 4.511652798549261e-06,
      "loss": 0.0802,
      "step": 597
    },
    {
      "epoch": 1.0640569395017794,
      "grad_norm": 3.5925423790833424,
      "learning_rate": 4.497746483864725e-06,
      "loss": 0.0641,
      "step": 598
    },
    {
      "epoch": 1.0658362989323844,
      "grad_norm": 2.7317047661684373,
      "learning_rate": 4.483844092826267e-06,
      "loss": 0.0506,
      "step": 599
    },
    {
      "epoch": 1.0676156583629894,
      "grad_norm": 2.3357672871007846,
      "learning_rate": 4.469945734040516e-06,
      "loss": 0.061,
      "step": 600
    },
    {
      "epoch": 1.0693950177935942,
      "grad_norm": 3.3921690976793215,
      "learning_rate": 4.456051516082603e-06,
      "loss": 0.0669,
      "step": 601
    },
    {
      "epoch": 1.0711743772241993,
      "grad_norm": 4.089559936207029,
      "learning_rate": 4.442161547495309e-06,
      "loss": 0.0881,
      "step": 602
    },
    {
      "epoch": 1.0729537366548043,
      "grad_norm": 4.180039570388092,
      "learning_rate": 4.42827593678822e-06,
      "loss": 0.0834,
      "step": 603
    },
    {
      "epoch": 1.0747330960854093,
      "grad_norm": 3.495798185465847,
      "learning_rate": 4.414394792436877e-06,
      "loss": 0.0681,
      "step": 604
    },
    {
      "epoch": 1.0765124555160142,
      "grad_norm": 3.3378420594876683,
      "learning_rate": 4.400518222881931e-06,
      "loss": 0.0838,
      "step": 605
    },
    {
      "epoch": 1.0782918149466192,
      "grad_norm": 3.131215350773149,
      "learning_rate": 4.386646336528291e-06,
      "loss": 0.0653,
      "step": 606
    },
    {
      "epoch": 1.0800711743772242,
      "grad_norm": 2.9032685674535705,
      "learning_rate": 4.372779241744282e-06,
      "loss": 0.062,
      "step": 607
    },
    {
      "epoch": 1.0818505338078293,
      "grad_norm": 3.042555205092551,
      "learning_rate": 4.358917046860799e-06,
      "loss": 0.0726,
      "step": 608
    },
    {
      "epoch": 1.083629893238434,
      "grad_norm": 2.9071801091458904,
      "learning_rate": 4.345059860170458e-06,
      "loss": 0.0765,
      "step": 609
    },
    {
      "epoch": 1.085409252669039,
      "grad_norm": 6.695432609048516,
      "learning_rate": 4.331207789926746e-06,
      "loss": 0.072,
      "step": 610
    },
    {
      "epoch": 1.0871886120996441,
      "grad_norm": 2.945964073158328,
      "learning_rate": 4.317360944343184e-06,
      "loss": 0.0662,
      "step": 611
    },
    {
      "epoch": 1.0889679715302492,
      "grad_norm": 3.733354431293984,
      "learning_rate": 4.303519431592479e-06,
      "loss": 0.062,
      "step": 612
    },
    {
      "epoch": 1.0907473309608542,
      "grad_norm": 3.9833317366454994,
      "learning_rate": 4.289683359805673e-06,
      "loss": 0.089,
      "step": 613
    },
    {
      "epoch": 1.092526690391459,
      "grad_norm": 3.5266700085633405,
      "learning_rate": 4.275852837071309e-06,
      "loss": 0.0776,
      "step": 614
    },
    {
      "epoch": 1.094306049822064,
      "grad_norm": 2.654294722059649,
      "learning_rate": 4.26202797143457e-06,
      "loss": 0.0645,
      "step": 615
    },
    {
      "epoch": 1.096085409252669,
      "grad_norm": 2.9999667440216884,
      "learning_rate": 4.248208870896456e-06,
      "loss": 0.0704,
      "step": 616
    },
    {
      "epoch": 1.097864768683274,
      "grad_norm": 3.27113446294257,
      "learning_rate": 4.234395643412925e-06,
      "loss": 0.0674,
      "step": 617
    },
    {
      "epoch": 1.099644128113879,
      "grad_norm": 3.255141312453534,
      "learning_rate": 4.220588396894055e-06,
      "loss": 0.0668,
      "step": 618
    },
    {
      "epoch": 1.101423487544484,
      "grad_norm": 3.1785993848046727,
      "learning_rate": 4.2067872392031965e-06,
      "loss": 0.0685,
      "step": 619
    },
    {
      "epoch": 1.103202846975089,
      "grad_norm": 3.356303973778546,
      "learning_rate": 4.192992278156141e-06,
      "loss": 0.0703,
      "step": 620
    },
    {
      "epoch": 1.104982206405694,
      "grad_norm": 3.4918532643293783,
      "learning_rate": 4.179203621520262e-06,
      "loss": 0.0772,
      "step": 621
    },
    {
      "epoch": 1.106761565836299,
      "grad_norm": 3.5930488736165747,
      "learning_rate": 4.165421377013691e-06,
      "loss": 0.0562,
      "step": 622
    },
    {
      "epoch": 1.1085409252669038,
      "grad_norm": 3.652015089949573,
      "learning_rate": 4.151645652304465e-06,
      "loss": 0.0699,
      "step": 623
    },
    {
      "epoch": 1.1103202846975089,
      "grad_norm": 3.464122740485754,
      "learning_rate": 4.137876555009684e-06,
      "loss": 0.0879,
      "step": 624
    },
    {
      "epoch": 1.112099644128114,
      "grad_norm": 3.1026520583321333,
      "learning_rate": 4.124114192694676e-06,
      "loss": 0.0614,
      "step": 625
    },
    {
      "epoch": 1.113879003558719,
      "grad_norm": 3.7271258051590372,
      "learning_rate": 4.110358672872158e-06,
      "loss": 0.0777,
      "step": 626
    },
    {
      "epoch": 1.1156583629893237,
      "grad_norm": 3.2840872759027397,
      "learning_rate": 4.0966101030013915e-06,
      "loss": 0.089,
      "step": 627
    },
    {
      "epoch": 1.1174377224199288,
      "grad_norm": 2.852238361314974,
      "learning_rate": 4.082868590487339e-06,
      "loss": 0.0668,
      "step": 628
    },
    {
      "epoch": 1.1192170818505338,
      "grad_norm": 2.636541631537876,
      "learning_rate": 4.069134242679837e-06,
      "loss": 0.0576,
      "step": 629
    },
    {
      "epoch": 1.1209964412811388,
      "grad_norm": 3.714611285468836,
      "learning_rate": 4.055407166872748e-06,
      "loss": 0.069,
      "step": 630
    },
    {
      "epoch": 1.1227758007117439,
      "grad_norm": 4.112584241624248,
      "learning_rate": 4.041687470303127e-06,
      "loss": 0.0934,
      "step": 631
    },
    {
      "epoch": 1.1245551601423487,
      "grad_norm": 3.2252768414370583,
      "learning_rate": 4.02797526015038e-06,
      "loss": 0.0957,
      "step": 632
    },
    {
      "epoch": 1.1263345195729537,
      "grad_norm": 3.3660979258039827,
      "learning_rate": 4.014270643535427e-06,
      "loss": 0.0627,
      "step": 633
    },
    {
      "epoch": 1.1281138790035588,
      "grad_norm": 3.5956426803534685,
      "learning_rate": 4.000573727519868e-06,
      "loss": 0.0825,
      "step": 634
    },
    {
      "epoch": 1.1298932384341638,
      "grad_norm": 3.103661800001212,
      "learning_rate": 3.9868846191051465e-06,
      "loss": 0.085,
      "step": 635
    },
    {
      "epoch": 1.1316725978647686,
      "grad_norm": 3.4011545814622224,
      "learning_rate": 3.973203425231715e-06,
      "loss": 0.0835,
      "step": 636
    },
    {
      "epoch": 1.1334519572953736,
      "grad_norm": 2.8305947706191548,
      "learning_rate": 3.959530252778187e-06,
      "loss": 0.0963,
      "step": 637
    },
    {
      "epoch": 1.1352313167259787,
      "grad_norm": 3.308586207005005,
      "learning_rate": 3.945865208560522e-06,
      "loss": 0.0924,
      "step": 638
    },
    {
      "epoch": 1.1370106761565837,
      "grad_norm": 3.0421378691997583,
      "learning_rate": 3.932208399331177e-06,
      "loss": 0.0953,
      "step": 639
    },
    {
      "epoch": 1.1387900355871885,
      "grad_norm": 3.7684823453299687,
      "learning_rate": 3.918559931778277e-06,
      "loss": 0.0783,
      "step": 640
    },
    {
      "epoch": 1.1405693950177935,
      "grad_norm": 2.7161766769209423,
      "learning_rate": 3.904919912524784e-06,
      "loss": 0.0517,
      "step": 641
    },
    {
      "epoch": 1.1423487544483986,
      "grad_norm": 2.7128113994663834,
      "learning_rate": 3.891288448127654e-06,
      "loss": 0.0659,
      "step": 642
    },
    {
      "epoch": 1.1441281138790036,
      "grad_norm": 3.495333170689702,
      "learning_rate": 3.877665645077017e-06,
      "loss": 0.0799,
      "step": 643
    },
    {
      "epoch": 1.1459074733096086,
      "grad_norm": 3.618492342183983,
      "learning_rate": 3.86405160979534e-06,
      "loss": 0.0885,
      "step": 644
    },
    {
      "epoch": 1.1476868327402134,
      "grad_norm": 3.0456993727793304,
      "learning_rate": 3.850446448636597e-06,
      "loss": 0.0588,
      "step": 645
    },
    {
      "epoch": 1.1494661921708185,
      "grad_norm": 2.825178944906573,
      "learning_rate": 3.8368502678854296e-06,
      "loss": 0.0545,
      "step": 646
    },
    {
      "epoch": 1.1512455516014235,
      "grad_norm": 3.0936271152928545,
      "learning_rate": 3.8232631737563306e-06,
      "loss": 0.0606,
      "step": 647
    },
    {
      "epoch": 1.1530249110320285,
      "grad_norm": 3.298535161622391,
      "learning_rate": 3.809685272392804e-06,
      "loss": 0.0691,
      "step": 648
    },
    {
      "epoch": 1.1548042704626336,
      "grad_norm": 3.8123086681773084,
      "learning_rate": 3.796116669866543e-06,
      "loss": 0.0922,
      "step": 649
    },
    {
      "epoch": 1.1565836298932384,
      "grad_norm": 2.7999871821454048,
      "learning_rate": 3.78255747217659e-06,
      "loss": 0.0604,
      "step": 650
    },
    {
      "epoch": 1.1583629893238434,
      "grad_norm": 2.971190570111659,
      "learning_rate": 3.769007785248523e-06,
      "loss": 0.059,
      "step": 651
    },
    {
      "epoch": 1.1601423487544484,
      "grad_norm": 3.0172485450763333,
      "learning_rate": 3.7554677149336186e-06,
      "loss": 0.0688,
      "step": 652
    },
    {
      "epoch": 1.1619217081850535,
      "grad_norm": 3.36341340138087,
      "learning_rate": 3.7419373670080284e-06,
      "loss": 0.0912,
      "step": 653
    },
    {
      "epoch": 1.1637010676156583,
      "grad_norm": 3.6639063883943046,
      "learning_rate": 3.7284168471719527e-06,
      "loss": 0.0886,
      "step": 654
    },
    {
      "epoch": 1.1654804270462633,
      "grad_norm": 3.151744033617198,
      "learning_rate": 3.7149062610488085e-06,
      "loss": 0.0779,
      "step": 655
    },
    {
      "epoch": 1.1672597864768683,
      "grad_norm": 3.1408174810804157,
      "learning_rate": 3.701405714184416e-06,
      "loss": 0.0753,
      "step": 656
    },
    {
      "epoch": 1.1690391459074734,
      "grad_norm": 2.682543747824902,
      "learning_rate": 3.687915312046166e-06,
      "loss": 0.0797,
      "step": 657
    },
    {
      "epoch": 1.1708185053380782,
      "grad_norm": 2.8104836027708266,
      "learning_rate": 3.6744351600221994e-06,
      "loss": 0.0527,
      "step": 658
    },
    {
      "epoch": 1.1725978647686832,
      "grad_norm": 3.177096326345197,
      "learning_rate": 3.6609653634205773e-06,
      "loss": 0.1183,
      "step": 659
    },
    {
      "epoch": 1.1743772241992882,
      "grad_norm": 4.480361218299358,
      "learning_rate": 3.647506027468467e-06,
      "loss": 0.0774,
      "step": 660
    },
    {
      "epoch": 1.1761565836298933,
      "grad_norm": 3.1937674421689395,
      "learning_rate": 3.6340572573113176e-06,
      "loss": 0.0776,
      "step": 661
    },
    {
      "epoch": 1.1779359430604983,
      "grad_norm": 2.7201992456922235,
      "learning_rate": 3.6206191580120346e-06,
      "loss": 0.0694,
      "step": 662
    },
    {
      "epoch": 1.1797153024911031,
      "grad_norm": 3.11645628384522,
      "learning_rate": 3.6071918345501655e-06,
      "loss": 0.0652,
      "step": 663
    },
    {
      "epoch": 1.1814946619217082,
      "grad_norm": 3.2538041786921186,
      "learning_rate": 3.5937753918210705e-06,
      "loss": 0.0695,
      "step": 664
    },
    {
      "epoch": 1.1832740213523132,
      "grad_norm": 3.140350660846512,
      "learning_rate": 3.5803699346351117e-06,
      "loss": 0.0692,
      "step": 665
    },
    {
      "epoch": 1.1850533807829182,
      "grad_norm": 2.4758910974036334,
      "learning_rate": 3.566975567716833e-06,
      "loss": 0.0667,
      "step": 666
    },
    {
      "epoch": 1.1868327402135233,
      "grad_norm": 3.1814443286799134,
      "learning_rate": 3.5535923957041374e-06,
      "loss": 0.0668,
      "step": 667
    },
    {
      "epoch": 1.188612099644128,
      "grad_norm": 2.704127431911285,
      "learning_rate": 3.540220523147474e-06,
      "loss": 0.0656,
      "step": 668
    },
    {
      "epoch": 1.190391459074733,
      "grad_norm": 3.137071663289649,
      "learning_rate": 3.5268600545090183e-06,
      "loss": 0.0623,
      "step": 669
    },
    {
      "epoch": 1.1921708185053381,
      "grad_norm": 2.8284964088390785,
      "learning_rate": 3.513511094161858e-06,
      "loss": 0.0668,
      "step": 670
    },
    {
      "epoch": 1.193950177935943,
      "grad_norm": 2.964749809920623,
      "learning_rate": 3.5001737463891793e-06,
      "loss": 0.0568,
      "step": 671
    },
    {
      "epoch": 1.195729537366548,
      "grad_norm": 2.826877684920855,
      "learning_rate": 3.4868481153834454e-06,
      "loss": 0.0888,
      "step": 672
    },
    {
      "epoch": 1.197508896797153,
      "grad_norm": 4.927703620196868,
      "learning_rate": 3.4735343052455905e-06,
      "loss": 0.0648,
      "step": 673
    },
    {
      "epoch": 1.199288256227758,
      "grad_norm": 3.243097957336036,
      "learning_rate": 3.4602324199842026e-06,
      "loss": 0.0683,
      "step": 674
    },
    {
      "epoch": 1.201067615658363,
      "grad_norm": 2.834599065216145,
      "learning_rate": 3.446942563514711e-06,
      "loss": 0.0701,
      "step": 675
    },
    {
      "epoch": 1.2028469750889679,
      "grad_norm": 3.1066371333102074,
      "learning_rate": 3.4336648396585777e-06,
      "loss": 0.0523,
      "step": 676
    },
    {
      "epoch": 1.204626334519573,
      "grad_norm": 2.8807198284484907,
      "learning_rate": 3.4203993521424774e-06,
      "loss": 0.0806,
      "step": 677
    },
    {
      "epoch": 1.206405693950178,
      "grad_norm": 3.2221596006021573,
      "learning_rate": 3.407146204597499e-06,
      "loss": 0.0778,
      "step": 678
    },
    {
      "epoch": 1.208185053380783,
      "grad_norm": 2.9214122156037163,
      "learning_rate": 3.3939055005583305e-06,
      "loss": 0.0597,
      "step": 679
    },
    {
      "epoch": 1.209964412811388,
      "grad_norm": 5.149213903953065,
      "learning_rate": 3.3806773434624475e-06,
      "loss": 0.0758,
      "step": 680
    },
    {
      "epoch": 1.2117437722419928,
      "grad_norm": 3.072774454153599,
      "learning_rate": 3.3674618366493117e-06,
      "loss": 0.0924,
      "step": 681
    },
    {
      "epoch": 1.2135231316725978,
      "grad_norm": 2.9624031675173774,
      "learning_rate": 3.3542590833595533e-06,
      "loss": 0.0701,
      "step": 682
    },
    {
      "epoch": 1.2153024911032029,
      "grad_norm": 3.099121775691056,
      "learning_rate": 3.341069186734176e-06,
      "loss": 0.0701,
      "step": 683
    },
    {
      "epoch": 1.217081850533808,
      "grad_norm": 2.6063212077427393,
      "learning_rate": 3.3278922498137455e-06,
      "loss": 0.0555,
      "step": 684
    },
    {
      "epoch": 1.2188612099644127,
      "grad_norm": 3.4022509056311194,
      "learning_rate": 3.314728375537587e-06,
      "loss": 0.0716,
      "step": 685
    },
    {
      "epoch": 1.2206405693950177,
      "grad_norm": 2.6434303672909567,
      "learning_rate": 3.3015776667429724e-06,
      "loss": 0.0867,
      "step": 686
    },
    {
      "epoch": 1.2224199288256228,
      "grad_norm": 3.5643054580686186,
      "learning_rate": 3.2884402261643296e-06,
      "loss": 0.0717,
      "step": 687
    },
    {
      "epoch": 1.2241992882562278,
      "grad_norm": 3.5698215173997814,
      "learning_rate": 3.2753161564324344e-06,
      "loss": 0.0721,
      "step": 688
    },
    {
      "epoch": 1.2259786476868326,
      "grad_norm": 4.373339611779974,
      "learning_rate": 3.262205560073605e-06,
      "loss": 0.0712,
      "step": 689
    },
    {
      "epoch": 1.2277580071174377,
      "grad_norm": 2.968350020755562,
      "learning_rate": 3.249108539508909e-06,
      "loss": 0.0507,
      "step": 690
    },
    {
      "epoch": 1.2295373665480427,
      "grad_norm": 4.4988607831566085,
      "learning_rate": 3.2360251970533527e-06,
      "loss": 0.1131,
      "step": 691
    },
    {
      "epoch": 1.2313167259786477,
      "grad_norm": 3.1911810466805024,
      "learning_rate": 3.2229556349150947e-06,
      "loss": 0.0765,
      "step": 692
    },
    {
      "epoch": 1.2330960854092528,
      "grad_norm": 2.792403915143028,
      "learning_rate": 3.2098999551946337e-06,
      "loss": 0.0524,
      "step": 693
    },
    {
      "epoch": 1.2348754448398576,
      "grad_norm": 3.7794063502876556,
      "learning_rate": 3.1968582598840234e-06,
      "loss": 0.087,
      "step": 694
    },
    {
      "epoch": 1.2366548042704626,
      "grad_norm": 4.057263636091964,
      "learning_rate": 3.183830650866068e-06,
      "loss": 0.0707,
      "step": 695
    },
    {
      "epoch": 1.2384341637010676,
      "grad_norm": 3.1557300823298964,
      "learning_rate": 3.1708172299135266e-06,
      "loss": 0.0656,
      "step": 696
    },
    {
      "epoch": 1.2402135231316727,
      "grad_norm": 4.245500645222655,
      "learning_rate": 3.1578180986883234e-06,
      "loss": 0.1018,
      "step": 697
    },
    {
      "epoch": 1.2419928825622777,
      "grad_norm": 3.0624323355157497,
      "learning_rate": 3.1448333587407486e-06,
      "loss": 0.0689,
      "step": 698
    },
    {
      "epoch": 1.2437722419928825,
      "grad_norm": 2.8737499216665308,
      "learning_rate": 3.131863111508667e-06,
      "loss": 0.0677,
      "step": 699
    },
    {
      "epoch": 1.2455516014234875,
      "grad_norm": 3.1462684388402784,
      "learning_rate": 3.118907458316722e-06,
      "loss": 0.0634,
      "step": 700
    },
    {
      "epoch": 1.2473309608540926,
      "grad_norm": 2.821577544766221,
      "learning_rate": 3.105966500375551e-06,
      "loss": 0.0614,
      "step": 701
    },
    {
      "epoch": 1.2491103202846976,
      "grad_norm": 3.709904045922049,
      "learning_rate": 3.0930403387809892e-06,
      "loss": 0.0974,
      "step": 702
    },
    {
      "epoch": 1.2508896797153026,
      "grad_norm": 3.282003926508322,
      "learning_rate": 3.080129074513285e-06,
      "loss": 0.0793,
      "step": 703
    },
    {
      "epoch": 1.2526690391459074,
      "grad_norm": 3.1785599261011965,
      "learning_rate": 3.067232808436299e-06,
      "loss": 0.072,
      "step": 704
    },
    {
      "epoch": 1.2544483985765125,
      "grad_norm": 3.240135791061997,
      "learning_rate": 3.0543516412967327e-06,
      "loss": 0.0653,
      "step": 705
    },
    {
      "epoch": 1.2562277580071175,
      "grad_norm": 2.457212859512729,
      "learning_rate": 3.041485673723331e-06,
      "loss": 0.0597,
      "step": 706
    },
    {
      "epoch": 1.2580071174377223,
      "grad_norm": 3.6580044362060216,
      "learning_rate": 3.0286350062261017e-06,
      "loss": 0.0759,
      "step": 707
    },
    {
      "epoch": 1.2597864768683273,
      "grad_norm": 2.472710826112781,
      "learning_rate": 3.0157997391955172e-06,
      "loss": 0.0621,
      "step": 708
    },
    {
      "epoch": 1.2615658362989324,
      "grad_norm": 3.2273089140358793,
      "learning_rate": 3.0029799729017518e-06,
      "loss": 0.074,
      "step": 709
    },
    {
      "epoch": 1.2633451957295374,
      "grad_norm": 2.5959014517277486,
      "learning_rate": 2.9901758074938797e-06,
      "loss": 0.0678,
      "step": 710
    },
    {
      "epoch": 1.2651245551601424,
      "grad_norm": 3.3129931092478357,
      "learning_rate": 2.977387342999103e-06,
      "loss": 0.0708,
      "step": 711
    },
    {
      "epoch": 1.2669039145907472,
      "grad_norm": 3.139602593892867,
      "learning_rate": 2.964614679321966e-06,
      "loss": 0.0632,
      "step": 712
    },
    {
      "epoch": 1.2686832740213523,
      "grad_norm": 2.575914299100435,
      "learning_rate": 2.951857916243574e-06,
      "loss": 0.0729,
      "step": 713
    },
    {
      "epoch": 1.2704626334519573,
      "grad_norm": 4.324662632577459,
      "learning_rate": 2.9391171534208185e-06,
      "loss": 0.1095,
      "step": 714
    },
    {
      "epoch": 1.2722419928825623,
      "grad_norm": 3.814249772985952,
      "learning_rate": 2.9263924903855932e-06,
      "loss": 0.0593,
      "step": 715
    },
    {
      "epoch": 1.2740213523131674,
      "grad_norm": 3.2565457360639507,
      "learning_rate": 2.9136840265440213e-06,
      "loss": 0.0728,
      "step": 716
    },
    {
      "epoch": 1.2758007117437722,
      "grad_norm": 3.347768465163155,
      "learning_rate": 2.9009918611756732e-06,
      "loss": 0.0846,
      "step": 717
    },
    {
      "epoch": 1.2775800711743772,
      "grad_norm": 2.917314484152546,
      "learning_rate": 2.8883160934327968e-06,
      "loss": 0.0575,
      "step": 718
    },
    {
      "epoch": 1.2793594306049823,
      "grad_norm": 3.487296999321987,
      "learning_rate": 2.8756568223395396e-06,
      "loss": 0.0863,
      "step": 719
    },
    {
      "epoch": 1.281138790035587,
      "grad_norm": 3.2344388829281066,
      "learning_rate": 2.8630141467911777e-06,
      "loss": 0.069,
      "step": 720
    },
    {
      "epoch": 1.282918149466192,
      "grad_norm": 2.687700200363273,
      "learning_rate": 2.8503881655533395e-06,
      "loss": 0.0564,
      "step": 721
    },
    {
      "epoch": 1.2846975088967971,
      "grad_norm": 3.1442375063568457,
      "learning_rate": 2.837778977261235e-06,
      "loss": 0.0707,
      "step": 722
    },
    {
      "epoch": 1.2864768683274022,
      "grad_norm": 3.0411894823360295,
      "learning_rate": 2.8251866804188875e-06,
      "loss": 0.0692,
      "step": 723
    },
    {
      "epoch": 1.2882562277580072,
      "grad_norm": 3.2309596260598217,
      "learning_rate": 2.812611373398365e-06,
      "loss": 0.0831,
      "step": 724
    },
    {
      "epoch": 1.290035587188612,
      "grad_norm": 3.550189581309326,
      "learning_rate": 2.8000531544390064e-06,
      "loss": 0.0746,
      "step": 725
    },
    {
      "epoch": 1.291814946619217,
      "grad_norm": 3.5224626116450204,
      "learning_rate": 2.7875121216466595e-06,
      "loss": 0.0722,
      "step": 726
    },
    {
      "epoch": 1.293594306049822,
      "grad_norm": 3.7414072940553353,
      "learning_rate": 2.7749883729929105e-06,
      "loss": 0.0842,
      "step": 727
    },
    {
      "epoch": 1.295373665480427,
      "grad_norm": 2.799569023038234,
      "learning_rate": 2.762482006314324e-06,
      "loss": 0.0537,
      "step": 728
    },
    {
      "epoch": 1.2971530249110321,
      "grad_norm": 3.4710664574133427,
      "learning_rate": 2.7499931193116692e-06,
      "loss": 0.0763,
      "step": 729
    },
    {
      "epoch": 1.298932384341637,
      "grad_norm": 3.2763577165092594,
      "learning_rate": 2.737521809549167e-06,
      "loss": 0.0685,
      "step": 730
    },
    {
      "epoch": 1.300711743772242,
      "grad_norm": 3.233112856246983,
      "learning_rate": 2.725068174453722e-06,
      "loss": 0.0801,
      "step": 731
    },
    {
      "epoch": 1.302491103202847,
      "grad_norm": 3.421960055168857,
      "learning_rate": 2.712632311314165e-06,
      "loss": 0.0704,
      "step": 732
    },
    {
      "epoch": 1.304270462633452,
      "grad_norm": 4.142403573270503,
      "learning_rate": 2.7002143172804875e-06,
      "loss": 0.0722,
      "step": 733
    },
    {
      "epoch": 1.306049822064057,
      "grad_norm": 3.2933947598025632,
      "learning_rate": 2.6878142893630904e-06,
      "loss": 0.0717,
      "step": 734
    },
    {
      "epoch": 1.3078291814946619,
      "grad_norm": 3.1464528792776627,
      "learning_rate": 2.6754323244320154e-06,
      "loss": 0.0924,
      "step": 735
    },
    {
      "epoch": 1.309608540925267,
      "grad_norm": 2.9583982740169743,
      "learning_rate": 2.6630685192161995e-06,
      "loss": 0.076,
      "step": 736
    },
    {
      "epoch": 1.311387900355872,
      "grad_norm": 3.7953907731232084,
      "learning_rate": 2.650722970302714e-06,
      "loss": 0.0739,
      "step": 737
    },
    {
      "epoch": 1.3131672597864767,
      "grad_norm": 3.613724804992424,
      "learning_rate": 2.638395774136009e-06,
      "loss": 0.0743,
      "step": 738
    },
    {
      "epoch": 1.3149466192170818,
      "grad_norm": 3.4616335229171367,
      "learning_rate": 2.6260870270171645e-06,
      "loss": 0.0758,
      "step": 739
    },
    {
      "epoch": 1.3167259786476868,
      "grad_norm": 3.8411543649093156,
      "learning_rate": 2.613796825103129e-06,
      "loss": 0.0897,
      "step": 740
    },
    {
      "epoch": 1.3185053380782918,
      "grad_norm": 2.703814922087166,
      "learning_rate": 2.60152526440598e-06,
      "loss": 0.0517,
      "step": 741
    },
    {
      "epoch": 1.3202846975088969,
      "grad_norm": 2.518740689572965,
      "learning_rate": 2.5892724407921667e-06,
      "loss": 0.0464,
      "step": 742
    },
    {
      "epoch": 1.3220640569395017,
      "grad_norm": 4.207831263637035,
      "learning_rate": 2.577038449981763e-06,
      "loss": 0.0823,
      "step": 743
    },
    {
      "epoch": 1.3238434163701067,
      "grad_norm": 3.485750602558497,
      "learning_rate": 2.564823387547716e-06,
      "loss": 0.0728,
      "step": 744
    },
    {
      "epoch": 1.3256227758007118,
      "grad_norm": 3.609194013538497,
      "learning_rate": 2.552627348915106e-06,
      "loss": 0.0884,
      "step": 745
    },
    {
      "epoch": 1.3274021352313168,
      "grad_norm": 3.758999181647603,
      "learning_rate": 2.5404504293603983e-06,
      "loss": 0.096,
      "step": 746
    },
    {
      "epoch": 1.3291814946619218,
      "grad_norm": 4.0265038795647765,
      "learning_rate": 2.528292724010697e-06,
      "loss": 0.1072,
      "step": 747
    },
    {
      "epoch": 1.3309608540925266,
      "grad_norm": 3.734355392504927,
      "learning_rate": 2.5161543278430055e-06,
      "loss": 0.0877,
      "step": 748
    },
    {
      "epoch": 1.3327402135231317,
      "grad_norm": 3.689796136331922,
      "learning_rate": 2.5040353356834756e-06,
      "loss": 0.0809,
      "step": 749
    },
    {
      "epoch": 1.3345195729537367,
      "grad_norm": 3.2170855042943782,
      "learning_rate": 2.4919358422066816e-06,
      "loss": 0.0631,
      "step": 750
    },
    {
      "epoch": 1.3362989323843417,
      "grad_norm": 2.619469783225848,
      "learning_rate": 2.4798559419348672e-06,
      "loss": 0.0519,
      "step": 751
    },
    {
      "epoch": 1.3380782918149468,
      "grad_norm": 2.7420674094590876,
      "learning_rate": 2.4677957292372166e-06,
      "loss": 0.0556,
      "step": 752
    },
    {
      "epoch": 1.3398576512455516,
      "grad_norm": 2.5943600453278495,
      "learning_rate": 2.455755298329107e-06,
      "loss": 0.0601,
      "step": 753
    },
    {
      "epoch": 1.3416370106761566,
      "grad_norm": 2.743153004114721,
      "learning_rate": 2.4437347432713838e-06,
      "loss": 0.0685,
      "step": 754
    },
    {
      "epoch": 1.3434163701067616,
      "grad_norm": 3.1661526518100853,
      "learning_rate": 2.431734157969619e-06,
      "loss": 0.0729,
      "step": 755
    },
    {
      "epoch": 1.3451957295373664,
      "grad_norm": 4.0649209966983335,
      "learning_rate": 2.4197536361733792e-06,
      "loss": 0.1042,
      "step": 756
    },
    {
      "epoch": 1.3469750889679715,
      "grad_norm": 3.177658098771234,
      "learning_rate": 2.407793271475495e-06,
      "loss": 0.0786,
      "step": 757
    },
    {
      "epoch": 1.3487544483985765,
      "grad_norm": 2.828108635798382,
      "learning_rate": 2.3958531573113223e-06,
      "loss": 0.0916,
      "step": 758
    },
    {
      "epoch": 1.3505338078291815,
      "grad_norm": 3.156047631289946,
      "learning_rate": 2.3839333869580243e-06,
      "loss": 0.0797,
      "step": 759
    },
    {
      "epoch": 1.3523131672597866,
      "grad_norm": 3.052971652831814,
      "learning_rate": 2.372034053533835e-06,
      "loss": 0.072,
      "step": 760
    },
    {
      "epoch": 1.3540925266903914,
      "grad_norm": 2.583297340709491,
      "learning_rate": 2.360155249997334e-06,
      "loss": 0.0703,
      "step": 761
    },
    {
      "epoch": 1.3558718861209964,
      "grad_norm": 2.814154788414897,
      "learning_rate": 2.348297069146715e-06,
      "loss": 0.0556,
      "step": 762
    },
    {
      "epoch": 1.3576512455516014,
      "grad_norm": 2.3990094071459866,
      "learning_rate": 2.3364596036190706e-06,
      "loss": 0.0543,
      "step": 763
    },
    {
      "epoch": 1.3594306049822065,
      "grad_norm": 2.9695763369826054,
      "learning_rate": 2.3246429458896637e-06,
      "loss": 0.0713,
      "step": 764
    },
    {
      "epoch": 1.3612099644128115,
      "grad_norm": 3.1876296719070885,
      "learning_rate": 2.312847188271203e-06,
      "loss": 0.0792,
      "step": 765
    },
    {
      "epoch": 1.3629893238434163,
      "grad_norm": 3.094414699358429,
      "learning_rate": 2.301072422913123e-06,
      "loss": 0.0512,
      "step": 766
    },
    {
      "epoch": 1.3647686832740213,
      "grad_norm": 3.138605354524385,
      "learning_rate": 2.2893187418008666e-06,
      "loss": 0.0667,
      "step": 767
    },
    {
      "epoch": 1.3665480427046264,
      "grad_norm": 2.81860899721894,
      "learning_rate": 2.2775862367551642e-06,
      "loss": 0.0643,
      "step": 768
    },
    {
      "epoch": 1.3683274021352312,
      "grad_norm": 3.113848421323806,
      "learning_rate": 2.265874999431318e-06,
      "loss": 0.0517,
      "step": 769
    },
    {
      "epoch": 1.3701067615658362,
      "grad_norm": 3.57890938516717,
      "learning_rate": 2.254185121318484e-06,
      "loss": 0.0611,
      "step": 770
    },
    {
      "epoch": 1.3718861209964412,
      "grad_norm": 3.6862311492709163,
      "learning_rate": 2.2425166937389596e-06,
      "loss": 0.0782,
      "step": 771
    },
    {
      "epoch": 1.3736654804270463,
      "grad_norm": 2.9691792997481627,
      "learning_rate": 2.2308698078474645e-06,
      "loss": 0.0752,
      "step": 772
    },
    {
      "epoch": 1.3754448398576513,
      "grad_norm": 2.96528166110836,
      "learning_rate": 2.219244554630438e-06,
      "loss": 0.0739,
      "step": 773
    },
    {
      "epoch": 1.3772241992882561,
      "grad_norm": 2.871137025793297,
      "learning_rate": 2.207641024905322e-06,
      "loss": 0.0602,
      "step": 774
    },
    {
      "epoch": 1.3790035587188612,
      "grad_norm": 3.170100627531155,
      "learning_rate": 2.1960593093198508e-06,
      "loss": 0.0571,
      "step": 775
    },
    {
      "epoch": 1.3807829181494662,
      "grad_norm": 2.8471048818138422,
      "learning_rate": 2.184499498351347e-06,
      "loss": 0.0579,
      "step": 776
    },
    {
      "epoch": 1.3825622775800712,
      "grad_norm": 2.9618537792640853,
      "learning_rate": 2.172961682306011e-06,
      "loss": 0.0782,
      "step": 777
    },
    {
      "epoch": 1.3843416370106763,
      "grad_norm": 3.065863887996152,
      "learning_rate": 2.1614459513182173e-06,
      "loss": 0.0662,
      "step": 778
    },
    {
      "epoch": 1.386120996441281,
      "grad_norm": 3.044096110376973,
      "learning_rate": 2.149952395349813e-06,
      "loss": 0.0602,
      "step": 779
    },
    {
      "epoch": 1.387900355871886,
      "grad_norm": 3.197075061336266,
      "learning_rate": 2.1384811041894055e-06,
      "loss": 0.0665,
      "step": 780
    },
    {
      "epoch": 1.3896797153024911,
      "grad_norm": 3.4657573249086386,
      "learning_rate": 2.1270321674516736e-06,
      "loss": 0.0759,
      "step": 781
    },
    {
      "epoch": 1.3914590747330962,
      "grad_norm": 2.9015691372249175,
      "learning_rate": 2.1156056745766593e-06,
      "loss": 0.0613,
      "step": 782
    },
    {
      "epoch": 1.3932384341637012,
      "grad_norm": 3.193920880951642,
      "learning_rate": 2.104201714829074e-06,
      "loss": 0.0756,
      "step": 783
    },
    {
      "epoch": 1.395017793594306,
      "grad_norm": 3.420197259493236,
      "learning_rate": 2.0928203772975917e-06,
      "loss": 0.0857,
      "step": 784
    },
    {
      "epoch": 1.396797153024911,
      "grad_norm": 2.823912315076262,
      "learning_rate": 2.081461750894166e-06,
      "loss": 0.0677,
      "step": 785
    },
    {
      "epoch": 1.398576512455516,
      "grad_norm": 3.3263264560286103,
      "learning_rate": 2.070125924353328e-06,
      "loss": 0.0615,
      "step": 786
    },
    {
      "epoch": 1.4003558718861209,
      "grad_norm": 3.0917130397326416,
      "learning_rate": 2.058812986231493e-06,
      "loss": 0.0755,
      "step": 787
    },
    {
      "epoch": 1.402135231316726,
      "grad_norm": 3.4736639155886158,
      "learning_rate": 2.0475230249062727e-06,
      "loss": 0.1149,
      "step": 788
    },
    {
      "epoch": 1.403914590747331,
      "grad_norm": 3.0908057225582715,
      "learning_rate": 2.0362561285757766e-06,
      "loss": 0.075,
      "step": 789
    },
    {
      "epoch": 1.405693950177936,
      "grad_norm": 2.772166008104422,
      "learning_rate": 2.0250123852579347e-06,
      "loss": 0.051,
      "step": 790
    },
    {
      "epoch": 1.407473309608541,
      "grad_norm": 2.351063239659787,
      "learning_rate": 2.013791882789801e-06,
      "loss": 0.0452,
      "step": 791
    },
    {
      "epoch": 1.4092526690391458,
      "grad_norm": 2.9572576519509712,
      "learning_rate": 2.0025947088268714e-06,
      "loss": 0.0439,
      "step": 792
    },
    {
      "epoch": 1.4110320284697508,
      "grad_norm": 2.6042273061256846,
      "learning_rate": 1.9914209508423943e-06,
      "loss": 0.0577,
      "step": 793
    },
    {
      "epoch": 1.4128113879003559,
      "grad_norm": 2.8962455430890697,
      "learning_rate": 1.9802706961266936e-06,
      "loss": 0.064,
      "step": 794
    },
    {
      "epoch": 1.414590747330961,
      "grad_norm": 3.285570164402755,
      "learning_rate": 1.969144031786483e-06,
      "loss": 0.0728,
      "step": 795
    },
    {
      "epoch": 1.416370106761566,
      "grad_norm": 2.9277955791620998,
      "learning_rate": 1.958041044744186e-06,
      "loss": 0.0828,
      "step": 796
    },
    {
      "epoch": 1.4181494661921707,
      "grad_norm": 2.6837375539002113,
      "learning_rate": 1.94696182173726e-06,
      "loss": 0.0472,
      "step": 797
    },
    {
      "epoch": 1.4199288256227758,
      "grad_norm": 3.351432120514172,
      "learning_rate": 1.9359064493175077e-06,
      "loss": 0.0748,
      "step": 798
    },
    {
      "epoch": 1.4217081850533808,
      "grad_norm": 3.2052134409659585,
      "learning_rate": 1.9248750138504176e-06,
      "loss": 0.062,
      "step": 799
    },
    {
      "epoch": 1.4234875444839858,
      "grad_norm": 3.291673078002292,
      "learning_rate": 1.9138676015144765e-06,
      "loss": 0.0647,
      "step": 800
    },
    {
      "epoch": 1.4252669039145909,
      "grad_norm": 3.5021711264474926,
      "learning_rate": 1.9028842983005036e-06,
      "loss": 0.0883,
      "step": 801
    },
    {
      "epoch": 1.4270462633451957,
      "grad_norm": 3.615562114459196,
      "learning_rate": 1.8919251900109697e-06,
      "loss": 0.0844,
      "step": 802
    },
    {
      "epoch": 1.4288256227758007,
      "grad_norm": 3.03231923061856,
      "learning_rate": 1.8809903622593395e-06,
      "loss": 0.0657,
      "step": 803
    },
    {
      "epoch": 1.4306049822064058,
      "grad_norm": 2.5455644674315963,
      "learning_rate": 1.870079900469392e-06,
      "loss": 0.0432,
      "step": 804
    },
    {
      "epoch": 1.4323843416370106,
      "grad_norm": 3.268568052317049,
      "learning_rate": 1.8591938898745593e-06,
      "loss": 0.0695,
      "step": 805
    },
    {
      "epoch": 1.4341637010676156,
      "grad_norm": 2.9280501960609824,
      "learning_rate": 1.8483324155172594e-06,
      "loss": 0.0643,
      "step": 806
    },
    {
      "epoch": 1.4359430604982206,
      "grad_norm": 2.6836702188505828,
      "learning_rate": 1.837495562248226e-06,
      "loss": 0.0771,
      "step": 807
    },
    {
      "epoch": 1.4377224199288257,
      "grad_norm": 3.2099204963671113,
      "learning_rate": 1.8266834147258577e-06,
      "loss": 0.0678,
      "step": 808
    },
    {
      "epoch": 1.4395017793594307,
      "grad_norm": 3.1239746513156375,
      "learning_rate": 1.8158960574155455e-06,
      "loss": 0.0613,
      "step": 809
    },
    {
      "epoch": 1.4412811387900355,
      "grad_norm": 3.5986579798312444,
      "learning_rate": 1.8051335745890196e-06,
      "loss": 0.0743,
      "step": 810
    },
    {
      "epoch": 1.4430604982206405,
      "grad_norm": 3.1949236108078827,
      "learning_rate": 1.7943960503236856e-06,
      "loss": 0.0885,
      "step": 811
    },
    {
      "epoch": 1.4448398576512456,
      "grad_norm": 2.6927615904221205,
      "learning_rate": 1.7836835685019732e-06,
      "loss": 0.0641,
      "step": 812
    },
    {
      "epoch": 1.4466192170818506,
      "grad_norm": 3.907359979503544,
      "learning_rate": 1.7729962128106787e-06,
      "loss": 0.0683,
      "step": 813
    },
    {
      "epoch": 1.4483985765124556,
      "grad_norm": 2.9896144901449384,
      "learning_rate": 1.7623340667403089e-06,
      "loss": 0.0554,
      "step": 814
    },
    {
      "epoch": 1.4501779359430604,
      "grad_norm": 3.9748071456946317,
      "learning_rate": 1.7516972135844352e-06,
      "loss": 0.096,
      "step": 815
    },
    {
      "epoch": 1.4519572953736655,
      "grad_norm": 3.8420721384063428,
      "learning_rate": 1.741085736439031e-06,
      "loss": 0.0666,
      "step": 816
    },
    {
      "epoch": 1.4537366548042705,
      "grad_norm": 3.333277492952436,
      "learning_rate": 1.730499718201838e-06,
      "loss": 0.0711,
      "step": 817
    },
    {
      "epoch": 1.4555160142348753,
      "grad_norm": 3.408798206176967,
      "learning_rate": 1.7199392415717064e-06,
      "loss": 0.068,
      "step": 818
    },
    {
      "epoch": 1.4572953736654806,
      "grad_norm": 3.36617255703665,
      "learning_rate": 1.7094043890479557e-06,
      "loss": 0.0808,
      "step": 819
    },
    {
      "epoch": 1.4590747330960854,
      "grad_norm": 2.638501185428134,
      "learning_rate": 1.698895242929725e-06,
      "loss": 0.0716,
      "step": 820
    },
    {
      "epoch": 1.4608540925266904,
      "grad_norm": 3.0334683003269918,
      "learning_rate": 1.6884118853153358e-06,
      "loss": 0.0516,
      "step": 821
    },
    {
      "epoch": 1.4626334519572954,
      "grad_norm": 3.6038498517902036,
      "learning_rate": 1.6779543981016478e-06,
      "loss": 0.0769,
      "step": 822
    },
    {
      "epoch": 1.4644128113879002,
      "grad_norm": 3.103413282674726,
      "learning_rate": 1.6675228629834133e-06,
      "loss": 0.0812,
      "step": 823
    },
    {
      "epoch": 1.4661921708185053,
      "grad_norm": 2.6176917056223585,
      "learning_rate": 1.657117361452651e-06,
      "loss": 0.0577,
      "step": 824
    },
    {
      "epoch": 1.4679715302491103,
      "grad_norm": 2.8231520427671595,
      "learning_rate": 1.6467379747980011e-06,
      "loss": 0.0714,
      "step": 825
    },
    {
      "epoch": 1.4697508896797153,
      "grad_norm": 2.3769464466168446,
      "learning_rate": 1.6363847841040914e-06,
      "loss": 0.0535,
      "step": 826
    },
    {
      "epoch": 1.4715302491103204,
      "grad_norm": 2.6605718243188488,
      "learning_rate": 1.626057870250906e-06,
      "loss": 0.0491,
      "step": 827
    },
    {
      "epoch": 1.4733096085409252,
      "grad_norm": 2.7833455747478184,
      "learning_rate": 1.6157573139131527e-06,
      "loss": 0.0474,
      "step": 828
    },
    {
      "epoch": 1.4750889679715302,
      "grad_norm": 2.8621281912110743,
      "learning_rate": 1.605483195559628e-06,
      "loss": 0.0756,
      "step": 829
    },
    {
      "epoch": 1.4768683274021353,
      "grad_norm": 3.0792898099939245,
      "learning_rate": 1.5952355954525966e-06,
      "loss": 0.0498,
      "step": 830
    },
    {
      "epoch": 1.4786476868327403,
      "grad_norm": 2.713548455534332,
      "learning_rate": 1.5850145936471607e-06,
      "loss": 0.0625,
      "step": 831
    },
    {
      "epoch": 1.4804270462633453,
      "grad_norm": 3.456186798770157,
      "learning_rate": 1.5748202699906335e-06,
      "loss": 0.0613,
      "step": 832
    },
    {
      "epoch": 1.4822064056939501,
      "grad_norm": 2.1732391322717204,
      "learning_rate": 1.5646527041219128e-06,
      "loss": 0.0446,
      "step": 833
    },
    {
      "epoch": 1.4839857651245552,
      "grad_norm": 2.926068862388731,
      "learning_rate": 1.5545119754708682e-06,
      "loss": 0.0644,
      "step": 834
    },
    {
      "epoch": 1.4857651245551602,
      "grad_norm": 3.702509008639658,
      "learning_rate": 1.544398163257711e-06,
      "loss": 0.0859,
      "step": 835
    },
    {
      "epoch": 1.487544483985765,
      "grad_norm": 2.8162939267064755,
      "learning_rate": 1.5343113464923808e-06,
      "loss": 0.0648,
      "step": 836
    },
    {
      "epoch": 1.48932384341637,
      "grad_norm": 2.3360866359821544,
      "learning_rate": 1.524251603973927e-06,
      "loss": 0.0499,
      "step": 837
    },
    {
      "epoch": 1.491103202846975,
      "grad_norm": 3.0741125035066483,
      "learning_rate": 1.5142190142898883e-06,
      "loss": 0.0684,
      "step": 838
    },
    {
      "epoch": 1.49288256227758,
      "grad_norm": 3.3390229120911674,
      "learning_rate": 1.5042136558156883e-06,
      "loss": 0.0764,
      "step": 839
    },
    {
      "epoch": 1.4946619217081851,
      "grad_norm": 3.673413800735187,
      "learning_rate": 1.4942356067140162e-06,
      "loss": 0.0895,
      "step": 840
    },
    {
      "epoch": 1.49644128113879,
      "grad_norm": 2.357242407022346,
      "learning_rate": 1.4842849449342195e-06,
      "loss": 0.0555,
      "step": 841
    },
    {
      "epoch": 1.498220640569395,
      "grad_norm": 2.8698416323425526,
      "learning_rate": 1.4743617482116896e-06,
      "loss": 0.0678,
      "step": 842
    },
    {
      "epoch": 1.5,
      "grad_norm": 3.5058328919413353,
      "learning_rate": 1.4644660940672628e-06,
      "loss": 0.0571,
      "step": 843
    },
    {
      "epoch": 1.501779359430605,
      "grad_norm": 2.8629055112826833,
      "learning_rate": 1.454598059806609e-06,
      "loss": 0.0717,
      "step": 844
    },
    {
      "epoch": 1.50355871886121,
      "grad_norm": 3.3771826545057193,
      "learning_rate": 1.4447577225196296e-06,
      "loss": 0.0741,
      "step": 845
    },
    {
      "epoch": 1.5053380782918149,
      "grad_norm": 3.3952376761798444,
      "learning_rate": 1.4349451590798564e-06,
      "loss": 0.0656,
      "step": 846
    },
    {
      "epoch": 1.50711743772242,
      "grad_norm": 2.9948095276247098,
      "learning_rate": 1.4251604461438444e-06,
      "loss": 0.0725,
      "step": 847
    },
    {
      "epoch": 1.508896797153025,
      "grad_norm": 2.829567415359015,
      "learning_rate": 1.4154036601505834e-06,
      "loss": 0.0589,
      "step": 848
    },
    {
      "epoch": 1.5106761565836297,
      "grad_norm": 3.359320312830941,
      "learning_rate": 1.4056748773208933e-06,
      "loss": 0.0787,
      "step": 849
    },
    {
      "epoch": 1.512455516014235,
      "grad_norm": 3.3879026897095876,
      "learning_rate": 1.3959741736568339e-06,
      "loss": 0.0734,
      "step": 850
    },
    {
      "epoch": 1.5142348754448398,
      "grad_norm": 2.8633107625733722,
      "learning_rate": 1.3863016249411027e-06,
      "loss": 0.0574,
      "step": 851
    },
    {
      "epoch": 1.5160142348754448,
      "grad_norm": 2.6015476924115477,
      "learning_rate": 1.376657306736453e-06,
      "loss": 0.0691,
      "step": 852
    },
    {
      "epoch": 1.5177935943060499,
      "grad_norm": 3.2359576376032493,
      "learning_rate": 1.3670412943850975e-06,
      "loss": 0.0758,
      "step": 853
    },
    {
      "epoch": 1.5195729537366547,
      "grad_norm": 2.702887084561589,
      "learning_rate": 1.3574536630081208e-06,
      "loss": 0.0413,
      "step": 854
    },
    {
      "epoch": 1.52135231316726,
      "grad_norm": 2.751620279366842,
      "learning_rate": 1.347894487504896e-06,
      "loss": 0.053,
      "step": 855
    },
    {
      "epoch": 1.5231316725978647,
      "grad_norm": 2.8174542333312016,
      "learning_rate": 1.3383638425524909e-06,
      "loss": 0.0524,
      "step": 856
    },
    {
      "epoch": 1.5249110320284698,
      "grad_norm": 2.9591397031217643,
      "learning_rate": 1.3288618026050943e-06,
      "loss": 0.0657,
      "step": 857
    },
    {
      "epoch": 1.5266903914590748,
      "grad_norm": 2.9861238023822305,
      "learning_rate": 1.31938844189343e-06,
      "loss": 0.0602,
      "step": 858
    },
    {
      "epoch": 1.5284697508896796,
      "grad_norm": 3.494342962918134,
      "learning_rate": 1.3099438344241777e-06,
      "loss": 0.0741,
      "step": 859
    },
    {
      "epoch": 1.5302491103202847,
      "grad_norm": 2.7632491745650873,
      "learning_rate": 1.3005280539793908e-06,
      "loss": 0.0611,
      "step": 860
    },
    {
      "epoch": 1.5320284697508897,
      "grad_norm": 2.9921663662280245,
      "learning_rate": 1.2911411741159273e-06,
      "loss": 0.0585,
      "step": 861
    },
    {
      "epoch": 1.5338078291814945,
      "grad_norm": 2.936707163896156,
      "learning_rate": 1.2817832681648712e-06,
      "loss": 0.0981,
      "step": 862
    },
    {
      "epoch": 1.5355871886120998,
      "grad_norm": 3.205997082163906,
      "learning_rate": 1.2724544092309581e-06,
      "loss": 0.0668,
      "step": 863
    },
    {
      "epoch": 1.5373665480427046,
      "grad_norm": 2.7924043036963986,
      "learning_rate": 1.2631546701920073e-06,
      "loss": 0.0658,
      "step": 864
    },
    {
      "epoch": 1.5391459074733096,
      "grad_norm": 2.961367619694483,
      "learning_rate": 1.2538841236983519e-06,
      "loss": 0.0691,
      "step": 865
    },
    {
      "epoch": 1.5409252669039146,
      "grad_norm": 3.0873931967961363,
      "learning_rate": 1.244642842172266e-06,
      "loss": 0.0501,
      "step": 866
    },
    {
      "epoch": 1.5427046263345194,
      "grad_norm": 3.290074922419228,
      "learning_rate": 1.2354308978074088e-06,
      "loss": 0.07,
      "step": 867
    },
    {
      "epoch": 1.5444839857651247,
      "grad_norm": 2.982642320949764,
      "learning_rate": 1.2262483625682514e-06,
      "loss": 0.0721,
      "step": 868
    },
    {
      "epoch": 1.5462633451957295,
      "grad_norm": 3.2642975184155323,
      "learning_rate": 1.2170953081895214e-06,
      "loss": 0.0723,
      "step": 869
    },
    {
      "epoch": 1.5480427046263345,
      "grad_norm": 2.8411847926833627,
      "learning_rate": 1.2079718061756369e-06,
      "loss": 0.0498,
      "step": 870
    },
    {
      "epoch": 1.5498220640569396,
      "grad_norm": 2.9954245162162976,
      "learning_rate": 1.1988779278001517e-06,
      "loss": 0.0605,
      "step": 871
    },
    {
      "epoch": 1.5516014234875444,
      "grad_norm": 3.3421440527059674,
      "learning_rate": 1.1898137441051982e-06,
      "loss": 0.075,
      "step": 872
    },
    {
      "epoch": 1.5533807829181496,
      "grad_norm": 3.0782906321932386,
      "learning_rate": 1.1807793259009282e-06,
      "loss": 0.0758,
      "step": 873
    },
    {
      "epoch": 1.5551601423487544,
      "grad_norm": 2.8547315513309774,
      "learning_rate": 1.1717747437649657e-06,
      "loss": 0.0649,
      "step": 874
    },
    {
      "epoch": 1.5569395017793595,
      "grad_norm": 2.6408885706995937,
      "learning_rate": 1.1628000680418533e-06,
      "loss": 0.0521,
      "step": 875
    },
    {
      "epoch": 1.5587188612099645,
      "grad_norm": 3.1797028373340885,
      "learning_rate": 1.1538553688425002e-06,
      "loss": 0.0586,
      "step": 876
    },
    {
      "epoch": 1.5604982206405693,
      "grad_norm": 3.0877701580176002,
      "learning_rate": 1.14494071604364e-06,
      "loss": 0.059,
      "step": 877
    },
    {
      "epoch": 1.5622775800711743,
      "grad_norm": 3.0256422186124876,
      "learning_rate": 1.1360561792872754e-06,
      "loss": 0.057,
      "step": 878
    },
    {
      "epoch": 1.5640569395017794,
      "grad_norm": 3.5516890779797277,
      "learning_rate": 1.127201827980145e-06,
      "loss": 0.0886,
      "step": 879
    },
    {
      "epoch": 1.5658362989323842,
      "grad_norm": 2.544505728020871,
      "learning_rate": 1.1183777312931748e-06,
      "loss": 0.0487,
      "step": 880
    },
    {
      "epoch": 1.5676156583629894,
      "grad_norm": 2.7123688338719574,
      "learning_rate": 1.1095839581609407e-06,
      "loss": 0.071,
      "step": 881
    },
    {
      "epoch": 1.5693950177935942,
      "grad_norm": 3.2490106311652758,
      "learning_rate": 1.1008205772811248e-06,
      "loss": 0.0758,
      "step": 882
    },
    {
      "epoch": 1.5711743772241993,
      "grad_norm": 3.3242674903450475,
      "learning_rate": 1.0920876571139843e-06,
      "loss": 0.0777,
      "step": 883
    },
    {
      "epoch": 1.5729537366548043,
      "grad_norm": 3.3849777312355607,
      "learning_rate": 1.0833852658818167e-06,
      "loss": 0.0745,
      "step": 884
    },
    {
      "epoch": 1.5747330960854091,
      "grad_norm": 3.066920337928523,
      "learning_rate": 1.0747134715684221e-06,
      "loss": 0.0599,
      "step": 885
    },
    {
      "epoch": 1.5765124555160144,
      "grad_norm": 2.5920726632949336,
      "learning_rate": 1.0660723419185776e-06,
      "loss": 0.0583,
      "step": 886
    },
    {
      "epoch": 1.5782918149466192,
      "grad_norm": 3.268075458180264,
      "learning_rate": 1.0574619444375017e-06,
      "loss": 0.0653,
      "step": 887
    },
    {
      "epoch": 1.5800711743772242,
      "grad_norm": 2.692293333066783,
      "learning_rate": 1.0488823463903341e-06,
      "loss": 0.0628,
      "step": 888
    },
    {
      "epoch": 1.5818505338078293,
      "grad_norm": 2.8703204110342218,
      "learning_rate": 1.0403336148016053e-06,
      "loss": 0.0831,
      "step": 889
    },
    {
      "epoch": 1.583629893238434,
      "grad_norm": 2.890182118830602,
      "learning_rate": 1.0318158164547159e-06,
      "loss": 0.0762,
      "step": 890
    },
    {
      "epoch": 1.585409252669039,
      "grad_norm": 2.5847985124267225,
      "learning_rate": 1.0233290178914096e-06,
      "loss": 0.0562,
      "step": 891
    },
    {
      "epoch": 1.5871886120996441,
      "grad_norm": 2.4619657014440186,
      "learning_rate": 1.014873285411262e-06,
      "loss": 0.0564,
      "step": 892
    },
    {
      "epoch": 1.5889679715302492,
      "grad_norm": 3.5526128877456347,
      "learning_rate": 1.006448685071154e-06,
      "loss": 0.0797,
      "step": 893
    },
    {
      "epoch": 1.5907473309608542,
      "grad_norm": 3.3376359590904907,
      "learning_rate": 9.980552826847635e-07,
      "loss": 0.0758,
      "step": 894
    },
    {
      "epoch": 1.592526690391459,
      "grad_norm": 3.9315348045876513,
      "learning_rate": 9.896931438220453e-07,
      "loss": 0.0884,
      "step": 895
    },
    {
      "epoch": 1.594306049822064,
      "grad_norm": 3.3389690131568655,
      "learning_rate": 9.813623338087181e-07,
      "loss": 0.0694,
      "step": 896
    },
    {
      "epoch": 1.596085409252669,
      "grad_norm": 4.051315747604687,
      "learning_rate": 9.730629177257623e-07,
      "loss": 0.0944,
      "step": 897
    },
    {
      "epoch": 1.5978647686832739,
      "grad_norm": 2.954233792128124,
      "learning_rate": 9.64794960408903e-07,
      "loss": 0.0583,
      "step": 898
    },
    {
      "epoch": 1.5996441281138791,
      "grad_norm": 3.058087887161698,
      "learning_rate": 9.565585264481092e-07,
      "loss": 0.0594,
      "step": 899
    },
    {
      "epoch": 1.601423487544484,
      "grad_norm": 2.790153105752596,
      "learning_rate": 9.483536801870835e-07,
      "loss": 0.0619,
      "step": 900
    },
    {
      "epoch": 1.603202846975089,
      "grad_norm": 2.6027059299921462,
      "learning_rate": 9.401804857227648e-07,
      "loss": 0.0472,
      "step": 901
    },
    {
      "epoch": 1.604982206405694,
      "grad_norm": 2.80287175799951,
      "learning_rate": 9.320390069048258e-07,
      "loss": 0.0702,
      "step": 902
    },
    {
      "epoch": 1.6067615658362988,
      "grad_norm": 3.264885039022311,
      "learning_rate": 9.239293073351735e-07,
      "loss": 0.0636,
      "step": 903
    },
    {
      "epoch": 1.608540925266904,
      "grad_norm": 3.520208344674982,
      "learning_rate": 9.158514503674543e-07,
      "loss": 0.0711,
      "step": 904
    },
    {
      "epoch": 1.6103202846975089,
      "grad_norm": 3.0619415280804994,
      "learning_rate": 9.078054991065532e-07,
      "loss": 0.0609,
      "step": 905
    },
    {
      "epoch": 1.612099644128114,
      "grad_norm": 2.94114057559002,
      "learning_rate": 8.997915164081095e-07,
      "loss": 0.07,
      "step": 906
    },
    {
      "epoch": 1.613879003558719,
      "grad_norm": 3.5247782825574143,
      "learning_rate": 8.918095648780195e-07,
      "loss": 0.0673,
      "step": 907
    },
    {
      "epoch": 1.6156583629893237,
      "grad_norm": 3.1303728843632164,
      "learning_rate": 8.838597068719518e-07,
      "loss": 0.066,
      "step": 908
    },
    {
      "epoch": 1.6174377224199288,
      "grad_norm": 3.5759872882101496,
      "learning_rate": 8.75942004494853e-07,
      "loss": 0.0684,
      "step": 909
    },
    {
      "epoch": 1.6192170818505338,
      "grad_norm": 3.2993834086741054,
      "learning_rate": 8.680565196004704e-07,
      "loss": 0.0771,
      "step": 910
    },
    {
      "epoch": 1.6209964412811388,
      "grad_norm": 3.081015623753002,
      "learning_rate": 8.602033137908666e-07,
      "loss": 0.057,
      "step": 911
    },
    {
      "epoch": 1.6227758007117439,
      "grad_norm": 2.9290247464754584,
      "learning_rate": 8.523824484159348e-07,
      "loss": 0.0634,
      "step": 912
    },
    {
      "epoch": 1.6245551601423487,
      "grad_norm": 3.357193073747661,
      "learning_rate": 8.445939845729245e-07,
      "loss": 0.0548,
      "step": 913
    },
    {
      "epoch": 1.6263345195729537,
      "grad_norm": 2.5104406410728943,
      "learning_rate": 8.368379831059592e-07,
      "loss": 0.0602,
      "step": 914
    },
    {
      "epoch": 1.6281138790035588,
      "grad_norm": 2.488875447410959,
      "learning_rate": 8.29114504605566e-07,
      "loss": 0.0529,
      "step": 915
    },
    {
      "epoch": 1.6298932384341636,
      "grad_norm": 2.743204948304154,
      "learning_rate": 8.21423609408199e-07,
      "loss": 0.0724,
      "step": 916
    },
    {
      "epoch": 1.6316725978647688,
      "grad_norm": 3.46182303576283,
      "learning_rate": 8.137653575957666e-07,
      "loss": 0.0608,
      "step": 917
    },
    {
      "epoch": 1.6334519572953736,
      "grad_norm": 3.1926014239166096,
      "learning_rate": 8.061398089951678e-07,
      "loss": 0.0784,
      "step": 918
    },
    {
      "epoch": 1.6352313167259787,
      "grad_norm": 2.9708617200089433,
      "learning_rate": 7.985470231778203e-07,
      "loss": 0.0623,
      "step": 919
    },
    {
      "epoch": 1.6370106761565837,
      "grad_norm": 3.491360267384296,
      "learning_rate": 7.909870594591951e-07,
      "loss": 0.0691,
      "step": 920
    },
    {
      "epoch": 1.6387900355871885,
      "grad_norm": 3.107261281539005,
      "learning_rate": 7.834599768983553e-07,
      "loss": 0.0608,
      "step": 921
    },
    {
      "epoch": 1.6405693950177938,
      "grad_norm": 2.6375368104799333,
      "learning_rate": 7.759658342974951e-07,
      "loss": 0.059,
      "step": 922
    },
    {
      "epoch": 1.6423487544483986,
      "grad_norm": 3.5420359756209128,
      "learning_rate": 7.685046902014747e-07,
      "loss": 0.0784,
      "step": 923
    },
    {
      "epoch": 1.6441281138790036,
      "grad_norm": 3.4708709973480665,
      "learning_rate": 7.61076602897371e-07,
      "loss": 0.0687,
      "step": 924
    },
    {
      "epoch": 1.6459074733096086,
      "grad_norm": 2.2676999368741413,
      "learning_rate": 7.536816304140177e-07,
      "loss": 0.0451,
      "step": 925
    },
    {
      "epoch": 1.6476868327402134,
      "grad_norm": 2.874354681727319,
      "learning_rate": 7.46319830521553e-07,
      "loss": 0.0386,
      "step": 926
    },
    {
      "epoch": 1.6494661921708185,
      "grad_norm": 2.606793751198401,
      "learning_rate": 7.389912607309662e-07,
      "loss": 0.0705,
      "step": 927
    },
    {
      "epoch": 1.6512455516014235,
      "grad_norm": 3.741006888586328,
      "learning_rate": 7.316959782936516e-07,
      "loss": 0.1022,
      "step": 928
    },
    {
      "epoch": 1.6530249110320283,
      "grad_norm": 2.4788377074128958,
      "learning_rate": 7.244340402009608e-07,
      "loss": 0.0526,
      "step": 929
    },
    {
      "epoch": 1.6548042704626336,
      "grad_norm": 3.726395929142042,
      "learning_rate": 7.172055031837572e-07,
      "loss": 0.0813,
      "step": 930
    },
    {
      "epoch": 1.6565836298932384,
      "grad_norm": 2.9573402931994233,
      "learning_rate": 7.100104237119676e-07,
      "loss": 0.0638,
      "step": 931
    },
    {
      "epoch": 1.6583629893238434,
      "grad_norm": 2.7156104271055055,
      "learning_rate": 7.028488579941506e-07,
      "loss": 0.0787,
      "step": 932
    },
    {
      "epoch": 1.6601423487544484,
      "grad_norm": 3.856294816091033,
      "learning_rate": 6.957208619770505e-07,
      "loss": 0.0757,
      "step": 933
    },
    {
      "epoch": 1.6619217081850532,
      "grad_norm": 3.3616803607287653,
      "learning_rate": 6.886264913451635e-07,
      "loss": 0.0717,
      "step": 934
    },
    {
      "epoch": 1.6637010676156585,
      "grad_norm": 3.3739032119754335,
      "learning_rate": 6.815658015203014e-07,
      "loss": 0.0716,
      "step": 935
    },
    {
      "epoch": 1.6654804270462633,
      "grad_norm": 3.304142605470946,
      "learning_rate": 6.745388476611553e-07,
      "loss": 0.0712,
      "step": 936
    },
    {
      "epoch": 1.6672597864768683,
      "grad_norm": 3.1182910077762434,
      "learning_rate": 6.67545684662873e-07,
      "loss": 0.0711,
      "step": 937
    },
    {
      "epoch": 1.6690391459074734,
      "grad_norm": 2.477453703652539,
      "learning_rate": 6.605863671566221e-07,
      "loss": 0.0465,
      "step": 938
    },
    {
      "epoch": 1.6708185053380782,
      "grad_norm": 2.735842344913042,
      "learning_rate": 6.536609495091695e-07,
      "loss": 0.0514,
      "step": 939
    },
    {
      "epoch": 1.6725978647686834,
      "grad_norm": 3.4967094064920032,
      "learning_rate": 6.467694858224488e-07,
      "loss": 0.0743,
      "step": 940
    },
    {
      "epoch": 1.6743772241992882,
      "grad_norm": 3.013792407275164,
      "learning_rate": 6.399120299331468e-07,
      "loss": 0.044,
      "step": 941
    },
    {
      "epoch": 1.6761565836298933,
      "grad_norm": 3.0318455949381824,
      "learning_rate": 6.330886354122768e-07,
      "loss": 0.0639,
      "step": 942
    },
    {
      "epoch": 1.6779359430604983,
      "grad_norm": 3.3015757627247924,
      "learning_rate": 6.262993555647617e-07,
      "loss": 0.0791,
      "step": 943
    },
    {
      "epoch": 1.6797153024911031,
      "grad_norm": 2.602340166992549,
      "learning_rate": 6.1954424342902e-07,
      "loss": 0.0611,
      "step": 944
    },
    {
      "epoch": 1.6814946619217082,
      "grad_norm": 3.9286221139552953,
      "learning_rate": 6.128233517765448e-07,
      "loss": 0.0851,
      "step": 945
    },
    {
      "epoch": 1.6832740213523132,
      "grad_norm": 3.1039748858767227,
      "learning_rate": 6.061367331114992e-07,
      "loss": 0.0587,
      "step": 946
    },
    {
      "epoch": 1.685053380782918,
      "grad_norm": 3.2480327591635345,
      "learning_rate": 5.994844396703025e-07,
      "loss": 0.0977,
      "step": 947
    },
    {
      "epoch": 1.6868327402135233,
      "grad_norm": 3.04444019356055,
      "learning_rate": 5.928665234212233e-07,
      "loss": 0.0595,
      "step": 948
    },
    {
      "epoch": 1.688612099644128,
      "grad_norm": 2.962169044589747,
      "learning_rate": 5.862830360639698e-07,
      "loss": 0.0681,
      "step": 949
    },
    {
      "epoch": 1.690391459074733,
      "grad_norm": 3.27841867657538,
      "learning_rate": 5.797340290292907e-07,
      "loss": 0.065,
      "step": 950
    },
    {
      "epoch": 1.6921708185053381,
      "grad_norm": 3.065293170079295,
      "learning_rate": 5.732195534785723e-07,
      "loss": 0.068,
      "step": 951
    },
    {
      "epoch": 1.693950177935943,
      "grad_norm": 2.4758166631219285,
      "learning_rate": 5.667396603034369e-07,
      "loss": 0.0454,
      "step": 952
    },
    {
      "epoch": 1.6957295373665482,
      "grad_norm": 3.8238749794872997,
      "learning_rate": 5.602944001253486e-07,
      "loss": 0.0793,
      "step": 953
    },
    {
      "epoch": 1.697508896797153,
      "grad_norm": 3.307022058993498,
      "learning_rate": 5.538838232952104e-07,
      "loss": 0.0732,
      "step": 954
    },
    {
      "epoch": 1.699288256227758,
      "grad_norm": 3.619277188884766,
      "learning_rate": 5.475079798929816e-07,
      "loss": 0.0985,
      "step": 955
    },
    {
      "epoch": 1.701067615658363,
      "grad_norm": 2.69556686479802,
      "learning_rate": 5.411669197272795e-07,
      "loss": 0.0492,
      "step": 956
    },
    {
      "epoch": 1.7028469750889679,
      "grad_norm": 2.843989527406096,
      "learning_rate": 5.348606923349903e-07,
      "loss": 0.0588,
      "step": 957
    },
    {
      "epoch": 1.704626334519573,
      "grad_norm": 3.1784014133641945,
      "learning_rate": 5.285893469808855e-07,
      "loss": 0.0503,
      "step": 958
    },
    {
      "epoch": 1.706405693950178,
      "grad_norm": 3.1274161482690435,
      "learning_rate": 5.223529326572352e-07,
      "loss": 0.0639,
      "step": 959
    },
    {
      "epoch": 1.708185053380783,
      "grad_norm": 3.247319111207832,
      "learning_rate": 5.161514980834232e-07,
      "loss": 0.074,
      "step": 960
    },
    {
      "epoch": 1.709964412811388,
      "grad_norm": 3.3281477946564095,
      "learning_rate": 5.099850917055709e-07,
      "loss": 0.0587,
      "step": 961
    },
    {
      "epoch": 1.7117437722419928,
      "grad_norm": 3.025382413791231,
      "learning_rate": 5.038537616961559e-07,
      "loss": 0.0585,
      "step": 962
    },
    {
      "epoch": 1.7135231316725978,
      "grad_norm": 3.1558019175838727,
      "learning_rate": 4.977575559536358e-07,
      "loss": 0.054,
      "step": 963
    },
    {
      "epoch": 1.7153024911032029,
      "grad_norm": 3.0150234778352076,
      "learning_rate": 4.916965221020753e-07,
      "loss": 0.0557,
      "step": 964
    },
    {
      "epoch": 1.7170818505338077,
      "grad_norm": 2.5695298257451866,
      "learning_rate": 4.856707074907729e-07,
      "loss": 0.0519,
      "step": 965
    },
    {
      "epoch": 1.718861209964413,
      "grad_norm": 3.1532489124060277,
      "learning_rate": 4.796801591938922e-07,
      "loss": 0.0693,
      "step": 966
    },
    {
      "epoch": 1.7206405693950177,
      "grad_norm": 3.652340329272302,
      "learning_rate": 4.737249240100911e-07,
      "loss": 0.0797,
      "step": 967
    },
    {
      "epoch": 1.7224199288256228,
      "grad_norm": 2.859965535619542,
      "learning_rate": 4.6780504846216155e-07,
      "loss": 0.0463,
      "step": 968
    },
    {
      "epoch": 1.7241992882562278,
      "grad_norm": 2.834189026511556,
      "learning_rate": 4.619205787966613e-07,
      "loss": 0.0491,
      "step": 969
    },
    {
      "epoch": 1.7259786476868326,
      "grad_norm": 2.5086691747534524,
      "learning_rate": 4.560715609835548e-07,
      "loss": 0.0449,
      "step": 970
    },
    {
      "epoch": 1.7277580071174379,
      "grad_norm": 2.6220965005293873,
      "learning_rate": 4.5025804071585464e-07,
      "loss": 0.0486,
      "step": 971
    },
    {
      "epoch": 1.7295373665480427,
      "grad_norm": 3.2705881855554804,
      "learning_rate": 4.4448006340926163e-07,
      "loss": 0.0727,
      "step": 972
    },
    {
      "epoch": 1.7313167259786477,
      "grad_norm": 3.029087362492905,
      "learning_rate": 4.3873767420181344e-07,
      "loss": 0.0629,
      "step": 973
    },
    {
      "epoch": 1.7330960854092528,
      "grad_norm": 3.2366251099862176,
      "learning_rate": 4.3303091795353024e-07,
      "loss": 0.0684,
      "step": 974
    },
    {
      "epoch": 1.7348754448398576,
      "grad_norm": 2.797352830206982,
      "learning_rate": 4.2735983924606596e-07,
      "loss": 0.0465,
      "step": 975
    },
    {
      "epoch": 1.7366548042704626,
      "grad_norm": 3.3015846994997085,
      "learning_rate": 4.2172448238235464e-07,
      "loss": 0.0471,
      "step": 976
    },
    {
      "epoch": 1.7384341637010676,
      "grad_norm": 2.736278414696688,
      "learning_rate": 4.161248913862731e-07,
      "loss": 0.0471,
      "step": 977
    },
    {
      "epoch": 1.7402135231316724,
      "grad_norm": 3.041296241444544,
      "learning_rate": 4.1056111000228937e-07,
      "loss": 0.0587,
      "step": 978
    },
    {
      "epoch": 1.7419928825622777,
      "grad_norm": 3.978286477786184,
      "learning_rate": 4.0503318169512417e-07,
      "loss": 0.0727,
      "step": 979
    },
    {
      "epoch": 1.7437722419928825,
      "grad_norm": 2.6630880708888856,
      "learning_rate": 3.9954114964941336e-07,
      "loss": 0.0537,
      "step": 980
    },
    {
      "epoch": 1.7455516014234875,
      "grad_norm": 2.9732792508107546,
      "learning_rate": 3.9408505676936327e-07,
      "loss": 0.0601,
      "step": 981
    },
    {
      "epoch": 1.7473309608540926,
      "grad_norm": 3.0372051348456073,
      "learning_rate": 3.886649456784253e-07,
      "loss": 0.0644,
      "step": 982
    },
    {
      "epoch": 1.7491103202846974,
      "grad_norm": 2.963285131988169,
      "learning_rate": 3.8328085871895624e-07,
      "loss": 0.0811,
      "step": 983
    },
    {
      "epoch": 1.7508896797153026,
      "grad_norm": 2.67325057107037,
      "learning_rate": 3.779328379518898e-07,
      "loss": 0.0574,
      "step": 984
    },
    {
      "epoch": 1.7526690391459074,
      "grad_norm": 2.680631692465237,
      "learning_rate": 3.7262092515640556e-07,
      "loss": 0.055,
      "step": 985
    },
    {
      "epoch": 1.7544483985765125,
      "grad_norm": 2.5577019764093714,
      "learning_rate": 3.673451618296081e-07,
      "loss": 0.0486,
      "step": 986
    },
    {
      "epoch": 1.7562277580071175,
      "grad_norm": 2.863191182646623,
      "learning_rate": 3.621055891861963e-07,
      "loss": 0.0655,
      "step": 987
    },
    {
      "epoch": 1.7580071174377223,
      "grad_norm": 3.1909945742397956,
      "learning_rate": 3.56902248158148e-07,
      "loss": 0.0627,
      "step": 988
    },
    {
      "epoch": 1.7597864768683276,
      "grad_norm": 3.1625303547166093,
      "learning_rate": 3.517351793943913e-07,
      "loss": 0.0629,
      "step": 989
    },
    {
      "epoch": 1.7615658362989324,
      "grad_norm": 4.546898130410527,
      "learning_rate": 3.4660442326049704e-07,
      "loss": 0.0599,
      "step": 990
    },
    {
      "epoch": 1.7633451957295374,
      "grad_norm": 2.902596422882486,
      "learning_rate": 3.4151001983835696e-07,
      "loss": 0.069,
      "step": 991
    },
    {
      "epoch": 1.7651245551601424,
      "grad_norm": 4.149935600470566,
      "learning_rate": 3.364520089258727e-07,
      "loss": 0.0748,
      "step": 992
    },
    {
      "epoch": 1.7669039145907472,
      "grad_norm": 3.0285672632178526,
      "learning_rate": 3.314304300366461e-07,
      "loss": 0.0634,
      "step": 993
    },
    {
      "epoch": 1.7686832740213523,
      "grad_norm": 3.358091494579457,
      "learning_rate": 3.2644532239966444e-07,
      "loss": 0.0727,
      "step": 994
    },
    {
      "epoch": 1.7704626334519573,
      "grad_norm": 2.4991312826524137,
      "learning_rate": 3.2149672495900286e-07,
      "loss": 0.056,
      "step": 995
    },
    {
      "epoch": 1.7722419928825621,
      "grad_norm": 2.9861249154170753,
      "learning_rate": 3.165846763735153e-07,
      "loss": 0.0691,
      "step": 996
    },
    {
      "epoch": 1.7740213523131674,
      "grad_norm": 2.5522113098477295,
      "learning_rate": 3.117092150165324e-07,
      "loss": 0.0559,
      "step": 997
    },
    {
      "epoch": 1.7758007117437722,
      "grad_norm": 3.122721200228551,
      "learning_rate": 3.068703789755606e-07,
      "loss": 0.0591,
      "step": 998
    },
    {
      "epoch": 1.7775800711743772,
      "grad_norm": 3.2480454610726603,
      "learning_rate": 3.020682060519886e-07,
      "loss": 0.0667,
      "step": 999
    },
    {
      "epoch": 1.7793594306049823,
      "grad_norm": 3.2175717139593067,
      "learning_rate": 2.9730273376078923e-07,
      "loss": 0.0552,
      "step": 1000
    },
    {
      "epoch": 1.7793594306049823,
      "eval_loss": 0.17049002647399902,
      "eval_runtime": 1.5601,
      "eval_samples_per_second": 29.485,
      "eval_steps_per_second": 7.692,
      "step": 1000
    },
    {
      "epoch": 1.781138790035587,
      "grad_norm": 3.0869264852679077,
      "learning_rate": 2.9257399933022737e-07,
      "loss": 0.0592,
      "step": 1001
    },
    {
      "epoch": 1.7829181494661923,
      "grad_norm": 2.8116703538320533,
      "learning_rate": 2.8788203970156805e-07,
      "loss": 0.0548,
      "step": 1002
    },
    {
      "epoch": 1.7846975088967971,
      "grad_norm": 2.527184878092922,
      "learning_rate": 2.832268915287878e-07,
      "loss": 0.0687,
      "step": 1003
    },
    {
      "epoch": 1.7864768683274022,
      "grad_norm": 3.4006193202614132,
      "learning_rate": 2.7860859117828985e-07,
      "loss": 0.074,
      "step": 1004
    },
    {
      "epoch": 1.7882562277580072,
      "grad_norm": 4.119995920447034,
      "learning_rate": 2.740271747286194e-07,
      "loss": 0.1209,
      "step": 1005
    },
    {
      "epoch": 1.790035587188612,
      "grad_norm": 3.1257719024558206,
      "learning_rate": 2.6948267797018145e-07,
      "loss": 0.0646,
      "step": 1006
    },
    {
      "epoch": 1.791814946619217,
      "grad_norm": 2.9747319632660676,
      "learning_rate": 2.649751364049613e-07,
      "loss": 0.0467,
      "step": 1007
    },
    {
      "epoch": 1.793594306049822,
      "grad_norm": 3.0475188492954683,
      "learning_rate": 2.6050458524624735e-07,
      "loss": 0.0597,
      "step": 1008
    },
    {
      "epoch": 1.795373665480427,
      "grad_norm": 2.905330612358647,
      "learning_rate": 2.560710594183552e-07,
      "loss": 0.054,
      "step": 1009
    },
    {
      "epoch": 1.7971530249110321,
      "grad_norm": 2.8775183805997773,
      "learning_rate": 2.5167459355635524e-07,
      "loss": 0.0666,
      "step": 1010
    },
    {
      "epoch": 1.798932384341637,
      "grad_norm": 3.1759660212425174,
      "learning_rate": 2.473152220058039e-07,
      "loss": 0.0579,
      "step": 1011
    },
    {
      "epoch": 1.800711743772242,
      "grad_norm": 3.7024556814179714,
      "learning_rate": 2.429929788224722e-07,
      "loss": 0.0753,
      "step": 1012
    },
    {
      "epoch": 1.802491103202847,
      "grad_norm": 2.9168881839757343,
      "learning_rate": 2.38707897772083e-07,
      "loss": 0.0776,
      "step": 1013
    },
    {
      "epoch": 1.8042704626334518,
      "grad_norm": 3.23452416835502,
      "learning_rate": 2.3446001233004333e-07,
      "loss": 0.0862,
      "step": 1014
    },
    {
      "epoch": 1.806049822064057,
      "grad_norm": 3.012537543174496,
      "learning_rate": 2.3024935568118745e-07,
      "loss": 0.0585,
      "step": 1015
    },
    {
      "epoch": 1.8078291814946619,
      "grad_norm": 2.729801999778358,
      "learning_rate": 2.2607596071951288e-07,
      "loss": 0.0459,
      "step": 1016
    },
    {
      "epoch": 1.809608540925267,
      "grad_norm": 2.7730296362492868,
      "learning_rate": 2.2193986004792667e-07,
      "loss": 0.0537,
      "step": 1017
    },
    {
      "epoch": 1.811387900355872,
      "grad_norm": 2.3623466616896724,
      "learning_rate": 2.1784108597799058e-07,
      "loss": 0.0384,
      "step": 1018
    },
    {
      "epoch": 1.8131672597864767,
      "grad_norm": 2.586437247378554,
      "learning_rate": 2.1377967052966685e-07,
      "loss": 0.0526,
      "step": 1019
    },
    {
      "epoch": 1.814946619217082,
      "grad_norm": 2.5914309002398332,
      "learning_rate": 2.0975564543107007e-07,
      "loss": 0.0607,
      "step": 1020
    },
    {
      "epoch": 1.8167259786476868,
      "grad_norm": 3.108843234304536,
      "learning_rate": 2.057690421182168e-07,
      "loss": 0.0577,
      "step": 1021
    },
    {
      "epoch": 1.8185053380782918,
      "grad_norm": 2.6437269638019583,
      "learning_rate": 2.01819891734783e-07,
      "loss": 0.0524,
      "step": 1022
    },
    {
      "epoch": 1.8202846975088969,
      "grad_norm": 2.923611406537203,
      "learning_rate": 1.979082251318576e-07,
      "loss": 0.0583,
      "step": 1023
    },
    {
      "epoch": 1.8220640569395017,
      "grad_norm": 2.827720028825018,
      "learning_rate": 1.9403407286770592e-07,
      "loss": 0.0551,
      "step": 1024
    },
    {
      "epoch": 1.8238434163701067,
      "grad_norm": 2.437785467084817,
      "learning_rate": 1.9019746520752502e-07,
      "loss": 0.054,
      "step": 1025
    },
    {
      "epoch": 1.8256227758007118,
      "grad_norm": 2.4422282580425057,
      "learning_rate": 1.8639843212321206e-07,
      "loss": 0.0493,
      "step": 1026
    },
    {
      "epoch": 1.8274021352313166,
      "grad_norm": 2.7654516115379386,
      "learning_rate": 1.826370032931285e-07,
      "loss": 0.0575,
      "step": 1027
    },
    {
      "epoch": 1.8291814946619218,
      "grad_norm": 2.971673971283802,
      "learning_rate": 1.789132081018674e-07,
      "loss": 0.0532,
      "step": 1028
    },
    {
      "epoch": 1.8309608540925266,
      "grad_norm": 2.799101936477553,
      "learning_rate": 1.7522707564002706e-07,
      "loss": 0.0519,
      "step": 1029
    },
    {
      "epoch": 1.8327402135231317,
      "grad_norm": 2.839432191061096,
      "learning_rate": 1.7157863470397718e-07,
      "loss": 0.0462,
      "step": 1030
    },
    {
      "epoch": 1.8345195729537367,
      "grad_norm": 2.9443195510317697,
      "learning_rate": 1.6796791379564138e-07,
      "loss": 0.0561,
      "step": 1031
    },
    {
      "epoch": 1.8362989323843415,
      "grad_norm": 2.8382796466975337,
      "learning_rate": 1.6439494112227173e-07,
      "loss": 0.049,
      "step": 1032
    },
    {
      "epoch": 1.8380782918149468,
      "grad_norm": 2.9194126225748316,
      "learning_rate": 1.6085974459622567e-07,
      "loss": 0.0549,
      "step": 1033
    },
    {
      "epoch": 1.8398576512455516,
      "grad_norm": 2.3384708289182394,
      "learning_rate": 1.573623518347517e-07,
      "loss": 0.0406,
      "step": 1034
    },
    {
      "epoch": 1.8416370106761566,
      "grad_norm": 2.600042139041554,
      "learning_rate": 1.5390279015977117e-07,
      "loss": 0.0649,
      "step": 1035
    },
    {
      "epoch": 1.8434163701067616,
      "grad_norm": 2.583746208290995,
      "learning_rate": 1.5048108659766693e-07,
      "loss": 0.0524,
      "step": 1036
    },
    {
      "epoch": 1.8451957295373664,
      "grad_norm": 2.5350941131708127,
      "learning_rate": 1.470972678790711e-07,
      "loss": 0.066,
      "step": 1037
    },
    {
      "epoch": 1.8469750889679717,
      "grad_norm": 3.2048699936886997,
      "learning_rate": 1.437513604386559e-07,
      "loss": 0.0878,
      "step": 1038
    },
    {
      "epoch": 1.8487544483985765,
      "grad_norm": 3.1790959376040466,
      "learning_rate": 1.404433904149266e-07,
      "loss": 0.0632,
      "step": 1039
    },
    {
      "epoch": 1.8505338078291815,
      "grad_norm": 3.484711910370608,
      "learning_rate": 1.3717338365001943e-07,
      "loss": 0.0587,
      "step": 1040
    },
    {
      "epoch": 1.8523131672597866,
      "grad_norm": 3.001843249193619,
      "learning_rate": 1.3394136568949834e-07,
      "loss": 0.0779,
      "step": 1041
    },
    {
      "epoch": 1.8540925266903914,
      "grad_norm": 4.572527523423257,
      "learning_rate": 1.307473617821553e-07,
      "loss": 0.0797,
      "step": 1042
    },
    {
      "epoch": 1.8558718861209964,
      "grad_norm": 3.542102764492959,
      "learning_rate": 1.275913968798137e-07,
      "loss": 0.0626,
      "step": 1043
    },
    {
      "epoch": 1.8576512455516014,
      "grad_norm": 2.8224387129917714,
      "learning_rate": 1.2447349563713186e-07,
      "loss": 0.0701,
      "step": 1044
    },
    {
      "epoch": 1.8594306049822062,
      "grad_norm": 3.062466641395086,
      "learning_rate": 1.213936824114137e-07,
      "loss": 0.0619,
      "step": 1045
    },
    {
      "epoch": 1.8612099644128115,
      "grad_norm": 4.513577237425157,
      "learning_rate": 1.1835198126241509e-07,
      "loss": 0.0749,
      "step": 1046
    },
    {
      "epoch": 1.8629893238434163,
      "grad_norm": 3.1706599627525076,
      "learning_rate": 1.1534841595215617e-07,
      "loss": 0.0639,
      "step": 1047
    },
    {
      "epoch": 1.8647686832740213,
      "grad_norm": 2.581834212372844,
      "learning_rate": 1.1238300994473983e-07,
      "loss": 0.0451,
      "step": 1048
    },
    {
      "epoch": 1.8665480427046264,
      "grad_norm": 3.2579815734616933,
      "learning_rate": 1.0945578640616183e-07,
      "loss": 0.1046,
      "step": 1049
    },
    {
      "epoch": 1.8683274021352312,
      "grad_norm": 2.598700783565905,
      "learning_rate": 1.0656676820413603e-07,
      "loss": 0.0402,
      "step": 1050
    },
    {
      "epoch": 1.8701067615658364,
      "grad_norm": 2.9838224983191837,
      "learning_rate": 1.0371597790791166e-07,
      "loss": 0.0608,
      "step": 1051
    },
    {
      "epoch": 1.8718861209964412,
      "grad_norm": 3.079678034578598,
      "learning_rate": 1.0090343778809908e-07,
      "loss": 0.061,
      "step": 1052
    },
    {
      "epoch": 1.8736654804270463,
      "grad_norm": 3.660010687726608,
      "learning_rate": 9.812916981649433e-08,
      "loss": 0.0681,
      "step": 1053
    },
    {
      "epoch": 1.8754448398576513,
      "grad_norm": 2.728755569535375,
      "learning_rate": 9.539319566590766e-08,
      "loss": 0.0674,
      "step": 1054
    },
    {
      "epoch": 1.8772241992882561,
      "grad_norm": 2.884166200313499,
      "learning_rate": 9.269553670999743e-08,
      "loss": 0.0687,
      "step": 1055
    },
    {
      "epoch": 1.8790035587188612,
      "grad_norm": 3.820520722207487,
      "learning_rate": 9.003621402309815e-08,
      "loss": 0.0588,
      "step": 1056
    },
    {
      "epoch": 1.8807829181494662,
      "grad_norm": 2.971830790709613,
      "learning_rate": 8.741524838005888e-08,
      "loss": 0.0548,
      "step": 1057
    },
    {
      "epoch": 1.8825622775800712,
      "grad_norm": 2.9714318502338584,
      "learning_rate": 8.483266025608061e-08,
      "loss": 0.0461,
      "step": 1058
    },
    {
      "epoch": 1.8843416370106763,
      "grad_norm": 2.6228289853570814,
      "learning_rate": 8.228846982655525e-08,
      "loss": 0.0535,
      "step": 1059
    },
    {
      "epoch": 1.886120996441281,
      "grad_norm": 2.7527747748008524,
      "learning_rate": 7.978269696691021e-08,
      "loss": 0.065,
      "step": 1060
    },
    {
      "epoch": 1.887900355871886,
      "grad_norm": 3.1060017505771125,
      "learning_rate": 7.731536125244965e-08,
      "loss": 0.0651,
      "step": 1061
    },
    {
      "epoch": 1.8896797153024911,
      "grad_norm": 3.5744823457124015,
      "learning_rate": 7.488648195820513e-08,
      "loss": 0.0959,
      "step": 1062
    },
    {
      "epoch": 1.891459074733096,
      "grad_norm": 2.9687660367391144,
      "learning_rate": 7.249607805878245e-08,
      "loss": 0.0534,
      "step": 1063
    },
    {
      "epoch": 1.8932384341637012,
      "grad_norm": 3.1899360716032117,
      "learning_rate": 7.014416822821557e-08,
      "loss": 0.0699,
      "step": 1064
    },
    {
      "epoch": 1.895017793594306,
      "grad_norm": 2.53903905091045,
      "learning_rate": 6.783077083981793e-08,
      "loss": 0.0428,
      "step": 1065
    },
    {
      "epoch": 1.896797153024911,
      "grad_norm": 3.031110393360956,
      "learning_rate": 6.55559039660425e-08,
      "loss": 0.0631,
      "step": 1066
    },
    {
      "epoch": 1.898576512455516,
      "grad_norm": 2.9343642524071942,
      "learning_rate": 6.331958537833693e-08,
      "loss": 0.0607,
      "step": 1067
    },
    {
      "epoch": 1.9003558718861209,
      "grad_norm": 2.9613665704541683,
      "learning_rate": 6.112183254700866e-08,
      "loss": 0.0581,
      "step": 1068
    },
    {
      "epoch": 1.9021352313167261,
      "grad_norm": 2.8082248015348616,
      "learning_rate": 5.8962662641083856e-08,
      "loss": 0.0588,
      "step": 1069
    },
    {
      "epoch": 1.903914590747331,
      "grad_norm": 3.1566430585686645,
      "learning_rate": 5.6842092528176516e-08,
      "loss": 0.0586,
      "step": 1070
    },
    {
      "epoch": 1.905693950177936,
      "grad_norm": 2.91605289189826,
      "learning_rate": 5.476013877435626e-08,
      "loss": 0.0605,
      "step": 1071
    },
    {
      "epoch": 1.907473309608541,
      "grad_norm": 3.167168155261568,
      "learning_rate": 5.271681764401848e-08,
      "loss": 0.0689,
      "step": 1072
    },
    {
      "epoch": 1.9092526690391458,
      "grad_norm": 2.421939118324357,
      "learning_rate": 5.071214509975775e-08,
      "loss": 0.0502,
      "step": 1073
    },
    {
      "epoch": 1.9110320284697508,
      "grad_norm": 3.3098149116228783,
      "learning_rate": 4.8746136802240716e-08,
      "loss": 0.0587,
      "step": 1074
    },
    {
      "epoch": 1.9128113879003559,
      "grad_norm": 3.1390644087699795,
      "learning_rate": 4.6818808110087875e-08,
      "loss": 0.0675,
      "step": 1075
    },
    {
      "epoch": 1.914590747330961,
      "grad_norm": 3.3804830630156486,
      "learning_rate": 4.493017407975087e-08,
      "loss": 0.0816,
      "step": 1076
    },
    {
      "epoch": 1.916370106761566,
      "grad_norm": 3.2024567967941318,
      "learning_rate": 4.308024946539424e-08,
      "loss": 0.0507,
      "step": 1077
    },
    {
      "epoch": 1.9181494661921707,
      "grad_norm": 2.8633195611594604,
      "learning_rate": 4.1269048718783344e-08,
      "loss": 0.049,
      "step": 1078
    },
    {
      "epoch": 1.9199288256227758,
      "grad_norm": 2.691588928859178,
      "learning_rate": 3.9496585989167726e-08,
      "loss": 0.056,
      "step": 1079
    },
    {
      "epoch": 1.9217081850533808,
      "grad_norm": 2.6432756386502314,
      "learning_rate": 3.776287512317345e-08,
      "loss": 0.0504,
      "step": 1080
    },
    {
      "epoch": 1.9234875444839856,
      "grad_norm": 3.178492051740319,
      "learning_rate": 3.606792966469375e-08,
      "loss": 0.0615,
      "step": 1081
    },
    {
      "epoch": 1.9252669039145909,
      "grad_norm": 2.7596193782930887,
      "learning_rate": 3.4411762854782426e-08,
      "loss": 0.0579,
      "step": 1082
    },
    {
      "epoch": 1.9270462633451957,
      "grad_norm": 3.97416264684535,
      "learning_rate": 3.279438763155174e-08,
      "loss": 0.0478,
      "step": 1083
    },
    {
      "epoch": 1.9288256227758007,
      "grad_norm": 2.891118295308179,
      "learning_rate": 3.121581663007134e-08,
      "loss": 0.0612,
      "step": 1084
    },
    {
      "epoch": 1.9306049822064058,
      "grad_norm": 3.2589311758118673,
      "learning_rate": 2.967606218226837e-08,
      "loss": 0.0679,
      "step": 1085
    },
    {
      "epoch": 1.9323843416370106,
      "grad_norm": 2.5028068219973676,
      "learning_rate": 2.8175136316832e-08,
      "loss": 0.0483,
      "step": 1086
    },
    {
      "epoch": 1.9341637010676158,
      "grad_norm": 3.682753960692582,
      "learning_rate": 2.6713050759120117e-08,
      "loss": 0.097,
      "step": 1087
    },
    {
      "epoch": 1.9359430604982206,
      "grad_norm": 2.890187176074847,
      "learning_rate": 2.528981693106558e-08,
      "loss": 0.0597,
      "step": 1088
    },
    {
      "epoch": 1.9377224199288257,
      "grad_norm": 2.8297796018874095,
      "learning_rate": 2.3905445951089013e-08,
      "loss": 0.0655,
      "step": 1089
    },
    {
      "epoch": 1.9395017793594307,
      "grad_norm": 3.446892192091957,
      "learning_rate": 2.2559948634011673e-08,
      "loss": 0.0597,
      "step": 1090
    },
    {
      "epoch": 1.9412811387900355,
      "grad_norm": 3.161353824731978,
      "learning_rate": 2.125333549096942e-08,
      "loss": 0.0692,
      "step": 1091
    },
    {
      "epoch": 1.9430604982206405,
      "grad_norm": 2.6058205044741434,
      "learning_rate": 1.9985616729332747e-08,
      "loss": 0.0594,
      "step": 1092
    },
    {
      "epoch": 1.9448398576512456,
      "grad_norm": 3.2577250864401734,
      "learning_rate": 1.8756802252625773e-08,
      "loss": 0.0728,
      "step": 1093
    },
    {
      "epoch": 1.9466192170818504,
      "grad_norm": 3.1195527646205305,
      "learning_rate": 1.75669016604485e-08,
      "loss": 0.0498,
      "step": 1094
    },
    {
      "epoch": 1.9483985765124556,
      "grad_norm": 2.7546770954561683,
      "learning_rate": 1.6415924248403547e-08,
      "loss": 0.0389,
      "step": 1095
    },
    {
      "epoch": 1.9501779359430604,
      "grad_norm": 3.297986081955345,
      "learning_rate": 1.5303879008021773e-08,
      "loss": 0.0726,
      "step": 1096
    },
    {
      "epoch": 1.9519572953736655,
      "grad_norm": 2.7043209129450596,
      "learning_rate": 1.4230774626691756e-08,
      "loss": 0.0583,
      "step": 1097
    },
    {
      "epoch": 1.9537366548042705,
      "grad_norm": 3.3670598032071357,
      "learning_rate": 1.3196619487594875e-08,
      "loss": 0.0749,
      "step": 1098
    },
    {
      "epoch": 1.9555160142348753,
      "grad_norm": 3.4479757119436956,
      "learning_rate": 1.2201421669636448e-08,
      "loss": 0.0738,
      "step": 1099
    },
    {
      "epoch": 1.9572953736654806,
      "grad_norm": 2.7797437178753257,
      "learning_rate": 1.1245188947384133e-08,
      "loss": 0.0595,
      "step": 1100
    },
    {
      "epoch": 1.9590747330960854,
      "grad_norm": 2.9721968581025555,
      "learning_rate": 1.0327928791006858e-08,
      "loss": 0.0613,
      "step": 1101
    },
    {
      "epoch": 1.9608540925266904,
      "grad_norm": 2.883396870053337,
      "learning_rate": 9.449648366217645e-09,
      "loss": 0.0613,
      "step": 1102
    },
    {
      "epoch": 1.9626334519572954,
      "grad_norm": 2.8548792391457902,
      "learning_rate": 8.61035453421588e-09,
      "loss": 0.0589,
      "step": 1103
    },
    {
      "epoch": 1.9644128113879002,
      "grad_norm": 2.6522688674350987,
      "learning_rate": 7.81005385163458e-09,
      "loss": 0.0583,
      "step": 1104
    },
    {
      "epoch": 1.9661921708185055,
      "grad_norm": 2.6226719727810486,
      "learning_rate": 7.048752570488205e-09,
      "loss": 0.0666,
      "step": 1105
    },
    {
      "epoch": 1.9679715302491103,
      "grad_norm": 3.3391367344313734,
      "learning_rate": 6.326456638125478e-09,
      "loss": 0.0845,
      "step": 1106
    },
    {
      "epoch": 1.9697508896797153,
      "grad_norm": 2.884116390700698,
      "learning_rate": 5.643171697183314e-09,
      "loss": 0.0466,
      "step": 1107
    },
    {
      "epoch": 1.9715302491103204,
      "grad_norm": 3.159346575228147,
      "learning_rate": 4.998903085539075e-09,
      "loss": 0.0625,
      "step": 1108
    },
    {
      "epoch": 1.9733096085409252,
      "grad_norm": 2.703289391601726,
      "learning_rate": 4.393655836272825e-09,
      "loss": 0.0453,
      "step": 1109
    },
    {
      "epoch": 1.9750889679715302,
      "grad_norm": 2.9705576949530133,
      "learning_rate": 3.8274346776262514e-09,
      "loss": 0.0512,
      "step": 1110
    },
    {
      "epoch": 1.9768683274021353,
      "grad_norm": 3.034172310094397,
      "learning_rate": 3.300244032966582e-09,
      "loss": 0.0615,
      "step": 1111
    },
    {
      "epoch": 1.97864768683274,
      "grad_norm": 3.4411625755017594,
      "learning_rate": 2.8120880207493928e-09,
      "loss": 0.0559,
      "step": 1112
    },
    {
      "epoch": 1.9804270462633453,
      "grad_norm": 2.9328070857649227,
      "learning_rate": 2.362970454491409e-09,
      "loss": 0.0571,
      "step": 1113
    },
    {
      "epoch": 1.9822064056939501,
      "grad_norm": 3.599687589291081,
      "learning_rate": 1.952894842735531e-09,
      "loss": 0.0641,
      "step": 1114
    },
    {
      "epoch": 1.9839857651245552,
      "grad_norm": 1.924893933931121,
      "learning_rate": 1.5818643890258555e-09,
      "loss": 0.0371,
      "step": 1115
    },
    {
      "epoch": 1.9857651245551602,
      "grad_norm": 4.1023077374690216,
      "learning_rate": 1.2498819918843609e-09,
      "loss": 0.0615,
      "step": 1116
    },
    {
      "epoch": 1.987544483985765,
      "grad_norm": 3.467329992082243,
      "learning_rate": 9.569502447837053e-10,
      "loss": 0.0578,
      "step": 1117
    },
    {
      "epoch": 1.9893238434163703,
      "grad_norm": 3.2506696707647564,
      "learning_rate": 7.03071436131686e-10,
      "loss": 0.0564,
      "step": 1118
    },
    {
      "epoch": 1.991103202846975,
      "grad_norm": 2.626357039522761,
      "learning_rate": 4.882475492506977e-10,
      "loss": 0.0613,
      "step": 1119
    },
    {
      "epoch": 1.99288256227758,
      "grad_norm": 3.0134175710882825,
      "learning_rate": 3.124802623627465e-10,
      "loss": 0.069,
      "step": 1120
    },
    {
      "epoch": 1.9946619217081851,
      "grad_norm": 3.2540037933351815,
      "learning_rate": 1.7577094857557097e-10,
      "loss": 0.0842,
      "step": 1121
    },
    {
      "epoch": 1.99644128113879,
      "grad_norm": 3.194741562439074,
      "learning_rate": 7.812067587487093e-11,
      "loss": 0.0759,
      "step": 1122
    },
    {
      "epoch": 1.998220640569395,
      "grad_norm": 2.565114932226767,
      "learning_rate": 1.9530207111539967e-11,
      "loss": 0.052,
      "step": 1123
    },
    {
      "epoch": 2.0,
      "grad_norm": 2.2375530713532674,
      "learning_rate": 0.0,
      "loss": 0.0542,
      "step": 1124
    },
    {
      "epoch": 2.0,
      "step": 1124,
      "total_flos": 2468145598464.0,
      "train_loss": 0.11595660425868429,
      "train_runtime": 680.0781,
      "train_samples_per_second": 13.213,
      "train_steps_per_second": 1.653
    }
  ],
  "logging_steps": 1,
  "max_steps": 1124,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 50000000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 2468145598464.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}